mirror of
https://github.com/flibusta-apps/book_bot.git
synced 2026-03-04 07:30:53 +01:00
Add register functions
This commit is contained in:
56
src/bots/bots_manager/utils.rs
Normal file
56
src/bots/bots_manager/utils.rs
Normal file
@@ -0,0 +1,56 @@
|
||||
use regex::Regex;
|
||||
|
||||
|
||||
pub fn get_token(message_text: &str) -> Option<&str> {
|
||||
let re = Regex::new("(?P<token>[0-9]+:[0-9a-zA-Z-_]+)").unwrap();
|
||||
|
||||
match re.find(message_text) {
|
||||
Some(v) => Some(v.as_str()),
|
||||
None => None
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn check_token_fail() {
|
||||
let message = "wrong_token";
|
||||
|
||||
let result = get_token(message);
|
||||
|
||||
assert!(result.is_none())
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn check_token_short() {
|
||||
let message = "
|
||||
Done! Congratulations on your new bot. You will find it at t.me/aaaa_bot.
|
||||
You can now add a description, about section and profile picture for your bot,
|
||||
see /help for a list of commands.
|
||||
By the way, when you've finished creating your cool bot, ping our Bot Support if you want a better username for it.
|
||||
Just make sure the bot is fully operational before you do this. \
|
||||
\
|
||||
Use this token to access the HTTP API: \
|
||||
5555555555:AAF-AAAAAAAA1239AA2AAsvy13Axp23RAa \
|
||||
Keep your token secure and store it safely, it can be used by anyone to control your bot. \
|
||||
\
|
||||
For a description of the Bot API, see this page: https://core.telegram.org/bots/api \
|
||||
";
|
||||
|
||||
let result = get_token(message);
|
||||
|
||||
assert_eq!(result.unwrap(), "5555555555:AAF-AAAAAAAA1239AA2AAsvy13Axp23RAa");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn check_token_long() {
|
||||
let message = "5555555555:AAF-AAAAAAAA1239AA2AAsvy13Axp23RAa";
|
||||
|
||||
let result = get_token(message);
|
||||
|
||||
assert_eq!(result.unwrap(), message);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user