* initial auth migration (@ji-yaqi) * clean up files * update imports * add type annotations * update docstrings * add tests