mdql_web/

server.rs

//! MDQL browser UI — axum server with embedded SPA.

use std::collections::HashMap;
use std::path::PathBuf;
use std::sync::{Arc, Mutex};

use axum::extract::{Path as AxumPath, State};
use axum::http::{header, StatusCode, Uri};
use axum::response::{IntoResponse, Json};
use axum::routing::{get, post};
use axum::Router;
use rust_embed::Embed;
use serde::{Deserialize, Serialize};
use tower_http::cors::CorsLayer;

use mdql_core::api::Table;
use mdql_core::loader;
use mdql_core::model::{Row, Value};
use mdql_core::projector::format_results;
use mdql_core::query_parser::{parse_query, Statement};
use mdql_core::schema::Schema;

#[derive(Embed)]
#[folder = "static/"]
struct StaticFiles;

#[derive(Clone)]
struct AppState {
    db_path: PathBuf,
    tables: Arc<Mutex<HashMap<String, (Schema, Vec<Row>)>>>,
    fk_errors: Arc<Mutex<Vec<mdql_core::errors::ValidationError>>>,
}

#[derive(Serialize)]
struct TableInfo {
    name: String,
    row_count: usize,
}

#[derive(Serialize)]
struct TablesResponse {
    tables: Vec<TableInfo>,
}

#[derive(Serialize)]
struct TableDetailResponse {
    table: String,
    primary_key: String,
    row_count: usize,
    frontmatter: HashMap<String, FieldInfo>,
    sections: HashMap<String, SectionInfo>,
}

#[derive(Serialize)]
struct FieldInfo {
    #[serde(rename = "type")]
    field_type: String,
    required: bool,
    enum_values: Option<Vec<String>>,
}

#[derive(Serialize)]
struct SectionInfo {
    content_type: String,
    required: bool,
}

#[derive(Deserialize)]
struct QueryRequest {
    sql: String,
    #[serde(default = "default_format")]
    format: String,
}

fn default_format() -> String {
    "table".into()
}

#[derive(Serialize)]
struct QueryResponse {
    columns: Option<Vec<String>>,
    rows: Option<Vec<HashMap<String, serde_json::Value>>>,
    output: Option<String>,
    error: Option<String>,
    row_count: Option<usize>,
}

pub async fn run_server(db_path: PathBuf, port: u16) {
    // Load the database
    let tables = match load_all_tables(&db_path) {
        Ok(t) => t,
        Err(e) => {
            eprintln!("Failed to load database: {}", e);
            std::process::exit(1);
        }
    };

    let fk_errors: Arc<Mutex<Vec<mdql_core::errors::ValidationError>>> =
        Arc::new(Mutex::new(Vec::new()));

    let state = AppState {
        db_path: db_path.clone(),
        tables: Arc::new(Mutex::new(tables)),
        fk_errors: fk_errors.clone(),
    };

    // Start filesystem watcher for FK validation
    {
        let tables_clone = state.tables.clone();
        let fk_errors_clone = fk_errors.clone();
        let db_path_clone = db_path.clone();
        tokio::task::spawn_blocking(move || {
            let watcher = match mdql_core::watcher::FkWatcher::start(db_path_clone.clone()) {
                Ok(w) => w,
                Err(e) => {
                    eprintln!("Warning: could not start FK watcher: {}", e);
                    return;
                }
            };
            loop {
                if let Some(errors) = watcher.poll() {
                    *fk_errors_clone.lock().unwrap() = errors;
                    // Also reload tables on file change
                    if let Ok(new_tables) = load_all_tables(&db_path_clone) {
                        *tables_clone.lock().unwrap() = new_tables;
                    }
                }
                std::thread::sleep(std::time::Duration::from_millis(200));
            }
        });
    }

    let app = Router::new()
        .route("/api/tables", get(list_tables))
        .route("/api/tables/{name}", get(table_detail))
        .route("/api/query", post(execute_query))
        .route("/api/reload", post(reload_tables))
        .route("/api/fk-errors", get(get_fk_errors))
        .fallback(static_handler)
        .layer(CorsLayer::permissive())
        .with_state(state);

    let listener = tokio::net::TcpListener::bind(format!("0.0.0.0:{}", port))
        .await
        .expect("Failed to bind");

    println!("MDQL client running at http://localhost:{}", port);

    axum::serve(listener, app).await.expect("Server failed");
}

fn load_all_tables(db_path: &std::path::Path) -> Result<HashMap<String, (Schema, Vec<Row>)>, String> {
    // Try as database first
    if let Ok((_config, tables, _errors)) = loader::load_database(db_path) {
        return Ok(tables);
    }

    // Try as single table
    match loader::load_table(db_path) {
        Ok((schema, rows, _errors)) => {
            let mut map = HashMap::new();
            map.insert(schema.table.clone(), (schema, rows));
            Ok(map)
        }
        Err(e) => Err(format!("Failed to load: {}", e)),
    }
}

async fn list_tables(State(state): State<AppState>) -> Json<TablesResponse> {
    let tables = state.tables.lock().unwrap();
    let mut infos: Vec<TableInfo> = tables
        .iter()
        .map(|(name, (_schema, rows))| TableInfo {
            name: name.clone(),
            row_count: rows.len(),
        })
        .collect();
    infos.sort_by(|a, b| a.name.cmp(&b.name));
    Json(TablesResponse { tables: infos })
}

async fn table_detail(
    State(state): State<AppState>,
    AxumPath(name): AxumPath<String>,
) -> Result<Json<TableDetailResponse>, StatusCode> {
    let tables = state.tables.lock().unwrap();
    let (schema, rows) = tables.get(&name).ok_or(StatusCode::NOT_FOUND)?;

    let frontmatter: HashMap<String, FieldInfo> = schema
        .frontmatter
        .iter()
        .map(|(k, v)| {
            (
                k.clone(),
                FieldInfo {
                    field_type: format!("{:?}", v.field_type),
                    required: v.required,
                    enum_values: v.enum_values.clone(),
                },
            )
        })
        .collect();

    let sections: HashMap<String, SectionInfo> = schema
        .sections
        .iter()
        .map(|(k, v)| {
            (
                k.clone(),
                SectionInfo {
                    content_type: v.content_type.clone(),
                    required: v.required,
                },
            )
        })
        .collect();

    Ok(Json(TableDetailResponse {
        table: schema.table.clone(),
        primary_key: schema.primary_key.clone(),
        row_count: rows.len(),
        frontmatter,
        sections,
    }))
}

async fn execute_query(
    State(state): State<AppState>,
    Json(req): Json<QueryRequest>,
) -> Json<QueryResponse> {
    let tables = state.tables.lock().unwrap();

    // Parse the SQL
    let stmt = match parse_query(&req.sql) {
        Ok(s) => s,
        Err(e) => {
            return Json(QueryResponse {
                columns: None,
                rows: None,
                output: None,
                error: Some(format!("Parse error: {}", e)),
                row_count: None,
            });
        }
    };

    match stmt {
        Statement::Select(query) => {
            // Determine table
            let table_name = &query.table;
            let (schema, rows) = match tables.get(table_name.as_str()) {
                Some(t) => t,
                None => {
                    return Json(QueryResponse {
                        columns: None,
                        rows: None,
                        output: None,
                        error: Some(format!("Unknown table '{}'", table_name)),
                        row_count: None,
                    });
                }
            };

            // Handle JOINs
            let result = if !query.joins.is_empty() {
                mdql_core::query_engine::execute_join_query(&query, &tables)
            } else {
                mdql_core::query_engine::execute_query(&query, rows, schema)
            };

            match result {
                Ok((result_rows, columns)) => {
                    if req.format == "json" || req.format == "csv" {
                        let output = format_results(
                            &result_rows,
                            Some(&columns),
                            &req.format,
                            80,
                        );
                        Json(QueryResponse {
                            columns: None,
                            rows: None,
                            output: Some(output),
                            error: None,
                            row_count: Some(result_rows.len()),
                        })
                    } else {
                        let json_rows: Vec<HashMap<String, serde_json::Value>> = result_rows
                            .iter()
                            .map(|row| {
                                columns
                                    .iter()
                                    .map(|col| {
                                        let val = row.get(col).unwrap_or(&Value::Null);
                                        (col.clone(), value_to_json(val))
                                    })
                                    .collect()
                            })
                            .collect();

                        Json(QueryResponse {
                            columns: Some(columns),
                            rows: Some(json_rows.clone()),
                            output: None,
                            error: None,
                            row_count: Some(json_rows.len()),
                        })
                    }
                }
                Err(e) => Json(QueryResponse {
                    columns: None,
                    rows: None,
                    output: None,
                    error: Some(e.to_string()),
                    row_count: None,
                }),
            }
        }
        _ => {
            // For write operations, use the Table API
            drop(tables); // Release lock before write operations
            let result = execute_write(&state, &req.sql);
            Json(QueryResponse {
                columns: None,
                rows: None,
                output: Some(result.clone()),
                error: if result.starts_with("Error") {
                    Some(result)
                } else {
                    None
                },
                row_count: None,
            })
        }
    }
}

fn execute_write(state: &AppState, sql: &str) -> String {
    // Find the table in the db_path
    let stmt = match parse_query(sql) {
        Ok(s) => s,
        Err(e) => return format!("Error: {}", e),
    };

    let table_name = match &stmt {
        Statement::Insert(q) => q.table.clone(),
        Statement::Update(q) => q.table.clone(),
        Statement::Delete(q) => q.table.clone(),
        Statement::AlterRename(q) => q.table.clone(),
        Statement::AlterDrop(q) => q.table.clone(),
        Statement::AlterMerge(q) => q.table.clone(),
        _ => return "Error: unsupported statement type".into(),
    };

    let table_path = state.db_path.join(&table_name);
    let mut table = match Table::new(&table_path) {
        Ok(t) => t,
        Err(e) => return format!("Error: {}", e),
    };

    let result = match table.execute_sql(sql) {
        Ok(s) => s,
        Err(e) => format!("Error: {}", e),
    };

    // Reload tables after write
    if let Ok(new_tables) = load_all_tables(&state.db_path) {
        let mut tables = state.tables.lock().unwrap();
        *tables = new_tables;
    }

    result
}

async fn get_fk_errors(State(state): State<AppState>) -> Json<serde_json::Value> {
    let errors = state.fk_errors.lock().unwrap();
    let error_list: Vec<serde_json::Value> = errors
        .iter()
        .map(|e| {
            serde_json::json!({
                "file": e.file_path,
                "field": e.field,
                "message": e.message,
            })
        })
        .collect();
    Json(serde_json::json!({ "errors": error_list }))
}

async fn reload_tables(State(state): State<AppState>) -> Json<serde_json::Value> {
    match load_all_tables(&state.db_path) {
        Ok(new_tables) => {
            let mut tables = state.tables.lock().unwrap();
            *tables = new_tables;
            Json(serde_json::json!({ "status": "ok" }))
        }
        Err(e) => Json(serde_json::json!({ "status": "error", "message": e })),
    }
}

fn value_to_json(val: &Value) -> serde_json::Value {
    match val {
        Value::Null => serde_json::Value::Null,
        Value::String(s) => serde_json::Value::String(s.clone()),
        Value::Int(n) => serde_json::json!(n),
        Value::Float(f) => serde_json::json!(f),
        Value::Bool(b) => serde_json::json!(b),
        Value::Date(d) => serde_json::Value::String(d.format("%Y-%m-%d").to_string()),
        Value::List(items) => serde_json::json!(items),
    }
}

async fn static_handler(uri: Uri) -> impl IntoResponse {
    let path = uri.path().trim_start_matches('/');
    let path = if path.is_empty() { "index.html" } else { path };

    match StaticFiles::get(path) {
        Some(content) => {
            let mime = mime_guess::from_path(path).first_or_octet_stream();
            (
                StatusCode::OK,
                [(header::CONTENT_TYPE, mime.as_ref().to_string())],
                content.data.into_owned(),
            )
                .into_response()
        }
        None => {
            // SPA fallback: serve index.html for unknown routes
            match StaticFiles::get("index.html") {
                Some(content) => (
                    StatusCode::OK,
                    [(header::CONTENT_TYPE, "text/html".to_string())],
                    content.data.into_owned(),
                )
                    .into_response(),
                None => (StatusCode::NOT_FOUND, "Not found").into_response(),
            }
        }
    }
}