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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
extern crate fern;
#[macro_use]
extern crate log;
extern crate clap;
use clap::App;
use clap::Arg;
use clap::SubCommand;
use diesel::connection::Connection;
use diesel::pg::PgConnection;
fn setup_logger() -> Result<(), fern::InitError> {
use fern::colors::ColoredLevelConfig;
let colors = ColoredLevelConfig::new();
fern::Dispatch::new()
.format(move |out, message, record| {
out.finish(format_args!("[{}] {}",
colors.color(record.level()),
message))
})
.level(log::LevelFilter::Info)
.chain(std::io::stdout())
.apply()?;
Ok(())
}
fn main() {
let matches = App::new("pjournal")
.version("0.1")
.author("KJ Ørbekk <kj@orbekk.com>")
.about("Practice Journaling")
.arg(
Arg::with_name("database_url")
.long("database_url")
.required(true)
.takes_value(true)
.help("URL to postgresql database"),
)
.arg(
Arg::with_name("base_url")
.long("base_url")
.takes_value(true)
.help("Endpoint for this web server"),
)
.subcommand(
SubCommand::with_name("init")
.about("initialize database config")
.arg(
Arg::with_name("rocket_secret_key")
.long("rocket_secret_key")
.takes_value(true)
.required(true)
.help("Secret passed to rocket for encrypted cookies"),
)
.arg(
Arg::with_name("strava_client_secret")
.long("strava_client_secret")
.takes_value(true)
.required(true)
.help("Client secret for strava authentication"),
)
.arg(
Arg::with_name("strava_client_id")
.long("strava_client_id")
.takes_value(true)
.required(true)
.help("Client id for strava authentication"),
),
)
.subcommand(
SubCommand::with_name("adduser")
.about("add a user account")
.arg(Arg::with_name("USERNAME").required(true).index(1))
.arg(Arg::with_name("PASSWORD").required(true).index(2)),
)
.get_matches();
setup_logger().expect("logger");
let base_url = matches
.value_of("base_url")
.unwrap_or("http://localhost:8000");
let db_url = matches.value_of("database_url").unwrap();
let conn = PgConnection::establish(db_url).unwrap();
if let Some(matches) = matches.subcommand_matches("init") {
let config = pjournal::models::Config {
strava_client_id: matches.value_of("strava_client_id").unwrap().to_string(),
strava_client_secret: matches
.value_of("strava_client_secret")
.unwrap()
.to_string(),
rocket_secret_key: matches.value_of("rocket_secret_key").unwrap().to_string(),
singleton: true,
};
pjournal::db::create_config(&conn, &config).unwrap();
} else if let Some(matches) = matches.subcommand_matches("adduser") {
let user = matches.value_of("USERNAME").unwrap();
let password = matches.value_of("PASSWORD").unwrap();
pjournal::db::adduser(&conn, user, password).unwrap();
} else {
info!("Start server");
pjournal::server::start(conn, db_url, base_url);
}
}
|