1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use clap::{ArgMatches, Command};

use super::SeaplaneFlightCommonArgMatches;
use crate::{
    cli::{
        cmds::flight::common,
        errors::wrap_cli_context,
        specs::IMAGE_SPEC,
        validator::{validate_flight_name, validate_name_id},
        CliCommand,
    },
    context::{Ctx, FlightCtx},
    error::Result,
};

#[derive(Copy, Clone, Debug)]
pub struct SeaplaneFlightCopy;

impl SeaplaneFlightCopy {
    pub fn command() -> Command<'static> {
        let validator = |s: &str| validate_name_id(validate_flight_name, s);

        // TODO: add --from
        Command::new("copy")
            .visible_alias("clone")
            .about("Copy a local Flight Plan (optionally make changes to the copy)")
            .after_help(IMAGE_SPEC)
            .override_usage(
                "seaplane flight copy <NAME|ID> --name=<DEST_NAME> [OPTIONS]
    seaplane flight copy <NAME|ID> [OPTIONS]",
            ) // TODO: DEST_NAME is never used?!
            .arg(
                arg!(name_id =["NAME|ID"] required)
                    .validator(validator)
                    .help("The source name or ID of the Flight Plan to copy"),
            )
            .arg(arg!(--exact - ('x')).help("The given SOURCE must be an exact match"))
            .args(common::args(false))
    }
}

impl CliCommand for SeaplaneFlightCopy {
    fn run(&self, ctx: &mut Ctx) -> Result<()> {
        if ctx.args.stateless {
            cli_eprint!(@Red, "error: ");
            cli_eprint!("'");
            cli_eprint!(@Yellow, "--stateless");
            cli_eprint!("' cannot be used with the '");
            cli_eprint!(@Yellow, "seaplane flight copy");
            cli_eprintln!("' command");
            cli_eprintln!("(hint: 'seaplane flight ...' only modifies local plans)");
            std::process::exit(1);
        }
        // name_id cannot be None in `flight copy`
        let mut dest_flight = match ctx
            .db
            .flights
            .clone_flight(ctx.args.name_id.as_ref().unwrap(), ctx.args.exact)
        {
            Ok(f) => f,
            Err(e) => return wrap_cli_context(e, ctx.args.exact, false),
        };

        // Now we just edit the newly copied Flight to match the given CLI params...
        dest_flight.update_from(ctx.flight_ctx.get_or_init(), false)?;

        let id = dest_flight.id.to_string();
        let name = dest_flight.model.name().to_owned();

        // Add the new Flight
        ctx.db.flights.add_flight(dest_flight);

        ctx.persist_flights()?;

        cli_print!("Successfully copied Flight Plan '");
        cli_print!(@Yellow, "{}", ctx.args.name_id.as_ref().unwrap());
        cli_print!("' to new Flight Plan '");
        cli_print!(@Green, "{name}");
        cli_print!("' with ID '");
        cli_print!(@Green, "{}", &id[..8]);
        cli_println!("'");

        Ok(())
    }

    fn update_ctx(&self, matches: &ArgMatches, ctx: &mut Ctx) -> Result<()> {
        // clap will not let "source" be None
        ctx.args.name_id = matches.get_one::<String>("name_id").map(ToOwned::to_owned);
        ctx.flight_ctx
            .init(FlightCtx::from_flight_common(&SeaplaneFlightCommonArgMatches(matches), "")?);
        Ok(())
    }
}