This commit is contained in:
Sakuhl 2018-01-02 00:07:11 +01:00
parent 56bf12b4de
commit 37c369cfcb

View File

@ -1,130 +1,17 @@
extern crate serenity;
extern crate wynncraft;
use wynncraft::{Guild, Player};
extern crate serenity;
use serenity::prelude::*;
use serenity::model::*;
extern crate reqwest;
extern crate serde;
#[macro_use]
extern crate serde_derive;
extern crate serde_json;
use serde_json::Value;
use std::collections::HashMap;
use std::env;
#[derive(Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Guild {
pub name: String,
pub prefix: String,
pub members: Vec<Member>,
pub xp: f64,
pub level: u64,
pub created: String,
pub created_friendly: String,
pub territories: u64
}
#[derive(Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Member {
pub name: String,
pub rank: Rank,
pub contributed: u64,
pub joined: String,
pub joined_friendly: String,
}
#[derive(Deserialize)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum Rank {
Owner,
Chief,
Captain,
Recruiter,
Recruit
}
#[derive(Deserialize)]
#[allow(non_snake_case)]
pub struct Player {
pub username: String,
pub rank: String,
pub tag: String,
pub displayTag: bool,
pub veteran: bool,
pub playtime: u64,
pub first_join: String,
pub last_join: String,
pub first_join_friendly: String,
pub last_join_friendly: String,
pub current_server: String,
pub global: GlobalPlayerInfo,
pub classes: HashMap<String, Class>,
pub wizard_fortress: WFPlayerInfo,
pub rankings: PlayerRankings,
pub guild: PlayerGuildInfo
}
#[derive(Deserialize)]
pub struct GlobalPlayerInfo {
pub items_identified: u64,
pub mobs_killed: u64,
pub pvp_kills: u64,
pub pvp_deaths: u64,
pub chests_found: u64,
pub blocks_walked: u64,
pub logins: u64,
pub deaths: u64,
pub total_level: u64
}
#[derive(Deserialize)]
#[allow(non_snake_case)]
pub struct Class {
pub level: u64,
pub xp: f64,
pub dungeons: Value,
pub quests: Vec<String>,
pub dungeonsAmount: String,
pub questsAmount: String,
pub skills: Value,
pub items_identified: u64,
pub mobs_killed: u64,
pub pvp_kills: u64,
pub pvp_deaths: u64,
pub chests_found: u64,
pub blocks_walked: u64,
pub logins: u64,
pub deaths: u64,
pub events_won: u64,
/// in minutes
pub playtime: u64
}
#[derive(Deserialize)]
pub struct WFPlayerInfo {
pub kills: u64,
pub deaths: u64,
pub wins: u64,
pub losses: u64
}
#[derive(Deserialize)]
pub struct PlayerRankings {
pub pvp: u64,
pub player: u64,
pub guild: u64
}
#[derive(Deserialize)]
pub struct PlayerGuildInfo {
pub name: String,
pub rank: String
}
struct Handler;
impl EventHandler for Handler {
@ -135,102 +22,13 @@ impl EventHandler for Handler {
// of a single event can be dispatched simultaneously.
fn on_message(&self, _: Context, msg: Message) {
if msg.content.starts_with("wc!guild ") {
let guild = &msg.content[9..];
let resp = reqwest::get(&format!("https://api.wynncraft.com/public_api.php?action=guildStats&command={}", guild)).unwrap();
assert!(resp.status().is_success());
if let Ok(guild) = serde_json::from_reader(resp) {
let guild: Guild = guild;
let resp = reqwest::get("https://api.wynncraft.com/public_api.php?action=territoryList").unwrap();
assert!(resp.status().is_success());
let territories: Value = serde_json::from_reader(resp).unwrap();
let mut message = String::new();
for value in territories.get("territories").unwrap().as_object().unwrap().values() {
if value.get("guild").unwrap().as_str().unwrap() == guild.name {
message.push_str(&format!("
**Territory**: {}",
value.get("territory").unwrap().as_str().unwrap()
))
}
}
msg.channel_id.say(
format!(
"**Guild**: {}
**prefix**: {}
**Created**: {}
**Level**: {}
**Members**: {}",
guild.name,
guild.prefix,
guild.created_friendly,
guild.level,
guild.members.len()
) + &message
).unwrap();
}
wc_guild(msg);
} else if msg.content.starts_with("wc!territory ") {
let territory = &msg.content[13..];
let resp = reqwest::get("https://api.wynncraft.com/public_api.php?action=territoryList").unwrap();
assert!(resp.status().is_success());
let territories: Value = serde_json::from_reader(resp).unwrap();
let mut message = String::new();
for value in territories.get("territories").unwrap().as_object().unwrap().values() {
if value.get("territory").unwrap().as_str().unwrap().contains(territory) {
message.push_str(&format!("
**Territory**: {}
**Owner**: {}
**Acquired at**: {}
",
value.get("territory").unwrap().as_str().unwrap(),
value.get("guild").unwrap().as_str().unwrap(),
value.get("acquired").unwrap().as_str().unwrap()
))
}
}
if !message.is_empty() {
msg.channel_id.say(
message
).unwrap();
}
wc_territory(msg);
} else if msg.content.starts_with("wc!status") {
let resp = reqwest::get("https://api.wynncraft.com/public_api.php?action=onlinePlayersSum").unwrap();
let value: Value = serde_json::from_reader(resp).unwrap();
msg.channel_id.say(
format!("{} players online", value.get("players_online").unwrap().as_u64().unwrap())
).unwrap();
wc_status(msg);
} else if msg.content.starts_with("wc!player ") {
let player = &msg.content[10..];
let resp = reqwest::get(&format!("https://api.wynncraft.com/public_api.php?action=playerStats&command={}", player)).unwrap();
let player: Player = serde_json::from_reader(resp).unwrap();
msg.channel_id.say(
format!(
"**Player name**: {}
**Guild**: {}
**Playtime**: {} hours
**Kills**: {} mobs & {} players
**Chests looted**: {}
**Total level**: {}",
player.username,
player.guild.name,
player.playtime / 60,
player.global.mobs_killed, player.global.pvp_kills,
player.global.chests_found,
player.classes.iter().map(|(_, x)| x.level).sum::<u64>()
)
).unwrap();
wc_player(msg);
}
}
@ -245,6 +43,114 @@ impl EventHandler for Handler {
}
}
fn wc_guild(msg: Message) {
let guild = &msg.content[9..];
if let Ok(guild) = wynncraft::guild(guild) {
let guild: Guild = guild;
let resp = reqwest::get("https://api.wynncraft.com/public_api.php?action=territoryList").unwrap();
assert!(resp.status().is_success());
let territories: Value = serde_json::from_reader(resp).unwrap();
let mut message = String::new();
for value in territories.get("territories").unwrap().as_object().unwrap().values() {
if value.get("guild").unwrap().as_str().unwrap() == guild.name {
message.push_str(&format!("
**Territory**: {}",
value.get("territory").unwrap().as_str().unwrap()
))
}
}
msg.channel_id.say(
format!(
"**Guild**: {}
**prefix**: {}
**Created**: {}
**Level**: {}
**Members**: {}",
guild.name,
guild.prefix,
guild.created_friendly,
guild.level,
guild.members.len()
) + &message
).unwrap();
}
}
fn wc_status(msg: Message) {
let resp = reqwest::get("https://api.wynncraft.com/public_api.php?action=onlinePlayersSum").unwrap();
let value: Value = serde_json::from_reader(resp).unwrap();
msg.channel_id.say(
format!("{} players online", value.get("players_online").unwrap().as_u64().unwrap())
).unwrap();
}
fn wc_territory(msg: Message) {
let territory = &msg.content[13..];
let resp = reqwest::get("https://api.wynncraft.com/public_api.php?action=territoryList").unwrap();
assert!(resp.status().is_success());
let territories: Value = serde_json::from_reader(resp).unwrap();
let mut message = String::new();
for value in territories.get("territories").unwrap().as_object().unwrap().values() {
if value.get("territory").unwrap().as_str().unwrap().contains(territory) {
message.push_str(&format!("
**Territory**: {}
**Owner**: {}
**Acquired at**: {}
",
value.get("territory").unwrap().as_str().unwrap(),
value.get("guild").unwrap().as_str().unwrap(),
value.get("acquired").unwrap().as_str().unwrap()
))
}
}
if !message.is_empty() {
msg.channel_id.say(
message
).unwrap();
}
}
fn wc_player(msg: Message) {
let player = &msg.content[10..];
let player: Player = wynncraft::player(player).unwrap();
let current_server_msg = if player.current_server != "null" {
format!("\n**Currently online on**: {}", player.current_server)
} else {
String::new()
};
msg.channel_id.say(
format!(
"**Player name**: {}
**Guild**: {}
**Playtime**: {} hours
**Kills**: {} mobs & {} players
**Chests looted**: {}
**Total level**: {}{}",
player.username,
player.guild.name,
player.playtime / 60,
player.global.mobs_killed, player.global.pvp_kills,
player.global.chests_found,
player.classes.iter().map(|(_, x)| x.level).sum::<u64>(),
current_server_msg
)
).unwrap();
}
fn main() {
// Configure the client with your Discord bot token in the environment.
let token = env::var("DISCORD_TOKEN").expect("DISCORD_TOKEN must be set");