projkt 0.2.1

Generate project related stuffs such as .gitignore, license, readme and more 
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

import sys
from os import path
import subprocess

code = """use crate::plugins::{{
    types::{{Plugin, ProjktResult}},
}};

pub struct {0}Options;

pub struct {0};

impl Plugin for {0} {{
    type Opts = {0}Options;
    type Fetch = ();

    fn fetch(_: &Self::Opts) -> ProjktResult<Self::Fetch> {{
        unimplemented!()
    }}

    fn exec(_: Self::Opts) -> ProjktResult<()> {{
        unimplemented!()
    }}
}}
"""


def create_plugin(name):
    dest = path.join(
        path.realpath(path.curdir), "src", "plugins")

    if not path.exists(dest):
        print("Invoke this from the root of directory")
        exit(1)

    file_name = f"{path.join(dest, str.lower(name))}.rs"

    with open(file_name, "w") as f:
        name = str.capitalize(name)
        f.write(code.format(name))

    with open(f"{dest}.rs", "a") as f:
        f.write(f"\npub mod {str.lower(name)};")


def main():
    match sys.argv[1:]:
        case []:
            print("Please provide a module name")

        case [plugin_name]:
            create_plugin(plugin_name)

            subprocess.run(["cargo", "fmt"])

        case _:
            print("Too many arguments provided, takes max 1 args.")


if __name__ == "__main__":
    main()