path: root/src/database.rs

use directories::ProjectDirs;
use rusqlite::{Connection, Result as SqlResult};
use std::fs;
use std::path::{Path, PathBuf};
use std::time::SystemTime;

#[derive(Debug, Clone)]
#[allow(dead_code)]
pub struct Artist {
    pub id: i64,
    pub name: String,
}

#[derive(Debug, Clone)]
#[allow(dead_code)]
pub struct Album {
    pub id: i64,
    pub artist_id: i64,
    pub title: String,
}

#[derive(Debug, Clone)]
#[allow(dead_code)]
pub struct Track {
    pub id: i64,
    pub album_id: i64,
    pub title: String,
    pub path: PathBuf,
    pub duration: Option<i64>,
    pub track_number: Option<i32>,
    pub artist_name: Option<String>,
    pub album_title: Option<String>,
}

pub struct Database {
    conn: Connection,
}

#[allow(dead_code)]
impl Database {
    pub fn open(_config: &crate::config::Config) -> SqlResult<Self> {
        let proj_dirs = ProjectDirs::from("com", "rsplayer", "rsplayer").ok_or_else(|| {
            rusqlite::Error::InvalidPath("Failed to get project directories".into())
        })?;

        let data_dir = if let Ok(xdg) = std::env::var("XDG_CONFIG_HOME") {
            PathBuf::from(xdg).join("rsplayer")
        } else {
            proj_dirs.data_dir().to_path_buf()
        };

        fs::create_dir_all(&data_dir).ok();
        let db_path = data_dir.join("library.db");

        let conn = Connection::open(&db_path)?;
        let db = Database { conn };
        db.init_schema()?;
        Ok(db)
    }

    fn init_schema(&self) -> SqlResult<()> {
        self.conn.execute(
            "CREATE TABLE IF NOT EXISTS artists (
                id INTEGER PRIMARY KEY,
                name TEXT UNIQUE NOT NULL
            )",
            [],
        )?;

        self.conn.execute(
            "CREATE TABLE IF NOT EXISTS albums (
                id INTEGER PRIMARY KEY,
                artist_id INTEGER NOT NULL,
                title TEXT NOT NULL,
                FOREIGN KEY (artist_id) REFERENCES artists(id),
                UNIQUE(artist_id, title)
            )",
            [],
        )?;

        self.conn.execute(
            "CREATE TABLE IF NOT EXISTS tracks (
                id INTEGER PRIMARY KEY,
                album_id INTEGER NOT NULL,
                title TEXT NOT NULL,
                path TEXT UNIQUE NOT NULL,
                duration INTEGER,
                track_number INTEGER,
                last_scanned INTEGER,
                FOREIGN KEY (album_id) REFERENCES albums(id)
            )",
            [],
        )?;

        Ok(())
    }

    pub fn scan_paths(&self, paths: &[PathBuf], force: bool) -> SqlResult<()> {
        for path in paths {
            self.scan_directory(path, force)?;
        }
        Ok(())
    }

    fn scan_directory(&self, dir: &Path, force: bool) -> SqlResult<()> {
        if !dir.exists() {
            return Ok(());
        }

        let now = SystemTime::now()
            .duration_since(SystemTime::UNIX_EPOCH)
            .unwrap()
            .as_secs() as i64;

        self.scan_recursive(dir, now, force)?;
        self.cleanup_old_tracks(now)?;
        Ok(())
    }

    fn scan_recursive(&self, dir: &Path, scan_time: i64, force: bool) -> SqlResult<()> {
        let entries = fs::read_dir(dir).ok();
        if let Some(entries) = entries {
            for entry in entries.flatten() {
                let path = entry.path();
                if path.is_dir() {
                    self.scan_recursive(&path, scan_time, force)?;
                } else if is_audio_file(&path) {
                    self.process_audio_file(&path, scan_time, force)?;
                }
            }
        }
        Ok(())
    }

    fn process_audio_file(&self, path: &Path, scan_time: i64, force: bool) -> SqlResult<()> {
        let path_str = path.to_string_lossy().to_string();

        if !force {
            let count: i64 = self
                .conn
                .query_row(
                    "SELECT COUNT(*) FROM tracks WHERE path = ? AND last_scanned = ?",
                    rusqlite::params![path_str, scan_time],
                    |row| row.get(0),
                )
                .unwrap_or(0);
            if count > 0 {
                return Ok(());
            }
        }

        let metadata = extract_metadata(path);
        let (artist_name, album_title, track_title, track_number, duration) = metadata;
        let duration_i64 = duration.map(|d| d as i64);

        let artist_id = self.get_or_create_artist(&artist_name)?;
        let album_id = self.get_or_create_album(artist_id, &album_title)?;

        self.conn.execute(
            "INSERT OR REPLACE INTO tracks (album_id, title, path, duration, track_number, last_scanned)
             VALUES (?, ?, ?, ?, ?, ?)",
            rusqlite::params![album_id, track_title, path_str, duration_i64, track_number, scan_time],
        )?;

        Ok(())
    }

    fn get_or_create_artist(&self, name: &str) -> SqlResult<i64> {
        self.conn
            .execute("INSERT OR IGNORE INTO artists (name) VALUES (?)", [name])?;
        self.conn
            .query_row("SELECT id FROM artists WHERE name = ?", [name], |row| {
                row.get(0)
            })
    }

    fn get_or_create_album(&self, artist_id: i64, title: &str) -> SqlResult<i64> {
        self.conn.execute(
            "INSERT OR IGNORE INTO albums (artist_id, title) VALUES (?, ?)",
            rusqlite::params![artist_id, title],
        )?;
        self.conn.query_row(
            "SELECT id FROM albums WHERE artist_id = ? AND title = ?",
            rusqlite::params![artist_id, title],
            |row| row.get(0),
        )
    }

    fn cleanup_old_tracks(&self, scan_time: i64) -> SqlResult<()> {
        self.conn
            .execute("DELETE FROM tracks WHERE last_scanned < ?", [scan_time])?;
        Ok(())
    }

    pub fn search_tracks(&self, query: &str) -> SqlResult<Vec<Track>> {
        let mut stmt = self.conn.prepare(
            "SELECT t.id, t.album_id, t.title, t.path, t.duration, t.track_number, ar.name, a.title
             FROM tracks t
             JOIN albums a ON t.album_id = a.id
             JOIN artists ar ON a.artist_id = ar.id
             WHERE t.title LIKE ? OR a.title LIKE ? OR ar.name LIKE ?
             ORDER BY ar.name, a.title, t.track_number",
        )?;

        let search = format!("%{}%", query);
        let rows = stmt.query_map([&search, &search, &search], |row| {
            Ok(Track {
                id: row.get(0)?,
                album_id: row.get(1)?,
                title: row.get(2)?,
                path: PathBuf::from(row.get::<_, String>(3)?),
                duration: row.get(4)?,
                track_number: row.get(5)?,
                artist_name: Some(row.get(6)?),
                album_title: Some(row.get(7)?),
            })
        })?;

        rows.collect()
    }
}

pub fn is_audio_file(path: &Path) -> bool {
    path.extension()
        .and_then(|ext| ext.to_str())
        .map(|ext| {
            matches!(
                ext.to_lowercase().as_str(),
                "mp3" | "flac" | "ogg" | "m4a" | "wav" | "opus"
            )
        })
        .unwrap_or(false)
}

pub fn extract_metadata(path: &Path) -> (String, String, String, Option<i32>, Option<u64>) {
    let file_stem = path
        .file_stem()
        .and_then(|s| s.to_str())
        .unwrap_or("Unknown");
    let parent = path
        .parent()
        .and_then(|p| p.file_name())
        .and_then(|s| s.to_str())
        .unwrap_or("Unknown");
    let grandparent = path
        .parent()
        .and_then(|p| p.parent())
        .and_then(|p| p.file_name())
        .and_then(|s| s.to_str())
        .unwrap_or("Unknown");

    (
        grandparent.to_string(),
        parent.to_string(),
        file_stem.to_string(),
        None,
        None,
    )
}