cargo-pipe 0.2.2

A command-line tool to generate and build data integration app using pipebase framework
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

use super::do_cargo::*;
use super::utils::*;
use crate::{
    commands::generate::GenerateOptions,
    config::Config,
    errors::{cargo_error, CmdResult},
    print::Printer,
};
use pipegen::models::App;
use std::fs;
use std::path::{Path, PathBuf};

fn do_apply_additional_dependency(
    app: &App,
    toml_path: &Path,
    printer: &mut Printer,
) -> CmdResult<()> {
    printer.status("Generate", "add toml manifest dependencies")?;
    // fetch package dependency
    let additionals = app.get_dependencies();
    let toml_content = fs::read_to_string(toml_path)?;
    let mut manifest = toml::from_str::<PipeTomlManifest>(&toml_content)?;
    manifest.init();
    for add in additionals.iter().cloned() {
        manifest.add_dependency(add.get_name(), add.into());
    }
    let toml_content = toml::to_string(&manifest)?;
    fs::write(toml_path, toml_content)?;
    printer.status("Generate", "toml manifest dependencies added")?;
    Ok(())
}

pub fn do_generate(
    app: &App,
    mut path_buf: PathBuf,
    opts: &GenerateOptions,
    printer: &mut Printer,
) -> CmdResult<()> {
    let main_path = path_buf
        .as_path()
        .to_str()
        .expect("invalid main path")
        .to_owned();
    printer.status(&"Generate", &main_path)?;
    let contents = match opts.get_pipe_name() {
        Some(pipe_name) => app.generate_pipes(pipe_name)?,
        None => app.generate(),
    };
    // pop main.rs
    path_buf.pop();
    // pop src
    path_buf.pop();
    path_buf.push(CARGO_MANIFEST_FILE);
    let toml_manifest_path = path_buf.as_path();
    if let Err(err) = fs::write(&main_path, contents) {
        if !toml_manifest_path.exists() {
            printer.error("no app found, try 'cargo pipe new' first")?
        }
        return Err(err.into());
    }
    // replace dependency in cargo.toml
    do_apply_additional_dependency(app, toml_manifest_path, printer)?;
    // cargo format
    let status_code = do_cargo_fmt(toml_manifest_path, printer)?;
    match status_code {
        0 => (),
        _ => return Err(cargo_error("fmt", status_code)),
    };
    printer.status(&"Generate", "succeed")
}

pub fn do_exec(config: &Config, opts: &GenerateOptions) -> CmdResult<()> {
    let mut printer = Printer::new();
    let pipe_manifest_path = config.get_pipe_manifest_path();
    let app = read_pipe_manifest(pipe_manifest_path.as_path(), &mut printer)?;
    let path_buf = config.get_app_main_path(opts.get_app_name());
    do_generate(&app, path_buf, opts, &mut printer)?;
    Ok(())
}