feat: function to receive lysand note

This commit is contained in:
April John 2024-05-21 04:59:47 +02:00
parent e42baf51e4
commit bcab516a1f
Signed by: aprl
GPG key ID: BCB934A2909C5460
4 changed files with 148 additions and 28 deletions

View file

@ -34,7 +34,7 @@ pub fn listen(config: &FederationConfig<StateHandle>) -> Result<(), Error> {
} }
pub fn lysand_inbox(note: web::Json<lysand::objects::Note>, id: web::Path<String>, data: Data<StateHandle>) -> Result<HttpResponse, Error> { pub fn lysand_inbox(note: web::Json<lysand::objects::Note>, id: web::Path<String>, data: Data<StateHandle>) -> Result<HttpResponse, Error> {
tokio::spawn(receive_lysand_note(note.into_inner(), id.into_inner(), data.app_data().clone())); tokio::spawn(receive_lysand_note(note.into_inner(), id.into_inner()));
Ok(HttpResponse::Created().finish()) Ok(HttpResponse::Created().finish())
} }

View file

@ -1,38 +1,107 @@
use activitypub_federation::fetch::object_id::ObjectId; use activitypub_federation::{fetch::object_id::ObjectId, http_signatures::generate_actor_keypair};
use activitystreams_kinds::public; use activitystreams_kinds::public;
use sea_orm::{EntityTrait, QueryFilter}; use chrono::{DateTime, Utc};
use sea_orm::{ActiveModelTrait, ColumnTrait, EntityTrait, QueryFilter, Set};
use anyhow::anyhow;
use url::Url; use url::Url;
use crate::{database::State, entities::{self, post, prelude}, objects::post::Mention, utils::{generate_object_id, generate_user_id}, FEDERATION_CONFIG}; use crate::{database::State, entities::{self, post, prelude, user}, objects::post::Mention, utils::{generate_object_id, generate_user_id}, API_DOMAIN, DB, FEDERATION_CONFIG, LYSAND_DOMAIN};
use super::objects::Note; use super::{objects::{ContentFormat, Note}, superx::request_client};
pub async fn receive_lysand_note(note: Note, db_id: String, db: State) { pub async fn fetch_user_from_url(url: Url) -> anyhow::Result<super::objects::User> {
let author: entities::user::Model = todo!(); let req_client = request_client();
let user_res = prelude::User::find_by_id(db_id).one(db.database_connection.as_ref()).await; let request = req_client.get(url).send().await?;
if user_res.is_err() { Ok(request.json::<super::objects::User>().await?)
println!("{}", user_res.unwrap_err()); }
return;
pub async fn option_content_format_text(opt: Option<ContentFormat>) -> Option<String> {
if let Some(format) = opt {
return Some(format.select_rich_text().await.unwrap());
} }
if let Some(target) = user_res.ok().unwrap() {
None
}
pub async fn db_user_from_url(url: Url) -> anyhow::Result<entities::user::Model> {
if !url.domain().eq(&Some(LYSAND_DOMAIN.as_str())) {
return Err(anyhow!("not lysands domain"));
}
let user_res = prelude::User::find().filter(entities::user::Column::Url.eq(url.to_string())).one(DB.get().unwrap()).await?;
if let Some(user) = user_res {
Ok(user)
} else {
let ls_user = fetch_user_from_url(url).await?;
let keypair = generate_actor_keypair()?;
let user = entities::user::ActiveModel {
id: Set(ls_user.id.to_string()),
username: Set(ls_user.username.clone()),
name: Set(ls_user.display_name.unwrap_or(ls_user.username)),
inbox: Set(ls_user.inbox.to_string()),
public_key: Set(keypair.public_key.clone()),
private_key: Set(Some(keypair.private_key.clone())),
last_refreshed_at: Set(Utc::now()),
follower_count: Set(0),
following_count: Set(0),
url: Set(ls_user.uri.to_string()),
local: Set(true),
created_at: Set(DateTime::from_timestamp(ls_user.created_at.unix_timestamp(), 0).unwrap()),
summary: Set(option_content_format_text(ls_user.bio).await),
updated_at: Set(Some(Utc::now())),
followers: Set(Some(ls_user.followers.to_string())),
following: Set(Some(ls_user.following.to_string())),
..Default::default()
};
let db = DB.get().unwrap();
Ok(user.insert(db).await?)
}
}
pub async fn fetch_note_from_url(url: Url) -> anyhow::Result<super::objects::Note> {
let req_client = request_client();
let request = req_client.get(url).send().await?;
Ok(request.json::<super::objects::Note>().await?)
}
pub async fn receive_lysand_note(note: Note, db_id: String) -> anyhow::Result<()> {
let author: entities::user::Model = db_user_from_url(note.author.clone()).await?;
let user_res = prelude::User::find_by_id(db_id).one(DB.get().unwrap()).await;
if user_res.is_err() {
println!("{}", user_res.as_ref().unwrap_err());
return Err(user_res.err().unwrap().into());
}
if let Some(target) = user_res? {
let data = FEDERATION_CONFIG.get().unwrap(); let data = FEDERATION_CONFIG.get().unwrap();
let id: ObjectId<post::Model> = generate_object_id(data.domain(), &note.id.to_string()).unwrap().into(); let id: ObjectId<post::Model> = generate_object_id(data.domain(), &note.id.to_string())?.into();
let user_id = generate_user_id(data.domain(), &target.id.to_string()).unwrap(); let user_id = generate_user_id(data.domain(), &target.id.to_string())?;
let to = match note.visibility.unwrap_or(super::objects::VisibilityType::Public) { let user = fetch_user_from_url(user_id).await?;
super::objects::VisibilityType::Public => vec![public(), Url::parse(&author.inbox).unwrap()], let mut tag: Vec<Mention> = Vec::new();
super::objects::VisibilityType::Followers => vec![Url::parse(&author.inbox).unwrap()], for l_tag in note.mentions.clone().unwrap_or_default() {
super::objects::VisibilityType::Direct => vec![user_id], tag.push(Mention { href: l_tag, //todo convert to ap url
super::objects::VisibilityType::Unlisted => vec![Url::parse(&author.inbox).unwrap()], kind: Default::default(), })
}
let to = match note.visibility.clone().unwrap_or(super::objects::VisibilityType::Public) {
super::objects::VisibilityType::Public => vec![public(), Url::parse(&author.followers.unwrap_or_default())?],
super::objects::VisibilityType::Followers => vec![Url::parse(&author.followers.unwrap_or_default())?],
super::objects::VisibilityType::Direct => note.mentions.unwrap_or_default(),
super::objects::VisibilityType::Unlisted => vec![Url::parse(&author.followers.unwrap_or_default())?],
}; };
let cc = match note.visibility.unwrap_or(super::objects::VisibilityType::Public) { let cc = match note.visibility.unwrap_or(super::objects::VisibilityType::Public) {
super::objects::VisibilityType::Unlisted => Some(vec![public()]), super::objects::VisibilityType::Unlisted => Some(vec![public()]),
_ => None _ => None
}; };
let mut tag: Vec<Mention> = Vec::new(); let reply: Option<ObjectId<entities::post::Model>> = if let Some(rep) = note.replies_to {
for l_tag in note.mentions.unwrap_or_default() { let note = fetch_note_from_url(rep).await?;
tag.push(Mention { href: l_tag, //todo convert to ap url let fake_rep_url = Url::parse(&format!(
kind: Default::default(), }) "https://{}/lysand/apnote/{}",
} API_DOMAIN.to_string(),
&note.id.to_string()
))?;
Some(fake_rep_url.into())
} else {
None
};
let ap_note = crate::objects::post::Note { let ap_note = crate::objects::post::Note {
kind: Default::default(), kind: Default::default(),
id, id,
@ -40,7 +109,12 @@ pub async fn receive_lysand_note(note: Note, db_id: String, db: State) {
cc, cc,
to, to,
tag, tag,
attributed_to: Url::parse(author.url.clone().as_str()).unwrap().into(),
} content: option_content_format_text(note.content).await.unwrap_or_default(),
in_reply_to: reply
};
} }
Ok(())
} }

View file

@ -108,10 +108,55 @@ pub struct ContentHash {
} }
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
struct ContentFormat { pub struct ContentFormat {
x: HashMap<String, ContentEntry>, x: HashMap<String, ContentEntry>,
} }
impl ContentFormat {
pub async fn select_rich_text(&self) -> anyhow::Result<String> {
if let Some(entry) = self.x.get("text/x.misskeymarkdown") {
return Ok(entry.content.clone())
}
if let Some(entry) = self.x.get("text/html") {
return Ok(entry.content.clone())
}
if let Some(entry) = self.x.get("text/markdown") {
return Ok(entry.content.clone())
}
if let Some(entry) = self.x.get("text/plain") {
return Ok(entry.content.clone())
}
Ok(self.x.clone().values().next().unwrap().content.clone())
}
pub async fn select_rich_img(&self) -> anyhow::Result<String> {
if let Some(entry) = self.x.get("image/webp") {
return Ok(entry.content.clone())
}
if let Some(entry) = self.x.get("image/png") {
return Ok(entry.content.clone())
}
if let Some(entry) = self.x.get("image/avif") {
return Ok(entry.content.clone())
}
if let Some(entry) = self.x.get("image/jxl") {
return Ok(entry.content.clone())
}
if let Some(entry) = self.x.get("image/jpeg") {
return Ok(entry.content.clone())
}
if let Some(entry) = self.x.get("image/gif") {
return Ok(entry.content.clone())
}
if let Some(entry) = self.x.get("image/bmp") {
return Ok(entry.content.clone())
}
Ok(self.x.clone().values().next().unwrap().content.clone())
}
}
impl Serialize for ContentFormat { impl Serialize for ContentFormat {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error> fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where where
@ -136,7 +181,7 @@ impl<'de> Deserialize<'de> for ContentFormat {
#[derive(Debug, Serialize, Deserialize, Clone)] #[derive(Debug, Serialize, Deserialize, Clone)]
struct FieldKV { struct FieldKV {
name: ContentFormat, key: ContentFormat,
value: ContentFormat, value: ContentFormat,
} }

View file

@ -94,6 +94,7 @@ async fn post_manually(
content: format!("{} {}", path.1, target.name), content: format!("{} {}", path.1, target.name),
tag: vec![mention], tag: vec![mention],
in_reply_to: None, in_reply_to: None,
cc: vec![].into()
}; };
CreatePost::send( CreatePost::send(