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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
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() {
//let ids = vec![Identity::create().unwrap()];
//println!(
//"{}",
//toml::to_string(&MasterArgs {
//master_name: String::from("PokeBot"),
//id: Identity::create().unwrap(),
//names: vec![String::from("test")],
//address: String::from("localhost"),
//channel: Some(String::from("Poke If Needed")),
//local: false,
//verbose: 0,
//ids,
//})
//.map_err(|e| panic!(e.to_string()))
//.unwrap()
//);
//panic!();
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(())
}
|
