[DOC-XXX][feature] -  $context support with ast transformation