mirror of https://github.com/LemmyNet/lemmy.git
Adding a user agent. Fixes #1769
parent
aee48f3f5d
commit
fe112f460d
|
@ -1,4 +1,4 @@
|
|||
use crate::{settings::structs::Settings, LemmyError};
|
||||
use crate::{settings::structs::Settings, version::VERSION, LemmyError};
|
||||
use anyhow::anyhow;
|
||||
use log::error;
|
||||
use percent_encoding::{utf8_percent_encode, NON_ALPHANUMERIC};
|
||||
|
@ -216,17 +216,30 @@ async fn is_image_content_type(client: &Client, test: &Url) -> Result<(), LemmyE
|
|||
}
|
||||
}
|
||||
|
||||
pub fn build_user_agent(settings: &Settings) -> String {
|
||||
format!(
|
||||
"Lemmy/{}; +{}",
|
||||
VERSION,
|
||||
settings.get_protocol_and_hostname()
|
||||
)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use crate::request::fetch_site_metadata;
|
||||
use crate::request::{build_user_agent, fetch_site_metadata};
|
||||
use url::Url;
|
||||
|
||||
use super::SiteMetadata;
|
||||
use crate::settings::structs::Settings;
|
||||
|
||||
// These helped with testing
|
||||
#[actix_rt::test]
|
||||
async fn test_site_metadata() {
|
||||
let client = reqwest::Client::default();
|
||||
let settings = Settings::init().unwrap();
|
||||
let client = reqwest::Client::builder()
|
||||
.user_agent(build_user_agent(&settings))
|
||||
.build()
|
||||
.unwrap();
|
||||
let sample_url = Url::parse("https://www.redspark.nu/en/peoples-war/district-leader-of-chand-led-cpn-arrested-in-bhojpur/").unwrap();
|
||||
let sample_res = fetch_site_metadata(&client, &sample_url).await.unwrap();
|
||||
assert_eq!(
|
||||
|
|
|
@ -17,6 +17,7 @@ use lemmy_routes::{feeds, images, nodeinfo, webfinger};
|
|||
use lemmy_server::{api_routes, code_migrations::run_advanced_migrations, scheduled_tasks};
|
||||
use lemmy_utils::{
|
||||
rate_limit::{rate_limiter::RateLimiter, RateLimit},
|
||||
request::build_user_agent,
|
||||
settings::structs::Settings,
|
||||
LemmyError,
|
||||
};
|
||||
|
@ -72,6 +73,10 @@ async fn main() -> Result<(), LemmyError> {
|
|||
settings.bind, settings.port
|
||||
);
|
||||
|
||||
let client = Client::builder()
|
||||
.user_agent(build_user_agent(&settings))
|
||||
.build()?;
|
||||
|
||||
let activity_queue = create_activity_queue();
|
||||
|
||||
let chat_server = ChatServer::startup(
|
||||
|
@ -79,7 +84,7 @@ async fn main() -> Result<(), LemmyError> {
|
|||
rate_limiter.clone(),
|
||||
|c, i, o, d| Box::pin(match_websocket_operation(c, i, o, d)),
|
||||
|c, i, o, d| Box::pin(match_websocket_operation_crud(c, i, o, d)),
|
||||
Client::default(),
|
||||
client.clone(),
|
||||
activity_queue.clone(),
|
||||
settings.clone(),
|
||||
secret.clone(),
|
||||
|
@ -92,7 +97,7 @@ async fn main() -> Result<(), LemmyError> {
|
|||
let context = LemmyContext::create(
|
||||
pool.clone(),
|
||||
chat_server.to_owned(),
|
||||
Client::default(),
|
||||
client.clone(),
|
||||
activity_queue.to_owned(),
|
||||
settings.to_owned(),
|
||||
secret.to_owned(),
|
||||
|
|
Loading…
Reference in New Issue