sourcetrait_cli 1.2.1

use crate::*;

pub const ENV_VAR_SRCTRAIT_NOTES_DIR: &'static str = "SRCTRAIT_NOTES_DIR";
pub const ENV_VAR_SRCTRAIT_CONFIG_DIR: &'static str = "SRCTRAIT_CONFIG_DIR";
pub const NOTE_CONFIG_DIR: &'static str = "srctrait/note";
pub const CONFIG_FILENAME: &'static str = "config.toml";
const HOME_CONFIG_DIR: &'static str = ".config";

pub(crate) fn user_config_dir() -> anyhow::Result<PathBuf> {
    if let Some(dir) = std::env::var_os(ENV_VAR_SRCTRAIT_CONFIG_DIR) {
        Ok(PathBuf::from(dir))
    } else {
        dirs::home_dir()
            .with_context(|| "User directory not found")
            .map(|d| d.join(HOME_CONFIG_DIR))
    }
}

pub(crate) fn note_config_file() -> anyhow::Result<PathBuf> {
    let note_config_dir = user_config_dir()?.join(NOTE_CONFIG_DIR);
    if !note_config_dir.is_dir() {
        fs::create_dir_all(&note_config_dir)
            .with_context(|| format!("Unable to create SourceTrait Note config dir: {}", note_config_dir.display()))?;
    }

    let note_config_file = note_config_dir.join(CONFIG_FILENAME);
    Ok(note_config_file)
}

pub(crate) fn user_notes_config() -> anyhow::Result<lib::NoteConfig> {
    let note_config_file = note_config_file()?;
    let input = if note_config_file.is_file() {
        lib::NoteConfigInput::from_toml_file(&note_config_file)?
    } else {
        lib::NoteConfigInput::default()
    };

    let config = lib::NoteConfig::from_input(
        input,
        dirs::document_dir()
            .with_context(|| format!("Unable to determine user documents directory"))?
    );

    Ok(config)
}

pub(crate) fn user_notes_dir(config: &lib::NoteConfig) -> anyhow::Result<lib::NotesDir> {
    if let Some(dir) = std::env::var_os(ENV_VAR_SRCTRAIT_NOTES_DIR) {
        let dir = shellexpand::path::full(&PathBuf::from(&dir))
            .with_context(|| format!("Failed to expand config.editor: {}", config.editor()))?
            .to_path_buf();

        Ok(lib::NotesDir::new(dir))
    } else {
        let dir = shellexpand::path::full(config.notes_dir())
            .with_context(|| format!("Failed to expand config.editor: {}", config.editor()))?
            .to_path_buf();

        Ok(lib::NotesDir::new(dir))
    }
}

pub fn run_editor(config: &lib::NoteConfig, file: &Path) -> anyhow::Result<()> {
    let editor = shellexpand::full(config.editor())
        .with_context(|| format!("Failed to expand config.editor: {}", config.editor()))?;

    let output = Command::new(&*editor)
        .arg(file)
        .stdin(Stdio::inherit())
        .stdout(Stdio::inherit())
        .stderr(Stdio::inherit())
        .output()
        .with_context(|| format!("Unable to run editor: `{}`", editor))?;

    if output.status.success() {
        Ok(())
    } else {
        let out = format!("{}\n{}",
            String::from_utf8_lossy(&output.stdout),
            String::from_utf8_lossy(&output.stderr)
        );

        Err(anyhow::anyhow!("Editor failure: {editor} :: {out}"))
    }
}

/// currently defaults to yazi for now
pub fn run_picker(config: &lib::NoteConfig, dir: &Path) -> anyhow::Result<Option<PathBuf>> {
    run_picker_yazi(config, dir)
}

pub fn run_picker_yazi(_config: &lib::NoteConfig, dir: &Path) -> anyhow::Result<Option<PathBuf>> {
    let tempfile = tempfile::NamedTempFile::new()
        .with_context(|| "Unable to create tempfile")?;
    let tempfile_path = tempfile.path();
    
    let output = Command::new("yazi")
        .arg(dir)
        .arg("--chooser-file").arg(tempfile_path)
        .stdin(Stdio::inherit())
        .stdout(Stdio::inherit())
        .stderr(Stdio::inherit())
        .output()
        .with_context(|| format!("{}\n{}\n{}",
            "Unable to run the `yazi` file picker.",
            "You may need to install `yazi`:",
            "    https://yazi-rs.github.io/docs/installation"))?;

    if output.status.success() {
        let s = fs::read_to_string(tempfile_path)
            .with_context(|| "Unable to read file picker results from tmp file")?
            .trim().to_string();
        
        if s.is_empty() {
            Ok(None)
        } else {
            let note_path = PathBuf::from(s);
            if note_path.is_file() {
                Ok(Some(note_path))
            } else {
                Err(anyhow::anyhow!("Invalid filepath picked: {}", note_path.display()))
            }
        }
    } else {
        let out = format!("{}\n{}",
            String::from_utf8_lossy(&output.stdout),
            String::from_utf8_lossy(&output.stderr)
        );

        Err(anyhow::anyhow!("Picker failure: yazi :: {out}"))
    }
}