forc/src/cli/commands/new.rs - master - githubmirror/sway

forc/src/cli/commands/new.rs
95 строк3 KB
Halimao
refactor: `forc init` generate unexpected project_name when project dir contains dot (#5455)
21 мар 2024, 05:25
21 мар 2024, 05:254d23798
use crate::{cli::init::Command as InitCommand, ops::forc_init::init};
use anyhow::anyhow;
use clap::Parser;
use forc_util::{forc_result_bail, validate_project_name, ForcResult};
use std::path::{Path, PathBuf};

forc_util::cli_examples! {
    crate::cli::Opt {
        [Create a new project => "forc new --contract --name my_project <PATH>"]
        [Create a new workspace => "forc new --workspace --name my_workspace <PATH>"]
        [Create a new Forc project with a predicate => "forc new <PATH> --predicate"]
        [Create a new Forc library project => "forc new <PATH> --library"]
    }
}

/// Create a new Forc project at `<path>`.
#[derive(Debug, Parser)]
#[clap(bin_name = "forc new", version, after_help = help())]
pub struct Command {
    /// The default program type. Excluding all flags or adding this flag creates a basic contract
    /// program.
    #[clap(long)]
    pub contract: bool,
    /// Adding this flag creates an empty script program.
    #[clap(long)]
    pub script: bool,
    /// Adding this flag creates an empty predicate program.
    #[clap(long)]
    pub predicate: bool,
    /// Adding this flag creates an empty library program.
    #[clap(long)]
    pub library: bool,
    /// Adding this flag creates an empty workspace.
    #[clap(long)]
    pub workspace: bool,
    /// Set the package name. Defaults to the directory name
    #[clap(long)]
    pub name: Option<String>,
    /// The path at which the project directory will be created.
    pub path: String,
}

pub(crate) fn exec(command: Command) -> ForcResult<()> {
    // `forc new` is roughly short-hand for `forc init`, but we create the directory first if it
    // doesn't yet exist. Here we create the directory if it doesn't exist then re-use the existing
    // `forc init` logic.
    let Command {
        contract,
        script,
        predicate,
        library,
        workspace,
        name,
        path,
    } = command;

    match &name {
        Some(name) => validate_project_name(name)?,
        None => {
            // If there is no name specified for the project, the last component of the `path` (directory name)
            // will be used by default so we should also check that.
            let project_path = PathBuf::from(&path);
            let directory_name = project_path
                .file_name()
                .ok_or_else(|| anyhow!("missing path for new command"))?
                .to_string_lossy();
            validate_project_name(&directory_name)?;
        }
    }

    let dir_path = Path::new(&path);
    if dir_path.exists() {
        forc_result_bail!(
            "Directory \"{}\" already exists.\nIf you wish to initialise a forc project inside \
            this directory, consider using `forc init --path {}`",
            dir_path.canonicalize()?.display(),
            dir_path.display(),
        );
    } else {
        std::fs::create_dir_all(dir_path)?;
    }

    let init_cmd = InitCommand {
        path: Some(path),
        contract,
        script,
        predicate,
        library,
        workspace,
        name,
    };

    init(init_cmd)?;
    Ok(())
}