TIL that zsh has a helper for a shell function to keep all the command completions of the original.

e.g. this compdef '_as_if jj log' j

allows j to run jj log such that I can tab-complete any of the options for jj log