use mitra_config::Instance;
use mitra_utils::{
datetime::get_min_datetime,
html::{clean_html_all, escape_html},
};
use crate::activitypub::identifiers::{local_actor_id, local_object_id};
use crate::models::{
posts::types::Post,
profiles::types::DbActorProfile,
};
const ENTRY_TITLE_MAX_LENGTH: usize = 75;
fn make_entry(
instance_url: &str,
post: &Post,
) -> String {
let object_id = local_object_id(instance_url, &post.id);
let content_escaped = escape_html(&post.content);
let content_cleaned = clean_html_all(&post.content);
// Use trimmed content for title
let mut title: String = content_cleaned.chars()
.take(ENTRY_TITLE_MAX_LENGTH)
.collect();
if title.len() == ENTRY_TITLE_MAX_LENGTH &&
content_cleaned.len() != ENTRY_TITLE_MAX_LENGTH {
title += "...";
};
format!(
r#"
{url}
{title}
{updated_at}
{author}
{content}
"#,
url=object_id,
title=title,
updated_at=post.created_at.to_rfc3339(),
author=post.author.username,
content=content_escaped,
)
}
fn get_feed_url(instance_url: &str, username: &str) -> String {
format!("{}/feeds/users/{}", instance_url, username)
}
pub fn make_feed(
instance: &Instance,
profile: &DbActorProfile,
posts: Vec,
) -> String {
let actor_id = local_actor_id(&instance.url(), &profile.username);
let actor_name = profile.display_name.as_ref()
.unwrap_or(&profile.username);
let actor_address = profile.actor_address(&instance.hostname());
let feed_url = get_feed_url(&instance.url(), &profile.username);
let feed_title = format!("{} (@{})", actor_name, actor_address);
let mut entries = vec![];
let mut feed_updated_at = get_min_datetime();
for post in posts {
let entry = make_entry(&instance.url(), &post);
entries.push(entry);
if post.created_at > feed_updated_at {
feed_updated_at = post.created_at;
};
};
format!(
r#"
{id}
{title}
{updated_at}
{entries}
"#,
id=actor_id,
url=feed_url,
title=feed_title,
updated_at=feed_updated_at.to_rfc3339(),
entries=entries.join("\n"),
)
}
#[cfg(test)]
mod tests {
use chrono::{TimeZone, Utc};
use uuid::uuid;
use super::*;
#[test]
fn test_make_entry() {
let instance_url = "https://example.org";
let author = DbActorProfile {
username: "username".to_string(),
..Default::default()
};
let post_id = uuid!("67e55044-10b1-426f-9247-bb680e5fe0c8");
let created_at = Utc.with_ymd_and_hms(2020, 3, 3, 3, 3, 3).unwrap();
let post = Post {
id: post_id,
author: author,
content: "title
text text text
".to_string(),
created_at: created_at,
..Default::default()
};
let entry = make_entry(instance_url, &post);
let expected_entry = concat!(
"\n",
" https://example.org/objects/67e55044-10b1-426f-9247-bb680e5fe0c8\n",
" titletext text text\n",
" 2020-03-03T03:03:03+00:00\n",
" username\n",
r#" <p>title</p><p>text text text</p>"#, "\n",
r#" "#, "\n",
"",
);
assert_eq!(entry, expected_entry);
}
}