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
|
#[macro_use]
extern crate clap;
use clap::App;
use clap::Arg;
use clap::SubCommand;
use diesel::connection::Connection;
use diesel::pg::PgConnection;
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();
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(),
strava_client_secret: matches.value_of("strava_client_secret").unwrap(),
rocket_secret_key: matches.value_of("rocket_secret_key").unwrap(),
};
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 {
let config = pjournal::server::Params {
base_url: base_url.to_string(),
};
pjournal::server::start(db_url, config);
}
}
|
