rust-postgres/codegen/src/sqlstate.rs
2016-12-20 16:07:45 -08:00

150 lines
3.8 KiB
Rust

use std::fs::File;
use std::io::{Write, BufWriter};
use std::path::Path;
use phf_codegen;
use snake_to_camel;
const ERRCODES_TXT: &'static str = include_str!("errcodes.txt");
struct Code {
code: String,
variant: String,
}
pub fn build(path: &Path) {
let mut file = BufWriter::new(File::create(path.join("postgres-shared/src/error/sqlstate.rs")).unwrap());
let codes = parse_codes();
make_header(&mut file);
make_enum(&codes, &mut file);
make_map(&codes, &mut file);
make_impl(&codes, &mut file);
}
fn parse_codes() -> Vec<Code> {
let mut codes = vec![];
for line in ERRCODES_TXT.lines() {
if line.starts_with("#") || line.starts_with("Section") || line.trim().is_empty() {
continue;
}
let mut it = line.split_whitespace();
let code = it.next().unwrap().to_owned();
it.next();
it.next();
// for 2202E
let name = match it.next() {
Some(name) => name,
None => continue,
};
let variant = match variant_name(&code) {
Some(variant) => variant,
None => snake_to_camel(&name),
};
codes.push(Code {
code: code,
variant: variant,
});
}
codes
}
fn variant_name(code: &str) -> Option<String> {
match code {
"01004" => Some("WarningStringDataRightTruncation".to_owned()),
"22001" => Some("DataStringDataRightTruncation".to_owned()),
"2F002" => Some("SqlRoutineModifyingSqlDataNotPermitted".to_owned()),
"38002" => Some("ForeignRoutineModifyingSqlDataNotPermitted".to_owned()),
"2F003" => Some("SqlRoutineProhibitedSqlStatementAttempted".to_owned()),
"38003" => Some("ForeignRoutineProhibitedSqlStatementAttempted".to_owned()),
"2F004" => Some("SqlRoutineReadingSqlDataNotPermitted".to_owned()),
"38004" => Some("ForeignRoutineReadingSqlDataNotPermitted".to_owned()),
"22004" => Some("DataNullValueNotAllowed".to_owned()),
"39004" => Some("ExternalRoutineInvocationNullValueNotAllowed".to_owned()),
_ => None,
}
}
fn make_header(file: &mut BufWriter<File>) {
write!(file,
"// Autogenerated file - DO NOT EDIT
use phf;
"
).unwrap();
}
fn make_enum(codes: &[Code], file: &mut BufWriter<File>) {
write!(file,
r#"/// SQLSTATE error codes
#[derive(PartialEq, Eq, Clone, Debug)]
#[allow(enum_variant_names)]
pub enum SqlState {{
"#
).unwrap();
for code in codes {
write!(file,
" /// `{}`
{},\n",
code.code, code.variant).unwrap();
}
write!(file,
" /// An unknown code
Other(String),
}}
"
).unwrap();
}
fn make_map(codes: &[Code], file: &mut BufWriter<File>) {
write!(file,
"#[cfg_attr(rustfmt, rustfmt_skip)]
static SQLSTATE_MAP: phf::Map<&'static str, SqlState> = ").unwrap();
let mut builder = phf_codegen::Map::new();
for code in codes {
builder.entry(&*code.code, &format!("SqlState::{}", code.variant));
}
builder.build(file).unwrap();
write!(file, ";\n").unwrap();
}
fn make_impl(codes: &[Code], file: &mut BufWriter<File>) {
write!(file, r#"
impl SqlState {{
/// Creates a `SqlState` from its error code.
pub fn from_code(s: &str) -> SqlState {{
match SQLSTATE_MAP.get(s) {{
Some(state) => state.clone(),
None => SqlState::Other(s.to_owned()),
}}
}}
/// Returns the error code corresponding to the `SqlState`.
pub fn code(&self) -> &str {{
match *self {{"#
).unwrap();
for code in codes {
write!(file, r#"
SqlState::{} => "{}","#,
code.variant, code.code).unwrap();
}
write!(file, r#"
SqlState::Other(ref s) => s,
}}
}}
}}
"#
).unwrap();
}