131 lines
3.6 KiB
Rust
131 lines
3.6 KiB
Rust
use crate::check_is_apub_id_valid;
|
|
use activitystreams::{
|
|
base::{AsBase, BaseExt, ExtendsExt},
|
|
markers::Base,
|
|
mime::{FromStrError, Mime},
|
|
object::{ApObjectExt, Object, ObjectExt, Tombstone, TombstoneExt},
|
|
};
|
|
use anyhow::{anyhow, Context};
|
|
use chrono::NaiveDateTime;
|
|
use lemmy_utils::{location_info, utils::convert_datetime, LemmyError};
|
|
use url::Url;
|
|
|
|
pub(crate) mod comment;
|
|
pub(crate) mod community;
|
|
pub(crate) mod post;
|
|
pub(crate) mod private_message;
|
|
pub(crate) mod user;
|
|
|
|
/// Updated is actually the deletion time
|
|
fn create_tombstone<T>(
|
|
deleted: bool,
|
|
object_id: &str,
|
|
updated: Option<NaiveDateTime>,
|
|
former_type: T,
|
|
) -> Result<Tombstone, LemmyError>
|
|
where
|
|
T: ToString,
|
|
{
|
|
if deleted {
|
|
if let Some(updated) = updated {
|
|
let mut tombstone = Tombstone::new();
|
|
tombstone.set_id(object_id.parse()?);
|
|
tombstone.set_former_type(former_type.to_string());
|
|
tombstone.set_deleted(convert_datetime(updated));
|
|
Ok(tombstone)
|
|
} else {
|
|
Err(anyhow!("Cant convert to tombstone because updated time was None.").into())
|
|
}
|
|
} else {
|
|
Err(anyhow!("Cant convert object to tombstone if it wasnt deleted").into())
|
|
}
|
|
}
|
|
|
|
pub(in crate::objects) fn check_object_domain<T, Kind>(
|
|
apub: &T,
|
|
expected_domain: Option<Url>,
|
|
) -> Result<String, LemmyError>
|
|
where
|
|
T: Base + AsBase<Kind>,
|
|
{
|
|
let actor_id = if let Some(url) = expected_domain {
|
|
check_is_apub_id_valid(&url)?;
|
|
let domain = url.domain().context(location_info!())?;
|
|
apub.id(domain)?.context(location_info!())?
|
|
} else {
|
|
let actor_id = apub.id_unchecked().context(location_info!())?;
|
|
check_is_apub_id_valid(&actor_id)?;
|
|
actor_id
|
|
};
|
|
Ok(actor_id.to_string())
|
|
}
|
|
|
|
pub(in crate::objects) fn set_content_and_source<T, Kind1, Kind2>(
|
|
object: &mut T,
|
|
markdown_text: &str,
|
|
) -> Result<(), LemmyError>
|
|
where
|
|
T: ApObjectExt<Kind1> + ObjectExt<Kind2> + AsBase<Kind2>,
|
|
{
|
|
let mut source = Object::<()>::new_none_type();
|
|
source
|
|
.set_content(markdown_text)
|
|
.set_media_type(mime_markdown()?);
|
|
object.set_source(source.into_any_base()?);
|
|
|
|
// set `content` to markdown for compatibility with older Lemmy versions
|
|
// TODO: change this to HTML in a while
|
|
object.set_content(markdown_text);
|
|
object.set_media_type(mime_markdown()?);
|
|
//object.set_content(markdown_to_html(markdown_text));
|
|
Ok(())
|
|
}
|
|
|
|
pub(in crate::objects) fn get_source_markdown_value<T, Kind1, Kind2>(
|
|
object: &T,
|
|
) -> Result<Option<String>, LemmyError>
|
|
where
|
|
T: ApObjectExt<Kind1> + ObjectExt<Kind2> + AsBase<Kind2>,
|
|
{
|
|
let content = object
|
|
.content()
|
|
.map(|s| s.as_single_xsd_string())
|
|
.flatten()
|
|
.map(|s| s.to_string());
|
|
if content.is_some() {
|
|
let source = object.source();
|
|
// updated lemmy version, read markdown from `source.content`
|
|
if let Some(source) = source {
|
|
let source = Object::<()>::from_any_base(source.to_owned())?.context(location_info!())?;
|
|
check_is_markdown(source.media_type())?;
|
|
let source_content = source
|
|
.content()
|
|
.map(|s| s.as_single_xsd_string())
|
|
.flatten()
|
|
.context(location_info!())?
|
|
.to_string();
|
|
return Ok(Some(source_content));
|
|
}
|
|
// older lemmy version, read markdown from `content`
|
|
// TODO: remove this after a while
|
|
else {
|
|
return Ok(content);
|
|
}
|
|
}
|
|
Ok(None)
|
|
}
|
|
|
|
pub(in crate::objects) fn mime_markdown() -> Result<Mime, FromStrError> {
|
|
"text/markdown".parse()
|
|
}
|
|
|
|
pub(in crate::objects) fn check_is_markdown(mime: Option<&Mime>) -> Result<(), LemmyError> {
|
|
let mime = mime.context(location_info!())?;
|
|
if !mime.eq(&mime_markdown()?) {
|
|
Err(LemmyError::from(anyhow!(
|
|
"Lemmy only supports markdown content"
|
|
)))
|
|
} else {
|
|
Ok(())
|
|
}
|
|
}
|