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

// #[macro_use]
// extern crate itertools;

extern crate syntex_syntax as syntax;
extern crate syntex_errors as errors;
extern crate regex;
extern crate mysql;
extern crate rustc_serialize;

// pub use rustc_serialize::json;

#[macro_use]
mod macros;
mod formatter;
mod visitor;
mod attr;
mod entity;
mod db;
// mod session;
mod cond;
mod value;
mod insert;
mod select;

pub mod init;
pub mod meta;

pub use entity::Entity;
pub use entity::EntityInner;
pub use entity::EntityInnerPointer;
pub use meta::FieldMeta;
pub use meta::EntityMeta;
pub use meta::OrmMeta;
pub use mysql::Value;
pub use mysql::Row;
pub use db::Db;
pub use insert::Insert;
pub use select::Select;
pub use cond::Cond;
pub use cond::JoinCond;
// pub use value::FieldValue;

use syntax::codemap::CodeMap;
use syntax::parse::{self, ParseSess};
use errors::emitter::ColorConfig;
use errors::Handler;

use std::rc::Rc;

fn create_parse_session() -> ParseSess {
    let codemap = Rc::new(CodeMap::new());
    let tty_handler =
        Handler::with_tty_emitter(ColorConfig::Auto, true, false, Some(codemap.clone()));
    let parse_session = ParseSess::with_span_handler(tty_handler, codemap.clone());
    parse_session
}

pub fn build(src: &str) -> String {
    let parse_session = create_parse_session();
    let krate =
        parse::parse_crate_from_source_str("stdin".to_string(), src.to_string(), &parse_session)
            .unwrap();
    let meta = visitor::visit_krate(&krate);
    let ret = formatter::format_meta(&meta);
    ret
}

pub fn open(user: &str,
            pwd: &str,
            host: &str,
            port: u16,
            db: &str,
            orm_meta: &'static OrmMeta)
            -> Result<Db, mysql::Error> {
    let conn_str = format!("mysql://{}:{}@{}:{}/{}", user, pwd, host, port, db);
    match mysql::Pool::new(conn_str.as_ref()) {
        Ok(pool) => Ok(Db::new(pool, orm_meta)),
        Err(err) => Err(err),
    }
}