continuwuity/src/service/resolver/well_known.rs

use std::str::FromStr;

use conduwuit::{
	Result, debug, debug_error, debug_info, implement, trace, utils::response::LimitReadExt,
};
use ruma::{OwnedServerName, ServerName};

#[implement(super::Service)]
#[tracing::instrument(name = "well-known", level = "debug", skip(self, dest))]
pub(super) async fn request_well_known(&self, dest: &str) -> Result<Option<String>> {
	trace!("Requesting well known for {dest}");
	let response = self
		.services
		.client
		.well_known
		.get(format!("https://{dest}/.well-known/matrix/server"))
		.send()
		.await;

	trace!("response: {response:?}");
	if let Err(e) = &response {
		debug!("error: {e:?}");
		return Ok(None);
	}

	let response = response?;
	if !response.status().is_success() {
		debug!("response not 2XX");
		return Ok(None);
	}

	let Ok(text) = response.limit_read_text(8192).await else {
		debug!("failed to read well-known response (too large or non-text content)");
		return Ok(None);
	};
	trace!("response text: {text:?}");

	let body: serde_json::Value = serde_json::from_str(&text).unwrap_or_default();

	let m_server = body
		.get("m.server")
		.unwrap_or(&serde_json::Value::Null)
		.as_str()
		.unwrap_or_default();

	if ServerName::parse(m_server).is_err() {
		debug_error!("response content missing or invalid");
		return Ok(None);
	}

	debug_info!("{dest:?} found at {m_server:?}");
	Ok(Some(m_server.to_owned()))
}