Another approach is to use a joint modeling analysis in which the “covariate” is treated as an outcome. For example, using joint models for longitudinal and survival data. In such models treatment enters in both models as a covariate and you can investigate how it affects the primary outcome as the time-varying/longitudinal covariate changes over time.