115 lines
4 KiB
Rust
115 lines
4 KiB
Rust
use mitra_config::Instance;
|
|
use mitra_models::{posts::types::Post, profiles::types::DbActorProfile};
|
|
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::webfinger::types::ActorAddress;
|
|
|
|
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#"<entry>
|
|
<id>{url}</id>
|
|
<title>{title}</title>
|
|
<updated>{updated_at}</updated>
|
|
<author><name>{author}</name></author>
|
|
<content type="html">{content}</content>
|
|
<link rel="alternate" href="{url}"/>
|
|
</entry>"#,
|
|
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<Post>) -> 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 = ActorAddress::from_profile(&instance.hostname(), profile);
|
|
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#"<?xml version="1.0" encoding="utf-8"?>
|
|
<feed xmlns="http://www.w3.org/2005/Atom">
|
|
<id>{id}</id>
|
|
<link rel="self" href="{url}"/>
|
|
<title>{title}</title>
|
|
<updated>{updated_at}</updated>
|
|
{entries}
|
|
</feed>"#,
|
|
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 super::*;
|
|
use chrono::{TimeZone, Utc};
|
|
use uuid::uuid;
|
|
|
|
#[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: "<p>title</p><p>text text text</p>".to_string(),
|
|
created_at: created_at,
|
|
..Default::default()
|
|
};
|
|
let entry = make_entry(instance_url, &post);
|
|
let expected_entry = concat!(
|
|
"<entry>\n",
|
|
" <id>https://example.org/objects/67e55044-10b1-426f-9247-bb680e5fe0c8</id>\n",
|
|
" <title>titletext text text</title>\n",
|
|
" <updated>2020-03-03T03:03:03+00:00</updated>\n",
|
|
" <author><name>username</name></author>\n",
|
|
r#" <content type="html"><p>title</p><p>text text text</p></content>"#,
|
|
"\n",
|
|
r#" <link rel="alternate" href="https://example.org/objects/67e55044-10b1-426f-9247-bb680e5fe0c8"/>"#,
|
|
"\n",
|
|
"</entry>",
|
|
);
|
|
assert_eq!(entry, expected_entry);
|
|
}
|
|
}
|