rvy 0.1.2

A modular Rust CLI framework for generating services, tools, and project scaffolds.
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

use crate::context::Context;
use crate::generator::{render, write_file, update_module_exports};

pub fn generate(ctx: &Context, name: &str, db_type: &str) {
    let template = match db_type.to_lowercase().as_str() {
        "postgres" => include_str!("../../templates/adapter/postgres.rs.tpl"),
        "mysql" => include_str!("../../templates/adapter/mysql.rs.tpl"),
        "mongodb" => include_str!("../../templates/adapter/mongodb.rs.tpl"),
        "sqlite" => include_str!("../../templates/adapter/sqlite.rs.tpl"),
        _ => {
            eprintln!("Error: Unsupported database type '{}'", db_type);
            eprintln!("Supported types: postgres, mysql, mongodb, sqlite");
            return;
        }
    };

    let content = render(template, name);

    let filename = format!("{}_{}.rs", name, db_type.to_lowercase());
    let path = format!("src/adapter/{}", filename);

    write_file(ctx, &path, &content);
    update_module_exports(ctx, "src/adapter/mod.rs", &format!("{}_{}", name, db_type.to_lowercase()));
}

pub fn generate_all(ctx: &Context, name: &str) {
    let db_types = ["postgres", "mysql", "mongodb", "sqlite"];
    
    for db_type in &db_types {
        generate(ctx, name, db_type);
    }
}

pub fn generate_config(ctx: &Context, name: &str) {
    let template = include_str!("../../templates/config/database.rs.tpl");
    let content = render(template, name);
    let path = "src/config/database.rs";
    
    write_file(ctx, path, &content);
    
    // Update mod.rs to export database
    update_module_exports(ctx, "src/config/mod.rs", "database");
}

pub fn generate_factory(ctx: &Context, name: &str) {
    let template = include_str!("../../templates/factory/repository_factory.rs.tpl");
    let content = render(template, name);
    let path = format!("src/factory/{}_factory.rs", name);
    
    write_file(ctx, &path, &content);
    
    // Update mod.rs to export this factory
    let module_name = format!("{}_factory", name);
    update_module_exports(ctx, "src/factory/mod.rs", &module_name);
}

pub fn generate_usage_docs(ctx: &Context, name: &str) {
    // Generate example main
    let main_template = include_str!("../../templates/examples/main_with_crud.rs.tpl");
    let main_content = render(main_template, name);
    let main_path = format!("examples/{}_example.rs", name);
    write_file(ctx, &main_path, &main_content);

    // Generate usage documentation
    let doc_template = include_str!("../../templates/examples/USAGE.md.tpl");
    let doc_content = render(doc_template, name);
    let doc_path = format!("docs/{}_USAGE.md", name);
    write_file(ctx, &doc_path, &doc_content);
}