74 lines
2.4 KiB
Rust
74 lines
2.4 KiB
Rust
use formatx::formatx;
|
|
|
|
use clap::{Parser, Subcommand};
|
|
use mwbot::parsoid::WikiMultinode;
|
|
|
|
#[derive(Parser)]
|
|
#[command(version, about, long_about = None)]
|
|
struct Cli {
|
|
/// What action to perform
|
|
#[command(subcommand)]
|
|
action: Action,
|
|
}
|
|
|
|
#[derive(Clone, PartialEq, Eq, PartialOrd, Ord, Subcommand)]
|
|
enum Action {
|
|
/// Move config to ~/.config
|
|
Setup {
|
|
/// Specify the username of the bot
|
|
#[arg(long, env = "MW_USERNAME")]
|
|
username: String,
|
|
/// Specify the botpassword of the bot
|
|
#[arg(long, env = "MW_BOTPASSWORD")]
|
|
botpassword: String,
|
|
/// Specify the OAuth2 token of the bot
|
|
#[arg(long, env = "MW_OAUTH2")]
|
|
oauth2_token: String,
|
|
/// Specify the API URL of the bot
|
|
#[arg(long, env = "MW_API_URL")]
|
|
api_url: String,
|
|
/// Specify the REST URL of the bot
|
|
#[arg(long, env = "MW_REST_URL")]
|
|
rest_url: String,
|
|
},
|
|
/// Run the bot
|
|
Run,
|
|
}
|
|
|
|
fn setup(action: Action) -> Result<(), std::io::Error> {
|
|
if let Ok(config_template) = std::fs::read_to_string("mwbot.toml") {
|
|
match action {
|
|
Action::Run => {panic!();}
|
|
Action::Setup { username, botpassword, oauth2_token, api_url, rest_url } => {
|
|
let filled_in_config = formatx!(config_template, api_url, rest_url, username, botpassword, oauth2_token).unwrap();
|
|
std::fs::write(shellexpand::tilde("~/.config/mwbot.toml").into_owned(), filled_in_config).unwrap();
|
|
}
|
|
}
|
|
Ok(())
|
|
} else {
|
|
Err(std::io::Error::new(std::io::ErrorKind::NotFound, "Unable to find mwbot.toml; did you execute the script from the same directory?"))
|
|
}
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), std::io::Error> {
|
|
dotenvy::dotenv().expect("Couldn't load .env file; please make sure to create one in the same directory as executing from!");
|
|
|
|
let cli = Cli::parse();
|
|
|
|
match cli.action {
|
|
action @ Action::Setup {..} => {
|
|
setup(action)?;
|
|
eprintln!("Successfully set up ~/.config/mwbot.toml.")
|
|
},
|
|
Action::Run => {
|
|
let bot = mwbot::Bot::from_default_config().await.unwrap();
|
|
let page = bot.page("Bulgaria").unwrap();
|
|
let html = page.html().await.unwrap().into_mutable();
|
|
println!("{:?}", html);
|
|
println!("{:?}", html.as_nodes());
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|