Skip to main content

server/
server.rs

1//! Example server: ensures _sys_* tables exist, then loads config from PACKAGE_PATH (package directory with manifest.json) or from DB (config APIs).
2//! If PACKAGE_PATH is set, config is loaded from that directory (must contain manifest.json + config JSONs) and migrations applied; otherwise config is loaded from _sys_* tables (empty until fed via config APIs or package install).
3
4use architect_sdk::{
5    apply_migrations, common_routes_with_ready, config_routes, create_pool, ensure_database_exists,
6    ensure_sys_tables, entity_routes, load_from_pool, load_registry_from_pool, resolve, AppState,
7    FullConfig, DEFAULT_PACKAGE_ID,
8};
9use axum::Router;
10use std::collections::HashMap;
11use std::path::{Path, PathBuf};
12use std::sync::{Arc, RwLock};
13use tokio::net::TcpListener;
14use tracing_subscriber::EnvFilter;
15
16#[tokio::main]
17async fn main() -> Result<(), Box<dyn std::error::Error>> {
18    dotenvy::dotenv().ok();
19    let filter =
20        EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new("architect_sdk=info"));
21    tracing_subscriber::fmt().with_env_filter(filter).init();
22
23    let database_url =
24        std::env::var("DATABASE_URL").unwrap_or_else(|_| "sqlite://architect.db".into());
25    ensure_database_exists(&database_url).await?;
26
27    // create_pool uses the compiled-in dialect automatically (sqlite by default).
28    let pool = create_pool(&database_url, 5).await?;
29
30    let dialect = architect_sdk::db::active_dialect();
31    ensure_sys_tables(&pool, dialect.as_ref()).await?;
32
33    let tenant_registry = load_registry_from_pool(&pool)
34        .await
35        .map_err(|e| -> Box<dyn std::error::Error> { Box::new(e) })?;
36    tracing::info!("loaded tenant registry (X-Tenant-ID required for config and entity APIs)");
37
38    let (config, package_id) = match std::env::var("PACKAGE_PATH") {
39        Ok(package_path) => {
40            tracing::info!("loading config from package path: {}", package_path);
41            let (cfg, id) = load_config_from_package_path(&package_path).await?;
42            (cfg, id)
43        }
44        Err(_) => {
45            tracing::info!("PACKAGE_PATH not set; loading config from _sys_* tables (use config APIs or POST /api/v1/config/package to insert)");
46            let cfg = load_from_pool(&pool, DEFAULT_PACKAGE_ID)
47                .await
48                .map_err(|e| -> Box<dyn std::error::Error> { Box::new(e) })?;
49            (cfg, DEFAULT_PACKAGE_ID.to_string())
50        }
51    };
52    apply_migrations(&pool, &config, None, None, dialect.as_ref()).await?;
53    let model = resolve(&config)?.with_package_id(&package_id);
54    let mut package_models = HashMap::new();
55    package_models.insert(package_id.clone(), model.clone());
56    let storage = architect_sdk::init_storage_provider().await;
57    let event_client = architect_sdk::events::DecisionHubClient::from_env();
58    let authrs_client = architect_sdk::authrs::AuthrsClient::from_env();
59    let state = AppState {
60        pool: pool.clone(),
61        model: Arc::new(RwLock::new(model)),
62        package_models: Arc::new(RwLock::new(package_models)),
63        tenant_pools: Arc::new(RwLock::new(HashMap::new())),
64        tenant_registry: Arc::new(tenant_registry),
65        storage,
66        event_client,
67        authrs_client,
68        dialect,
69    };
70
71    let api = Router::new()
72        .merge(common_routes_with_ready(state.clone()))
73        .nest("/api/v1", config_routes(state.clone()))
74        .nest("/api/v1", entity_routes(state));
75
76    let app = Router::new().nest("/", api);
77
78    let listener = TcpListener::bind("0.0.0.0:3000").await?;
79    tracing::info!("listening on {}", listener.local_addr()?);
80    axum::serve(listener, app).await?;
81    Ok(())
82}
83
84/// Read all JSON records for a config kind from a package directory.
85/// Tries `{kind}.json` first (flat file), then scans `{kind}/*.json` (subdirectory),
86/// merging all arrays in alphabetical order. Returns an empty vec if neither exists.
87async fn read_kind_from_dir(
88    dir: &Path,
89    kind: &str,
90) -> Result<Vec<serde_json::Value>, Box<dyn std::error::Error>> {
91    let flat = dir.join(format!("{}.json", kind));
92    if flat.exists() {
93        let content = tokio::fs::read_to_string(&flat).await?;
94        return Ok(serde_json::from_str(&content)?);
95    }
96
97    let subdir = dir.join(kind);
98    if subdir.is_dir() {
99        let mut read_dir = tokio::fs::read_dir(&subdir).await?;
100        let mut files: Vec<PathBuf> = Vec::new();
101        while let Some(entry) = read_dir.next_entry().await? {
102            let path = entry.path();
103            if path.extension().and_then(|e| e.to_str()) == Some("json") {
104                files.push(path);
105            }
106        }
107        files.sort();
108        let mut merged: Vec<serde_json::Value> = Vec::new();
109        for path in files {
110            let content = tokio::fs::read_to_string(&path).await?;
111            let mut items: Vec<serde_json::Value> = serde_json::from_str(&content)?;
112            merged.append(&mut items);
113        }
114        return Ok(merged);
115    }
116
117    Ok(vec![])
118}
119
120async fn load_config_from_package_path(
121    dir: &str,
122) -> Result<(FullConfig, String), Box<dyn std::error::Error>> {
123    let dir = PathBuf::from(dir);
124    let manifest_path = dir.join("manifest.json");
125    let manifest_json = tokio::fs::read_to_string(&manifest_path)
126        .await
127        .map_err(|e| format!("package path must contain manifest.json: {}", e))?;
128    let manifest: serde_json::Value = serde_json::from_str(&manifest_json)?;
129    let manifest_obj = manifest.as_object().ok_or_else(|| {
130        std::io::Error::new(
131            std::io::ErrorKind::InvalidData,
132            "manifest.json must be an object",
133        )
134    })?;
135    let package_id = manifest_obj
136        .get("id")
137        .and_then(|v| v.as_str())
138        .ok_or_else(|| {
139            std::io::Error::new(
140                std::io::ErrorKind::InvalidData,
141                "manifest must have 'id' (string)",
142            )
143        })?
144        .to_string();
145    let _name = manifest_obj
146        .get("name")
147        .and_then(|v| v.as_str())
148        .ok_or_else(|| {
149            std::io::Error::new(
150                std::io::ErrorKind::InvalidData,
151                "manifest must have 'name' (string)",
152            )
153        })?;
154    let _version = manifest_obj
155        .get("version")
156        .and_then(|v| v.as_str())
157        .ok_or_else(|| {
158            std::io::Error::new(
159                std::io::ErrorKind::InvalidData,
160                "manifest must have 'version' (string)",
161            )
162        })?;
163    let schema_name = manifest_obj
164        .get("schema")
165        .and_then(|v| v.as_str())
166        .ok_or_else(|| {
167            std::io::Error::new(
168                std::io::ErrorKind::InvalidData,
169                "manifest must have 'schema' (string)",
170            )
171        })?;
172    tracing::info!(
173        "package manifest: id={:?} name={:?} version={:?} schema={:?}",
174        package_id,
175        _name,
176        _version,
177        schema_name
178    );
179
180    let schemas = vec![serde_json::json!({ "id": "default", "name": schema_name })];
181    let schemas: Vec<architect_sdk::config::SchemaConfig> =
182        serde_json::from_value(serde_json::Value::Array(schemas))?;
183
184    let mut enums = read_kind_from_dir(&dir, "enums").await?;
185    for o in enums.iter_mut() {
186        if let Some(obj) = o.as_object_mut() {
187            obj.entry("schema_id")
188                .or_insert_with(|| serde_json::Value::String("default".into()));
189        }
190    }
191    let enums: Vec<architect_sdk::config::EnumConfig> =
192        serde_json::from_value(serde_json::Value::Array(enums))?;
193
194    let mut tables = read_kind_from_dir(&dir, "tables").await?;
195    for o in tables.iter_mut() {
196        if let Some(obj) = o.as_object_mut() {
197            obj.entry("schema_id")
198                .or_insert_with(|| serde_json::Value::String("default".into()));
199        }
200    }
201    let tables: Vec<architect_sdk::config::TableConfig> =
202        serde_json::from_value(serde_json::Value::Array(tables))?;
203
204    let columns_raw = read_kind_from_dir(&dir, "columns").await?;
205    let columns: Vec<architect_sdk::config::ColumnConfig> =
206        serde_json::from_value(serde_json::Value::Array(columns_raw))?;
207
208    let mut indexes = read_kind_from_dir(&dir, "indexes").await?;
209    for o in indexes.iter_mut() {
210        if let Some(obj) = o.as_object_mut() {
211            obj.entry("schema_id")
212                .or_insert_with(|| serde_json::Value::String("default".into()));
213        }
214    }
215    let indexes: Vec<architect_sdk::config::IndexConfig> =
216        serde_json::from_value(serde_json::Value::Array(indexes))?;
217
218    let mut relationships = read_kind_from_dir(&dir, "relationships").await?;
219    for o in relationships.iter_mut() {
220        if let Some(obj) = o.as_object_mut() {
221            obj.entry("from_schema_id")
222                .or_insert_with(|| serde_json::Value::String("default".into()));
223            obj.entry("to_schema_id")
224                .or_insert_with(|| serde_json::Value::String("default".into()));
225        }
226    }
227    let relationships: Vec<architect_sdk::config::RelationshipConfig> =
228        serde_json::from_value(serde_json::Value::Array(relationships))?;
229
230    let api_entities_raw = read_kind_from_dir(&dir, "api_entities").await?;
231    let api_entities: Vec<architect_sdk::config::ApiEntityConfig> =
232        serde_json::from_value(serde_json::Value::Array(api_entities_raw))?;
233
234    let kv_stores_raw = read_kind_from_dir(&dir, "kv_stores").await?;
235    let kv_stores: Vec<architect_sdk::config::KvStoreConfig> =
236        serde_json::from_value(serde_json::Value::Array(kv_stores_raw))?;
237
238    Ok((
239        FullConfig {
240            schemas,
241            enums,
242            tables,
243            columns,
244            indexes,
245            relationships,
246            api_entities,
247            kv_stores,
248        },
249        package_id,
250    ))
251}