1 unstable release
0.1.0 | Oct 17, 2022 |
---|
#10 in #mt-proto
Used in teleser
24KB
628 lines
Teleser
A telegram's MTProto bot framework based grammers, supported login with phone number and bot token
Make a bot
At first
Get an api id form https://core.telegram.org/api/obtaining_api_id
Add teleser to dependencies
Both add proxy feature is good idea if your in China main land network
teleser = { version = "0", features = ["proxy"] }
And look up bot template source code
Main
- Set logger
- Using
tokio
async runtime run async main
fn main() -> Result<()> {
init_tracing_subscriber();
return runtime::Builder::new_multi_thread()
.enable_all()
.max_blocking_threads(30)
.worker_threads(50)
.build()
.unwrap()
.block_on(async_main());
}
Logger
tracing_subscriber
to set log level, your need modify "teleser_template"
to your self crate name.
fn init_tracing_subscriber() {
tracing_subscriber::registry()
.with(
tracing_subscriber::fmt::layer()
.with_target(true)
.without_time(),
)
.with(
tracing_subscriber::filter::Targets::new()
.with_target("teleser", Level::DEBUG)
.with_target("teleser_template", Level::DEBUG),
)
.init();
}
Async_main
async fn async_main() -> Result<()> {
// Using `ClientBuilder` build a bot client
let client = Arc::new(
teleser::ClientBuilder::new()
// read api id from env (on build)
.with_api_id(env!("API_ID").parse()?)
// read api hash from env (on build)
.with_api_hash(env!("API_HASH").to_string())
// auth
.with_auth(Auth::AuthWithPhoneAndCode(Box::new(Input {}))
// load session file on startup
// save session to file teleser.session if login
.with_session_store(Box::new(FileSessionStore{
path: "teleser.session".to_string(),
}))
// modules
.with_modules(vec![raw_plugin::module(), proc_plugin::module()])
// connect to server via proxy url, like socks5://127.0.0.1:1080 (runtime)
// please delete this code if you not add feature named proxy
.with_init_params(match std::env::var("TELESER_PROXY") {
Ok(url) => {
let mut ip = InitParams::default();
ip.proxy_url = Some(url);
Some(ip)
}
Err(_) => Some(InitParams::default()),
})
.build()?,
);
//////////////////////////////////////
// can start some timer task like this
let copy_client = client.clone();
tokio::spawn(async move {
let lock = copy_client.inner_client.lock().await;
let ic = lock.clone();
drop(lock);
if let Some(_ic) = ic {
// _ic.send_message();
}
});
//////////////////////////////////////
// run client
teleser::run_client_and_reconnect(client).await?;
/////////////////////////////////////
Ok(())
}
Input
Only read line from console
pub struct Input {}
#[async_trait]
impl AuthWithPhoneAndCode for Input {
async fn input_phone(&self) -> Result<String> {
input("Input your phone number ( like +112345678 )")
}
async fn input_code(&self) -> Result<String> {
input("Input your device or sms code ( like 12345 )")
}
async fn input_password(&self) -> Result<String> {
input("Input your password")
}
}
fn input(tips: &str) -> Result<String> {
let mut s = String::new();
print!("{tips}: ");
let _ = stdout().flush();
stdin().read_line(&mut s)?;
if let Some('\n') = s.chars().next_back() {
s.pop();
}
if let Some('\r') = s.chars().next_back() {
s.pop();
}
Ok(s)
}
Handler
This example has handler grammers Update::OnNewMessage
. You can use these handlers:
new_message
/ message_edited
/ message_deleted
/ callback_query
/ inline_query
/ raw
But I can't got message_edited event on grammers, and got the raw.
use crate::Result;
use teleser::re_exports::grammers_client::types::Message;
use teleser::{new_message, Handler, InnerClient};
#[new_message]
async fn proc_new_message(_: &mut InnerClient, message: &Message) -> Result<bool> {
println!("PROC : A NEW MESSAGE : {}", message.text());
Ok(false)
}
pub(crate) fn module() -> Module {
Module {
id: "proc_new_message".to_owned(),
name: "proc_new_message".to_owned(),
handlers: vec![Handler {
id: "proc_new_message".to_owned(),
process: proc_new_message {}.into(),
}],
}
}
parse to handler
pub(crate) fn module() -> Module {
Module {
id: "proc_new_message".to_owned(),
name: "proc_new_message".to_owned(),
handlers: vec![proc_new_message {}.into()],
}
}
parse to handlers
pub(crate) fn module() -> Module {
Module {
id: "proc_new_message".to_owned(),
name: "proc_new_message".to_owned(),
handlers: proc_new_message {}.into(),
}
}
parse to module
pub(crate) fn module() -> Module {
proc_new_message {}.into()
}
Manually write a handler
use teleser::re_exports::async_trait::async_trait;
use teleser::re_exports::grammers_client::types::Message;
use teleser::{Handler, InnerClient, NewMessageProcess, Process};
pub(crate) struct RawPlugin {}
#[async_trait]
impl NewMessageProcess for RawPlugin {
async fn handle(&self, _: &mut InnerClient, event: &Message) -> crate::Result<bool> {
println!("RAW : A NEW MESSAGE : {}", event.text());
Ok(false)
}
}
pub(crate) fn module() -> Module {
Module {
id: "RawPlugin".to_owned(),
name: "RawPlugin".to_owned(),
handlers: vec![Handler {
id: "RawPlugin".to_owned(),
process: Process::NewMessageProcess(Box::new(RawPlugin {})),
}],
}
}
Dependencies
~2MB
~42K SLoC