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
84

#[macro_use]
extern crate log;
#[macro_use]
extern crate codeviz_macros;
extern crate codeviz_java;
extern crate reproto_backend;

mod builder;
mod constructor_properties;
mod fasterxml;
mod java_backend;
mod java_compiler;
mod java_options;
mod listeners;
mod lombok;
mod models;
mod mutable;
mod nullable;

pub(crate) use codeviz_java::*;
pub(crate) use reproto_backend::errors::*;
pub(crate) use reproto_backend::imports::*;
pub(crate) use self::java_backend::*;
pub(crate) use self::java_compiler::*;
pub(crate) use self::java_options::*;
pub(crate) use self::listeners::*;
pub(crate) use self::models::*;

pub const JAVA_CONTEXT: &str = "java";

fn setup_module(module: &str) -> Result<Box<Listeners>> {
    let module: Box<Listeners> = match module {
        "builder" => Box::new(builder::Module::new()),
        "constructor_properties" => Box::new(constructor_properties::Module::new()),
        "fasterxml" => Box::new(fasterxml::Module::new()),
        "lombok" => Box::new(lombok::Module::new()),
        "mutable" => Box::new(mutable::Module::new()),
        "nullable" => Box::new(nullable::Module::new()),
        _ => return Err(format!("No such module: {}", module).into()),
    };

    Ok(module)
}

pub fn setup_listeners(options: Options) -> Result<(JavaOptions, Box<Listeners>)> {
    let mut listeners: Vec<Box<Listeners>> = Vec::new();

    for module in &options.modules {
        listeners.push(setup_module(module)?);
    }

    let mut options = JavaOptions::new();

    for listener in &listeners {
        listener.configure(&mut options)?;
    }

    Ok((options, Box::new(listeners)))
}

pub fn compile_options<'a, 'b>(out: App<'a, 'b>) -> App<'a, 'b> {
    out.about("Compile for Java")
}

pub fn verify_options<'a, 'b>(out: App<'a, 'b>) -> App<'a, 'b> {
    out.about("Verify for Java")
}

pub fn compile(env: Environment,
               options: Options,
               compiler_options: CompilerOptions,
               _matches: &ArgMatches)
               -> Result<()> {
    let (options, listeners) = setup_listeners(options)?;
    let backend = JavaBackend::new(env, options, listeners);
    let compiler = backend.compiler(compiler_options)?;
    compiler.compile()
}

pub fn verify(env: Environment, options: Options, _matches: &ArgMatches) -> Result<()> {
    let (options, listeners) = setup_listeners(options)?;
    let backend = JavaBackend::new(env, options, listeners);
    backend.verify()
}