sui_move/
new.rs

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
// Copyright (c) Mysten Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

use clap::Parser;
use move_cli::base::new;
use move_package::source_package::layout::SourcePackageLayout;
use std::{fs::create_dir_all, io::Write, path::Path};

#[derive(Parser)]
#[group(id = "sui-move-new")]
pub struct New {
    #[clap(flatten)]
    pub new: new::New,
}

impl New {
    pub fn execute(self, path: Option<&Path>) -> anyhow::Result<()> {
        let name = &self.new.name.to_lowercase();
        let provided_name = &self.new.name.to_string();

        self.new
            .execute(path, [] as [(&str, &str); 0], [(name, "0x0")], "")?;
        let p = path.unwrap_or_else(|| Path::new(&provided_name));
        let mut w = std::fs::File::create(
            p.join(SourcePackageLayout::Sources.path())
                .join(format!("{name}.move")),
        )?;
        writeln!(
            w,
            r#"/*
/// Module: {name}
module {name}::{name};
*/

// For Move coding conventions, see
// https://docs.sui.io/concepts/sui-move-concepts/conventions

"#,
            name = name
        )?;

        create_dir_all(p.join(SourcePackageLayout::Tests.path()))?;
        let mut w = std::fs::File::create(
            p.join(SourcePackageLayout::Tests.path())
                .join(format!("{name}_tests.move")),
        )?;
        writeln!(
            w,
            r#"/*
#[test_only]
module {name}::{name}_tests;
// uncomment this line to import the module
// use {name}::{name};

const ENotImplemented: u64 = 0;

#[test]
fun test_{name}() {{
    // pass
}}

#[test, expected_failure(abort_code = ::{name}::{name}_tests::ENotImplemented)]
fun test_{name}_fail() {{
    abort ENotImplemented
}}
*/"#,
            name = name
        )?;

        Ok(())
    }
}