ReVanced/revanced-discord-bot
1
0
Fork 0
mirror of https://github.com/revanced/revanced-discord-bot.git synced 2025-06-13 05:37:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: send code embeds as files

Browse Source
This commit is contained in:
oSumAtrIX
2023-03-30 01:57:08 +02:00
parent 991a90cfa9
commit f6d9817bb1
2 changed files with 14 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/utils/code_embed/url_parser.rs
View File

@ -68,9 +68,8 @@ impl CodeUrlParser for GitHubCodeUrl {
path.push_str(segment);
}
let raw_url = format!(
"https://raw.githubusercontent.com/{user}/{repo}/{branch_or_sha}{path}"
);
let raw_url =
format!("https://raw.githubusercontent.com/{user}/{repo}/{branch_or_sha}{path}");
let mut code_url = CodeUrl {
raw_code_url: raw_url,
@ -135,19 +134,7 @@ impl CodeUrlParser for GitHubCodeUrl {
return Err(ParserError::InvalidFragment(format!("{start}-{end}")));
}
let mut code_block = String::new();
code_block.push_str("```");
if let Some(language) = code_url.language.clone() {
code_block.push_str(&language);
code_block.push('\n');
}
code_block.push_str(&lines[start..=end].join("\n"));
code_block.push_str("```");
Some(code_block)
Some(lines[start..=end].join("\n"))
} else {
None
};

26
src/utils/code_embed/utils.rs
View File

@ -2,13 +2,9 @@ use reqwest::Url;
use tracing::{debug, error, trace};
use super::*;
use crate::utils::bot::get_data_lock;
use crate::utils::code_embed::url_parser::{CodePreview, CodeUrlParser, GitHubCodeUrl};
pub async fn code_preview(ctx: &serenity::Context, new_message: &serenity::Message) {
let data_lock = get_data_lock(ctx).await;
let configuration = &data_lock.read().await.configuration;
let mut urls: Vec<Url> = Vec::new();
fn get_all_http_urls(string: &str, out: &mut Vec<Url>) {
@ -71,18 +67,18 @@ pub async fn code_preview(ctx: &serenity::Context, new_message: &serenity::Messa
.channel_id
.send_message(&ctx.http, |message| {
let mut message = message.reference_message(new_message);
let icon_url = &new_message.guild(&ctx.cache).unwrap().icon_url();
for code_preview in code_previews {
message = message.add_embed(|embed| {
if let Some(url) = icon_url {
embed.footer(|f| f.text("ReVanced").icon_url(url))
} else {
embed
}
.color(configuration.general.embed_color)
.description(code_preview.preview.unwrap())
});
for preview in code_previews.iter() {
let language = match preview.code.language.as_ref() {
Some(language) => language,
None => "txt",
};
let name = format!("{}.{}", &preview.code.branch_or_sha, language);
let content = preview.preview.as_ref().unwrap().as_bytes();
message = message.add_file((content, name.as_str()));
}
message