MarshalX · MarshalX · Jan 10, 2024 · Jan 9, 2024 · Jan 10, 2024
diff --git a/docs/source/atproto/atproto_client.client.rst b/docs/source/atproto/atproto_client.client.rst
@@ -22,7 +22,6 @@ Submodules
 
    atproto_client.client.async_client
    atproto_client.client.async_raw
-   atproto_client.client.auth
    atproto_client.client.base
    atproto_client.client.client
    atproto_client.client.raw
diff --git a/docs/source/atproto/atproto_crypto.rst b/docs/source/atproto/atproto_crypto.rst
@@ -0,0 +1,15 @@
+atproto\_crypto
+===============
+
+.. automodule:: atproto_crypto
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+Submodules
+----------
+
+.. toctree::
+   :maxdepth: 4
+
+   atproto_crypto.verify
diff --git a/docs/source/atproto/atproto_crypto.verify.rst b/docs/source/atproto/atproto_crypto.verify.rst
@@ -0,0 +1,7 @@
+atproto\_crypto.verify
+======================
+
+.. automodule:: atproto_crypto.verify
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/docs/source/atproto/atproto_server.auth.jwt.rst b/docs/source/atproto/atproto_server.auth.jwt.rst
@@ -0,0 +1,7 @@
+atproto\_server.auth.jwt
+========================
+
+.. automodule:: atproto_server.auth.jwt
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/docs/source/atproto/atproto_server.auth.rst b/docs/source/atproto/atproto_server.auth.rst
@@ -0,0 +1,16 @@
+atproto\_server.auth
+====================
+
+.. automodule:: atproto_server.auth
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+Submodules
+----------
+
+.. toctree::
+   :maxdepth: 4
+
+   atproto_server.auth.jwt
+   atproto_server.auth.utils
diff --git a/docs/source/atproto/atproto_server.auth.utils.rst b/docs/source/atproto/atproto_server.auth.utils.rst
@@ -0,0 +1,7 @@
+atproto\_server.auth.utils
+==========================
+
+.. automodule:: atproto_server.auth.utils
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/docs/source/atproto/atproto_server.exceptions.rst b/docs/source/atproto/atproto_server.exceptions.rst
@@ -0,0 +1,7 @@
+atproto\_server.exceptions
+==========================
+
+.. automodule:: atproto_server.exceptions
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/docs/source/atproto/atproto_server.rst b/docs/source/atproto/atproto_server.rst
@@ -5,3 +5,19 @@ atproto\_server
    :members:
    :undoc-members:
    :show-inheritance:
+
+Subpackages
+-----------
+
+.. toctree::
+   :maxdepth: 4
+
+   atproto_server.auth
+
+Submodules
+----------
+
+.. toctree::
+   :maxdepth: 4
+
+   atproto_server.exceptions
diff --git a/docs/source/atproto/modules.rst b/docs/source/atproto/modules.rst
@@ -9,6 +9,7 @@ packages
    atproto_client
    atproto_codegen
    atproto_core
+   atproto_crypto
    atproto_firehose
    atproto_identity
    atproto_lexicon

diff --git a/docs/source/atproto_server/auth.rst b/docs/source/atproto_server/auth.rst
@@ -0,0 +1,7 @@
+Auth
+====
+
+.. automodule:: atproto_server.auth.jwt
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/docs/source/atproto_server/index.rst b/docs/source/atproto_server/index.rst
@@ -0,0 +1,12 @@
+Server
+======
+
+.. automodule:: atproto_server
+   :members:
+   :undoc-members:
+   :inherited-members:
+
+.. toctree::
+   :maxdepth: 4
+
+   auth
diff --git a/docs/source/exceptions.rst b/docs/source/exceptions.rst
@@ -37,6 +37,14 @@ Identity
    :show-inheritance:
 
 
+Server
+######
+
+.. automodule:: atproto_server.exceptions
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
 Lexicon
 #######
 

diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -23,6 +23,7 @@ Documentation
    namespace
    models
    atproto_identity/identity
+   atproto_server/index
 
 .. toctree::
    :caption: Core

diff --git a/packages/atproto/__init__.py b/packages/atproto/__init__.py
@@ -16,27 +16,48 @@
 from atproto_firehose import models as firehose_models
 from atproto_identity.cache.in_memory_cache import AsyncDidInMemoryCache, DidInMemoryCache
 from atproto_identity.resolver import AsyncIdResolver, IdResolver
+from atproto_server.auth.jwt import (
+    JwtPayload,
+    decode_jwt_payload,
+    get_jwt_payload,
+    parse_jwt,
+    validate_jwt_payload,
+    verify_jwt,
+    verify_jwt_async,
+)
 
 __all__ = [
+    # client
     'AsyncClient',
     'Client',
     'client_utils',
     'models',
+    # core
     'CAR',
     'CID',
     'CIDType',
     'DidDocument',
     'NSID',
     'AtUri',
+    # firehose
     'AsyncFirehoseSubscribeLabelsClient',
     'AsyncFirehoseSubscribeReposClient',
     'FirehoseSubscribeLabelsClient',
     'FirehoseSubscribeReposClient',
     'parse_subscribe_labels_message',
     'parse_subscribe_repos_message',
     'firehose_models',
+    # identity
     'AsyncDidInMemoryCache',
     'DidInMemoryCache',
     'AsyncIdResolver',
     'IdResolver',
+    # server jwt
+    'JwtPayload',
+    'decode_jwt_payload',
+    'get_jwt_payload',
+    'parse_jwt',
+    'validate_jwt_payload',
+    'verify_jwt',
+    'verify_jwt_async',
 ]
diff --git a/packages/atproto/exceptions.py b/packages/atproto/exceptions.py
@@ -3,3 +3,4 @@
 from atproto_firehose.exceptions import *
 from atproto_identity.exceptions import *
 from atproto_lexicon.exceptions import *
+from atproto_server.exceptions import *
diff --git a/packages/atproto_client/client/auth.py b/packages/atproto_client/client/auth.py
diff --git a/packages/atproto_client/client/methods_mixin/session.py b/packages/atproto_client/client/methods_mixin/session.py
@@ -3,12 +3,12 @@
 from datetime import timedelta
 
 import typing_extensions as te
-
-from atproto_client.client.auth import get_jwt_payload
+from atproto_server.auth.jwt import get_jwt_payload
 
 if t.TYPE_CHECKING:
+    from atproto_server.auth.jwt import JwtPayload
+
     from atproto_client import models
-    from atproto_client.client.auth import JwtPayload
 
 
 @dataclass

diff --git a/packages/atproto_core/did_doc/did_doc.py b/packages/atproto_core/did_doc/did_doc.py
@@ -80,6 +80,7 @@ def get_service_endpoint(did_doc: 'DidDocument', id_: str, type_: str) -> t.Opti
     """Returns the service endpoint of the given DID document.
 
     Args:
+        did_doc: The DID document.
         id_: The service ID.
         type_: The service type.
 

diff --git a/packages/atproto_crypto/__init__.py b/packages/atproto_crypto/__init__.py
diff --git a/packages/atproto_crypto/verify.py b/packages/atproto_crypto/verify.py
@@ -0,0 +1,13 @@
+import typing as t
+import warnings
+
+
+def verify_signature(did_key: str, signing_input: t.Union[str, bytes], signature: t.Union[str, bytes]) -> bool:
+    # TODO(MarshalX): implement
+    warnings.warn(
+        'verify_signature is not implemented yet. Do not trust to this signing_input',
+        RuntimeWarning,
+        stacklevel=0,
+    )
+
+    return True
diff --git a/packages/atproto_server/__init__.py b/packages/atproto_server/__init__.py
@@ -1 +0,0 @@
-# TODO(MarshalX): When ATProto will stable

diff --git a/packages/atproto_server/auth/__init__.py b/packages/atproto_server/auth/__init__.py