94 lines
2.5 KiB
Rust
94 lines
2.5 KiB
Rust
|
use serde::Deserialize;
|
||
|
|
use std::fs;
|
||
|
|
use std::path::Path;
|
||
|
|
use std::sync::Arc;
|
||
|
|
|
||
|
|
use crate::law::{self, responsible::*};
|
||
|
|
use crate::law::{ClassifierApplicable, LawBuilder};
|
||
|
|
use crate::misc::Error;
|
||
|
|
use crate::risparser::paragraph::Parser;
|
||
|
|
|
||
|
|
// TODO: more generic
|
||
|
|
fn create_classifier(match_function: &str) -> Result<ClassifierApplicable, Error> {
|
||
|
|
let func: ClassifierApplicable = match match_function {
|
||
|
|
"contains" => Arc::new(contains),
|
||
|
|
"starts_with_roman_number" => Arc::new(starts_with_roman_number),
|
||
|
|
"contains_at_start" => Arc::new(contains_at_start),
|
||
|
|
"starts_with_number" => Arc::new(starts_with_number),
|
||
|
|
"starts_with_letter" => Arc::new(starts_with_letter),
|
||
|
|
"starts_with_uppercaseletter" => Arc::new(starts_with_letter),
|
||
|
|
"contains_without_unter" => Arc::new(contains_without_unter),
|
||
|
|
_ => {
|
||
|
|
return Err(Error::new(&format!(
|
||
|
|
"Unknown match function: {}",
|
||
|
|
match_function
|
||
|
|
)))
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
Ok(func)
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Deserialize)]
|
||
|
|
pub struct Config {
|
||
|
|
law: Law,
|
||
|
|
parser: ParserConfig,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Config {
|
||
|
|
pub fn load<P: AsRef<Path>>(path: P) -> Result<(usize, LawBuilder, Parser), Error> {
|
||
|
|
let config_str = fs::read_to_string(path)?;
|
||
|
|
let config: Config = toml::from_str(&config_str)?;
|
||
|
|
|
||
|
|
let mut builder = LawBuilder::new();
|
||
|
|
for classifier in config.law.classifiers {
|
||
|
|
let to_add = law::Classifier::new(
|
||
|
|
&classifier.name,
|
||
|
|
create_classifier(&classifier.match_function)?,
|
||
|
|
);
|
||
|
|
if classifier.is_root {
|
||
|
|
builder.add_classifier(to_add.root());
|
||
|
|
} else {
|
||
|
|
builder.add_classifier(to_add);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
let mut parser = Parser::new();
|
||
|
|
|
||
|
|
for to_remove in config.parser.remove_strings {
|
||
|
|
parser.add_string_to_remove(&to_remove);
|
||
|
|
}
|
||
|
|
|
||
|
|
for to_replace in config.parser.replace_rules {
|
||
|
|
parser.add_string_to_replace(&to_replace.find, &to_replace.replace_with);
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok((config.law.id, builder, parser))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Deserialize)]
|
||
|
|
struct Law {
|
||
|
|
id: usize,
|
||
|
|
classifiers: Vec<Classifier>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Deserialize)]
|
||
|
|
struct Classifier {
|
||
|
|
name: String,
|
||
|
|
is_root: bool,
|
||
|
|
match_function: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Deserialize)]
|
||
|
|
struct ParserConfig {
|
||
|
|
remove_strings: Vec<String>,
|
||
|
|
replace_rules: Vec<ReplaceRule>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Deserialize)]
|
||
|
|
struct ReplaceRule {
|
||
|
|
find: String,
|
||
|
|
replace_with: String,
|
||
|
|
}
|