Compare commits

..

No commits in common. "master" and "new" have entirely different histories.
master ... new

8 changed files with 41 additions and 54 deletions

View File

@ -1,5 +1,6 @@
kind: pipeline
name: leetbot
type: kubernetes
name: default
clone:
skip_verify: true
@ -18,4 +19,3 @@ steps:
insecure: true
dockerfile: Dockerfile
tags: latest
cache_from: registry.local/northcode/leetbot

View File

@ -1,7 +1,7 @@
FROM debian
RUN apt update
RUN apt install -y openssl ca-certificates
RUN apt install -y libssl1.1 openssl ca-certificates
ADD target/release/telegram-leetbot /bin/.

View File

@ -7,7 +7,7 @@ type: application
# This is the chart version. This version number should be incremented each time you make changes
# to the chart and its templates, including the app version.
# Versions are expected to follow Semantic Versioning (https://semver.org/)
version: 0.1.1
version: 0.1.0
# This is the version number of the application being deployed. This version number should be
# incremented each time you make changes to the application. Versions are not expected to

View File

@ -19,7 +19,7 @@ spec:
name: leetbot
envFrom:
- secretRef:
name: {{ .Values.app.secret_name }}
name: {{ .Release.Name }}-token
- configMapRef:
name: {{ .Release.Name }}-cm
{{- if .Values.storage.enabled }}

View File

@ -0,0 +1,7 @@
apiVersion: v1
kind: Secret
metadata:
name: {{ .Release.Name }}-token
data:
TELEGRAM_BOT_TOKEN: {{ .Values.app.token | b64enc }}
DB_URL: {{ .Values.app.db | b64enc }}

View File

@ -2,7 +2,8 @@ image:
repository: registry.local/northcode/leetbot
app:
secret_name: leetbot-token
token: blabla
db: sqlite:leet.db
tz: Europe/Oslo
storage:

View File

@ -1,17 +1,17 @@
use chrono::NaiveTime;
use chrono::{Local, NaiveTime};
#[must_use]
pub fn text_is_leet(text: &str) -> bool {
text.eq_ignore_ascii_case("leet") || text == "l33t" || text == "1337"
}
#[must_use]
pub fn time_is_leet(time: NaiveTime) -> bool {
pub fn time_is_leet() -> bool {
let now = Local::now().time();
let leet = NaiveTime::from_hms(13,37,30);
let allowed_timegap = 30;
let distance = time - leet;
let distance = now - leet;
distance.num_seconds().abs() <= allowed_timegap
}

View File

@ -1,7 +1,7 @@
mod lib;
use crate::lib::{text_is_leet, time_is_leet};
use crate::lib::*;
use chrono::{NaiveDateTime, NaiveTime, Local};
use chrono::{NaiveDateTime, Local};
use sqlx::AnyPool;
use sqlx::any::AnyPoolOptions;
use teloxide::{prelude::*, RequestError, dispatching::UpdateFilterExt, utils::command::BotCommands};
@ -48,20 +48,22 @@ enum LeetError {
DbError(#[from] sqlx::Error)
}
async fn check_leet(username: &str, sent_time: NaiveDateTime, pool: &AnyPool) -> Result<(), LeetError> {
if ! time_is_leet(sent_time.time()) {
async fn check_leet(username: &str, pool: &AnyPool) -> Result<(), LeetError> {
if ! time_is_leet() {
return Err(LeetError::NotLeet);
}
let now = Local::now().naive_local();
let latest_leet : Option<(i32, String, NaiveDateTime)> = sqlx::query_as("select * from leet_log where username = $1 order by log_time desc")
.bind(username)
.fetch_optional(pool)
.await?;
if let Some(latest_leet) = latest_leet {
let distance = sent_time - latest_leet.2;
let distance = now - latest_leet.2;
if distance.num_hours() < 1 {
if distance.num_hours() < 24 {
return Err(LeetError::AlreadyHitLeet);
}
}
@ -69,10 +71,16 @@ async fn check_leet(username: &str, sent_time: NaiveDateTime, pool: &AnyPool) ->
Ok(())
}
async fn log_leet(username: &str, sent_time: NaiveDateTime, pool: &AnyPool) -> Result<(), sqlx::Error> {
sqlx::query("insert into leet_log (username, log_time) values ($1, $2)")
async fn log_leet(username: &str, pool: &AnyPool) -> Result<(), sqlx::Error> {
let sql_now = match pool.any_kind() {
sqlx::any::AnyKind::Postgres => "now()",
sqlx::any::AnyKind::Sqlite => "datetime()"
};
let query_sql = format!("insert into leet_log (username, log_time) values ($1, {})", sql_now);
sqlx::query(&query_sql)
.bind(username)
.bind(sent_time)
.execute(pool)
.await?;
@ -99,11 +107,7 @@ fn message_ok() -> Result<(), MessageHandlerError> {
#[command(rename = "lowercase", description = "score commands")]
enum LeetCommand {
#[command(description = "get scores")]
Scores,
#[command(description = "get current time info")]
Time,
#[command(description = "show help about leetbot")]
Help
Scores
}
#[derive(Error,Debug)]
@ -136,11 +140,7 @@ async fn leet_message_handler(message: Message, bot: AutoSend<Bot>, pool: AnyPoo
.username.clone()
.ok_or(MessageHandlerError::NoUsername)?;
let message_sent_time = message.date
.with_timezone(&Local)
.naive_local();
let leet_check = check_leet(&username, message_sent_time, &pool).await;
let leet_check = check_leet(&username, &pool).await;
if let Err(e) = leet_check {
match e {
@ -154,7 +154,7 @@ async fn leet_message_handler(message: Message, bot: AutoSend<Bot>, pool: AnyPoo
return message_ok();
}
log_leet(&username, message_sent_time, &pool).await?;
log_leet(&username, &pool).await?;
let scores = get_scores(&pool).await?;
@ -178,28 +178,6 @@ async fn leet_command_handler(message: Message, cmd: LeetCommand, bot: AutoSend<
let msg = format!("Leet scores:\n{}", scores_str);
bot.send_message(message.chat.id, msg).await?;
},
LeetCommand::Time => {
let message_time = message.date
.with_timezone(&Local)
.naive_local();
let current_time = Local::now().time();
let leet_time = NaiveTime::from_hms(13,37,30);
let distance_to_leet = leet_time - current_time;
let msg = format!("My current time is: {}
message was sent at: {}
leet is at: {}, in {} seconds",
current_time,
message_time,
leet_time,
distance_to_leet.num_seconds()
);
bot.send_message(message.chat.id, msg).await?;
},
LeetCommand::Help => {
bot.send_message(message.chat.id, LeetCommand::descriptions().to_string()).await?;
}
}
@ -234,7 +212,8 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
Dispatcher::builder(bot, handler)
.dependencies(dptree::deps![db_pool])
.default_handler(|_upd| async move {
.default_handler(|upd| async move {
println!("unhandled update: {:?}", upd);
})
.error_handler(LoggingErrorHandler::with_custom_text("error during dispatch"))
.enable_ctrlc_handler()