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(
53        &pool,
54        &config,
55        None,
56        None,
57        dialect.as_ref(),
58        &HashMap::new(),
59    )
60    .await?;
61    let model = resolve(&config)?.with_package_id(&package_id);
62    let mut package_models = HashMap::new();
63    package_models.insert(package_id.clone(), model.clone());
64    let storage = architect_sdk::init_storage_provider().await;
65    let event_client = architect_sdk::events::DecisionHubClient::from_env();
66    let authrs_client = architect_sdk::authrs::AuthrsClient::from_env();
67    let state = AppState {
68        pool: pool.clone(),
69        model: Arc::new(RwLock::new(model)),
70        package_models: Arc::new(RwLock::new(package_models)),
71        tenant_pools: Arc::new(RwLock::new(HashMap::new())),
72        tenant_registry: Arc::new(tenant_registry),
73        storage,
74        event_client,
75        authrs_client,
76        dialect,
77        extensible_cache: Default::default(),
78    };
79
80    let api = Router::new()
81        .merge(common_routes_with_ready(state.clone()))
82        .nest("/api/v1", config_routes(state.clone()))
83        .nest("/api/v1", entity_routes(state));
84
85    let app = Router::new().nest("/", api);
86
87    let listener = TcpListener::bind("0.0.0.0:3000").await?;
88    tracing::info!("listening on {}", listener.local_addr()?);
89    axum::serve(listener, app).await?;
90    Ok(())
91}
92
93/// Read all JSON records for a config kind from a package directory.
94/// Tries `{kind}.json` first (flat file), then scans `{kind}/*.json` (subdirectory),
95/// merging all arrays in alphabetical order. Returns an empty vec if neither exists.
96async fn read_kind_from_dir(
97    dir: &Path,
98    kind: &str,
99) -> Result<Vec<serde_json::Value>, Box<dyn std::error::Error>> {
100    let flat = dir.join(format!("{}.json", kind));
101    if flat.exists() {
102        let content = tokio::fs::read_to_string(&flat).await?;
103        return Ok(serde_json::from_str(&content)?);
104    }
105
106    let subdir = dir.join(kind);
107    if subdir.is_dir() {
108        let mut read_dir = tokio::fs::read_dir(&subdir).await?;
109        let mut files: Vec<PathBuf> = Vec::new();
110        while let Some(entry) = read_dir.next_entry().await? {
111            let path = entry.path();
112            if path.extension().and_then(|e| e.to_str()) == Some("json") {
113                files.push(path);
114            }
115        }
116        files.sort();
117        let mut merged: Vec<serde_json::Value> = Vec::new();
118        for path in files {
119            let content = tokio::fs::read_to_string(&path).await?;
120            let mut items: Vec<serde_json::Value> = serde_json::from_str(&content)?;
121            merged.append(&mut items);
122        }
123        return Ok(merged);
124    }
125
126    Ok(vec![])
127}
128
129async fn load_config_from_package_path(
130    dir: &str,
131) -> Result<(FullConfig, String), Box<dyn std::error::Error>> {
132    let dir = PathBuf::from(dir);
133    let manifest_path = dir.join("manifest.json");
134    let manifest_json = tokio::fs::read_to_string(&manifest_path)
135        .await
136        .map_err(|e| format!("package path must contain manifest.json: {}", e))?;
137    let manifest: serde_json::Value = serde_json::from_str(&manifest_json)?;
138    let manifest_obj = manifest.as_object().ok_or_else(|| {
139        std::io::Error::new(
140            std::io::ErrorKind::InvalidData,
141            "manifest.json must be an object",
142        )
143    })?;
144    let package_id = manifest_obj
145        .get("id")
146        .and_then(|v| v.as_str())
147        .ok_or_else(|| {
148            std::io::Error::new(
149                std::io::ErrorKind::InvalidData,
150                "manifest must have 'id' (string)",
151            )
152        })?
153        .to_string();
154    let _name = manifest_obj
155        .get("name")
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 'name' (string)",
161            )
162        })?;
163    let _version = manifest_obj
164        .get("version")
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 'version' (string)",
170            )
171        })?;
172    let schema_name = manifest_obj
173        .get("schema")
174        .and_then(|v| v.as_str())
175        .ok_or_else(|| {
176            std::io::Error::new(
177                std::io::ErrorKind::InvalidData,
178                "manifest must have 'schema' (string)",
179            )
180        })?;
181    tracing::info!(
182        "package manifest: id={:?} name={:?} version={:?} schema={:?}",
183        package_id,
184        _name,
185        _version,
186        schema_name
187    );
188
189    let schemas = vec![serde_json::json!({ "id": "default", "name": schema_name })];
190    let schemas: Vec<architect_sdk::config::SchemaConfig> =
191        serde_json::from_value(serde_json::Value::Array(schemas))?;
192
193    let mut enums = read_kind_from_dir(&dir, "enums").await?;
194    for o in enums.iter_mut() {
195        if let Some(obj) = o.as_object_mut() {
196            obj.entry("schema_id")
197                .or_insert_with(|| serde_json::Value::String("default".into()));
198        }
199    }
200    let enums: Vec<architect_sdk::config::EnumConfig> =
201        serde_json::from_value(serde_json::Value::Array(enums))?;
202
203    let mut tables = read_kind_from_dir(&dir, "tables").await?;
204    for o in tables.iter_mut() {
205        if let Some(obj) = o.as_object_mut() {
206            obj.entry("schema_id")
207                .or_insert_with(|| serde_json::Value::String("default".into()));
208        }
209    }
210    let tables: Vec<architect_sdk::config::TableConfig> =
211        serde_json::from_value(serde_json::Value::Array(tables))?;
212
213    let columns_raw = read_kind_from_dir(&dir, "columns").await?;
214    let columns: Vec<architect_sdk::config::ColumnConfig> =
215        serde_json::from_value(serde_json::Value::Array(columns_raw))?;
216
217    let mut indexes = read_kind_from_dir(&dir, "indexes").await?;
218    for o in indexes.iter_mut() {
219        if let Some(obj) = o.as_object_mut() {
220            obj.entry("schema_id")
221                .or_insert_with(|| serde_json::Value::String("default".into()));
222        }
223    }
224    let indexes: Vec<architect_sdk::config::IndexConfig> =
225        serde_json::from_value(serde_json::Value::Array(indexes))?;
226
227    let mut relationships = read_kind_from_dir(&dir, "relationships").await?;
228    for o in relationships.iter_mut() {
229        if let Some(obj) = o.as_object_mut() {
230            obj.entry("from_schema_id")
231                .or_insert_with(|| serde_json::Value::String("default".into()));
232            obj.entry("to_schema_id")
233                .or_insert_with(|| serde_json::Value::String("default".into()));
234        }
235    }
236    let relationships: Vec<architect_sdk::config::RelationshipConfig> =
237        serde_json::from_value(serde_json::Value::Array(relationships))?;
238
239    let api_entities_raw = read_kind_from_dir(&dir, "api_entities").await?;
240    let api_entities: Vec<architect_sdk::config::ApiEntityConfig> =
241        serde_json::from_value(serde_json::Value::Array(api_entities_raw))?;
242
243    let kv_stores_raw = read_kind_from_dir(&dir, "kv_stores").await?;
244    let kv_stores: Vec<architect_sdk::config::KvStoreConfig> =
245        serde_json::from_value(serde_json::Value::Array(kv_stores_raw))?;
246
247    Ok((
248        FullConfig {
249            schemas,
250            enums,
251            tables,
252            columns,
253            indexes,
254            relationships,
255            api_entities,
256            kv_stores,
257        },
258        package_id,
259    ))
260}