From a16243cb511ccd3210bf2d42fc0c3a98293d4e21 Mon Sep 17 00:00:00 2001
From: Mia <mia@parakeet.at>
Date: Mon, 20 Jan 2025 21:20:24 +0000
Subject: [PATCH] lexgen lexicon types

---
 parakeet-lexgen/src/main.rs  |   2 +
 parakeet-lexgen/src/types.rs | 225 +++++++++++++++++++++++++++++++++++
 2 files changed, 227 insertions(+)
 create mode 100644 parakeet-lexgen/src/types.rs

diff --git a/parakeet-lexgen/src/main.rs b/parakeet-lexgen/src/main.rs
index 4e74456d..9332860f 100644
--- a/parakeet-lexgen/src/main.rs
+++ b/parakeet-lexgen/src/main.rs
@@ -1,6 +1,8 @@
 use clap::Parser;
 use walkdir::WalkDir;
 
+mod types;
+
 #[derive(Debug, Parser)]
 struct Cli {
     sources: Vec<String>,
diff --git a/parakeet-lexgen/src/types.rs b/parakeet-lexgen/src/types.rs
new file mode 100644
index 00000000..255906fd
--- /dev/null
+++ b/parakeet-lexgen/src/types.rs
@@ -0,0 +1,225 @@
+use serde::Deserialize;
+use std::collections::BTreeMap;
+
+#[derive(Debug, Deserialize)]
+pub struct Lexicon {
+    /// Lexicon language version
+    pub lexicon: i32,
+    // Lexicon NSID
+    pub id: String,
+    pub revision: Option<i32>,
+    pub description: Option<String>,
+    pub defs: BTreeMap<String, LexiconDef>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(tag = "type")]
+#[serde(rename_all = "lowercase")]
+pub enum LexiconDef {
+    Query(LexiconQuery),
+    Procedure(LexiconProcedure),
+    Subscription(LexiconSubscription),
+    Record(LexiconRecord),
+    String(LexiconString),
+    Array(LexiconArray),
+    Object(LexiconObject),
+    Token { description: Option<String> },
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(tag = "type")]
+#[serde(rename_all = "kebab-case")]
+pub enum LexiconSchema {
+    Null {
+        description: Option<String>,
+    },
+    Boolean {
+        description: Option<String>,
+        default: Option<bool>,
+        #[serde(rename = "const")]
+        constant: Option<bool>,
+    },
+    Integer(LexiconInteger),
+    String(LexiconString),
+    Bytes {
+        description: Option<String>,
+        #[serde(rename = "maxLength")]
+        max_length: Option<i32>,
+        #[serde(rename = "minLength")]
+        min_length: Option<i32>,
+    },
+    CidLink {
+        description: Option<String>,
+    },
+    Array(LexiconArray),
+    Object(LexiconObject),
+    Blob {
+        description: Option<String>,
+        accept: Option<Vec<String>>,
+        #[serde(rename = "maxSize")]
+        max_size: Option<i32>,
+    },
+    Ref {
+        description: Option<String>,
+        #[serde(rename = "ref")]
+        ref_to: String,
+    },
+    Union(LexiconUnion),
+    Unknown {
+        description: Option<String>,
+    },
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "lowercase")]
+pub enum RKey {
+    Tid,
+    Nsid,
+    Any,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(untagged)]
+pub enum RecordKey {
+    Typed(RKey),
+    Other(String),
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "kebab-case")]
+pub enum StringFormats {
+    AtIdentifier,
+    AtUri,
+    Cid,
+    Datetime,
+    Did,
+    Handle,
+    Nsid,
+    Tid,
+    RecordKey,
+    Uri,
+    Language,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct HttpApiExchange {
+    pub description: Option<String>,
+    pub encoding: String,
+    pub schema: Option<LexiconSchema>,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct LexiconError {
+    pub name: String,
+    pub description: Option<String>,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct LexiconSubMessage {
+    pub description: Option<String>,
+    pub schema: Option<LexiconUnion>,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct LexiconRecord {
+    pub description: Option<String>,
+    pub key: RecordKey,
+    pub record: LexiconObject,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct LexiconQuery {
+    pub description: Option<String>,
+    pub parameters: Option<LexiconParams>,
+    pub output: Option<HttpApiExchange>,
+    pub errors: Option<Vec<LexiconError>>,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct LexiconProcedure {
+    pub description: Option<String>,
+    pub parameters: Option<LexiconParams>,
+    pub output: Option<HttpApiExchange>,
+    pub input: Option<HttpApiExchange>,
+    pub errors: Option<Vec<LexiconError>>,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct LexiconSubscription {
+    pub description: Option<String>,
+    pub parameters: Option<LexiconParams>,
+    pub message: BTreeMap<String, LexiconSubMessage>,
+    pub errors: Option<Vec<LexiconError>>,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct LexiconInteger {
+    pub description: Option<String>,
+
+    pub minimum: Option<i64>,
+    pub maximum: Option<i64>,
+    #[serde(rename = "enum")]
+    pub values: Option<Vec<i64>>,
+    pub default: Option<i64>,
+    #[serde(rename = "const")]
+    pub constant: Option<i64>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct LexiconString {
+    pub description: Option<String>,
+
+    pub format: Option<StringFormats>,
+
+    pub max_length: Option<i32>,
+    pub min_length: Option<i32>,
+    pub max_graphemes: Option<i32>,
+    pub min_graphemes: Option<i32>,
+
+    pub known_values: Option<Vec<String>>,
+    #[serde(rename = "enum")]
+    pub values: Option<Vec<String>>,
+
+    pub default: Option<String>,
+    #[serde(rename = "const")]
+    pub constant: Option<String>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct LexiconArray {
+    pub description: Option<String>,
+
+    pub min_length: Option<i32>,
+    pub max_length: Option<i32>,
+    pub items: Box<LexiconSchema>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(tag = "type")]
+#[serde(rename = "object")]
+pub struct LexiconObject {
+    pub description: Option<String>,
+    pub properties: BTreeMap<String, LexiconSchema>,
+    pub required: Option<Vec<String>>,
+    pub nullable: Option<Vec<String>>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(tag = "type")]
+#[serde(rename = "params")]
+pub struct LexiconParams {
+    pub description: Option<String>,
+    pub properties: BTreeMap<String, LexiconSchema>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(tag = "type")]
+#[serde(rename = "union")]
+pub struct LexiconUnion {
+    pub description: Option<String>,
+    pub refs: Vec<String>,
+    #[serde(default)]
+    pub closed: bool,
+}
-- 
2.51.2