shep 0.1.22

//! `shep import`: reading a pm2 dump into a Flockfile.
//!
//! [`dump`] parses a `dump.pm2` document into rows; [`mod@env`] splits a
//! row's environment into what a Flockfile should carry and what the
//! operator has to decide about; [`convert`] collapses rows into apps,
//! wiring `env`'s split into every field this importer knows how to map;
//! [`render`] serializes the result as Flockfile TOML. [`import`] is the
//! verb itself, gluing all four together and reporting what happened.

pub(crate) mod convert;
pub(crate) mod dump;
pub(crate) mod env;
mod render;

use std::path::PathBuf;

use convert::ImportNote;

use crate::cli::ImportArgs;
use crate::exit::ExitCode;
use crate::output::{ImportRow, ImportRows, Streams, emit, write_outcome};

/// Reads a pm2 dump into apps and writes them out as a Flockfile.
///
/// Takes no [`shep_client::Client`] and connects to nothing: it reads one
/// file and writes another, so there is nothing to ask the socket.
/// `commands::logs::flush_daemon` is the other verb that finishes without a
/// client, for the same reason.
///
/// # Source
/// `args.from`, or `$HOME/.pm2/dump.pm2` if `args.from` names nothing.
/// Neither resolving is [`ExitCode::Usage`], naming both.
///
/// # Output
/// `args.out`, or `./Flockfile.toml`. Refuses to overwrite an existing file
/// unless `args.force` — also [`ExitCode::Usage`]. `args.dry_run` prints
/// the rendered Flockfile to stdout instead, with no envelope, and writes
/// nothing (decision 12): `shep import --dry-run > Flockfile.toml` must
/// produce a file `shep` can read back.
///
/// # Notes
/// Every [`ImportNote`] the conversion produced is written to stderr, one
/// line each: a clustered app that must set `SO_REUSEPORT` itself, an
/// ambiguous env key left out, an env value a Flockfile cannot hold, an
/// instance variable turned into an `env` entry templated with
/// `{{instance}}`. Both `--dry-run` and normal runs write them, and both
/// `--format table`/`--format json` (decision 13). The first line is the
/// read summary: how many instance rows, for how many apps, from where.
pub fn import(streams: &mut Streams<'_>, args: &ImportArgs) -> ExitCode {
    let source = match resolve_source(args) {
        Ok(source) => source,
        Err(message) => {
            return streams.fail(ExitCode::Usage, &message);
        }
    };

    let text = match std::fs::read_to_string(&source) {
        Ok(text) => text,
        Err(err) => {
            let message = if err.kind() == std::io::ErrorKind::NotFound {
                format!(
                    "no dump at {}; `pm2 save` writes one there",
                    source.display()
                )
            } else {
                format!("could not read {}: {err}", source.display())
            };
            return streams.fail(ExitCode::Usage, &message);
        }
    };

    let rows = match dump::parse(&text) {
        Ok(rows) => rows,
        Err(err) => {
            return streams.fail(ExitCode::InvalidConfig, &err.to_string());
        }
    };
    let instance_count = rows.len();

    let imported = match convert::convert(rows) {
        Ok(imported) => imported,
        Err(err) => {
            return streams.fail(ExitCode::InvalidConfig, &err.to_string());
        }
    };

    streams.aside(
        "read",
        &format!(
            "read {instance_count} instance rows for {} apps from {}",
            imported.apps.len(),
            source.display(),
        ),
    );
    for note in &imported.notes {
        let (code, message) = describe_note(note);
        streams.aside(code, &message);
    }

    let rendered = match render::flockfile(&imported.apps) {
        Ok(rendered) => rendered,
        Err(err) => {
            return streams.fail(ExitCode::Failure, &err.to_string());
        }
    };

    if args.dry_run {
        return write_outcome(streams.out.write_all(rendered.as_bytes()));
    }

    let out_path = args
        .out
        .clone()
        .unwrap_or_else(|| PathBuf::from("Flockfile.toml"));
    if !args.force && out_path.exists() {
        let message = format!(
            "{} already exists; pass --force to overwrite it",
            out_path.display()
        );
        return streams.fail(ExitCode::Usage, &message);
    }
    if let Err(err) = std::fs::write(&out_path, &rendered) {
        return streams.fail(ExitCode::Failure, &err.to_string());
    }

    let rows = ImportRows(
        imported
            .apps
            .iter()
            .map(|app| ImportRow {
                name: app.name.clone(),
                script: app.script.clone(),
                instances: app.instances,
                reuse_port: app.reuse_port,
            })
            .collect(),
    );
    write_outcome(emit(
        &mut *streams.out,
        streams.fmt,
        "import",
        rows,
        streams.style,
    ))
}

/// `args.from`, or `$HOME/.pm2/dump.pm2` if it names nothing.
///
/// # Errors
/// A message naming both `--from` and `$HOME`, when neither resolves a
/// path — this is the one case `resolve_paths` (`main.rs`) does not already
/// catch on this verb's behalf: `--home`/`$SHEP_HOME` can resolve
/// `ShepPaths` (which this verb never uses) while the real `$HOME` this
/// function reads is still unset.
fn resolve_source(args: &ImportArgs) -> Result<PathBuf, String> {
    if let Some(from) = &args.from {
        return Ok(from.clone());
    }
    std::env::var_os("HOME")
        .map(|home| PathBuf::from(home).join(".pm2").join("dump.pm2"))
        .ok_or_else(|| {
            "neither --from nor $HOME names a dump to read; pass --from, or set $HOME so \
             ~/.pm2/dump.pm2 resolves"
                .to_string()
        })
}

/// [`ImportNote`]'s stderr rendering: a stable code for `--format json`'s
/// `notice.code`, and the human message.
fn describe_note(note: &ImportNote) -> (&'static str, String) {
    match note {
        ImportNote::ClusterMode { app, instances } => (
            "cluster_mode",
            format!(
                "{app} ran {instances} instances in pm2 cluster mode; shep binds nothing, so \
                 {app} must set SO_REUSEPORT itself (Node's reusePort: true, needing Node >= \
                 22.12) or every instance past the first hits EADDRINUSE at start"
            ),
        ),
        ImportNote::InheritedEnv { app, key } => (
            "inherited_env",
            format!(
                "{app}: {key} was running but neither declared nor recognized session junk; \
                 decide whether it belongs in the Flockfile, the unit, or nowhere"
            ),
        ),
        ImportNote::UnrepresentableEnv { app, key } => (
            "unrepresentable_env",
            format!(
                "{app}: {key}'s value is not a string, number, or boolean, which a Flockfile \
                 env cannot hold, so it was dropped"
            ),
        ),
        ImportNote::InstanceVar { app, var } => (
            "instance_var",
            format!(
                "{app}: reads its instance number from ${var}; imported as \
                 {var} = \"{{{{instance}}}}\" under [app.env] rather than copied as a value"
            ),
        ),
    }
}

#[cfg(test)]
mod tests {
    use std::path::Path;

    use shep_core::config::{FlockFormat, Flockfile};

    use super::*;
    use crate::cli::Format;

    /// Mirrors `commands/trigger.rs`'s own one-line `args(..)` helper.
    fn args(dump: &Path, out: &Path, dry_run: bool, force: bool) -> ImportArgs {
        ImportArgs {
            from: Some(dump.to_path_buf()),
            out: Some(out.to_path_buf()),
            dry_run,
            force,
        }
    }

    /// fails if `import` writes over a Flockfile without being asked. The
    /// default output path is one an operator is likely to already have, and
    /// clobbering a hand-written one has nothing to undo it.
    #[test]
    fn an_existing_flockfile_is_not_overwritten_without_force() {
        let dir = tempfile::tempdir().unwrap();
        let out = dir.path().join("Flockfile.toml");
        std::fs::write(&out, "[[app]]\nname = \"mine\"\nscript = \"./mine\"\n").unwrap();
        let dump = dir.path().join("dump.pm2");
        std::fs::write(&dump, include_str!("testdata/dump.pm2.json")).unwrap();

        let mut out_buf = Vec::new();
        let mut err_buf = Vec::new();
        let code = {
            let mut streams = Streams {
                out: &mut out_buf,
                err: &mut err_buf,
                style: crate::style::Presentation::BARE,
                fmt: Format::Table,
            };
            import(&mut streams, &args(&dump, &out, false, false))
        };
        assert_eq!(code, ExitCode::Usage);
        assert!(std::fs::read_to_string(&out).unwrap().contains("mine"));
    }

    /// fails if `--dry-run` writes a file, or wraps the Flockfile in an
    /// envelope. `shep import --dry-run > Flockfile.toml` must produce a
    /// file shep can parse; an envelope makes it one shep cannot.
    #[test]
    fn dry_run_prints_a_parseable_flockfile_and_writes_nothing() {
        let dir = tempfile::tempdir().unwrap();
        let dump = dir.path().join("dump.pm2");
        std::fs::write(&dump, include_str!("testdata/dump.pm2.json")).unwrap();
        let out = dir.path().join("Flockfile.toml");

        let mut out_buf = Vec::new();
        let mut err_buf = Vec::new();
        let code = {
            let mut streams = Streams {
                out: &mut out_buf,
                err: &mut err_buf,
                style: crate::style::Presentation::BARE,
                fmt: Format::Table,
            };
            import(&mut streams, &args(&dump, &out, true, false))
        };
        assert_eq!(code, ExitCode::Success);
        assert!(!out.exists(), "--dry-run writes nothing");
        let printed = String::from_utf8(out_buf).unwrap();
        assert_eq!(
            Flockfile::parse(&printed, FlockFormat::Toml)
                .unwrap()
                .apps
                .len(),
            3,
            "`shep import --dry-run > Flockfile.toml` must produce a file \
             shep can read back: {printed}"
        );
    }

    /// fails if the cluster warning or an ambiguous env key stops reaching
    /// the operator. Both are cutover blockers the design names: one
    /// surfaces as EADDRINUSE at first start otherwise, the other as an
    /// app missing half its configuration after a reboot.
    #[test]
    fn the_report_names_every_cluster_app_and_every_ambiguous_key() {
        let dir = tempfile::tempdir().unwrap();
        let dump = dir.path().join("dump.pm2");
        std::fs::write(&dump, include_str!("testdata/dump.pm2.json")).unwrap();
        let out = dir.path().join("Flockfile.toml");

        let mut out_buf = Vec::new();
        let mut err_buf = Vec::new();
        {
            let mut streams = Streams {
                out: &mut out_buf,
                err: &mut err_buf,
                style: crate::style::Presentation::BARE,
                fmt: Format::Table,
            };
            let _ = import(&mut streams, &args(&dump, &out, true, false));
        }
        let report = String::from_utf8(err_buf).unwrap();
        assert!(report.contains("api"), "{report}");
        assert!(report.contains("SO_REUSEPORT"), "{report}");
        for key in ["BUN_INSTALL", "JAVA_HOME", "DATABASE_URL"] {
            assert!(report.contains(key), "{key} was never named: {report}");
        }
    }

    /// fails if a missing dump exits anything but `Usage`, or the message
    /// stops naming both the path and `pm2 save`.
    #[test]
    fn a_missing_dump_is_a_named_usage_error() {
        let dir = tempfile::tempdir().unwrap();
        let dump = dir.path().join("nope.pm2");
        let out = dir.path().join("Flockfile.toml");

        let mut out_buf = Vec::new();
        let mut err_buf = Vec::new();
        let code = {
            let mut streams = Streams {
                out: &mut out_buf,
                err: &mut err_buf,
                style: crate::style::Presentation::BARE,
                fmt: Format::Table,
            };
            import(&mut streams, &args(&dump, &out, false, false))
        };
        assert_eq!(code, ExitCode::Usage);
        let report = String::from_utf8(err_buf).unwrap();
        assert!(report.contains(&dump.display().to_string()), "{report}");
        assert!(report.contains("pm2 save"), "{report}");
    }

    /// fails if a dump that fails to parse exits anything but
    /// `InvalidConfig`.
    #[test]
    fn a_malformed_dump_is_invalid_config() {
        let dir = tempfile::tempdir().unwrap();
        let dump = dir.path().join("dump.pm2");
        std::fs::write(&dump, "not json").unwrap();
        let out = dir.path().join("Flockfile.toml");

        let mut out_buf = Vec::new();
        let mut err_buf = Vec::new();
        let code = {
            let mut streams = Streams {
                out: &mut out_buf,
                err: &mut err_buf,
                style: crate::style::Presentation::BARE,
                fmt: Format::Table,
            };
            import(&mut streams, &args(&dump, &out, false, false))
        };
        assert_eq!(code, ExitCode::InvalidConfig);
    }
}