loco-rs 0.4.0

The one-person framework for Rust
Documentation 
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
80
81
82
83

use rrgen::RRgen;
use serde_json::json;

use crate::{app::Hooks, gen};

const API_CONTROLLER_SCAFFOLD_T: &str = include_str!("templates/scaffold/api/controller.t");

const HTMX_CONTROLLER_SCAFFOLD_T: &str = include_str!("templates/scaffold/htmx/controller.t");
const HTMX_VIEW_SCAFFOLD_T: &str = include_str!("templates/scaffold/htmx/view.t");
const HTMX_VIEW_EDIT_SCAFFOLD_T: &str = include_str!("templates/scaffold/htmx/view_edit.t");
const HTMX_VIEW_CREATE_SCAFFOLD_T: &str = include_str!("templates/scaffold/htmx/view_create.t");
const HTMX_VIEW_SHOW_SCAFFOLD_T: &str = include_str!("templates/scaffold/htmx/view_show.t");
const HTMX_VIEW_LIST_SCAFFOLD_T: &str = include_str!("templates/scaffold/htmx/view_list.t");

const HTML_CONTROLLER_SCAFFOLD_T: &str = include_str!("templates/scaffold/html/controller.t");
const HTML_VIEW_SCAFFOLD_T: &str = include_str!("templates/scaffold/html/view.t");
const HTML_VIEW_EDIT_SCAFFOLD_T: &str = include_str!("templates/scaffold/html/view_edit.t");
const HTML_VIEW_CREATE_SCAFFOLD_T: &str = include_str!("templates/scaffold/html/view_create.t");
const HTML_VIEW_SHOW_SCAFFOLD_T: &str = include_str!("templates/scaffold/html/view_show.t");
const HTML_VIEW_LIST_SCAFFOLD_T: &str = include_str!("templates/scaffold/html/view_list.t");

use super::{collect_messages, model, CONTROLLER_TEST_T, MAPPINGS};
use crate::{errors::Error, Result};

pub fn generate<H: Hooks>(
    rrgen: &RRgen,
    name: &str,
    fields: &[(String, String)],
    kind: &gen::ScaffoldKind,
) -> Result<String> {
    // - scaffold is never a link table
    // - never run with migration_only, because the controllers will refer to the
    //   models. the models only arrive after migration and entities sync.
    let model_messages = model::generate::<H>(rrgen, name, false, false, fields)?;

    let mut columns = Vec::new();
    for (fname, ftype) in fields {
        if gen::model::IGNORE_FIELDS.contains(&fname.as_str()) {
            tracing::warn!(
                field = fname,
                "note that a redundant field was specified, it is already generated automatically"
            );
            continue;
        }
        if ftype != "references" {
            let schema_type = MAPPINGS.rust_field(ftype.as_str()).ok_or_else(|| {
                Error::Message(format!(
                    "type: {} not found. try any of: {:?}",
                    ftype,
                    MAPPINGS.rust_fields()
                ))
            })?;
            columns.push((fname.to_string(), schema_type.as_str(), ftype));
        }
    }
    let vars = json!({"name": name, "columns": columns, "pkg_name": H::app_name()});
    match kind {
        gen::ScaffoldKind::Api => {
            let res1 = rrgen.generate(API_CONTROLLER_SCAFFOLD_T, &vars)?;
            let res2 = rrgen.generate(CONTROLLER_TEST_T, &vars)?;
            let messages = collect_messages(vec![res1, res2]);
            Ok(format!("{model_messages}{messages}"))
        }
        gen::ScaffoldKind::Html => {
            rrgen.generate(HTML_CONTROLLER_SCAFFOLD_T, &vars)?;
            rrgen.generate(HTML_VIEW_EDIT_SCAFFOLD_T, &vars)?;
            rrgen.generate(HTML_VIEW_CREATE_SCAFFOLD_T, &vars)?;
            rrgen.generate(HTML_VIEW_SHOW_SCAFFOLD_T, &vars)?;
            rrgen.generate(HTML_VIEW_LIST_SCAFFOLD_T, &vars)?;
            rrgen.generate(HTML_VIEW_SCAFFOLD_T, &vars)?;
            Ok(model_messages)
        }
        gen::ScaffoldKind::Htmx => {
            rrgen.generate(HTMX_CONTROLLER_SCAFFOLD_T, &vars)?;
            rrgen.generate(HTMX_VIEW_EDIT_SCAFFOLD_T, &vars)?;
            rrgen.generate(HTMX_VIEW_CREATE_SCAFFOLD_T, &vars)?;
            rrgen.generate(HTMX_VIEW_SHOW_SCAFFOLD_T, &vars)?;
            rrgen.generate(HTMX_VIEW_LIST_SCAFFOLD_T, &vars)?;
            rrgen.generate(HTMX_VIEW_SCAFFOLD_T, &vars)?;
            Ok(model_messages)
        }
    }
}