Compare commits

...

2 commits

Author SHA1 Message Date
5fc84c54e7
chore(deps): Remove unused dependencies
All checks were successful
Write changelog / write-changelog (push) Successful in 3s
Rust CI / Rust CI (push) Successful in 3m50s
Signed-off-by: Awiteb <a@4rs.nl>
2024-06-03 13:16:33 +03:00
d3b4ebfd53
chore: Pass the pinged bots to the threads as mutex
Signed-off-by: Awiteb <a@4rs.nl>
2024-06-03 13:16:02 +03:00
5 changed files with 38 additions and 37 deletions

14
Cargo.lock generated
View file

@ -339,12 +339,6 @@ dependencies = [
"winapi", "winapi",
] ]
[[package]]
name = "dotenv"
version = "0.15.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "77c90badedccf4105eca100756a0b1289e191f6fcbdadd3cee1d2f614f97da8f"
[[package]] [[package]]
name = "encoding_rs" name = "encoding_rs"
version = "0.8.34" version = "0.8.34"
@ -1009,12 +1003,6 @@ dependencies = [
"simple_asn1", "simple_asn1",
] ]
[[package]]
name = "lazy_static"
version = "1.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e2abad23fbc42b3700f2f279844dc832adb2b2eb069b2df918f455c4e18cc646"
[[package]] [[package]]
name = "libc" name = "libc"
version = "0.2.155" version = "0.2.155"
@ -2182,10 +2170,8 @@ name = "telepingbot"
version = "0.1.0" version = "0.1.0"
dependencies = [ dependencies = [
"chrono", "chrono",
"dotenv",
"grammers-client", "grammers-client",
"grammers-session", "grammers-session",
"lazy_static",
"log", "log",
"onlyerror", "onlyerror",
"pretty_env_logger", "pretty_env_logger",

View file

@ -13,10 +13,8 @@ version = "0.1.0"
[dependencies] [dependencies]
chrono = "0.4.38" chrono = "0.4.38"
dotenv = "0.15.0"
grammers-client = "0.6.0" grammers-client = "0.6.0"
grammers-session = "0.5.2" grammers-session = "0.5.2"
lazy_static = "1.4.0"
log = "0.4.21" log = "0.4.21"
onlyerror = "0.1.4" onlyerror = "0.1.4"
pretty_env_logger = "0.5.0" pretty_env_logger = "0.5.0"

View file

@ -14,7 +14,7 @@
// You should have received a copy of the GNU Affero General Public License // You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/agpl-3.0>. // along with this program. If not, see <https://www.gnu.org/licenses/agpl-3.0>.
use std::sync::Arc; use std::sync::{Arc, Mutex};
use salvo::{catcher::Catcher, http::HeaderValue, hyper::header, logging::Logger, prelude::*}; use salvo::{catcher::Catcher, http::HeaderValue, hyper::header, logging::Logger, prelude::*};
@ -91,13 +91,18 @@ async fn ping(req: &Request, res: &mut Response, depot: &mut Depot) {
let app_state = depot let app_state = depot
.obtain::<Arc<AppState>>() .obtain::<Arc<AppState>>()
.expect("The app state is injected"); .expect("The app state is injected");
let bots = depot
.obtain::<Arc<Mutex<Vec<crate::PingedBot>>>>()
.expect("The bots is injected");
log::debug!("Bots: {bots:?}");
let msg = if !app_state.bots.contains(&bot_username) { let msg = if !app_state.bots.contains(&bot_username) {
MessageSchema::new("Is not authorized to check the status of this bot") MessageSchema::new("Is not authorized to check the status of this bot")
.code(StatusCode::BAD_REQUEST) .code(StatusCode::BAD_REQUEST)
} else if let Ok(telegram_id) = superbot::send_start(&app_state.tg_client, &bot_username).await } else if let Ok(telegram_id) =
superbot::send_start(&app_state.tg_client, &bot_username, bots).await
{ {
if crate::PINGED_BOTS.check(telegram_id) { if bots.check(telegram_id) {
MessageSchema::new("Alive") MessageSchema::new("Alive")
} else { } else {
MessageSchema::new("No response from the bot").code(StatusCode::NOT_FOUND) MessageSchema::new("No response from the bot").code(StatusCode::NOT_FOUND)
@ -177,10 +182,11 @@ async fn add_server_headers(res: &mut Response) {
headers.insert("X-Powered-By", HeaderValue::from_static("Rust/Salvo")); headers.insert("X-Powered-By", HeaderValue::from_static("Rust/Salvo"));
} }
pub(crate) fn service(app_state: AppState) -> Service { pub(crate) fn service(app_state: AppState, bots: Arc<Mutex<Vec<crate::PingedBot>>>) -> Service {
log::debug!("Bots: {bots:?}");
let router = Router::new() let router = Router::new()
.hoop(Logger::new()) .hoop(Logger::new())
.hoop(affix::inject(Arc::new(app_state))) .hoop(affix::inject(Arc::new(app_state)).inject(Arc::clone(&bots)))
.hoop(add_server_headers) .hoop(add_server_headers)
.hoop(auth) .hoop(auth)
.push(Router::with_path("ping/@<bot_username>").get(ping)); .push(Router::with_path("ping/@<bot_username>").get(ping));

View file

@ -16,9 +16,11 @@
#![doc = include_str!("../README.md")] #![doc = include_str!("../README.md")]
use std::{process::ExitCode, sync::Mutex}; use std::{
process::ExitCode,
sync::{Arc, Mutex},
};
use lazy_static::lazy_static;
use salvo::{conn::TcpListener, Listener}; use salvo::{conn::TcpListener, Listener};
mod api; mod api;
@ -32,7 +34,7 @@ pub(crate) use errors::{Error as ServerError, Result as ServerResult};
use tokio::signal; use tokio::signal;
pub(crate) use traits::PingList; pub(crate) use traits::PingList;
#[derive(Default, Clone)] #[derive(Default, Clone, Debug)]
pub(crate) struct PingedBot { pub(crate) struct PingedBot {
telegram_id: u64, telegram_id: u64,
ping_in: i64, ping_in: i64,
@ -55,13 +57,8 @@ impl PingedBot {
} }
} }
lazy_static! {
static ref PINGED_BOTS: Mutex<Vec<PingedBot>> = Mutex::new(Vec::new());
}
async fn try_main() -> ServerResult<()> { async fn try_main() -> ServerResult<()> {
pretty_env_logger::init(); pretty_env_logger::init();
dotenv::dotenv().ok();
log::info!("Starting the API"); log::info!("Starting the API");
let cli_args = cli_parser::Args::parse()?; let cli_args = cli_parser::Args::parse()?;
@ -70,6 +67,9 @@ async fn try_main() -> ServerResult<()> {
} }
let config = config::Config::from_toml_file(&cli_args.config_file)?; let config = config::Config::from_toml_file(&cli_args.config_file)?;
let pinged_bots = Arc::new(Mutex::new(Vec::<PingedBot>::new()));
let client_bots = Arc::clone(&pinged_bots);
let server_bots = Arc::clone(&pinged_bots);
let (client, sign_out) = superbot::login(config.client.api_hash, config.client.api_id).await?; let (client, sign_out) = superbot::login(config.client.api_hash, config.client.api_id).await?;
let app_state = api::AppState::new(config.bots, config.tokens, client.clone()); let app_state = api::AppState::new(config.bots, config.tokens, client.clone());
@ -78,10 +78,11 @@ async fn try_main() -> ServerResult<()> {
let acceptor = TcpListener::new(format!("{}:{}", config.api.host, config.api.port)) let acceptor = TcpListener::new(format!("{}:{}", config.api.host, config.api.port))
.bind() .bind()
.await; .await;
let client_handler = tokio::spawn(async move { superbot::handler(handler_client).await }); let client_handler =
tokio::spawn(async move { superbot::handler(handler_client, client_bots).await });
let server_handler = tokio::spawn(async move { let server_handler = tokio::spawn(async move {
salvo::Server::new(acceptor) salvo::Server::new(acceptor)
.serve(api::service(app_state)) .serve(api::service(app_state, server_bots))
.await .await
}); });
log::info!("Bind the API to {}:{}", config.api.host, config.api.port); log::info!("Bind the API to {}:{}", config.api.host, config.api.port);

View file

@ -14,6 +14,8 @@
// You should have received a copy of the GNU Affero General Public License // You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/agpl-3.0>. // along with this program. If not, see <https://www.gnu.org/licenses/agpl-3.0>.
use std::sync::{Arc, Mutex};
use grammers_client::{Client, Config, InitParams, SignInError, Update}; use grammers_client::{Client, Config, InitParams, SignInError, Update};
use grammers_session::Session; use grammers_session::Session;
use tokio::{signal, time}; use tokio::{signal, time};
@ -72,16 +74,19 @@ pub(crate) async fn login(api_hash: String, api_id: i32) -> ServerResult<(Client
Ok((client, sign_out)) Ok((client, sign_out))
} }
fn update_handler(upd: Update) { fn update_handler(upd: Update, bots: Arc<Mutex<Vec<crate::PingedBot>>>) {
log::debug!("Bots: {bots:?}");
if let Update::NewMessage(msg) = upd { if let Update::NewMessage(msg) = upd {
if let Some(sender) = msg.sender() { if let Some(sender) = msg.sender() {
crate::PINGED_BOTS.new_res(sender.id() as u64) bots.new_res(sender.id() as u64)
} }
} }
} }
pub(crate) async fn handler(client: Client) { pub(crate) async fn handler(client: Client, bots: Arc<Mutex<Vec<crate::PingedBot>>>) {
log::debug!("Bots: {bots:?}");
loop { loop {
let cbots = Arc::clone(&bots);
tokio::select! { tokio::select! {
_ = signal::ctrl_c() => { _ = signal::ctrl_c() => {
break; break;
@ -89,17 +94,22 @@ pub(crate) async fn handler(client: Client) {
Ok(Some(update)) = client.next_update() => { Ok(Some(update)) = client.next_update() => {
log::debug!("New update: {update:?}"); log::debug!("New update: {update:?}");
tokio::spawn(async move { tokio::spawn(async move {
update_handler(update) update_handler(update, cbots)
}); });
} }
} }
} }
} }
pub(crate) async fn send_start(client: &Client, bot_username: &str) -> ServerResult<u64> { pub(crate) async fn send_start(
client: &Client,
bot_username: &str,
bots: &Arc<Mutex<Vec<crate::PingedBot>>>,
) -> ServerResult<u64> {
if let Some(chat) = client.resolve_username(bot_username).await? { if let Some(chat) = client.resolve_username(bot_username).await? {
log::debug!("Bots: {bots:?}");
let telegram_id = chat.id() as u64; let telegram_id = chat.id() as u64;
crate::PINGED_BOTS.add_new(telegram_id); bots.add_new(telegram_id);
client.send_message(chat, "/start").await?; client.send_message(chat, "/start").await?;
// Sleep, wating the response // Sleep, wating the response
time::sleep(time::Duration::from_secs(2)).await; time::sleep(time::Duration::from_secs(2)).await;