tuitbot-core 0.1.47

Core library for Tuitbot autonomous X growth assistant
Documentation 
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

//! Posting queue adapter implementations.

use tokio::sync::mpsc;

use super::super::loop_helpers::{LoopError, PostSender};
use super::super::posting_queue::{ApprovalQueue, PostAction};
use crate::storage::{self, DbPool};

/// Adapts `mpsc::Sender<PostAction>` to the `PostSender` port trait.
pub struct PostSenderAdapter {
    tx: mpsc::Sender<PostAction>,
}

impl PostSenderAdapter {
    pub fn new(tx: mpsc::Sender<PostAction>) -> Self {
        Self { tx }
    }
}

#[async_trait::async_trait]
impl PostSender for PostSenderAdapter {
    async fn send_reply(&self, tweet_id: &str, content: &str) -> Result<(), LoopError> {
        let (result_tx, result_rx) = tokio::sync::oneshot::channel();
        self.tx
            .send(PostAction::Reply {
                tweet_id: tweet_id.to_string(),
                content: content.to_string(),
                media_ids: vec![],
                result_tx: Some(result_tx),
            })
            .await
            .map_err(|e| LoopError::Other(format!("posting queue send failed: {e}")))?;

        result_rx
            .await
            .map_err(|e| LoopError::Other(format!("posting queue result recv failed: {e}")))?
            .map_err(|e| LoopError::Other(format!("post action failed: {e}")))?;

        Ok(())
    }
}

/// Adapts `DbPool` to the `ApprovalQueue` port trait.
pub struct ApprovalQueueAdapter {
    pool: DbPool,
}

impl ApprovalQueueAdapter {
    pub fn new(pool: DbPool) -> Self {
        Self { pool }
    }
}

#[async_trait::async_trait]
impl ApprovalQueue for ApprovalQueueAdapter {
    async fn queue_reply(
        &self,
        tweet_id: &str,
        content: &str,
        media_paths: &[String],
    ) -> Result<i64, String> {
        let media_json = serde_json::to_string(media_paths).unwrap_or_else(|_| "[]".to_string());
        storage::approval_queue::enqueue(
            &self.pool,
            "reply",
            tweet_id,
            "", // target_author not available here
            content,
            "",  // topic
            "",  // archetype
            0.0, // score
            &media_json,
        )
        .await
        .map_err(|e| e.to_string())
    }

    async fn queue_tweet(&self, content: &str, media_paths: &[String]) -> Result<i64, String> {
        let media_json = serde_json::to_string(media_paths).unwrap_or_else(|_| "[]".to_string());
        storage::approval_queue::enqueue(
            &self.pool,
            "tweet",
            "", // no target tweet
            "", // no target author
            content,
            "",  // topic
            "",  // archetype
            0.0, // score
            &media_json,
        )
        .await
        .map_err(|e| e.to_string())
    }
}