146 lines
4.6 KiB
Rust
146 lines
4.6 KiB
Rust
use teloxide::prelude::*;
|
|
use chrono::Local;
|
|
use std::ops::Add;
|
|
use std::cmp::Ordering::Equal;
|
|
use std::str;
|
|
use crate::telegram_utils::*;
|
|
mod database;
|
|
|
|
fn change_day(last_day: &str) -> bool{
|
|
last_day.cmp(&Local::now().add(chrono::TimeDelta::hours(2)).format("%Y-%m-%d").to_string()) != Equal
|
|
}
|
|
|
|
fn check_pole(group_id: &str) -> bool {
|
|
let data: database::DatabasePole = database::DatabasePole::get_database();
|
|
let slast_pole = match data.last_pole(group_id) {
|
|
Ok(s) => s,
|
|
Err(_e) => return true,
|
|
};
|
|
change_day(&slast_pole)
|
|
}
|
|
|
|
fn do_pole(group_id: &str, user_id: &str, user_name: &str){
|
|
let data: database::DatabasePole = database::DatabasePole::get_database();
|
|
data.write_points(group_id, user_id, user_name, &Local::now().format("%Y-%m-%d").to_string(),Rewards::POLE as i64);
|
|
}
|
|
|
|
fn do_plata(group_id: &str, user_id: &str, user_name: &str){
|
|
let data: database::DatabasePole = database::DatabasePole::get_database();
|
|
data.write_points(group_id, user_id, user_name, &Local::now().format("%Y-%m-%d").to_string(),Rewards::PLATA as i64);
|
|
}
|
|
|
|
fn do_fail(group_id: &str, user_id: &str, user_name: &str){
|
|
let data: database::DatabasePole = database::DatabasePole::get_database();
|
|
data.write_points(group_id, user_id, user_name, &Local::now().format("%Y-%m-%d").to_string(),Rewards::FAIL as i64);
|
|
}
|
|
|
|
fn check_user_points(msg: &teloxide::prelude::Message, rw: Rewards) -> bool{
|
|
let data: database::DatabasePole = database::DatabasePole::get_database();
|
|
let ret = data.check_user_pole(&msg.chat.id.to_string(),
|
|
&msg.from().unwrap().id.to_string(),
|
|
&Local::now().format("%Y-%m-%d").to_string());
|
|
check_group_points(msg, rw) && (ret == 0)
|
|
}
|
|
|
|
enum Rewards {
|
|
POLE = 3,
|
|
PLATA = 2,
|
|
FAIL = 1,
|
|
}
|
|
|
|
fn check_group_points(msg: &teloxide::prelude::Message, rw: Rewards) -> bool {
|
|
let data: database::DatabasePole = database::DatabasePole::get_database();
|
|
let ret = data.check_group_points(&msg.chat.id.to_string(),
|
|
&Local::now().format("%Y-%m-%d").to_string());
|
|
match rw {
|
|
Rewards::PLATA => ret == (Rewards::POLE as i64),
|
|
Rewards::FAIL => ret == (Rewards::PLATA as i64 + Rewards::POLE as i64),
|
|
_=> false,
|
|
}
|
|
}
|
|
|
|
pub async fn exe_pole(
|
|
msg: Message,
|
|
bot: Bot,
|
|
) -> anyhow::Result<()>{
|
|
let text_lower = match msg.text(){
|
|
Some(t) => t.to_lowercase(),
|
|
None => return Ok(()),
|
|
};
|
|
if pole_conditions(msg.clone()){
|
|
do_pole(&msg.chat.id.to_string(),
|
|
&*msg.from().unwrap().id.to_string(),
|
|
&get_alias(&msg));
|
|
bot.send_message(msg.chat.id, format!("{} ha hecho la pole",get_alias(&msg))).await?;
|
|
} else if plata_conditions(msg.clone()) {
|
|
do_plata(&msg.chat.id.to_string(),
|
|
&*msg.from().unwrap().id.to_string(),
|
|
&get_alias(&msg));
|
|
bot.send_message(msg.chat.id, format!("{} ha hecho la plata", get_alias(&msg))).await?;
|
|
} else if bronce_conditions(msg.clone()) {
|
|
do_fail(&msg.chat.id.to_string(),
|
|
&*msg.from().unwrap().id.to_string(),
|
|
&get_alias(&msg));
|
|
bot.send_message(msg.chat.id, format!("{} buen fail", get_alias(&msg))).await?;
|
|
}
|
|
return Ok(());
|
|
}
|
|
|
|
fn pole_conditions(msg: Message) -> bool{
|
|
let text_lower = match msg.text(){
|
|
Some(t) => t.to_lowercase(),
|
|
None => return false,
|
|
};
|
|
if text_lower.contains("pole") || text_lower.contains("oro") {
|
|
if check_pole(&msg.chat.id.to_string()) {
|
|
return true;
|
|
}
|
|
}
|
|
false
|
|
}
|
|
|
|
fn plata_conditions(msg: Message) -> bool{
|
|
let text_lower = match msg.text(){
|
|
Some(t) => t.to_lowercase(),
|
|
None => return false,
|
|
};
|
|
if text_lower.contains("plata") || text_lower.contains("subpole") {
|
|
if check_user_points(&msg, Rewards::PLATA) {
|
|
return true;
|
|
}
|
|
}
|
|
false
|
|
}
|
|
|
|
fn bronce_conditions(msg: Message) -> bool{
|
|
let text_lower = match msg.text(){
|
|
Some(t) => t.to_lowercase(),
|
|
None => return false,
|
|
};
|
|
if text_lower.contains("fail") || text_lower.contains("bronce") {
|
|
if check_user_points(&msg, Rewards::FAIL) {
|
|
return true;
|
|
}
|
|
}
|
|
false
|
|
}
|
|
|
|
pub fn get_top(msg: Message, bot: Bot) -> <Bot as Requester>::SendMessage{
|
|
let db = database::DatabasePole::get_database();
|
|
let top = db.get_top_users(&msg.chat.id.0.to_string());
|
|
let mut repl = String::new();
|
|
for u in top {
|
|
repl.push_str(&(u.1 + ": " + &u.0.to_string() + " puntos\n"));
|
|
}
|
|
bot.send_message(msg.chat.id, format!("{}", repl))
|
|
}
|
|
|
|
#[cfg(test)]
|
|
#[test]
|
|
fn compare_dates(){
|
|
assert_eq!(false, change_day("2020-01-01"));
|
|
assert_eq!(true, change_day("3025-01-01"));
|
|
}
|
|
|
|
|