drizzle-migrations 0.1.5

//! SQLite snapshot types matching drizzle-kit format
//!
//! Version 7 uses the DDL entity array format.

use super::ddl::SqliteEntity;
use crate::version::{ORIGIN_UUID, SQLITE_SNAPSHOT_VERSION};
use serde::{Deserialize, Serialize};

// =============================================================================
// V7 Snapshot Format (DDL Entity Array)
// =============================================================================

/// SQLite schema snapshot - matches drizzle-kit beta v7 format
///
/// This uses the new DDL entity array format where all schema elements
/// are stored as flat entities with `entityType` discriminators.
#[derive(Serialize, Deserialize, Clone, Debug)]
#[serde(rename_all = "camelCase")]
pub struct SQLiteSnapshot {
    /// Schema version (currently "7")
    pub version: String,
    /// Dialect identifier
    pub dialect: String,
    /// Unique ID for this snapshot
    pub id: String,
    /// IDs of previous snapshots in the chain (v7 uses array instead of single string)
    pub prev_ids: Vec<String>,
    /// DDL entities (tables, columns, indexes, etc.)
    pub ddl: Vec<SqliteEntity>,
    /// Tracked renames for migration generation
    #[serde(default)]
    pub renames: Vec<String>,
}

impl Default for SQLiteSnapshot {
    fn default() -> Self {
        Self::new()
    }
}

impl SQLiteSnapshot {
    /// Create a new empty SQLite snapshot
    pub fn new() -> Self {
        Self {
            version: SQLITE_SNAPSHOT_VERSION.to_string(),
            dialect: "sqlite".to_string(),
            id: uuid::Uuid::new_v4().to_string(),
            prev_ids: vec![ORIGIN_UUID.to_string()],
            ddl: Vec::new(),
            renames: Vec::new(),
        }
    }

    /// Create a new snapshot with specific previous IDs
    pub fn with_prev_ids(prev_ids: Vec<String>) -> Self {
        let mut snapshot = Self::new();
        snapshot.prev_ids = prev_ids;
        snapshot
    }

    /// Add an entity to the DDL array
    pub fn add_entity(&mut self, entity: SqliteEntity) {
        self.ddl.push(entity);
    }

    /// Load snapshot from JSON string
    pub fn from_json(json: &str) -> Result<Self, serde_json::Error> {
        serde_json::from_str(json)
    }

    /// Serialize snapshot to JSON string
    pub fn to_json(&self) -> Result<String, serde_json::Error> {
        serde_json::to_string_pretty(self)
    }

    /// Load snapshot from file
    pub fn load(path: &std::path::Path) -> std::io::Result<Self> {
        let contents = std::fs::read_to_string(path)?;
        serde_json::from_str(&contents)
            .map_err(|e| std::io::Error::new(std::io::ErrorKind::InvalidData, e))
    }

    /// Save snapshot to file
    pub fn save(&self, path: &std::path::Path) -> std::io::Result<()> {
        let json = serde_json::to_string_pretty(self)
            .map_err(|e| std::io::Error::new(std::io::ErrorKind::InvalidData, e))?;

        // Ensure parent directory exists
        if let Some(parent) = path.parent() {
            std::fs::create_dir_all(parent)?;
        }

        std::fs::write(path, json)
    }

    /// Check if this snapshot is empty (no DDL entities)
    pub fn is_empty(&self) -> bool {
        self.ddl.is_empty()
    }
}

// =============================================================================
// Legacy V6 Snapshot Format (for reading old snapshots)
// =============================================================================

use super::ddl::{Table, View};
use std::collections::HashMap;

/// Schema metadata for tracking renames (legacy v6 format)
#[derive(Serialize, Deserialize, Clone, Debug, Default)]
pub struct Meta {
    /// Table renames: old_name -> new_name
    #[serde(default)]
    pub tables: HashMap<String, String>,
    /// Column renames: "table.old_column" -> "table.new_column"
    #[serde(default)]
    pub columns: HashMap<String, String>,
}

/// Internal kit metadata (legacy v6 format)
#[derive(Serialize, Deserialize, Clone, Debug, Default)]
pub struct Internal {
    /// Index-specific internals
    #[serde(default)]
    pub indexes: HashMap<String, IndexInternal>,
}

/// Internal index metadata (legacy v6 format)
#[derive(Serialize, Deserialize, Clone, Debug, Default)]
pub struct IndexInternal {
    /// Column-specific metadata
    #[serde(default)]
    pub columns: HashMap<String, ColumnInternal>,
}

/// Internal column metadata (legacy v6 format)
#[derive(Serialize, Deserialize, Clone, Debug, Default)]
#[serde(rename_all = "camelCase")]
pub struct ColumnInternal {
    /// Is this column an expression?
    #[serde(skip_serializing_if = "Option::is_none")]
    pub is_expression: Option<bool>,
}

/// Legacy SQLite schema snapshot - v6 format (for reading old snapshots)
#[derive(Serialize, Deserialize, Clone, Debug)]
#[serde(rename_all = "camelCase")]
pub struct SQLiteSnapshotV6 {
    /// Schema version ("6")
    pub version: String,
    /// Dialect identifier
    pub dialect: String,
    /// Unique ID for this snapshot
    pub id: String,
    /// ID of the previous snapshot in the chain
    pub prev_id: String,
    /// Tables in the schema
    pub tables: HashMap<String, Table>,
    /// Views in the schema
    #[serde(default)]
    pub views: HashMap<String, View>,
    /// Enums (empty for SQLite, kept for compatibility)
    #[serde(default)]
    pub enums: HashMap<String, ()>,
    /// Metadata for tracking renames
    #[serde(rename = "_meta")]
    pub meta: Meta,
    /// Internal kit metadata
    #[serde(skip_serializing_if = "Option::is_none")]
    pub internal: Option<Internal>,
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::sqlite::ddl::{Column, Table};

    #[test]
    fn test_new_snapshot() {
        let snapshot = SQLiteSnapshot::new();
        assert_eq!(snapshot.version, "7");
        assert_eq!(snapshot.dialect, "sqlite");
        assert_eq!(snapshot.prev_ids[0], crate::ORIGIN_UUID);
        assert!(snapshot.ddl.is_empty());
    }

    #[test]
    fn test_add_entity() {
        let mut snapshot = SQLiteSnapshot::new();

        // Add a table entity
        let table = Table::new("users");
        snapshot.add_entity(SqliteEntity::Table(table));

        // Add column entities
        let id_col = Column::new("users", "id", "integer")
            .not_null()
            .autoincrement();
        let name_col = Column::new("users", "name", "text").not_null();

        snapshot.add_entity(SqliteEntity::Column(id_col));
        snapshot.add_entity(SqliteEntity::Column(name_col));

        assert_eq!(snapshot.ddl.len(), 3);
    }

    #[test]
    fn test_snapshot_serialization() {
        let mut snapshot = SQLiteSnapshot::new();

        let table = Table::new("users");
        snapshot.add_entity(SqliteEntity::Table(table));

        let col = Column::new("users", "id", "integer").not_null();
        snapshot.add_entity(SqliteEntity::Column(col));

        let json = snapshot.to_json().unwrap();

        // Verify it contains expected fields (pretty print adds spaces)
        assert!(json.contains("\"version\": \"7\""));
        assert!(json.contains("\"dialect\": \"sqlite\""));
        assert!(json.contains("\"entityType\": \"tables\""));
        assert!(json.contains("\"entityType\": \"columns\""));

        // Verify round-trip
        let parsed = SQLiteSnapshot::from_json(&json).unwrap();
        assert_eq!(parsed.version, snapshot.version);
        assert_eq!(parsed.ddl.len(), 2);
    }

    #[test]
    fn test_column_json_format_matches_drizzle_kit() {
        // Create a column with autoincrement to verify field naming
        let col = Column::new("users", "id", "integer")
            .not_null()
            .autoincrement();

        let json = serde_json::to_string_pretty(&col).unwrap();

        // Verify field names match drizzle-kit exactly:
        // - autoincrement (not autoIncrement)
        // - notNull (camelCase)
        // - type (renamed from sql_type)
        assert!(
            json.contains("\"autoincrement\""),
            "Expected 'autoincrement' field, got: {}",
            json
        );
        assert!(
            json.contains("\"notNull\""),
            "Expected 'notNull' field, got: {}",
            json
        );
        assert!(
            json.contains("\"type\""),
            "Expected 'type' field, got: {}",
            json
        );
        assert!(
            json.contains("\"table\""),
            "Expected 'table' field, got: {}",
            json
        );
        assert!(
            json.contains("\"name\""),
            "Expected 'name' field, got: {}",
            json
        );

        // Verify it doesn't contain snake_case versions
        assert!(
            !json.contains("\"sql_type\""),
            "Should not contain 'sql_type'"
        );
        assert!(
            !json.contains("\"not_null\""),
            "Should not contain 'not_null'"
        );
    }
}