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
107
108
109
110
111
|
use std::fs::File;
use std::io::{Read, Write};
use std::path::PathBuf;
use futures::future::{FutureExt, TryFutureExt};
use futures01::future::Future as Future01;
use log::{debug, info};
use structopt::clap::AppSettings;
use structopt::StructOpt;
use tsclientlib::Identity;
mod audio_player;
mod bot;
mod command;
mod playlist;
mod teamspeak;
mod youtube_dl;
use bot::{MasterArgs, MasterBot};
#[derive(StructOpt, Debug)]
#[structopt(raw(global_settings = "&[AppSettings::ColoredHelp]"))]
pub struct Args {
#[structopt(short = "l", long = "local", help = "Run locally in text mode")]
local: bool,
#[structopt(
short = "g",
long = "generate-identities",
help = "Generate 'count' identities"
)]
gen_id_count: Option<u8>,
#[structopt(
short = "a",
long = "address",
help = "The address of the server to connect to"
)]
address: Option<String>,
#[structopt(
help = "Configuration file",
parse(from_os_str),
default_value = "config.toml"
)]
config_path: PathBuf,
#[structopt(
short = "d",
long = "master_channel",
help = "The channel the master bot should connect to"
)]
master_channel: Option<String>,
#[structopt(
short = "v",
long = "verbose",
help = "Print the content of all packets",
parse(from_occurrences)
)]
verbose: u8,
// 0. Print nothing
// 1. Print command string
// 2. Print packets
// 3. Print udp packets
}
fn main() {
if let Err(e) = run() {
println!("Error: {}", e);
}
}
fn run() -> Result<(), Box<dyn std::error::Error>> {
log4rs::init_file("log4rs.yml", Default::default()).unwrap();
info!("Starting PokeBot!");
// Parse command line options
let args = Args::from_args();
let mut file = File::open(&args.config_path)?;
let mut toml = String::new();
file.read_to_string(&mut toml)?;
let mut config: MasterArgs = toml::from_str(&toml)?;
if let Some(count) = args.gen_id_count {
for _ in 0..count {
let id = Identity::create().expect("Failed to create id");
config.ids.push(id);
}
let toml = toml::to_string(&config)?;
let mut file = File::create(&args.config_path)?;
file.write_all(toml.as_bytes())?;
return Ok(());
}
let bot_args = config.merge(args);
debug!("Received CLI arguments: {:?}", std::env::args());
tokio::run(
async {
let (_, fut) = MasterBot::new(bot_args).await;
tokio::spawn(fut.unit_error().boxed().compat().map(|_| ()));
}
.unit_error()
.boxed()
.compat(),
);
Ok(())
}
|
