use std::collections::HashMap;
use std::fs;
use std::path::Path;

use crate::db::DbConnection as Connection;
use uuid::Uuid;

use crate::db::queries::project as q;
use crate::engine::site_assets::copy_bundled_site_assets;
use crate::engine::{EngineError, EngineResult};
use crate::model::Project;
use crate::model::metadata::ProjectMetadata;
use crate::util::{atomic_write_str, ensure_unique, now_unix_ms, slugify};

/// The well-known ID of the default project (spec: DefaultProjectExists).
pub const DEFAULT_PROJECT_ID: &str = "default";

/// Create a new project: insert into DB, create directory structure, write default meta files.
pub fn create_project(
    conn: &Connection,
    name: &str,
    data_path: Option<&str>,
) -> EngineResult<Project> {
    let id = Uuid::new_v4().to_string();
    let base_slug = slugify(name);
    let slug = ensure_unique(&base_slug, |candidate| {
        q::get_project_by_slug(conn, candidate).is_ok()
    });

    let now = now_unix_ms();
    let mut project = Project {
        id: id.clone(),
        name: name.to_string(),
        slug: slug.clone(),
        description: None,
        data_path: data_path.map(|s| s.to_string()),
        is_active: false,
        created_at: now,
        updated_at: now,
    };
    q::insert_project(conn, &project)?;

    // Determine data directory
    let data_dir = match data_path {
        Some(p) => std::path::PathBuf::from(p),
        None => std::path::PathBuf::from("projects").join(&id),
    };

    // Create directory structure
    create_directory_structure(&data_dir)?;

    // Store a stable, absolute machine-local pointer after the folder exists.
    if data_path.is_some() {
        project.data_path = Some(data_dir.canonicalize()?.to_string_lossy().to_string());
        q::update_project(conn, &project)?;
    }

    // Write default meta files
    write_default_meta_files(&data_dir, name)?;

    Ok(project)
}

/// Open an existing portable project folder and remember its current location
/// in the machine-local project registry (the application database).
///
/// The folder itself is authoritative: `data_path` is derived from the
/// location of `meta/project.json` and is never written into that file.
pub fn open_project(conn: &Connection, folder_path: &Path) -> EngineResult<Project> {
    let folder_path = folder_path.canonicalize().map_err(|error| {
        EngineError::Validation(format!(
            "cannot open project folder '{}': {error}",
            folder_path.display()
        ))
    })?;
    let metadata_path = folder_path.join("meta/project.json");
    if !metadata_path.is_file() {
        return Err(EngineError::Validation(format!(
            "project folder must contain meta/project.json: {}",
            folder_path.display()
        )));
    }

    let metadata: ProjectMetadata = serde_json::from_str(&fs::read_to_string(&metadata_path)?)?;
    metadata.validate().map_err(EngineError::Validation)?;
    let resolved_path = folder_path.to_string_lossy().to_string();
    let projects = q::list_projects(conn)?;

    // Reopening a registered folder is idempotent. If a registered folder no
    // longer exists and its portable metadata name matches uniquely, treat the
    // selected folder as that project's new location.
    let exact = projects.iter().find(|project| {
        project.data_path.as_deref().is_some_and(|path| {
            Path::new(path)
                .canonicalize()
                .map(|registered| registered == folder_path)
                .unwrap_or(false)
        })
    });
    let moved_matches: Vec<&Project> = projects
        .iter()
        .filter(|project| {
            project.name == metadata.name
                && project
                    .data_path
                    .as_deref()
                    .is_some_and(|path| !Path::new(path).join("meta/project.json").is_file())
        })
        .collect();

    let moved = if moved_matches.len() == 1 {
        Some(moved_matches[0])
    } else {
        None
    };
    if let Some(existing) = exact.or(moved) {
        let mut project = existing.clone();
        project.name = metadata.name;
        project.description = metadata.description;
        project.data_path = Some(resolved_path);
        project.updated_at = now_unix_ms();
        q::update_project(conn, &project)?;
        return Ok(project);
    }

    let now = now_unix_ms();
    let base_slug = slugify(&metadata.name);
    let project = Project {
        id: Uuid::new_v4().to_string(),
        name: metadata.name,
        slug: ensure_unique(&base_slug, |candidate| {
            q::get_project_by_slug(conn, candidate).is_ok()
        }),
        description: metadata.description,
        data_path: Some(resolved_path),
        is_active: false,
        created_at: now,
        updated_at: now,
    };
    q::insert_project(conn, &project)?;
    Ok(project)
}

/// Ensure the default project (id="default") exists.
/// Creates it on first launch if missing, per the DefaultProjectExists invariant.
/// Returns the project (existing or newly created).
pub fn ensure_default_project(
    conn: &Connection,
    default_data_dir: Option<&Path>,
) -> EngineResult<Project> {
    match q::get_project_by_id(conn, DEFAULT_PROJECT_ID) {
        Ok(p) => Ok(p),
        Err(diesel::result::Error::NotFound) => {
            let now = now_unix_ms();
            let project = Project {
                id: DEFAULT_PROJECT_ID.to_string(),
                name: "My Blog".to_string(),
                slug: "my-blog".to_string(),
                description: None,
                data_path: default_data_dir.map(|p| p.to_string_lossy().to_string()),
                is_active: true,
                created_at: now,
                updated_at: now,
            };
            q::insert_project(conn, &project)?;

            let data_dir = match default_data_dir {
                Some(p) => p.to_path_buf(),
                None => std::path::PathBuf::from("projects").join(DEFAULT_PROJECT_ID),
            };
            create_directory_structure(&data_dir)?;
            write_default_meta_files(&data_dir, "My Blog")?;
            Ok(project)
        }
        Err(e) => Err(EngineError::Db(e)),
    }
}

/// Get the currently active project, if any.
pub fn get_active_project(conn: &Connection) -> EngineResult<Option<Project>> {
    match q::get_active_project(conn) {
        Ok(p) => Ok(Some(p)),
        Err(diesel::result::Error::NotFound) => Ok(None),
        Err(e) => Err(EngineError::Db(e)),
    }
}

/// Deactivate all projects, then activate the given one.
pub fn set_active_project(conn: &Connection, project_id: &str) -> EngineResult<()> {
    q::set_active_project(conn, project_id)?;
    Ok(())
}

/// List all projects ordered by name.
pub fn list_projects(conn: &Connection) -> EngineResult<Vec<Project>> {
    Ok(q::list_projects(conn)?)
}

/// Delete a project row (cascading handled by queries).
/// Rejects deletion of the default project and the currently active project.
/// Only cleans up the internal project data directory (not external custom paths).
pub fn delete_project(
    conn: &Connection,
    project_id: &str,
    internal_data_dir: Option<&Path>,
) -> EngineResult<()> {
    // Cannot delete the default project
    if project_id == DEFAULT_PROJECT_ID {
        return Err(EngineError::Validation(
            "cannot delete the default project".to_string(),
        ));
    }

    // Check if this is the active project (don't delete active)
    if let Ok(active) = q::get_active_project(conn)
        && active.id == project_id
    {
        return Err(EngineError::Validation(
            "cannot delete the active project".to_string(),
        ));
    }

    // Only delete internal data directory, never external custom data_path.
    // The caller must pass the internal path only.
    let project = q::get_project_by_id(conn, project_id)
        .map_err(|_| EngineError::NotFound(format!("project {project_id}")))?;
    let is_custom_path = project.data_path.is_some();

    q::delete_project(conn, project_id)?;

    // Clean up internal filesystem only (not custom external paths per spec)
    if !is_custom_path
        && let Some(dir) = internal_data_dir
        && dir.exists()
    {
        let _ = fs::remove_dir_all(dir);
    }

    Ok(())
}

// ── helpers ──────────────────────────────────────────────────────────

fn create_directory_structure(data_dir: &Path) -> EngineResult<()> {
    let subdirs = [
        "posts",
        "media",
        "meta",
        "thumbnails",
        "templates",
        "scripts",
        "assets",
    ];
    for sub in &subdirs {
        fs::create_dir_all(data_dir.join(sub))?;
    }
    Ok(())
}

fn write_default_meta_files(data_dir: &Path, project_name: &str) -> EngineResult<()> {
    let meta_dir = data_dir.join("meta");

    // project.json
    let project_meta = ProjectMetadata {
        name: project_name.to_string(),
        description: None,
        public_url: None,
        main_language: None,
        default_author: None,
        max_posts_per_page: 50,
        image_import_concurrency: 4,
        blogmark_category: None,
        pico_theme: None,
        semantic_similarity_enabled: false,
        blog_languages: Vec::new(),
    };
    let json = serde_json::to_string_pretty(&project_meta)?;
    atomic_write_str(&meta_dir.join("project.json"), &json)?;

    // categories.json — default categories
    let categories = vec!["article", "aside", "page", "picture"];
    let json = serde_json::to_string_pretty(&categories)?;
    atomic_write_str(&meta_dir.join("categories.json"), &json)?;

    // category-meta.json — empty object
    let empty_map: HashMap<String, serde_json::Value> = HashMap::new();
    let json = serde_json::to_string_pretty(&empty_map)?;
    atomic_write_str(&meta_dir.join("category-meta.json"), &json)?;

    // publishing.json — empty object
    atomic_write_str(&meta_dir.join("publishing.json"), "{}")?;

    // tags.json — empty array
    atomic_write_str(&meta_dir.join("tags.json"), "[]")?;

    // menu.opml — default empty menu per menu.allium HomeAlwaysPresent
    let default_opml = crate::engine::menu::default_menu_opml();
    atomic_write_str(&meta_dir.join("menu.opml"), &default_opml)?;

    copy_bundled_site_assets(data_dir)?;

    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::db::Database;
    use tempfile::TempDir;

    fn setup() -> (Database, TempDir) {
        let db = Database::open_in_memory().unwrap();
        db.migrate().unwrap();
        let dir = TempDir::new().unwrap();
        (db, dir)
    }

    #[test]
    fn create_project_inserts_and_creates_dirs() {
        let (db, dir) = setup();
        let data_path = dir.path().join("my-blog");
        let project =
            create_project(db.conn(), "My Blog", Some(data_path.to_str().unwrap())).unwrap();

        assert_eq!(project.name, "My Blog");
        assert_eq!(project.slug, "my-blog");
        assert!(!project.is_active);

        // Verify directories
        assert!(data_path.join("posts").is_dir());
        assert!(data_path.join("media").is_dir());
        assert!(data_path.join("meta").is_dir());
        assert!(data_path.join("thumbnails").is_dir());
        assert!(data_path.join("templates").is_dir());
        assert!(data_path.join("scripts").is_dir());
        assert!(data_path.join("assets").is_dir());
        assert!(data_path.join("assets/pico.min.css").is_file());
        assert!(data_path.join("assets/tag-cloud.js").is_file());

        // Bundled defaults stay in the application. Project templates are
        // reserved for user-managed overrides.
        assert!(
            fs::read_dir(data_path.join("templates"))
                .unwrap()
                .next()
                .is_none()
        );

        // Verify meta files
        let project_json: serde_json::Value =
            serde_json::from_str(&fs::read_to_string(data_path.join("meta/project.json")).unwrap())
                .unwrap();
        assert_eq!(project_json["name"], "My Blog");
        assert_eq!(project_json["maxPostsPerPage"], 50);

        let cats: Vec<String> = serde_json::from_str(
            &fs::read_to_string(data_path.join("meta/categories.json")).unwrap(),
        )
        .unwrap();
        assert_eq!(cats, vec!["article", "aside", "page", "picture"]);

        let cat_meta: serde_json::Value = serde_json::from_str(
            &fs::read_to_string(data_path.join("meta/category-meta.json")).unwrap(),
        )
        .unwrap();
        assert!(cat_meta.as_object().unwrap().is_empty());

        let tags: Vec<String> =
            serde_json::from_str(&fs::read_to_string(data_path.join("meta/tags.json")).unwrap())
                .unwrap();
        assert!(tags.is_empty());
    }

    #[test]
    fn create_project_unique_slug() {
        let (db, dir) = setup();
        let p1_path = dir.path().join("blog1");
        create_project(db.conn(), "Blog", Some(p1_path.to_str().unwrap())).unwrap();
        let p2_path = dir.path().join("blog2");
        let p2 = create_project(db.conn(), "Blog", Some(p2_path.to_str().unwrap())).unwrap();
        assert_eq!(p2.slug, "blog-2");
    }

    #[test]
    fn open_project_registers_existing_folder_without_rewriting_metadata() {
        let (db, dir) = setup();
        let data_path = dir.path().join("portable-blog");
        fs::create_dir_all(data_path.join("meta")).unwrap();
        let metadata = r#"{"name":"Portable Blog","description":"Moved safely"}"#;
        fs::write(data_path.join("meta/project.json"), metadata).unwrap();

        let project = open_project(db.conn(), &data_path).unwrap();

        assert_eq!(project.name, "Portable Blog");
        assert_eq!(project.description.as_deref(), Some("Moved safely"));
        assert_eq!(
            project.data_path.as_deref(),
            data_path.canonicalize().unwrap().to_str()
        );
        assert_eq!(
            fs::read_to_string(data_path.join("meta/project.json")).unwrap(),
            metadata
        );
    }

    #[test]
    fn open_project_updates_moved_matching_project_location() {
        let (db, dir) = setup();
        let old_path = dir.path().join("old-location");
        let project =
            create_project(db.conn(), "Movable Blog", Some(old_path.to_str().unwrap())).unwrap();
        let new_path = dir.path().join("new-location");
        fs::rename(&old_path, &new_path).unwrap();

        let reopened = open_project(db.conn(), &new_path).unwrap();

        assert_eq!(reopened.id, project.id);
        assert_eq!(
            reopened.data_path.as_deref(),
            new_path.canonicalize().unwrap().to_str()
        );
        assert_eq!(list_projects(db.conn()).unwrap().len(), 1);
    }

    #[test]
    fn open_project_rejects_folder_without_project_metadata() {
        let (db, dir) = setup();
        let error = open_project(db.conn(), dir.path()).unwrap_err();
        assert!(error.to_string().contains("meta/project.json"));
    }

    #[test]
    fn get_active_project_none() {
        let (db, _dir) = setup();
        assert!(get_active_project(db.conn()).unwrap().is_none());
    }

    #[test]
    fn set_and_get_active_project() {
        let (db, dir) = setup();
        let p1_path = dir.path().join("p1");
        let p1 = create_project(db.conn(), "P1", Some(p1_path.to_str().unwrap())).unwrap();
        set_active_project(db.conn(), &p1.id).unwrap();
        let active = get_active_project(db.conn()).unwrap().unwrap();
        assert_eq!(active.id, p1.id);
    }

    #[test]
    fn list_projects_returns_all() {
        let (db, dir) = setup();
        let p1_path = dir.path().join("alpha");
        let p2_path = dir.path().join("beta");
        create_project(db.conn(), "Beta", Some(p2_path.to_str().unwrap())).unwrap();
        create_project(db.conn(), "Alpha", Some(p1_path.to_str().unwrap())).unwrap();
        let list = list_projects(db.conn()).unwrap();
        assert_eq!(list.len(), 2);
        assert_eq!(list[0].name, "Alpha");
        assert_eq!(list[1].name, "Beta");
    }

    #[test]
    fn delete_project_removes_row() {
        let (db, dir) = setup();
        // Simulate an internal project (no custom data_path) by inserting directly
        let now = crate::util::now_unix_ms();
        let project = Project {
            id: uuid::Uuid::new_v4().to_string(),
            name: "P".to_string(),
            slug: "p".to_string(),
            description: None,
            data_path: None,
            is_active: false,
            created_at: now,
            updated_at: now,
        };
        crate::db::queries::project::insert_project(db.conn(), &project).unwrap();
        // Create the internal directory structure under the temp dir
        let internal_dir = dir.path().join("projects").join(&project.id);
        create_directory_structure(&internal_dir).unwrap();
        assert!(internal_dir.join("posts").is_dir());
        delete_project(db.conn(), &project.id, Some(&internal_dir)).unwrap();
        assert!(list_projects(db.conn()).unwrap().is_empty());
        assert!(
            !internal_dir.exists(),
            "internal directory should be cleaned up"
        );
    }

    #[test]
    fn delete_active_project_rejected() {
        let (db, dir) = setup();
        let p_path = dir.path().join("p");
        let p = create_project(db.conn(), "P", Some(p_path.to_str().unwrap())).unwrap();
        set_active_project(db.conn(), &p.id).unwrap();
        let result = delete_project(db.conn(), &p.id, None);
        assert!(result.is_err());
    }

    #[test]
    fn ensure_default_project_creates_on_first_call() {
        let (db, dir) = setup();
        let data_path = dir.path().join("default-data");
        let p = ensure_default_project(db.conn(), Some(&data_path)).unwrap();
        assert_eq!(p.id, DEFAULT_PROJECT_ID);
        assert_eq!(p.name, "My Blog");
        assert!(p.is_active);
        assert!(data_path.join("posts").is_dir());
        assert!(data_path.join("meta/project.json").exists());
    }

    #[test]
    fn ensure_default_project_idempotent() {
        let (db, dir) = setup();
        let data_path = dir.path().join("default-data");
        let p1 = ensure_default_project(db.conn(), Some(&data_path)).unwrap();
        let p2 = ensure_default_project(db.conn(), Some(&data_path)).unwrap();
        assert_eq!(p1.id, p2.id);
    }

    #[test]
    fn delete_default_project_rejected() {
        let (db, dir) = setup();
        let data_path = dir.path().join("default-data");
        ensure_default_project(db.conn(), Some(&data_path)).unwrap();
        let result = delete_project(db.conn(), DEFAULT_PROJECT_ID, None);
        assert!(result.is_err());
    }
}