Function rewrite_index

fn rewrite_index(
    expr: &Expr,
    index: &Expr,
    context: &RewriteContext<'_>,
    shape: Shape,
) -> Result<String, RewriteError>