scythe_codegen/backends/
ruby_trilogy.rs

1use std::fmt::Write;
2use std::path::Path;
3
4use scythe_backend::manifest::{BackendManifest, load_manifest};
5use scythe_backend::naming::{
6    enum_type_name, enum_variant_name, fn_name, row_struct_name, to_pascal_case,
7};
8
9use scythe_core::analyzer::{AnalyzedQuery, CompositeInfo, EnumInfo};
10use scythe_core::errors::{ErrorCode, ScytheError};
11use scythe_core::parser::QueryCommand;
12
13use crate::backend_trait::{CodegenBackend, ResolvedColumn, ResolvedParam};
14
15const DEFAULT_MANIFEST_TOML: &str = include_str!("../../manifests/ruby-trilogy.toml");
16
17pub struct RubyTrilogyBackend {
18    manifest: BackendManifest,
19}
20
21impl RubyTrilogyBackend {
22    pub fn new(engine: &str) -> Result<Self, ScytheError> {
23        match engine {
24            "mysql" | "mariadb" => {}
25            _ => {
26                return Err(ScytheError::new(
27                    ErrorCode::InternalError,
28                    format!("ruby-trilogy only supports MySQL, got engine '{}'", engine),
29                ));
30            }
31        }
32        let manifest_path = Path::new("backends/ruby-trilogy/manifest.toml");
33        let manifest = if manifest_path.exists() {
34            load_manifest(manifest_path)
35                .map_err(|e| ScytheError::new(ErrorCode::InternalError, format!("manifest: {e}")))?
36        } else {
37            toml::from_str(DEFAULT_MANIFEST_TOML)
38                .map_err(|e| ScytheError::new(ErrorCode::InternalError, format!("manifest: {e}")))?
39        };
40        Ok(Self { manifest })
41    }
42}
43
44/// Map a neutral type to a Ruby type coercion method for trilogy.
45fn ruby_coercion(neutral_type: &str) -> &'static str {
46    match neutral_type {
47        "int16" | "int32" | "int64" => ".to_i",
48        "float32" | "float64" => ".to_f",
49        "bool" => " == 1",
50        _ => "",
51    }
52}
53
54impl CodegenBackend for RubyTrilogyBackend {
55    fn name(&self) -> &str {
56        "ruby-trilogy"
57    }
58
59    fn manifest(&self) -> &scythe_backend::manifest::BackendManifest {
60        &self.manifest
61    }
62
63    fn supported_engines(&self) -> &[&str] {
64        &["mysql"]
65    }
66
67    fn file_header(&self) -> String {
68        "# frozen_string_literal: true\n\n# Auto-generated by scythe. Do not edit.\n\nmodule Queries"
69            .to_string()
70    }
71
72    fn file_footer(&self) -> String {
73        "end".to_string()
74    }
75
76    fn generate_row_struct(
77        &self,
78        query_name: &str,
79        columns: &[ResolvedColumn],
80    ) -> Result<String, ScytheError> {
81        let struct_name = row_struct_name(query_name, &self.manifest.naming);
82        let fields = columns
83            .iter()
84            .map(|c| format!(":{}", c.field_name))
85            .collect::<Vec<_>>()
86            .join(", ");
87        let mut out = String::new();
88        let _ = writeln!(out, "  {} = Data.define({})", struct_name, fields);
89        Ok(out)
90    }
91
92    fn generate_model_struct(
93        &self,
94        table_name: &str,
95        columns: &[ResolvedColumn],
96    ) -> Result<String, ScytheError> {
97        let name = to_pascal_case(table_name);
98        self.generate_row_struct(&name, columns)
99    }
100
101    fn generate_query_fn(
102        &self,
103        analyzed: &AnalyzedQuery,
104        struct_name: &str,
105        columns: &[ResolvedColumn],
106        params: &[ResolvedParam],
107    ) -> Result<String, ScytheError> {
108        let func_name = fn_name(&analyzed.name, &self.manifest.naming);
109        let sql = super::clean_sql_with_optional(
110            &analyzed.sql,
111            &analyzed.optional_params,
112            &analyzed.params,
113        );
114        let mut out = String::new();
115
116        let param_list = params
117            .iter()
118            .map(|p| p.field_name.clone())
119            .collect::<Vec<_>>()
120            .join(", ");
121        let sep = if param_list.is_empty() { "" } else { ", " };
122
123        let param_array = if params.is_empty() {
124            "[]".to_string()
125        } else {
126            format!(
127                "[{}]",
128                params
129                    .iter()
130                    .map(|p| p.field_name.clone())
131                    .collect::<Vec<_>>()
132                    .join(", ")
133            )
134        };
135
136        match &analyzed.command {
137            QueryCommand::One => {
138                let _ = writeln!(out, "  def self.{}(client{}{})", func_name, sep, param_list);
139                let _ = writeln!(out, "    stmt = client.prepare(\"{}\")", sql);
140                let _ = writeln!(
141                    out,
142                    "    results = stmt.execute({})",
143                    param_array.trim_start_matches('[').trim_end_matches(']')
144                );
145                let _ = writeln!(out, "    row = results.first");
146                let _ = writeln!(out, "    return nil if row.nil?");
147
148                let fields = columns
149                    .iter()
150                    .enumerate()
151                    .map(|(i, c)| {
152                        let coercion = ruby_coercion(&c.neutral_type);
153                        if c.nullable {
154                            format!("{}: row[{}]&.then {{ |v| v{} }}", c.field_name, i, coercion)
155                        } else {
156                            format!("{}: row[{}]{}", c.field_name, i, coercion)
157                        }
158                    })
159                    .collect::<Vec<_>>()
160                    .join(", ");
161                let _ = writeln!(out, "    {}.new({})", struct_name, fields);
162            }
163            QueryCommand::Batch => {
164                let batch_fn_name = format!("{}_batch", func_name);
165                // Batch writes its own function signature (not the outer one)
166                let _ = writeln!(out, "  def self.{}(client, items)", batch_fn_name);
167                let _ = writeln!(out, "    stmt = client.prepare(\"{}\")", sql);
168                let _ = writeln!(out, "    items.each do |item|");
169                if params.len() > 1 {
170                    let _ = writeln!(out, "      stmt.execute(*item)");
171                } else if params.len() == 1 {
172                    let _ = writeln!(out, "      stmt.execute(item)");
173                } else {
174                    let _ = writeln!(out, "      stmt.execute");
175                }
176                let _ = writeln!(out, "    end");
177                let _ = write!(out, "  end");
178                return Ok(out);
179            }
180            QueryCommand::Many => {
181                let _ = writeln!(out, "  def self.{}(client{}{})", func_name, sep, param_list);
182                let _ = writeln!(out, "    stmt = client.prepare(\"{}\")", sql);
183                let _ = writeln!(
184                    out,
185                    "    results = stmt.execute({})",
186                    param_array.trim_start_matches('[').trim_end_matches(']')
187                );
188                let _ = writeln!(out, "    results.map do |row|");
189                let fields = columns
190                    .iter()
191                    .enumerate()
192                    .map(|(i, c)| {
193                        let coercion = ruby_coercion(&c.neutral_type);
194                        if c.nullable {
195                            format!("{}: row[{}]&.then {{ |v| v{} }}", c.field_name, i, coercion)
196                        } else {
197                            format!("{}: row[{}]{}", c.field_name, i, coercion)
198                        }
199                    })
200                    .collect::<Vec<_>>()
201                    .join(", ");
202                let _ = writeln!(out, "      {}.new({})", struct_name, fields);
203                let _ = writeln!(out, "    end");
204            }
205            QueryCommand::Exec => {
206                let _ = writeln!(out, "  def self.{}(client{}{})", func_name, sep, param_list);
207                let _ = writeln!(out, "    stmt = client.prepare(\"{}\")", sql);
208                let _ = writeln!(
209                    out,
210                    "    stmt.execute({})",
211                    param_array.trim_start_matches('[').trim_end_matches(']')
212                );
213                let _ = writeln!(out, "    nil");
214            }
215            QueryCommand::ExecResult | QueryCommand::ExecRows => {
216                let _ = writeln!(out, "  def self.{}(client{}{})", func_name, sep, param_list);
217                let _ = writeln!(out, "    stmt = client.prepare(\"{}\")", sql);
218                let _ = writeln!(
219                    out,
220                    "    stmt.execute({})",
221                    param_array.trim_start_matches('[').trim_end_matches(']')
222                );
223                let _ = writeln!(out, "    client.affected_rows");
224            }
225        }
226
227        let _ = write!(out, "  end");
228        Ok(out)
229    }
230
231    fn generate_enum_def(&self, enum_info: &EnumInfo) -> Result<String, ScytheError> {
232        let type_name = enum_type_name(&enum_info.sql_name, &self.manifest.naming);
233        let mut out = String::new();
234        let _ = writeln!(out, "  module {}", type_name);
235        for value in &enum_info.values {
236            let variant = enum_variant_name(value, &self.manifest.naming);
237            let _ = writeln!(out, "    {} = \"{}\"", variant, value);
238        }
239        let all_values = enum_info
240            .values
241            .iter()
242            .map(|v| enum_variant_name(v, &self.manifest.naming))
243            .collect::<Vec<_>>()
244            .join(", ");
245        let _ = writeln!(out, "    ALL = [{}].freeze", all_values);
246        let _ = write!(out, "  end");
247        Ok(out)
248    }
249
250    fn generate_composite_def(&self, composite: &CompositeInfo) -> Result<String, ScytheError> {
251        let name = to_pascal_case(&composite.sql_name);
252        let mut out = String::new();
253        if composite.fields.is_empty() {
254            let _ = writeln!(out, "  {} = Data.define()", name);
255        } else {
256            let fields = composite
257                .fields
258                .iter()
259                .map(|f| format!(":{}", f.name))
260                .collect::<Vec<_>>()
261                .join(", ");
262            let _ = writeln!(out, "  {} = Data.define({})", name, fields);
263        }
264        Ok(out)
265    }
266}
scythe_codegen/backends/ruby_trilogy.rs

scythe_codegen/backends/
ruby_trilogy.rs
