1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
use async_trait::async_trait;
use serde::{Deserialize, Serialize};
use std::borrow::Cow;
#[async_trait]
pub trait TaskDef: Serialize + std::fmt::Debug + Sync {
const KIND: &'static str;
const MAX_ATTEMPTS: i16 = 8;
async fn perform(self, ctx: &crate::BaseContext) -> Result<(), crate::Error>;
}
#[derive(Deserialize, Serialize, Debug)]
pub struct DeliverToInbox<'a> {
pub inbox: Cow<'a, url::Url>,
pub sign_as: Option<crate::ActorLocalRef>,
pub object: String,
}
#[async_trait]
impl<'a> TaskDef for DeliverToInbox<'a> {
const KIND: &'static str = "deliver_to_inbox";
async fn perform(self, ctx: &crate::BaseContext) -> Result<(), crate::Error> {
let db = ctx.db_pool.get().await?;
let signing_info: Option<(_, _)> = match self.sign_as {
None => None,
Some(actor_ref) => Some(
crate::apub_util::fetch_or_create_local_actor_privkey(
actor_ref,
&db,
&ctx.host_url_apub,
)
.await?,
),
};
let mut req = hyper::Request::post(self.inbox.as_str().parse::<hyper::Uri>()?)
.header(hyper::header::CONTENT_TYPE, crate::apub_util::ACTIVITY_TYPE)
.body(self.object.into())?;
if let Ok(path_and_query) = crate::get_path_and_query(&self.inbox) {
req.headers_mut()
.insert(hyper::header::DATE, crate::apub_util::now_http_date());
if let Some((privkey, key_id)) = signing_info {
let signature = hancock::Signature::create_legacy(
key_id.as_str(),
&hyper::Method::POST,
&path_and_query,
req.headers(),
|src| crate::apub_util::do_sign(&privkey, &src),
)?;
req.headers_mut().insert("Signature", signature.to_header());
}
}
let res = crate::res_to_error(ctx.http_client.request(req).await?).await?;
println!("{:?}", res);
Ok(())
}
}
#[derive(Deserialize, Serialize, Debug)]
pub struct DeliverToFollowers {
pub actor: crate::ActorLocalRef,
pub sign: bool,
pub object: String,
}
#[async_trait]
impl TaskDef for DeliverToFollowers {
const KIND: &'static str = "deliver_to_followers";
async fn perform(self, ctx: &crate::BaseContext) -> Result<(), crate::Error> {
let community_id = match self.actor {
crate::ActorLocalRef::Community(id) => id,
crate::ActorLocalRef::Person(_) => return Ok(()), // We don't have user followers at this point
};
let db = ctx.db_pool.get().await?;
db.execute(
"INSERT INTO task (kind, params, max_attempts, created_at) SELECT $1, json_build_object('sign_as', $2::JSON, 'object', $3::TEXT, 'inbox', inbox), $4, current_timestamp FROM (SELECT DISTINCT COALESCE(ap_shared_inbox, ap_inbox) AS inbox FROM community_follow, person WHERE person.id = community_follow.follower AND person.local = FALSE AND community = $5) AS result",
&[&DeliverToInbox::KIND, &postgres_types::Json(&if self.sign { Some(self.actor) } else { None }), &self.object, &DeliverToInbox::MAX_ATTEMPTS, &community_id],
).await?;
Ok(())
}
}
#[derive(Deserialize, Serialize, Debug)]
pub struct FetchActor<'a> {
pub actor_ap_id: Cow<'a, url::Url>,
}
#[async_trait]
impl<'a> TaskDef for FetchActor<'a> {
const KIND: &'static str = "fetch_actor";
async fn perform(self, ctx: &crate::BaseContext) -> Result<(), crate::Error> {
let db = ctx.db_pool.get().await?;
crate::apub_util::fetch_actor(&self.actor_ap_id, &db, &ctx.http_client).await?;
Ok(())
}
}
|
