activitypub/src/activities/follow.rs

use activitypub_federation::{
    activity_sending::SendActivityTask,
    config::Data,
    fetch::object_id::ObjectId,
    protocol::context::WithContext,
    traits::{ActivityHandler, Actor, Object},
};
use activitystreams_kinds::activity::{AcceptType, FollowType};
use sea_orm::{ActiveModelTrait, ColumnTrait, EntityOrSelect, EntityTrait, QueryFilter, Set};
use serde::{Deserialize, Serialize};
use url::Url;

use crate::{
    database::StateHandle,
    entities::{
        follow_relation::{self, Entity},
        post, prelude, user,
    },
    error,
    lysand::funcs::send_follow_accept_to_lysand,
    utils::{generate_follow_accept_id, generate_random_object_id},
    DB,
};

#[derive(Deserialize, Serialize, Debug)]
pub struct Follow {
    pub actor: ObjectId<user::Model>,
    pub object: ObjectId<user::Model>,
    #[serde(rename = "type")]
    pub kind: FollowType,
    pub id: Url,
}

impl Follow {
    pub async fn send(
        local_user: ObjectId<user::Model>,
        followee: ObjectId<user::Model>,
        inbox: Url,
        data: &Data<StateHandle>,
    ) -> Result<(), error::Error> {
        print!("Sending follow request to {}", &followee);
        let create = Follow {
            actor: local_user.clone(),
            object: followee.clone(),
            kind: FollowType::Follow,
            id: generate_random_object_id(data.domain())?,
        };
        let create_with_context = WithContext::new_default(create);
        let sends = SendActivityTask::prepare(
            &create_with_context,
            &data.local_user().await?,
            vec![inbox],
            data,
        )
        .await?;
        for send in sends {
            send.sign_and_send(data).await?;
        }
        Ok(())
    }
}

#[derive(Deserialize, Serialize, Debug)]
pub struct Accept {
    actor: ObjectId<user::Model>,
    object: Follow,
    #[serde(rename = "type")]
    kind: AcceptType,
    id: Url,
}

impl Accept {
    pub async fn send(
        follow_relation: follow_relation::Model,
        follow_req: Follow,
        inbox: Url,
        data: &Data<StateHandle>,
    ) -> Result<(), error::Error> {
        print!("Sending accept to {}", &follow_relation.follower_id);
        let create = Accept {
            actor: follow_req.object.clone(),
            object: follow_req,
            kind: AcceptType::Accept,
            id: generate_follow_accept_id(data.domain(), follow_relation.id.to_string().as_str())?,
        };
        let create_with_context = WithContext::new_default(create);
        let sends = SendActivityTask::prepare(
            &create_with_context,
            &data.local_user().await?,
            vec![inbox],
            data,
        )
        .await?;
        for send in sends {
            send.sign_and_send(data).await?;
        }
        Ok(())
    }
}

#[async_trait::async_trait]
impl ActivityHandler for Follow {
    type DataType = StateHandle;
    type Error = crate::error::Error;

    fn id(&self) -> &Url {
        &self.id
    }

    fn actor(&self) -> &Url {
        self.actor.inner()
    }

    async fn verify(&self, data: &Data<Self::DataType>) -> Result<(), Self::Error> {
        Ok(())
    }

    async fn receive(self, data: &Data<Self::DataType>) -> Result<(), Self::Error> {
        //accept_follow(self, data).await?; TODO replace w/ lysand forward
        Ok(())
    }
}

#[async_trait::async_trait]
impl ActivityHandler for Accept {
    type DataType = StateHandle;
    type Error = crate::error::Error;

    fn id(&self) -> &Url {
        &self.id
    }

    fn actor(&self) -> &Url {
        self.actor.inner()
    }

    async fn verify(&self, data: &Data<Self::DataType>) -> Result<(), Self::Error> {
        Ok(())
    }

    async fn receive(self, data: &Data<Self::DataType>) -> Result<(), Self::Error> {
        let user = self.actor.dereference(data).await?;
        let follower_id;
        let follower_bridge_url = self.object.actor.clone().to_string();
        let split = follower_bridge_url.split("/").collect::<Vec<&str>>();
        if split[split.len() - 1].is_empty() {
            follower_id = split[split.len() - 2];
        } else {
            follower_id = split[split.len() - 1];
        }
        let follower = prelude::User::find()
            .filter(user::Column::Id.eq(follower_id))
            .one(data.database_connection.as_ref())
            .await?;
        save_accept_follow(user, follower.unwrap(), self).await?;
        Ok(())
    }
}

/*
async fn accept_follow(
    follow_req: Follow,
    data: &Data<StateHandle>,
) -> Result<(), crate::error::Error> {
    let local_user = follow_req.actor.dereference(data).await?;
    let follower = follow_req.object.dereference(data).await?;
    let follow_relation = save_follow(local_user, follower.clone()).await?;
    Accept::send(follow_relation, follow_req, follower.inbox().clone(), data).await?;
    Ok(())
}
*/

async fn save_accept_follow(
    followee: user::Model,
    follower: user::Model,
    accept_activity: Accept,
) -> Result<follow_relation::Model, crate::error::Error> {
    let db = DB.get().unwrap();
    let query = prelude::FollowRelation::find()
        .filter(follow_relation::Column::FollowerId.eq(follower.id.as_str()))
        .filter(follow_relation::Column::FolloweeId.eq(followee.id.as_str()))
        .one(db)
        .await?;
    if query.is_none() {
        return Err(crate::error::Error(anyhow::anyhow!("oopsie woopise")));
    }
    let lysand_accept_id = uuid::Uuid::now_v7().to_string();
    // all values in the ActiveModel that are set, except the id, will be updated
    let active_query = follow_relation::ActiveModel {
        id: Set(query.unwrap().id),
        ap_accept_id: Set(Some(accept_activity.id.to_string())),
        ap_accept_json: Set(Some(serde_json::to_string(&accept_activity).unwrap())),
        accept_id: Set(Some(lysand_accept_id)),
        ..Default::default()
    };
    // modify db entry
    let res = prelude::FollowRelation::update(active_query);
    let model = res.exec(db).await?;

    let _ = send_follow_accept_to_lysand(model.clone()).await?;

    Ok(model)
}