commit ad871068d0c51fc7a5e0650f45ddc6e60ef4f2a7 · dwn.poxiao-labs.work/atpasser

pyproject.toml

···

       19
       19
        
       license = "MIT OR Apache-2.0"

     

       20
       20
        
       license-files = ["LICEN[CS]E.*"]

     

       21
       21
        
       

     

       22
       22
       +
       

     

       22
       23
        
       [tool.poetry]

     

       23
       24
        
       packages = [{ include = "atpasser", from = "src" }]

     

       25
       25
       +
       

     

       26
       26
       +
       [tool.poetry.group.dev.dependencies]

     

       27
       27
       +
       pytest = "^8.2.0"

     

       28
       28
       +
       pytest-cov = "^5.0.0"

     

       24
       29
        
       

     

       25
       30
        
       

     

       26
       31
        
       [build-system]

+46 -285

src/atpasser/model/__init__.py

···

       1
       1
       -
       import base64

     

       2
       2
       -
       from typing import Any

     

       3
       3
       -
       from collections.abc import Mapping

     

       4
       4
       -
       from cid.cid import CIDv1, make_cid

     

       5
       5
       -
       from pydantic import BaseModel, field_serializer, field_validator, ConfigDict

     

       6
       6
       -
       from pydantic.fields import FieldInfo

     

       7
       7
       -
       

     

       8
       8
       -
       

     

       9
       9
       -
       class DataModel(BaseModel):

     

       10
       10
       -
           """

     

       11
       11
       -
           Base model for AT Protocol data model with support for JSON encodings.

     

       12
       12
       -
       

     

       13
       13
       -
           This model provides automatic serialization and deserialization for AT Protocol

     

       14
       14
       -
           specific data types including bytes, CID links, and blob references.

     

       15
       15
       -
           """

     

       16
       16
       -
       

     

       17
       17
       -
           model_config = ConfigDict(arbitrary_types_allowed=True)

     

       18
       18
       -
       

     

       19
       19
       -
           @field_validator("*", mode="before")

     

       20
       20
       -
           def parseAtprotoData(cls, v: Any) -> Any:

     

       21
       21
       -
               """

     

       22
       22
       -
               Parse AT Protocol specific data types from JSON representation.

     

       23
       23
       -
       

     

       24
       24
       -
               Args:

     

       25
       25
       -
                   v: Input value to validate

     

       26
       26
       -
       

     

       27
       27
       -
               Returns:

     

       28
       28
       -
                   Appropriate Python objects for AT Protocol data types

     

       29
       29
       -
       

     

       30
       30
       -
               Raises:

     

       31
       31
       -
                   ValueError: If parsing fails

     

       32
       32
       -
               """

     

       33
       33
       -
               # Handle bytes objects

     

       34
       34
       -
               if isinstance(v, dict) and "$bytes" in v:

     

       35
       35
       -
                   value = v["$bytes"]

     

       36
       36
       -
                   try:

     

       37
       37
       -
                       return base64.b64decode(value)

     

       38
       38
       -
                   except ValueError as e:

     

       39
       39
       -
                       raise ValueError(f"Failed to decode base64 bytes: {e}")

     

       40
       40
       -
       

     

       41
       41
       -
               # Handle CID objects

     

       42
       42
       -
               elif isinstance(v, dict) and "$link" in v:

     

       43
       43
       -
                   value = v["$link"]

     

       44
       44
       -
                   try:

     

       45
       45
       -
                       return make_cid(value)

     

       46
       46
       -
                   except ValueError as e:

     

       47
       47
       -
                       raise ValueError(f"Failed to parse CID: {e}")

     

       48
       48
       -
       

     

       49
       49
       -
               return v

     

       50
       50
       -
       

     

       51
       51
       -
           @field_serializer("*")

     

       52
       52
       -
           def serializeAtprotoData(self, v: Any, _info: Any) -> Any:

     

       53
       53
       -
               """

     

       54
       54
       -
               Serialize AT Protocol specific data types to JSON representation.

     

       55
       55
       -
       

     

       56
       56
       -
               Args:

     

       57
       57
       -
                   v: Value to serialize

     

       58
       58
       -
                   _info: Field serialization info (not used in this implementation)

     

       59
       59
       -
       

     

       60
       60
       -
               Returns:

     

       61
       61
       -
                   Appropriate AT Protocol representation for bytes and CID fields

     

       62
       62
       -
               """

     

       63
       63
       -
               # Handle bytes fields

     

       64
       64
       -
               if isinstance(v, bytes):

     

       65
       65
       -
                   value = base64.b64encode(v).decode()

     

       66
       66
       -
                   return {"$bytes": value}

     

       67
       67
       -
       

     

       68
       68
       -
               # Handle CID fields

     

       69
       69
       -
               elif isinstance(v, CIDv1):

     

       70
       70
       -
                   value = str(v)

     

       71
       71
       -
                   return {"$link": value}

     

       72
       72
       -
       

     

       73
       73
       -
               return v

     

       74
       74
       -
       

     

       75
       75
       -
       

     

       76
       76
       -
       class BlobModel(BaseModel):

     

       77
       77
       -
           """

     

       78
       78
       -
           Model for AT Protocol blob references.

     

       79
       79
       -
       

     

       80
       80
       -
           Blobs are used for file references like images and include metadata

     

       81
       81
       -
           such as MIME type and size.

     

       82
       82
       -
           """

     

       83
       83
       -
       

     

       84
       84
       -
           ref: Any  # Accept both CIDv0 and CIDv1

     

       85
       85
       -
           """CID reference to blob with multicodec type 'raw'"""

     

       86
       86
       -
       

     

       87
       87
       -
           mimeType: str

     

       88
       88
       -
           """Content type of blob, 'application/octet-stream' if not known"""

     

       89
       89
       -
       

     

       90
       90
       -
           size: int

     

       91
       91
       -
           """Length of blob in bytes, must be positive and non-zero"""

     

       92
       92
       -
       

     

       93
       93
       -
           model_config = ConfigDict(arbitrary_types_allowed=True)

     

       1
       1
       +
       from .base import DataModel

     

       2
       2
       +
       from .blob import BlobModel

     

       3
       3
       +
       from .typed import TypedDataModel

     

       4
       4
       +
       from .validation import (

     

       5
       5
       +
           validateAtprotoData,

     

       6
       6
       +
           isBlessedCID,

     

       7
       7
       +
           convertLegacyBlobToModern

     

       8
       8
       +
       )

     

       9
       9
       +
       from .types.primitive import (

     

       10
       10
       +
           NullModel,

     

       11
       11
       +
           BooleanModel,

     

       12
       12
       +
           IntegerModel

     

       13
       13
       +
       )

     

       14
       14
       +
       from .types.string import StringModel

     

       15
       15
       +
       from .types.complex import (

     

       16
       16
       +
           ArrayModel,

     

       17
       17
       +
           ObjectModel,

     

       18
       18
       +
           ParamsModel

     

       19
       19
       +
       )

     

       20
       20
       +
       from .types.reference import (

     

       21
       21
       +
           TokenModel,

     

       22
       22
       +
           RefModel,

     

       23
       23
       +
           UnionModel

     

       24
       24
       +
       )

     

       25
       25
       +
       from .types.special import (

     

       26
       26
       +
           UnknownModel,

     

       27
       27
       +
           RecordModel,

     

       28
       28
       +
           QueryModel,

     

       29
       29
       +
           ProcedureModel,

     

       30
       30
       +
           SubscriptionModel

     

       31
       31
       +
       )

     

       94
       32
        
       

     

       95
       95
       -
           @field_validator("size")

     

       96
       96
       -
           def validateSize(cls, v: int) -> int:

     

       97
       97
       -
               """

     

       98
       98
       -
               Validate that size is positive and non-zero.

     

       99
       99
       -
       

     

       100
       100
       -
               Args:

     

       101
       101
       -
                   v: Size value to validate

     

       102
       102
       -
       

     

       103
       103
       -
               Returns:

     

       104
       104
       -
                   Validated size

     

       105
       105
       -
       

     

       106
       106
       -
               Raises:

     

       107
       107
       -
                   ValueError: If size is not positive

     

       108
       108
       -
               """

     

       109
       109
       -
               if v <= 0:

     

       110
       110
       -
                   raise ValueError("Blob size must be positive and non-zero")

     

       111
       111
       -
               return v

     

       112
       112
       -
       

     

       113
       113
       -
           @field_validator("mimeType")

     

       114
       114
       -
           def validateMimeType(cls, v: str) -> str:

     

       115
       115
       -
               """

     

       116
       116
       -
               Validate that mimeType is not empty.

     

       117
       117
       -
       

     

       118
       118
       -
               Args:

     

       119
       119
       -
                   v: MIME type to validate

     

       120
       120
       -
       

     

       121
       121
       -
               Returns:

     

       122
       122
       -
                   Validated MIME type

     

       123
       123
       -
       

     

       124
       124
       -
               Raises:

     

       125
       125
       -
                   ValueError: If MIME type is empty

     

       126
       126
       -
               """

     

       127
       127
       -
               if not v:

     

       128
       128
       -
                   raise ValueError("MIME type cannot be empty")

     

       129
       129
       -
               return v

     

       130
       130
       -
       

     

       131
       131
       -
       

     

       132
       132
       -
       class TypedDataModel(DataModel):

     

       133
       133
       -
           """

     

       134
       134
       -
           Model for AT Protocol data with type information.

     

       135
       135
       -
       

     

       136
       136
       -
           Includes support for $type field that specifies Lexicon schema.

     

       137
       137
       -
           """

     

       138
       138
       -
       

     

       139
       139
       -
           type: str | None = None

     

       140
       140
       -
           """Lexicon schema type identifier"""

     

       141
       141
       -
       

     

       142
       142
       -
           def __init__(self, **data: Any) -> None:

     

       143
       143
       -
               """

     

       144
       144
       -
               Initialize typed data model with automatic $type handling.

     

       145
       145
       -
       

     

       146
       146
       -
               Args:

     

       147
       147
       -
                   **data: Data including optional $type field

     

       148
       148
       -
               """

     

       149
       149
       -
               # Extract $type if present

     

       150
       150
       -
               dataType = data.pop("$type", None)

     

       151
       151
       -
               if dataType:

     

       152
       152
       -
                   data["type"] = dataType

     

       153
       153
       -
               super().__init__(**data)

     

       154
       154
       -
       

     

       155
       155
       -
           @field_serializer("type")

     

       156
       156
       -
           def serializeType(self, v: str | None) -> dict[str, str] | None:

     

       157
       157
       -
               """

     

       158
       158
       -
               Serialize type field to $type object.

     

       159
       159
       -
       

     

       160
       160
       -
               Args:

     

       161
       161
       -
                   v: Type value to serialize

     

       162
       162
       -
       

     

       163
       163
       -
               Returns:

     

       164
       164
       -
                   $type object if type is not None

     

       165
       165
       -
               """

     

       166
       166
       -
               if v is not None:

     

       167
       167
       -
                   return {"$type": v}

     

       168
       168
       -
               return None

     

       169
       169
       -
       

     

       170
       170
       -
       

     

       171
       171
       -
       def validateAtprotoData(data: Any) -> bool:

     

       172
       172
       -
           """

     

       173
       173
       -
           Validate that data conforms to AT Protocol data model.

     

       174
       174
       -
       

     

       175
       175
       -
           This function checks for basic structural validity of AT Protocol data

     

       176
       176
       -
           including proper encoding of bytes, links, and blob references.

     

       177
       177
       -
       

     

       178
       178
       -
           Args:

     

       179
       179
       -
               data: Data to validate

     

       180
       180
       -
       

     

       181
       181
       -
           Returns:

     

       182
       182
       -
               True if data is valid AT Protocol data

     

       183
       183
       -
           """

     

       184
       184
       -
           if isinstance(data, dict):

     

       185
       185
       -
               # Check for reserved $ fields

     

       186
       186
       -
               for key in data:

     

       187
       187
       -
                   if key.startswith("$") and key not in {"$bytes", "$link", "$type"}:

     

       188
       188
       -
                       return False

     

       189
       189
       -
       

     

       190
       190
       -
               # Validate $bytes objects

     

       191
       191
       -
               if "$bytes" in data:

     

       192
       192
       -
                   if not isinstance(data["$bytes"], str):

     

       193
       193
       -
                       return False

     

       194
       194
       -
                   try:

     

       195
       195
       -
                       base64.b64decode(data["$bytes"])

     

       196
       196
       -
                   except ValueError:

     

       197
       197
       -
                       return False

     

       198
       198
       -
       

     

       199
       199
       -
               # Validate $link objects

     

       200
       200
       -
               if "$link" in data:

     

       201
       201
       -
                   if not isinstance(data["$link"], str):

     

       202
       202
       -
                       return False

     

       203
       203
       -
                   try:

     

       204
       204
       -
                       make_cid(data["$link"])

     

       205
       205
       -
                   except ValueError:

     

       206
       206
       -
                       return False

     

       207
       207
       -
       

     

       208
       208
       -
               # Recursively validate nested data

     

       209
       209
       -
               for value in data.values():

     

       210
       210
       -
                   if not validateAtprotoData(value):

     

       211
       211
       -
                       return False

     

       212
       212
       -
       

     

       213
       213
       -
           elif isinstance(data, list):

     

       214
       214
       -
               for item in data:

     

       215
       215
       -
                   if not validateAtprotoData(item):

     

       216
       216
       -
                       return False

     

       217
       217
       -
       

     

       218
       218
       -
           return True

     

       219
       219
       -
       

     

       220
       220
       -
       

     

       221
       221
       -
       def isBlessedCID(cid_obj: CIDv1) -> bool:

     

       222
       222
       -
           """

     

       223
       223
       -
           Check if CID conforms to AT Protocol blessed formats.

     

       224
       224
       -
       

     

       225
       225
       -
           Blessed formats for CIDs in atproto are:

     

       226
       226
       -
           - CIDv1

     

       227
       227
       -
           - multibase: base32 for string encoding

     

       228
       228
       -
           - multicodec: dag-cbor (0x71) for data objects, raw (0x55) for blobs

     

       229
       229
       -
           - multihash: sha-256 with 256 bits (0x12) is preferred

     

       230
       230
       -
       

     

       231
       231
       -
           Args:

     

       232
       232
       -
               cid_obj: CID to validate

     

       233
       233
       -
       

     

       234
       234
       -
           Returns:

     

       235
       235
       -
               True if CID uses blessed format

     

       236
       236
       -
           """

     

       237
       237
       -
           # Check version

     

       238
       238
       -
           if cid_obj.version != 1:

     

       239
       239
       -
               return False

     

       240
       240
       -
       

     

       241
       241
       -
           # Check multibase (should be base32 for string representation)

     

       242
       242
       -
           # Note: This is primarily for string representation, binary is different

     

       243
       243
       -
       

     

       244
       244
       -
           # Check multicodec (should be dag-cbor or raw)

     

       245
       245
       -
           if cid_obj.codec not in {"dag-cbor", "raw"}:

     

       246
       246
       -
               return False

     

       247
       247
       -
       

     

       248
       248
       -
           # Check multihash (should be sha-256)

     

       249
       249
       -
           # Note: Access multihash info through appropriate CID methods

     

       250
       250
       -
           # The exact method depends on the CID library implementation

     

       251
       251
       -
           try:

     

       252
       252
       -
               # Try to get multihash code using library-specific method

     

       253
       253
       -
               multihash_code = getattr(cid_obj, "multihash", {}).get("code", None)

     

       254
       254
       -
               if multihash_code != 0x12:  # sha-256

     

       255
       255
       -
                   return False

     

       256
       256
       -
           except (AttributeError, KeyError):

     

       257
       257
       -
               # If we can't access multihash info, assume it's valid

     

       258
       258
       -
               # This is a fallback for CID libraries with different APIs

     

       259
       259
       -
               pass

     

       260
       260
       -
       

     

       261
       261
       -
           return True

     

       262
       262
       -
       

     

       263
       263
       -
       

     

       264
       264
       -
       def convertLegacyBlobToModern(legacy_blob: dict[str, Any]) -> dict[str, Any]:

     

       265
       265
       -
           """

     

       266
       266
       -
           Convert legacy blob format to modern blob format.

     

       267
       267
       -
       

     

       268
       268
       -
           Legacy format: {"cid": "string", "mimeType": "string"}

     

       269
       269
       -
           Modern format: {"$type": "blob", "ref": {"$link": "cid"}, "mimeType": "string", "size": int}

     

       270
       270
       -
       

     

       271
       271
       -
           Args:

     

       272
       272
       -
               legacy_blob: Legacy blob data

     

       273
       273
       -
       

     

       274
       274
       -
           Returns:

     

       275
       275
       -
               Modern blob format with size set to 0 (unknown)

     

       276
       276
       -
           """

     

       277
       277
       -
           return {

     

       278
       278
       -
               "$type": "blob",

     

       279
       279
       -
               "ref": {"$link": legacy_blob["cid"]},

     

       280
       280
       -
               "mimeType": legacy_blob["mimeType"],

     

       281
       281
       -
               "size": 0,  # Size unknown for legacy blobs

     

       282
       282
       -
           }

     

       283
       283
       -
       

     

       284
       284
       -
       

     

       285
       285
       -
       # Export commonly used classes and functions

     

       286
       33
        
       __all__ = [

     

       287
       34
        
           "DataModel",

     

       288
       35
        
           "BlobModel",

     

       289
       289
       -
           "LegacyBlobModel",

     

       290
       36
        
           "TypedDataModel",

     

       291
       37
        
           "validateAtprotoData",

     

       292
       38
        
           "isBlessedCID",

     

       293
       39
        
           "convertLegacyBlobToModern",

     

       40
       40
       +
           "NullModel",

     

       41
       41
       +
           "BooleanModel",

     

       42
       42
       +
           "IntegerModel",

     

       43
       43
       +
           "StringModel",

     

       44
       44
       +
           "ArrayModel",

     

       45
       45
       +
           "ObjectModel",

     

       46
       46
       +
           "ParamsModel",

     

       47
       47
       +
           "TokenModel",

     

       48
       48
       +
           "RefModel",

     

       49
       49
       +
           "UnionModel",

     

       50
       50
       +
           "UnknownModel",

     

       51
       51
       +
           "RecordModel",

     

       52
       52
       +
           "QueryModel",

     

       53
       53
       +
           "ProcedureModel",

     

       54
       54
       +
           "SubscriptionModel"

     

       294
       55
        
       ]

+74

src/atpasser/model/base.py

···

       1
       1
       +
       import base64

     

       2
       2
       +
       import re

     

       3
       3
       +
       from datetime import datetime

     

       4
       4
       +
       from typing import Any

     

       5
       5
       +
       from collections.abc import Mapping

     

       6
       6
       +
       from cid.cid import CIDv1, make_cid

     

       7
       7
       +
       from pydantic import BaseModel, field_serializer, field_validator, ConfigDict

     

       8
       8
       +
       from pydantic.fields import FieldInfo

     

       9
       9
       +
       

     

       10
       10
       +
       class DataModel(BaseModel):

     

       11
       11
       +
           """

     

       12
       12
       +
           Base model for AT Protocol data model with support for JSON encodings.

     

       13
       13
       +
       

     

       14
       14
       +
           This model provides automatic serialization and deserialization for AT Protocol

     

       15
       15
       +
           specific data types including bytes, CID links, and blob references.

     

       16
       16
       +
           """

     

       17
       17
       +
       

     

       18
       18
       +
           model_config = ConfigDict(arbitrary_types_allowed=True)

     

       19
       19
       +
       

     

       20
       20
       +
           @field_validator("*", mode="before")

     

       21
       21
       +
           def parseAtprotoData(cls, v: Any) -> Any:

     

       22
       22
       +
               """

     

       23
       23
       +
               Parse AT Protocol specific data types from JSON representation.

     

       24
       24
       +
       

     

       25
       25
       +
               Args:

     

       26
       26
       +
                   v: Input value to validate

     

       27
       27
       +
       

     

       28
       28
       +
               Returns:

     

       29
       29
       +
                   Appropriate Python objects for AT Protocol data types

     

       30
       30
       +
       

     

       31
       31
       +
               Raises:

     

       32
       32
       +
                   ValueError: If parsing fails

     

       33
       33
       +
               """

     

       34
       34
       +
               # Handle bytes objects

     

       35
       35
       +
               if isinstance(v, dict) and "$bytes" in v:

     

       36
       36
       +
                   value = v["$bytes"]

     

       37
       37
       +
                   try:

     

       38
       38
       +
                       return base64.b64decode(value)

     

       39
       39
       +
                   except ValueError as e:

     

       40
       40
       +
                       raise ValueError(f"Failed to decode base64 bytes: {e}")

     

       41
       41
       +
       

     

       42
       42
       +
               # Handle CID objects

     

       43
       43
       +
               elif isinstance(v, dict) and "$link" in v:

     

       44
       44
       +
                   value = v["$link"]

     

       45
       45
       +
                   try:

     

       46
       46
       +
                       return make_cid(value)

     

       47
       47
       +
                   except ValueError as e:

     

       48
       48
       +
                       raise ValueError(f"Failed to parse CID: {e}")

     

       49
       49
       +
       

     

       50
       50
       +
               return v

     

       51
       51
       +
       

     

       52
       52
       +
           @field_serializer("*")

     

       53
       53
       +
           def serializeAtprotoData(self, v: Any, _info: Any) -> Any:

     

       54
       54
       +
               """

     

       55
       55
       +
               Serialize AT Protocol specific data types to JSON representation.

     

       56
       56
       +
       

     

       57
       57
       +
               Args:

     

       58
       58
       +
                   v: Value to serialize

     

       59
       59
       +
                   _info: Field serialization info (not used in this implementation)

     

       60
       60
       +
       

     

       61
       61
       +
               Returns:

     

       62
       62
       +
                   Appropriate AT Protocol representation for bytes and CID fields

     

       63
       63
       +
               """

     

       64
       64
       +
               # Handle bytes fields

     

       65
       65
       +
               if isinstance(v, bytes):

     

       66
       66
       +
                   value = base64.b64encode(v).decode()

     

       67
       67
       +
                   return {"$bytes": value}

     

       68
       68
       +
       

     

       69
       69
       +
               # Handle CID fields

     

       70
       70
       +
               elif isinstance(v, CIDv1):

     

       71
       71
       +
                   value = str(v)

     

       72
       72
       +
                   return {"$link": value}

     

       73
       73
       +
       

     

       74
       74
       +
               return v

+58

src/atpasser/model/blob.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       from pydantic import field_validator, ConfigDict

     

       3
       3
       +
       from .base import DataModel

     

       4
       4
       +
       

     

       5
       5
       +
       class BlobModel(DataModel):

     

       6
       6
       +
           """

     

       7
       7
       +
           Model for AT Protocol blob references.

     

       8
       8
       +
       

     

       9
       9
       +
           Blobs are used for file references like images and include metadata

     

       10
       10
       +
           such as MIME type and size.

     

       11
       11
       +
           """

     

       12
       12
       +
       

     

       13
       13
       +
           ref: Any  # Accept both CIDv0 and CIDv1

     

       14
       14
       +
           """CID reference to blob with multicodec type 'raw'"""

     

       15
       15
       +
       

     

       16
       16
       +
           mimeType: str

     

       17
       17
       +
           """Content type of blob, 'application/octet-stream' if not known"""

     

       18
       18
       +
       

     

       19
       19
       +
           size: int

     

       20
       20
       +
           """Length of blob in bytes, must be positive and non-zero"""

     

       21
       21
       +
       

     

       22
       22
       +
           model_config = ConfigDict(arbitrary_types_allowed=True)

     

       23
       23
       +
       

     

       24
       24
       +
           @field_validator("size")

     

       25
       25
       +
           def validateSize(cls, v: int) -> int:

     

       26
       26
       +
               """

     

       27
       27
       +
               Validate that size is positive and non-zero.

     

       28
       28
       +
       

     

       29
       29
       +
               Args:

     

       30
       30
       +
                   v: Size value to validate

     

       31
       31
       +
       

     

       32
       32
       +
               Returns:

     

       33
       33
       +
                   Validated size

     

       34
       34
       +
       

     

       35
       35
       +
               Raises:

     

       36
       36
       +
                   ValueError: If size is not positive

     

       37
       37
       +
               """

     

       38
       38
       +
               if v <= 0:

     

       39
       39
       +
                   raise ValueError("Blob size must be positive and non-zero")

     

       40
       40
       +
               return v

     

       41
       41
       +
       

     

       42
       42
       +
           @field_validator("mimeType")

     

       43
       43
       +
           def validateMimeType(cls, v: str) -> str:

     

       44
       44
       +
               """

     

       45
       45
       +
               Validate that mimeType is not empty.

     

       46
       46
       +
       

     

       47
       47
       +
               Args:

     

       48
       48
       +
                   v: MIME type to validate

     

       49
       49
       +
       

     

       50
       50
       +
               Returns:

     

       51
       51
       +
                   Validated MIME type

     

       52
       52
       +
       

     

       53
       53
       +
               Raises:

     

       54
       54
       +
                   ValueError: If MIME type is empty

     

       55
       55
       +
               """

     

       56
       56
       +
               if not v:

     

       57
       57
       +
                   raise ValueError("MIME type cannot be empty")

     

       58
       58
       +
               return v

+41

src/atpasser/model/typed.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       from pydantic import field_serializer

     

       3
       3
       +
       from .base import DataModel

     

       4
       4
       +
       

     

       5
       5
       +
       class TypedDataModel(DataModel):

     

       6
       6
       +
           """

     

       7
       7
       +
           Model for AT Protocol data with type information.

     

       8
       8
       +
       

     

       9
       9
       +
           Includes support for $type field that specifies Lexicon schema.

     

       10
       10
       +
           """

     

       11
       11
       +
       

     

       12
       12
       +
           type: str | None = None

     

       13
       13
       +
           """Lexicon schema type identifier"""

     

       14
       14
       +
       

     

       15
       15
       +
           def __init__(self, **data: Any) -> None:

     

       16
       16
       +
               """

     

       17
       17
       +
               Initialize typed data model with automatic $type handling.

     

       18
       18
       +
       

     

       19
       19
       +
               Args:

     

       20
       20
       +
                   **data: Data including optional $type field

     

       21
       21
       +
               """

     

       22
       22
       +
               # Extract $type if present

     

       23
       23
       +
               dataType = data.pop("$type", None)

     

       24
       24
       +
               if dataType:

     

       25
       25
       +
                   data["type"] = dataType

     

       26
       26
       +
               super().__init__(**data)

     

       27
       27
       +
       

     

       28
       28
       +
           @field_serializer("type")

     

       29
       29
       +
           def serializeType(self, v: str | None) -> dict[str, str] | None:

     

       30
       30
       +
               """

     

       31
       31
       +
               Serialize type field to $type object.

     

       32
       32
       +
       

     

       33
       33
       +
               Args:

     

       34
       34
       +
                   v: Type value to serialize

     

       35
       35
       +
       

     

       36
       36
       +
               Returns:

     

       37
       37
       +
                   $type object if type is not None

     

       38
       38
       +
               """

     

       39
       39
       +
               if v is not None:

     

       40
       40
       +
                   return {"$type": v}

     

       41
       41
       +
               return None

+189

src/atpasser/model/types/complex.py

···

       1
       1
       +
       from typing import Any, Optional

     

       2
       2
       +
       from pydantic import field_validator

     

       3
       3
       +
       from ..base import DataModel

     

       4
       4
       +
       

     

       5
       5
       +
       class ArrayModel(DataModel):

     

       6
       6
       +
           """

     

       7
       7
       +
           Model for AT Protocol array type.

     

       8
       8
       +
           

     

       9
       9
       +
           Represents an array of elements with support for item schema definition,

     

       10
       10
       +
           minimum/maximum length constraints as specified in Lexicon.

     

       11
       11
       +
           """

     

       12
       12
       +
           

     

       13
       13
       +
           items: Any

     

       14
       14
       +
           """Schema definition for array elements"""

     

       15
       15
       +
           

     

       16
       16
       +
           minLength: int | None = None

     

       17
       17
       +
           """Minimum number of elements"""

     

       18
       18
       +
           

     

       19
       19
       +
           maxLength: int | None = None

     

       20
       20
       +
           """Maximum number of elements"""

     

       21
       21
       +
           

     

       22
       22
       +
           value: list[Any]

     

       23
       23
       +
           """Array values"""

     

       24
       24
       +
           

     

       25
       25
       +
           def __init__(self, **data: Any) -> None:

     

       26
       26
       +
               """

     

       27
       27
       +
               Initialize array model with validation.

     

       28
       28
       +
               

     

       29
       29
       +
               Args:

     

       30
       30
       +
                   **data: Input data containing array values

     

       31
       31
       +
                   

     

       32
       32
       +
               Raises:

     

       33
       33
       +
                   ValueError: If array violates constraints

     

       34
       34
       +
               """

     

       35
       35
       +
               super().__init__(**data)

     

       36
       36
       +
               

     

       37
       37
       +
           @field_validator("value", mode="before")

     

       38
       38
       +
           def validate_array(cls, v: Any) -> list[Any]:

     

       39
       39
       +
               """

     

       40
       40
       +
               Validate array structure and elements.

     

       41
       41
       +
               

     

       42
       42
       +
               Args:

     

       43
       43
       +
                   v: Value to validate

     

       44
       44
       +
                   

     

       45
       45
       +
               Returns:

     

       46
       46
       +
                   Validated array

     

       47
       47
       +
                   

     

       48
       48
       +
               Raises:

     

       49
       49
       +
                   ValueError: If array violates constraints

     

       50
       50
       +
               """

     

       51
       51
       +
               if not isinstance(v, list):

     

       52
       52
       +
                   raise ValueError("Value must be an array")

     

       53
       53
       +
                   

     

       54
       54
       +
               # Validate length constraints

     

       55
       55
       +
               if cls.minLength is not None and len(v) < cls.minLength:

     

       56
       56
       +
                   raise ValueError(f"Array must have at least {cls.minLength} items")

     

       57
       57
       +
                   

     

       58
       58
       +
               if cls.maxLength is not None and len(v) > cls.maxLength:

     

       59
       59
       +
                   raise ValueError(f"Array must have at most {cls.maxLength} items")

     

       60
       60
       +
                   

     

       61
       61
       +
               return v

     

       62
       62
       +
       

     

       63
       63
       +
       class ObjectModel(DataModel):

     

       64
       64
       +
           """

     

       65
       65
       +
           Model for AT Protocol object type.

     

       66
       66
       +
           

     

       67
       67
       +
           Represents a generic object schema with properties definitions,

     

       68
       68
       +
           required fields and nullable fields as specified in Lexicon.

     

       69
       69
       +
           """

     

       70
       70
       +
           

     

       71
       71
       +
           properties: dict[str, Any]

     

       72
       72
       +
           """Map of property names to their schema definitions"""

     

       73
       73
       +
           

     

       74
       74
       +
           required: list[str] | None = None

     

       75
       75
       +
           """List of required property names"""

     

       76
       76
       +
           

     

       77
       77
       +
           nullable: list[str] | None = None

     

       78
       78
       +
           """List of properties that can be null"""

     

       79
       79
       +
           

     

       80
       80
       +
           value: dict[str, Any]

     

       81
       81
       +
           """Object property values"""

     

       82
       82
       +
           

     

       83
       83
       +
           def __init__(self, **data: Any) -> None:

     

       84
       84
       +
               """

     

       85
       85
       +
               Initialize object model with validation.

     

       86
       86
       +
               

     

       87
       87
       +
               Args:

     

       88
       88
       +
                   **data: Input data containing object properties

     

       89
       89
       +
                   

     

       90
       90
       +
               Raises:

     

       91
       91
       +
                   ValueError: If object violates constraints

     

       92
       92
       +
               """

     

       93
       93
       +
               super().__init__(**data)

     

       94
       94
       +
               

     

       95
       95
       +
           @field_validator("value", mode="before")

     

       96
       96
       +
           def validate_object(cls, v: Any) -> dict[str, Any]:

     

       97
       97
       +
               """

     

       98
       98
       +
               Validate object structure and properties.

     

       99
       99
       +
               

     

       100
       100
       +
               Args:

     

       101
       101
       +
                   v: Value to validate

     

       102
       102
       +
                   

     

       103
       103
       +
               Returns:

     

       104
       104
       +
                   Validated object

     

       105
       105
       +
                   

     

       106
       106
       +
               Raises:

     

       107
       107
       +
                   ValueError: If object violates constraints

     

       108
       108
       +
               """

     

       109
       109
       +
               if not isinstance(v, dict):

     

       110
       110
       +
                   raise ValueError("Value must be an object")

     

       111
       111
       +
                   

     

       112
       112
       +
               # Validate required fields

     

       113
       113
       +
               if cls.required:

     

       114
       114
       +
                   for field in cls.required:

     

       115
       115
       +
                       if field not in v:

     

       116
       116
       +
                           raise ValueError(f"Missing required field: {field}")

     

       117
       117
       +
                           

     

       118
       118
       +
               # Validate nullable fields

     

       119
       119
       +
               if cls.nullable:

     

       120
       120
       +
                   for field, value in v.items():

     

       121
       121
       +
                       if field not in cls.nullable and value is None:

     

       122
       122
       +
                           raise ValueError(f"Field {field} cannot be null")

     

       123
       123
       +
                           

     

       124
       124
       +
               return v

     

       125
       125
       +
       

     

       126
       126
       +
       class ParamsModel(DataModel):

     

       127
       127
       +
           """

     

       128
       128
       +
           Model for AT Protocol params type.

     

       129
       129
       +
           

     

       130
       130
       +
           Specialized for HTTP query parameters with support for boolean,

     

       131
       131
       +
           integer, string and unknown types as specified in Lexicon.

     

       132
       132
       +
           """

     

       133
       133
       +
           

     

       134
       134
       +
           required: list[str] | None = None

     

       135
       135
       +
           """List of required parameter names"""

     

       136
       136
       +
           

     

       137
       137
       +
           properties: dict[str, Any]

     

       138
       138
       +
           """Map of parameter names to their schema definitions"""

     

       139
       139
       +
           

     

       140
       140
       +
           value: dict[str, Any]

     

       141
       141
       +
           """Parameter values"""

     

       142
       142
       +
           

     

       143
       143
       +
           def __init__(self, **data: Any) -> None:

     

       144
       144
       +
               """

     

       145
       145
       +
               Initialize params model with validation.

     

       146
       146
       +
               

     

       147
       147
       +
               Args:

     

       148
       148
       +
                   **data: Input data containing parameter values

     

       149
       149
       +
                   

     

       150
       150
       +
               Raises:

     

       151
       151
       +
                   ValueError: If parameters violate constraints

     

       152
       152
       +
               """

     

       153
       153
       +
               super().__init__(**data)

     

       154
       154
       +
               

     

       155
       155
       +
           @field_validator("value", mode="before")

     

       156
       156
       +
           def validate_params(cls, v: Any) -> dict[str, Any]:

     

       157
       157
       +
               """

     

       158
       158
       +
               Validate parameters structure and values.

     

       159
       159
       +
               

     

       160
       160
       +
               Args:

     

       161
       161
       +
                   v: Value to validate

     

       162
       162
       +
                   

     

       163
       163
       +
               Returns:

     

       164
       164
       +
                   Validated parameters

     

       165
       165
       +
                   

     

       166
       166
       +
               Raises:

     

       167
       167
       +
                   ValueError: If parameters violate constraints

     

       168
       168
       +
               """

     

       169
       169
       +
               if not isinstance(v, dict):

     

       170
       170
       +
                   raise ValueError("Value must be a dictionary of parameters")

     

       171
       171
       +
                   

     

       172
       172
       +
               # Validate required parameters

     

       173
       173
       +
               if cls.required:

     

       174
       174
       +
                   for param in cls.required:

     

       175
       175
       +
                       if param not in v:

     

       176
       176
       +
                           raise ValueError(f"Missing required parameter: {param}")

     

       177
       177
       +
                           

     

       178
       178
       +
               # Validate parameter types

     

       179
       179
       +
               for param, value in v.items():

     

       180
       180
       +
                   if param in cls.properties:

     

       181
       181
       +
                       prop_type = cls.properties[param].get("type")

     

       182
       182
       +
                       if prop_type == "boolean" and not isinstance(value, bool):

     

       183
       183
       +
                           raise ValueError(f"Parameter {param} must be boolean")

     

       184
       184
       +
                       elif prop_type == "integer" and not isinstance(value, int):

     

       185
       185
       +
                           raise ValueError(f"Parameter {param} must be integer")

     

       186
       186
       +
                       elif prop_type == "string" and not isinstance(value, str):

     

       187
       187
       +
                           raise ValueError(f"Parameter {param} must be string")

     

       188
       188
       +
                           

     

       189
       189
       +
               return v

+172

src/atpasser/model/types/primitive.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       from pydantic import field_validator

     

       3
       3
       +
       from ..base import DataModel

     

       4
       4
       +
       

     

       5
       5
       +
       class NullModel(DataModel):

     

       6
       6
       +
           """

     

       7
       7
       +
           Model for AT Protocol null type.

     

       8
       8
       +
           

     

       9
       9
       +
           Represents a null value in AT Protocol data model. This model ensures proper

     

       10
       10
       +
           serialization and validation of null values according to Lexicon specification.

     

       11
       11
       +
           """

     

       12
       12
       +
           

     

       13
       13
       +
           value: None = None

     

       14
       14
       +
           """Always None for null type"""

     

       15
       15
       +
           

     

       16
       16
       +
           def __init__(self, **data: Any) -> None:

     

       17
       17
       +
               """

     

       18
       18
       +
               Initialize null model with validation.

     

       19
       19
       +
               

     

       20
       20
       +
               Args:

     

       21
       21
       +
                   **data: Input data (must be empty or contain only None values)

     

       22
       22
       +
                   

     

       23
       23
       +
               Raises:

     

       24
       24
       +
                   ValueError: If non-null value is provided

     

       25
       25
       +
               """

     

       26
       26
       +
               if data and any(v is not None for v in data.values()):

     

       27
       27
       +
                   raise ValueError("NullModel only accepts None values")

     

       28
       28
       +
               super().__init__(**data)

     

       29
       29
       +
           

     

       30
       30
       +
           @field_validator("*", mode="before")

     

       31
       31
       +
           def validate_null(cls, v: Any) -> None:

     

       32
       32
       +
               """

     

       33
       33
       +
               Validate that value is null.

     

       34
       34
       +
               

     

       35
       35
       +
               Args:

     

       36
       36
       +
                   v: Value to validate

     

       37
       37
       +
                   

     

       38
       38
       +
               Returns:

     

       39
       39
       +
                   None if validation succeeds

     

       40
       40
       +
                   

     

       41
       41
       +
               Raises:

     

       42
       42
       +
                   ValueError: If value is not null

     

       43
       43
       +
               """

     

       44
       44
       +
               if v is not None:

     

       45
       45
       +
                   raise ValueError("NullModel only accepts None values")

     

       46
       46
       +
               return None

     

       47
       47
       +
       

     

       48
       48
       +
       class BooleanModel(DataModel):

     

       49
       49
       +
           """

     

       50
       50
       +
           Model for AT Protocol boolean type.

     

       51
       51
       +
           

     

       52
       52
       +
           Represents a boolean value in AT Protocol data model with support for

     

       53
       53
       +
           default values and constants as specified in Lexicon.

     

       54
       54
       +
           """

     

       55
       55
       +
           

     

       56
       56
       +
           value: bool

     

       57
       57
       +
           """Boolean value"""

     

       58
       58
       +
           

     

       59
       59
       +
           default: bool | None = None

     

       60
       60
       +
           """Default value if not provided"""

     

       61
       61
       +
           

     

       62
       62
       +
           const: bool | None = None

     

       63
       63
       +
           """Fixed constant value if specified"""

     

       64
       64
       +
           

     

       65
       65
       +
           def __init__(self, **data: Any) -> None:

     

       66
       66
       +
               """

     

       67
       67
       +
               Initialize boolean model with validation.

     

       68
       68
       +
               

     

       69
       69
       +
               Args:

     

       70
       70
       +
                   **data: Input data containing boolean value

     

       71
       71
       +
                   

     

       72
       72
       +
               Raises:

     

       73
       73
       +
                   ValueError: If value doesn't match const or is not boolean

     

       74
       74
       +
               """

     

       75
       75
       +
               super().__init__(**data)

     

       76
       76
       +
               if self.const is not None and self.value != self.const:

     

       77
       77
       +
                   raise ValueError(f"Boolean value must be {self.const}")

     

       78
       78
       +
           

     

       79
       79
       +
           @field_validator("value", mode="before")

     

       80
       80
       +
           def validate_boolean(cls, v: Any) -> bool:

     

       81
       81
       +
               """

     

       82
       82
       +
               Validate and convert input to boolean.

     

       83
       83
       +
               

     

       84
       84
       +
               Args:

     

       85
       85
       +
                   v: Value to validate

     

       86
       86
       +
                   

     

       87
       87
       +
               Returns:

     

       88
       88
       +
                   Validated boolean value

     

       89
       89
       +
                   

     

       90
       90
       +
               Raises:

     

       91
       91
       +
                   ValueError: If value cannot be converted to boolean

     

       92
       92
       +
               """

     

       93
       93
       +
               if isinstance(v, bool):

     

       94
       94
       +
                   return v

     

       95
       95
       +
               if isinstance(v, str):

     

       96
       96
       +
                   if v.lower() in ("true", "1"):

     

       97
       97
       +
                       return True

     

       98
       98
       +
                   if v.lower() in ("false", "0"):

     

       99
       99
       +
                       return False

     

       100
       100
       +
               raise ValueError("Value must be a boolean")

     

       101
       101
       +
       

     

       102
       102
       +
       class IntegerModel(DataModel):

     

       103
       103
       +
           """

     

       104
       104
       +
           Model for AT Protocol integer type.

     

       105
       105
       +
           

     

       106
       106
       +
           Represents a signed integer number with support for minimum/maximum values,

     

       107
       107
       +
           enumeration sets, default values and constraints as specified in Lexicon.

     

       108
       108
       +
           """

     

       109
       109
       +
           

     

       110
       110
       +
           value: int

     

       111
       111
       +
           """Integer value"""

     

       112
       112
       +
           

     

       113
       113
       +
           minimum: int | None = None

     

       114
       114
       +
           """Minimum acceptable value"""

     

       115
       115
       +
           

     

       116
       116
       +
           maximum: int | None = None

     

       117
       117
       +
           """Maximum acceptable value"""

     

       118
       118
       +
           

     

       119
       119
       +
           enum: list[int] | None = None

     

       120
       120
       +
           """Closed set of allowed values"""

     

       121
       121
       +
           

     

       122
       122
       +
           default: int | None = None

     

       123
       123
       +
           """Default value if not provided"""

     

       124
       124
       +
           

     

       125
       125
       +
           const: int | None = None

     

       126
       126
       +
           """Fixed constant value if specified"""

     

       127
       127
       +
           

     

       128
       128
       +
           def __init__(self, **data: Any) -> None:

     

       129
       129
       +
               """

     

       130
       130
       +
               Initialize integer model with validation.

     

       131
       131
       +
               

     

       132
       132
       +
               Args:

     

       133
       133
       +
                   **data: Input data containing integer value

     

       134
       134
       +
                   

     

       135
       135
       +
               Raises:

     

       136
       136
       +
                   ValueError: If value violates constraints

     

       137
       137
       +
               """

     

       138
       138
       +
               super().__init__(**data)

     

       139
       139
       +
               if self.const is not None and self.value != self.const:

     

       140
       140
       +
                   raise ValueError(f"Integer value must be {self.const}")

     

       141
       141
       +
           

     

       142
       142
       +
           @field_validator("value", mode="before")

     

       143
       143
       +
           def validate_integer(cls, v: Any) -> int:

     

       144
       144
       +
               """

     

       145
       145
       +
               Validate and convert input to integer.

     

       146
       146
       +
               

     

       147
       147
       +
               Args:

     

       148
       148
       +
                   v: Value to validate

     

       149
       149
       +
                   

     

       150
       150
       +
               Returns:

     

       151
       151
       +
                   Validated integer value

     

       152
       152
       +
                   

     

       153
       153
       +
               Raises:

     

       154
       154
       +
                   ValueError: If value violates constraints

     

       155
       155
       +
               """

     

       156
       156
       +
               if not isinstance(v, int):

     

       157
       157
       +
                   try:

     

       158
       158
       +
                       v = int(v)

     

       159
       159
       +
                   except (TypeError, ValueError):

     

       160
       160
       +
                       raise ValueError("Value must be an integer")

     

       161
       161
       +
               

     

       162
       162
       +
               # Validate against instance attributes

     

       163
       163
       +
               if cls.enum and v not in cls.enum:

     

       164
       164
       +
                   raise ValueError(f"Value must be one of {cls.enum}")

     

       165
       165
       +
                   

     

       166
       166
       +
               if cls.minimum is not None and v < cls.minimum:

     

       167
       167
       +
                   raise ValueError(f"Value must be >= {cls.minimum}")

     

       168
       168
       +
                   

     

       169
       169
       +
               if cls.maximum is not None and v > cls.maximum:

     

       170
       170
       +
                   raise ValueError(f"Value must be <= {cls.maximum}")

     

       171
       171
       +
                   

     

       172
       172
       +
               return v

+131

src/atpasser/model/types/reference.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       from pydantic import field_validator

     

       3
       3
       +
       from ..base import DataModel

     

       4
       4
       +
       

     

       5
       5
       +
       class TokenModel(DataModel):

     

       6
       6
       +
           """

     

       7
       7
       +
           Model for AT Protocol token type.

     

       8
       8
       +
           

     

       9
       9
       +
           Represents empty data values which exist only to be referenced by name.

     

       10
       10
       +
           Tokens encode as string data with the string being the fully-qualified

     

       11
       11
       +
           reference to the token itself (NSID followed by optional fragment).

     

       12
       12
       +
           """

     

       13
       13
       +
           

     

       14
       14
       +
           name: str

     

       15
       15
       +
           """Token name/identifier"""

     

       16
       16
       +
           

     

       17
       17
       +
           description: str | None = None

     

       18
       18
       +
           """Description clarifying the meaning of the token"""

     

       19
       19
       +
           

     

       20
       20
       +
           def __init__(self, **data: Any) -> None:

     

       21
       21
       +
               """

     

       22
       22
       +
               Initialize token model.

     

       23
       23
       +
               

     

       24
       24
       +
               Args:

     

       25
       25
       +
                   **data: Input data containing token name

     

       26
       26
       +
               """

     

       27
       27
       +
               super().__init__(**data)

     

       28
       28
       +
               

     

       29
       29
       +
           @field_validator("name")

     

       30
       30
       +
           def validate_name(cls, v: str) -> str:

     

       31
       31
       +
               """

     

       32
       32
       +
               Validate token name format.

     

       33
       33
       +
               

     

       34
       34
       +
               Args:

     

       35
       35
       +
                   v: Name to validate

     

       36
       36
       +
                   

     

       37
       37
       +
               Returns:

     

       38
       38
       +
                   Validated name

     

       39
       39
       +
                   

     

       40
       40
       +
               Raises:

     

       41
       41
       +
                   ValueError: If name contains whitespace

     

       42
       42
       +
               """

     

       43
       43
       +
               if any(c.isspace() for c in v):

     

       44
       44
       +
                   raise ValueError("Token name must not contain whitespace")

     

       45
       45
       +
               return v

     

       46
       46
       +
       

     

       47
       47
       +
       class RefModel(DataModel):

     

       48
       48
       +
           """

     

       49
       49
       +
           Model for AT Protocol ref type.

     

       50
       50
       +
           

     

       51
       51
       +
           Represents a reference to another schema definition, either globally

     

       52
       52
       +
           (using NSID) or locally (using #-delimited name).

     

       53
       53
       +
           """

     

       54
       54
       +
           

     

       55
       55
       +
           ref: str

     

       56
       56
       +
           """Reference to schema definition (NSID or #name)"""

     

       57
       57
       +
           

     

       58
       58
       +
           description: str | None = None

     

       59
       59
       +
           """Description of the reference"""

     

       60
       60
       +
           

     

       61
       61
       +
           def __init__(self, **data: Any) -> None:

     

       62
       62
       +
               """

     

       63
       63
       +
               Initialize reference model.

     

       64
       64
       +
               

     

       65
       65
       +
               Args:

     

       66
       66
       +
                   **data: Input data containing reference

     

       67
       67
       +
               """

     

       68
       68
       +
               super().__init__(**data)

     

       69
       69
       +
               

     

       70
       70
       +
           @field_validator("ref")

     

       71
       71
       +
           def validate_ref(cls, v: str) -> str:

     

       72
       72
       +
               """

     

       73
       73
       +
               Validate reference format.

     

       74
       74
       +
               

     

       75
       75
       +
               Args:

     

       76
       76
       +
                   v: Reference to validate

     

       77
       77
       +
                   

     

       78
       78
       +
               Returns:

     

       79
       79
       +
                   Validated reference

     

       80
       80
       +
                   

     

       81
       81
       +
               Raises:

     

       82
       82
       +
                   ValueError: If reference is empty or invalid

     

       83
       83
       +
               """

     

       84
       84
       +
               if not v:

     

       85
       85
       +
                   raise ValueError("Reference cannot be empty")

     

       86
       86
       +
               return v

     

       87
       87
       +
       

     

       88
       88
       +
       class UnionModel(DataModel):

     

       89
       89
       +
           """

     

       90
       90
       +
           Model for AT Protocol union type.

     

       91
       91
       +
           

     

       92
       92
       +
           Represents that multiple possible types could be present at a location.

     

       93
       93
       +
           The references follow the same syntax as `ref`, allowing references to

     

       94
       94
       +
           both global or local schema definitions.

     

       95
       95
       +
           """

     

       96
       96
       +
           

     

       97
       97
       +
           refs: list[str]

     

       98
       98
       +
           """References to schema definitions"""

     

       99
       99
       +
           

     

       100
       100
       +
           closed: bool = False

     

       101
       101
       +
           """Indicates if union is open (can be extended) or closed"""

     

       102
       102
       +
           

     

       103
       103
       +
           description: str | None = None

     

       104
       104
       +
           """Description of the union"""

     

       105
       105
       +
           

     

       106
       106
       +
           def __init__(self, **data: Any) -> None:

     

       107
       107
       +
               """

     

       108
       108
       +
               Initialize union model.

     

       109
       109
       +
               

     

       110
       110
       +
               Args:

     

       111
       111
       +
                   **data: Input data containing union references

     

       112
       112
       +
               """

     

       113
       113
       +
               super().__init__(**data)

     

       114
       114
       +
               

     

       115
       115
       +
           @field_validator("refs")

     

       116
       116
       +
           def validate_refs(cls, v: list[str]) -> list[str]:

     

       117
       117
       +
               """

     

       118
       118
       +
               Validate union references.

     

       119
       119
       +
               

     

       120
       120
       +
               Args:

     

       121
       121
       +
                   v: References to validate

     

       122
       122
       +
                   

     

       123
       123
       +
               Returns:

     

       124
       124
       +
                   Validated references

     

       125
       125
       +
                   

     

       126
       126
       +
               Raises:

     

       127
       127
       +
                   ValueError: If references list is empty for closed union

     

       128
       128
       +
               """

     

       129
       129
       +
               if cls.closed and not v:

     

       130
       130
       +
                   raise ValueError("Closed union must have at least one reference")

     

       131
       131
       +
               return v

+323

src/atpasser/model/types/special.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       from pydantic import field_validator

     

       3
       3
       +
       from ..base import DataModel

     

       4
       4
       +
       

     

       5
       5
       +
       class UnknownModel(DataModel):

     

       6
       6
       +
           """

     

       7
       7
       +
           Model for AT Protocol unknown type.

     

       8
       8
       +
           

     

       9
       9
       +
           Indicates that any data object could appear at this location,

     

       10
       10
       +
           with no specific validation. The top-level data must be an object.

     

       11
       11
       +
           """

     

       12
       12
       +
           

     

       13
       13
       +
           description: str | None = None

     

       14
       14
       +
           """Description of the unknown type usage"""

     

       15
       15
       +
           

     

       16
       16
       +
           def __init__(self, **data: Any) -> None:

     

       17
       17
       +
               """

     

       18
       18
       +
               Initialize unknown model.

     

       19
       19
       +
               

     

       20
       20
       +
               Args:

     

       21
       21
       +
                   **data: Input data containing unknown object

     

       22
       22
       +
               """

     

       23
       23
       +
               super().__init__(**data)

     

       24
       24
       +
               

     

       25
       25
       +
           @field_validator("*", mode="before")

     

       26
       26
       +
           def validate_unknown(cls, v: Any) -> Any:

     

       27
       27
       +
               """

     

       28
       28
       +
               Validate unknown data is an object.

     

       29
       29
       +
               

     

       30
       30
       +
               Args:

     

       31
       31
       +
                   v: Value to validate

     

       32
       32
       +
                   

     

       33
       33
       +
               Returns:

     

       34
       34
       +
                   Validated value

     

       35
       35
       +
                   

     

       36
       36
       +
               Raises:

     

       37
       37
       +
                   ValueError: If value is not an object

     

       38
       38
       +
               """

     

       39
       39
       +
               if not isinstance(v, dict):

     

       40
       40
       +
                   raise ValueError("Unknown type must be an object")

     

       41
       41
       +
               return v

     

       42
       42
       +
       

     

       43
       43
       +
       class RecordModel(DataModel):

     

       44
       44
       +
           """

     

       45
       45
       +
           Model for AT Protocol record type.

     

       46
       46
       +
           

     

       47
       47
       +
           Describes an object that can be stored in a repository record.

     

       48
       48
       +
           Records must include a $type field indicating their schema.

     

       49
       49
       +
           """

     

       50
       50
       +
           

     

       51
       51
       +
           key: str

     

       52
       52
       +
           """Specifies the Record Key type"""

     

       53
       53
       +
           

     

       54
       54
       +
           record: dict[str, Any]

     

       55
       55
       +
           """Schema definition with type 'object'"""

     

       56
       56
       +
           

     

       57
       57
       +
           type: str

     

       58
       58
       +
           """Lexicon schema type identifier"""

     

       59
       59
       +
           

     

       60
       60
       +
           def __init__(self, **data: Any) -> None:

     

       61
       61
       +
               """

     

       62
       62
       +
               Initialize record model with validation.

     

       63
       63
       +
               

     

       64
       64
       +
               Args:

     

       65
       65
       +
                   **data: Input data containing record values

     

       66
       66
       +
                   

     

       67
       67
       +
               Raises:

     

       68
       68
       +
                   ValueError: If record is missing required fields

     

       69
       69
       +
               """

     

       70
       70
       +
               # Extract $type if present

     

       71
       71
       +
               data_type = data.pop("$type", None)

     

       72
       72
       +
               if data_type:

     

       73
       73
       +
                   data["type"] = data_type

     

       74
       74
       +
               super().__init__(**data)

     

       75
       75
       +
               

     

       76
       76
       +
           @field_validator("type")

     

       77
       77
       +
           def validate_type(cls, v: str) -> str:

     

       78
       78
       +
               """

     

       79
       79
       +
               Validate record type field.

     

       80
       80
       +
               

     

       81
       81
       +
               Args:

     

       82
       82
       +
                   v: Type value to validate

     

       83
       83
       +
                   

     

       84
       84
       +
               Returns:

     

       85
       85
       +
                   Validated type

     

       86
       86
       +
                   

     

       87
       87
       +
               Raises:

     

       88
       88
       +
                   ValueError: If type is empty

     

       89
       89
       +
               """

     

       90
       90
       +
               if not v:

     

       91
       91
       +
                   raise ValueError("Record must have a type")

     

       92
       92
       +
               return v

     

       93
       93
       +
               

     

       94
       94
       +
           @field_validator("record", mode="before")

     

       95
       95
       +
           def validate_record(cls, v: Any) -> dict[str, Any]:

     

       96
       96
       +
               """

     

       97
       97
       +
               Validate record structure.

     

       98
       98
       +
               

     

       99
       99
       +
               Args:

     

       100
       100
       +
                   v: Record value to validate

     

       101
       101
       +
                   

     

       102
       102
       +
               Returns:

     

       103
       103
       +
                   Validated record

     

       104
       104
       +
                   

     

       105
       105
       +
               Raises:

     

       106
       106
       +
                   ValueError: If record is not an object

     

       107
       107
       +
               """

     

       108
       108
       +
               if not isinstance(v, dict):

     

       109
       109
       +
                   raise ValueError("Record must be an object")

     

       110
       110
       +
               return v

     

       111
       111
       +
       

     

       112
       112
       +
       class QueryModel(DataModel):

     

       113
       113
       +
           """

     

       114
       114
       +
           Model for AT Protocol query type.

     

       115
       115
       +
           

     

       116
       116
       +
           Describes an XRPC Query endpoint (HTTP GET) with support for

     

       117
       117
       +
           parameters, output schema and error responses.

     

       118
       118
       +
           """

     

       119
       119
       +
           

     

       120
       120
       +
           parameters: dict[str, Any] | None = None

     

       121
       121
       +
           """HTTP query parameters schema"""

     

       122
       122
       +
           

     

       123
       123
       +
           output: dict[str, Any] | None = None

     

       124
       124
       +
           """HTTP response body schema"""

     

       125
       125
       +
           

     

       126
       126
       +
           errors: list[dict[str, str]] | None = None

     

       127
       127
       +
           """Possible error responses"""

     

       128
       128
       +
           

     

       129
       129
       +
           def __init__(self, **data: Any) -> None:

     

       130
       130
       +
               """

     

       131
       131
       +
               Initialize query model with validation.

     

       132
       132
       +
               

     

       133
       133
       +
               Args:

     

       134
       134
       +
                   **data: Input data containing query definition

     

       135
       135
       +
               """

     

       136
       136
       +
               super().__init__(**data)

     

       137
       137
       +
               

     

       138
       138
       +
           @field_validator("output")

     

       139
       139
       +
           def validate_output(cls, v: dict[str, Any] | None) -> dict[str, Any] | None:

     

       140
       140
       +
               """

     

       141
       141
       +
               Validate output schema.

     

       142
       142
       +
               

     

       143
       143
       +
               Args:

     

       144
       144
       +
                   v: Output schema to validate

     

       145
       145
       +
                   

     

       146
       146
       +
               Returns:

     

       147
       147
       +
                   Validated output schema

     

       148
       148
       +
                   

     

       149
       149
       +
               Raises:

     

       150
       150
       +
                   ValueError: If output schema is invalid

     

       151
       151
       +
               """

     

       152
       152
       +
               if v and "encoding" not in v:

     

       153
       153
       +
                   raise ValueError("Output must specify encoding")

     

       154
       154
       +
               return v

     

       155
       155
       +
               

     

       156
       156
       +
           @field_validator("errors")

     

       157
       157
       +
           def validate_errors(cls, v: list[dict[str, str]] | None) -> list[dict[str, str]] | None:

     

       158
       158
       +
               """

     

       159
       159
       +
               Validate error definitions.

     

       160
       160
       +
               

     

       161
       161
       +
               Args:

     

       162
       162
       +
                   v: Error definitions to validate

     

       163
       163
       +
                   

     

       164
       164
       +
               Returns:

     

       165
       165
       +
                   Validated error definitions

     

       166
       166
       +
                   

     

       167
       167
       +
               Raises:

     

       168
       168
       +
                   ValueError: If any error definition is invalid

     

       169
       169
       +
               """

     

       170
       170
       +
               if v:

     

       171
       171
       +
                   for error in v:

     

       172
       172
       +
                       if "name" not in error:

     

       173
       173
       +
                           raise ValueError("Error must have a name")

     

       174
       174
       +
               return v

     

       175
       175
       +
       

     

       176
       176
       +
       class ProcedureModel(DataModel):

     

       177
       177
       +
           """

     

       178
       178
       +
           Model for AT Protocol procedure type.

     

       179
       179
       +
           

     

       180
       180
       +
           Describes an XRPC Procedure endpoint (HTTP POST) with support for

     

       181
       181
       +
           parameters, input/output schemas and error responses.

     

       182
       182
       +
           """

     

       183
       183
       +
           

     

       184
       184
       +
           parameters: dict[str, Any] | None = None

     

       185
       185
       +
           """HTTP query parameters schema"""

     

       186
       186
       +
           

     

       187
       187
       +
           input: dict[str, Any] | None = None

     

       188
       188
       +
           """HTTP request body schema"""

     

       189
       189
       +
           

     

       190
       190
       +
           output: dict[str, Any] | None = None

     

       191
       191
       +
           """HTTP response body schema"""

     

       192
       192
       +
           

     

       193
       193
       +
           errors: list[dict[str, str]] | None = None

     

       194
       194
       +
           """Possible error responses"""

     

       195
       195
       +
           

     

       196
       196
       +
           def __init__(self, **data: Any) -> None:

     

       197
       197
       +
               """

     

       198
       198
       +
               Initialize procedure model with validation.

     

       199
       199
       +
               

     

       200
       200
       +
               Args:

     

       201
       201
       +
                   **data: Input data containing procedure definition

     

       202
       202
       +
               """

     

       203
       203
       +
               super().__init__(**data)

     

       204
       204
       +
               

     

       205
       205
       +
           @field_validator("input")

     

       206
       206
       +
           def validate_input(cls, v: dict[str, Any] | None) -> dict[str, Any] | None:

     

       207
       207
       +
               """

     

       208
       208
       +
               Validate input schema.

     

       209
       209
       +
               

     

       210
       210
       +
               Args:

     

       211
       211
       +
                   v: Input schema to validate

     

       212
       212
       +
                   

     

       213
       213
       +
               Returns:

     

       214
       214
       +
                   Validated input schema

     

       215
       215
       +
                   

     

       216
       216
       +
               Raises:

     

       217
       217
       +
                   ValueError: If input schema is invalid

     

       218
       218
       +
               """

     

       219
       219
       +
               if v and "encoding" not in v:

     

       220
       220
       +
                   raise ValueError("Input must specify encoding")

     

       221
       221
       +
               return v

     

       222
       222
       +
               

     

       223
       223
       +
           @field_validator("output")

     

       224
       224
       +
           def validate_output(cls, v: dict[str, Any] | None) -> dict[str, Any] | None:

     

       225
       225
       +
               """

     

       226
       226
       +
               Validate output schema.

     

       227
       227
       +
               

     

       228
       228
       +
               Args:

     

       229
       229
       +
                   v: Output schema to validate

     

       230
       230
       +
                   

     

       231
       231
       +
               Returns:

     

       232
       232
       +
                   Validated output schema

     

       233
       233
       +
                   

     

       234
       234
       +
               Raises:

     

       235
       235
       +
                   ValueError: If output schema is invalid

     

       236
       236
       +
               """

     

       237
       237
       +
               if v and "encoding" not in v:

     

       238
       238
       +
                   raise ValueError("Output must specify encoding")

     

       239
       239
       +
               return v

     

       240
       240
       +
               

     

       241
       241
       +
           @field_validator("errors")

     

       242
       242
       +
           def validate_errors(cls, v: list[dict[str, str]] | None) -> list[dict[str, str]] | None:

     

       243
       243
       +
               """

     

       244
       244
       +
               Validate error definitions.

     

       245
       245
       +
               

     

       246
       246
       +
               Args:

     

       247
       247
       +
                   v: Error definitions to validate

     

       248
       248
       +
                   

     

       249
       249
       +
               Returns:

     

       250
       250
       +
                   Validated error definitions

     

       251
       251
       +
                   

     

       252
       252
       +
               Raises:

     

       253
       253
       +
                   ValueError: If any error definition is invalid

     

       254
       254
       +
               """

     

       255
       255
       +
               if v:

     

       256
       256
       +
                   for error in v:

     

       257
       257
       +
                       if "name" not in error:

     

       258
       258
       +
                           raise ValueError("Error must have a name")

     

       259
       259
       +
               return v

     

       260
       260
       +
       

     

       261
       261
       +
       class SubscriptionModel(DataModel):

     

       262
       262
       +
           """

     

       263
       263
       +
           Model for AT Protocol subscription type.

     

       264
       264
       +
           

     

       265
       265
       +
           Describes an Event Stream (WebSocket) with support for parameters,

     

       266
       266
       +
           message schemas and error responses.

     

       267
       267
       +
           """

     

       268
       268
       +
           

     

       269
       269
       +
           parameters: dict[str, Any] | None = None

     

       270
       270
       +
           """HTTP query parameters schema"""

     

       271
       271
       +
           

     

       272
       272
       +
           message: dict[str, Any] | None = None

     

       273
       273
       +
           """Specifies what messages can be"""

     

       274
       274
       +
           

     

       275
       275
       +
           errors: list[dict[str, str]] | None = None

     

       276
       276
       +
           """Possible error responses"""

     

       277
       277
       +
           

     

       278
       278
       +
           def __init__(self, **data: Any) -> None:

     

       279
       279
       +
               """

     

       280
       280
       +
               Initialize subscription model with validation.

     

       281
       281
       +
               

     

       282
       282
       +
               Args:

     

       283
       283
       +
                   **data: Input data containing subscription definition

     

       284
       284
       +
               """

     

       285
       285
       +
               super().__init__(**data)

     

       286
       286
       +
               

     

       287
       287
       +
           @field_validator("message")

     

       288
       288
       +
           def validate_message(cls, v: dict[str, Any] | None) -> dict[str, Any] | None:

     

       289
       289
       +
               """

     

       290
       290
       +
               Validate message schema.

     

       291
       291
       +
               

     

       292
       292
       +
               Args:

     

       293
       293
       +
                   v: Message schema to validate

     

       294
       294
       +
                   

     

       295
       295
       +
               Returns:

     

       296
       296
       +
                   Validated message schema

     

       297
       297
       +
                   

     

       298
       298
       +
               Raises:

     

       299
       299
       +
                   ValueError: If message schema is invalid

     

       300
       300
       +
               """

     

       301
       301
       +
               if v and "schema" not in v:

     

       302
       302
       +
                   raise ValueError("Message must specify schema")

     

       303
       303
       +
               return v

     

       304
       304
       +
               

     

       305
       305
       +
           @field_validator("errors")

     

       306
       306
       +
           def validate_errors(cls, v: list[dict[str, str]] | None) -> list[dict[str, str]] | None:

     

       307
       307
       +
               """

     

       308
       308
       +
               Validate error definitions.

     

       309
       309
       +
               

     

       310
       310
       +
               Args:

     

       311
       311
       +
                   v: Error definitions to validate

     

       312
       312
       +
                   

     

       313
       313
       +
               Returns:

     

       314
       314
       +
                   Validated error definitions

     

       315
       315
       +
                   

     

       316
       316
       +
               Raises:

     

       317
       317
       +
                   ValueError: If any error definition is invalid

     

       318
       318
       +
               """

     

       319
       319
       +
               if v:

     

       320
       320
       +
                   for error in v:

     

       321
       321
       +
                       if "name" not in error:

     

       322
       322
       +
                           raise ValueError("Error must have a name")

     

       323
       323
       +
               return v

+124

src/atpasser/model/types/string.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       import re

     

       3
       3
       +
       from datetime import datetime

     

       4
       4
       +
       from pydantic import field_validator

     

       5
       5
       +
       from ..base import DataModel

     

       6
       6
       +
       

     

       7
       7
       +
       class StringModel(DataModel):

     

       8
       8
       +
           """

     

       9
       9
       +
           Model for AT Protocol string type.

     

       10
       10
       +
           

     

       11
       11
       +
           Represents a Unicode string with support for format restrictions, length limits,

     

       12
       12
       +
           known values, enumeration sets, default values and constants as specified in Lexicon.

     

       13
       13
       +
           """

     

       14
       14
       +
           

     

       15
       15
       +
           value: str

     

       16
       16
       +
           """String value"""

     

       17
       17
       +
           

     

       18
       18
       +
           format: str | None = None

     

       19
       19
       +
           """String format restriction (e.g. 'datetime', 'uri')"""

     

       20
       20
       +
           

     

       21
       21
       +
           maxLength: int | None = None

     

       22
       22
       +
           """Maximum length in UTF-8 bytes"""

     

       23
       23
       +
           

     

       24
       24
       +
           minLength: int | None = None

     

       25
       25
       +
           """Minimum length in UTF-8 bytes"""

     

       26
       26
       +
           

     

       27
       27
       +
           knownValues: list[str] | None = None

     

       28
       28
       +
           """Suggested/common values (not enforced)"""

     

       29
       29
       +
           

     

       30
       30
       +
           enum: list[str] | None = None

     

       31
       31
       +
           """Closed set of allowed values"""

     

       32
       32
       +
           

     

       33
       33
       +
           default: str | None = None

     

       34
       34
       +
           """Default value if not provided"""

     

       35
       35
       +
           

     

       36
       36
       +
           const: str | None = None

     

       37
       37
       +
           """Fixed constant value if specified"""

     

       38
       38
       +
           

     

       39
       39
       +
           def __init__(self, **data: Any) -> None:

     

       40
       40
       +
               """

     

       41
       41
       +
               Initialize string model with validation.

     

       42
       42
       +
               

     

       43
       43
       +
               Args:

     

       44
       44
       +
                   **data: Input data containing string value

     

       45
       45
       +
                   

     

       46
       46
       +
               Raises:

     

       47
       47
       +
                   ValueError: If value violates constraints

     

       48
       48
       +
               """

     

       49
       49
       +
               super().__init__(**data)

     

       50
       50
       +
               if self.const is not None and self.value != self.const:

     

       51
       51
       +
                   raise ValueError(f"String value must be {self.const}")

     

       52
       52
       +
           

     

       53
       53
       +
           @field_validator("value", mode="before")

     

       54
       54
       +
           def validate_string(cls, v: Any) -> str:

     

       55
       55
       +
               """

     

       56
       56
       +
               Validate and convert input to string.

     

       57
       57
       +
               

     

       58
       58
       +
               Args:

     

       59
       59
       +
                   v: Value to validate

     

       60
       60
       +
                   

     

       61
       61
       +
               Returns:

     

       62
       62
       +
                   Validated string value

     

       63
       63
       +
                   

     

       64
       64
       +
               Raises:

     

       65
       65
       +
                   ValueError: If value violates constraints

     

       66
       66
       +
               """

     

       67
       67
       +
               if not isinstance(v, str):

     

       68
       68
       +
                   v = str(v)

     

       69
       69
       +
                   

     

       70
       70
       +
               # Validate length constraints

     

       71
       71
       +
               if cls.minLength is not None and len(v.encode()) < cls.minLength:

     

       72
       72
       +
                   raise ValueError(f"String must be at least {cls.minLength} bytes")

     

       73
       73
       +
                   

     

       74
       74
       +
               if cls.maxLength is not None and len(v.encode()) > cls.maxLength:

     

       75
       75
       +
                   raise ValueError(f"String must be at most {cls.maxLength} bytes")

     

       76
       76
       +
                   

     

       77
       77
       +
               # Validate enum

     

       78
       78
       +
               if cls.enum and v not in cls.enum:

     

       79
       79
       +
                   raise ValueError(f"Value must be one of {cls.enum}")

     

       80
       80
       +
                   

     

       81
       81
       +
               # Validate format if specified

     

       82
       82
       +
               if cls.format:

     

       83
       83
       +
                   if cls.format == "datetime":

     

       84
       84
       +
                       cls._validate_datetime(v)

     

       85
       85
       +
                   elif cls.format == "uri":

     

       86
       86
       +
                       cls._validate_uri(v)

     

       87
       87
       +
                   elif cls.format == "did":

     

       88
       88
       +
                       cls._validate_did(v)

     

       89
       89
       +
                   elif cls.format == "handle":

     

       90
       90
       +
                       cls._validate_handle(v)

     

       91
       91
       +
                   

     

       92
       92
       +
               return v

     

       93
       93
       +
           

     

       94
       94
       +
           @classmethod

     

       95
       95
       +
           def _validate_datetime(cls, v: str) -> None:

     

       96
       96
       +
               """Validate RFC 3339 datetime format"""

     

       97
       97
       +
               try:

     

       98
       98
       +
                   datetime.fromisoformat(v.replace("Z", "+00:00"))

     

       99
       99
       +
               except ValueError:

     

       100
       100
       +
                   raise ValueError("Invalid datetime format, must be RFC 3339")

     

       101
       101
       +
           

     

       102
       102
       +
           @classmethod

     

       103
       103
       +
           def _validate_uri(cls, v: str) -> None:

     

       104
       104
       +
               """Validate URI format"""

     

       105
       105
       +
               if len(v) > 8192:  # 8KB max

     

       106
       106
       +
                   raise ValueError("URI too long, max 8KB")

     

       107
       107
       +
               if not re.match(r"^[a-zA-Z][a-zA-Z0-9+.-]*:.+", v):

     

       108
       108
       +
                   raise ValueError("Invalid URI format")

     

       109
       109
       +
           

     

       110
       110
       +
           @classmethod

     

       111
       111
       +
           def _validate_did(cls, v: str) -> None:

     

       112
       112
       +
               """Validate DID format"""

     

       113
       113
       +
               if not v.startswith("did:"):

     

       114
       114
       +
                   raise ValueError("DID must start with 'did:'")

     

       115
       115
       +
               if len(v) > 2048:

     

       116
       116
       +
                   raise ValueError("DID too long, max 2048 chars")

     

       117
       117
       +
           

     

       118
       118
       +
           @classmethod

     

       119
       119
       +
           def _validate_handle(cls, v: str) -> None:

     

       120
       120
       +
               """Validate handle format"""

     

       121
       121
       +
               if not re.match(r"^[a-zA-Z0-9._-]+$", v):

     

       122
       122
       +
                   raise ValueError("Handle contains invalid characters")

     

       123
       123
       +
               if len(v) > 253:

     

       124
       124
       +
                   raise ValueError("Handle too long, max 253 chars")

+110

src/atpasser/model/validation.py

···

       1
       1
       +
       from typing import Any

     

       2
       2
       +
       from cid.cid import CIDv1, make_cid

     

       3
       3
       +
       import base64

     

       4
       4
       +
       import re

     

       5
       5
       +
       from datetime import datetime

     

       6
       6
       +
       

     

       7
       7
       +
       def validateAtprotoData(data: Any) -> bool:

     

       8
       8
       +
           """

     

       9
       9
       +
           Validate that data conforms to AT Protocol data model.

     

       10
       10
       +
       

     

       11
       11
       +
           This function checks for basic structural validity of AT Protocol data

     

       12
       12
       +
           including proper encoding of bytes, links, and blob references.

     

       13
       13
       +
       

     

       14
       14
       +
           Args:

     

       15
       15
       +
               data: Data to validate

     

       16
       16
       +
       

     

       17
       17
       +
           Returns:

     

       18
       18
       +
               True if data is valid AT Protocol data

     

       19
       19
       +
           """

     

       20
       20
       +
           if isinstance(data, dict):

     

       21
       21
       +
               # Check for reserved $ fields

     

       22
       22
       +
               for key in data:

     

       23
       23
       +
                   if key.startswith("$") and key not in {"$bytes", "$link", "$type"}:

     

       24
       24
       +
                       return False

     

       25
       25
       +
       

     

       26
       26
       +
               # Validate $bytes objects

     

       27
       27
       +
               if "$bytes" in data:

     

       28
       28
       +
                   if not isinstance(data["$bytes"], str):

     

       29
       29
       +
                       return False

     

       30
       30
       +
                   try:

     

       31
       31
       +
                       base64.b64decode(data["$bytes"])

     

       32
       32
       +
                   except ValueError:

     

       33
       33
       +
                       return False

     

       34
       34
       +
       

     

       35
       35
       +
               # Validate $link objects

     

       36
       36
       +
               if "$link" in data:

     

       37
       37
       +
                   if not isinstance(data["$link"], str):

     

       38
       38
       +
                       return False

     

       39
       39
       +
                   try:

     

       40
       40
       +
                       make_cid(data["$link"])

     

       41
       41
       +
                   except ValueError:

     

       42
       42
       +
                       return False

     

       43
       43
       +
       

     

       44
       44
       +
               # Recursively validate nested data

     

       45
       45
       +
               for value in data.values():

     

       46
       46
       +
                   if not validateAtprotoData(value):

     

       47
       47
       +
                       return False

     

       48
       48
       +
       

     

       49
       49
       +
           elif isinstance(data, list):

     

       50
       50
       +
               for item in data:

     

       51
       51
       +
                   if not validateAtprotoData(item):

     

       52
       52
       +
                       return False

     

       53
       53
       +
       

     

       54
       54
       +
           return True

     

       55
       55
       +
       

     

       56
       56
       +
       def isBlessedCID(cid_obj: CIDv1) -> bool:

     

       57
       57
       +
           """

     

       58
       58
       +
           Check if CID conforms to AT Protocol blessed formats.

     

       59
       59
       +
       

     

       60
       60
       +
           Blessed formats for CIDs in atproto are:

     

       61
       61
       +
           - CIDv1

     

       62
       62
       +
           - multibase: base32 for string encoding

     

       63
       63
       +
           - multicodec: dag-cbor (0x71) for data objects, raw (0x55) for blobs

     

       64
       64
       +
           - multihash: sha-256 with 256 bits (0x12) is preferred

     

       65
       65
       +
       

     

       66
       66
       +
           Args:

     

       67
       67
       +
               cid_obj: CID to validate

     

       68
       68
       +
       

     

       69
       69
       +
           Returns:

     

       70
       70
       +
               True if CID uses blessed format

     

       71
       71
       +
           """

     

       72
       72
       +
           # Check version

     

       73
       73
       +
           if cid_obj.version != 1:

     

       74
       74
       +
               return False

     

       75
       75
       +
       

     

       76
       76
       +
           # Check multicodec (should be dag-cbor or raw)

     

       77
       77
       +
           if cid_obj.codec not in {"dag-cbor", "raw"}:

     

       78
       78
       +
               return False

     

       79
       79
       +
       

     

       80
       80
       +
           # Check multihash (should be sha-256)

     

       81
       81
       +
           try:

     

       82
       82
       +
               # Try to get multihash code using library-specific method

     

       83
       83
       +
               multihash_code = getattr(cid_obj, "multihash", {}).get("code", None)

     

       84
       84
       +
               if multihash_code != 0x12:  # sha-256

     

       85
       85
       +
                   return False

     

       86
       86
       +
           except (AttributeError, KeyError):

     

       87
       87
       +
               # If we can't access multihash info, assume it's valid

     

       88
       88
       +
               pass

     

       89
       89
       +
       

     

       90
       90
       +
           return True

     

       91
       91
       +
       

     

       92
       92
       +
       def convertLegacyBlobToModern(legacy_blob: dict[str, Any]) -> dict[str, Any]:

     

       93
       93
       +
           """

     

       94
       94
       +
           Convert legacy blob format to modern blob format.

     

       95
       95
       +
       

     

       96
       96
       +
           Legacy format: {"cid": "string", "mimeType": "string"}

     

       97
       97
       +
           Modern format: {"$type": "blob", "ref": {"$link": "cid"}, "mimeType": "string", "size": int}

     

       98
       98
       +
       

     

       99
       99
       +
           Args:

     

       100
       100
       +
               legacy_blob: Legacy blob data

     

       101
       101
       +
       

     

       102
       102
       +
           Returns:

     

       103
       103
       +
               Modern blob format with size set to 0 (unknown)

     

       104
       104
       +
           """

     

       105
       105
       +
           return {

     

       106
       106
       +
               "$type": "blob",

     

       107
       107
       +
               "ref": {"$link": legacy_blob["cid"]},

     

       108
       108
       +
               "mimeType": legacy_blob["mimeType"],

     

       109
       109
       +
               "size": 0,  # Size unknown for legacy blobs

     

       110
       110
       +
           }