Add janitor plugin that would suggest mingpt plugin to reply to newcomers messages
authorunc0rr
Wed, 14 Jul 2021 23:50:44 +0200
changeset 15837 a803bfa3f56c
parent 15836 d9db7b763bd1
child 15839 0defc1d3261d
Add janitor plugin that would suggest mingpt plugin to reply to newcomers messages
tools/ubot-plugins/ubot-plugin-janitor/Cargo.toml
tools/ubot-plugins/ubot-plugin-janitor/src/main.rs
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tools/ubot-plugins/ubot-plugin-janitor/Cargo.toml	Wed Jul 14 23:50:44 2021 +0200
@@ -0,0 +1,12 @@
+[package]
+name = "ubot-plugin-janitor"
+version = "0.1.0"
+edition = "2018"
+
+[dependencies]
+anyhow = "1.0"
+tokio-amqp = "1.0"
+lapin = "1.7"
+tokio = {version="1.6", features = ["full"]}
+rand = "0.8"
+futures = "0.3"
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tools/ubot-plugins/ubot-plugin-janitor/src/main.rs	Wed Jul 14 23:50:44 2021 +0200
@@ -0,0 +1,84 @@
+use anyhow::Result as AHResult;
+
+use futures::prelude::*;
+use lapin::{options::*, types::FieldTable, BasicProperties, Connection, ConnectionProperties};
+
+use tokio_amqp::*;
+
+#[tokio::main]
+async fn main() -> AHResult<()> {
+    let amqp_url = std::env::var("AMQP_URL").expect("expected AMQP_URL env variabe");
+    let conn = Connection::connect(&amqp_url, ConnectionProperties::default().with_tokio()).await?;
+
+    let pub_channel = conn.create_channel().await?;
+    let sub_channel = conn.create_channel().await?;
+
+    let queue = sub_channel
+        .queue_declare(
+            &"",
+            QueueDeclareOptions {
+                exclusive: true,
+                auto_delete: true,
+                ..QueueDeclareOptions::default()
+            },
+            FieldTable::default(),
+        )
+        .await?;
+
+    sub_channel
+        .queue_bind(
+            queue.name().as_str(),
+            "irc",
+            "*.hedgewars",
+            QueueBindOptions::default(),
+            FieldTable::default(),
+        )
+        .await?;
+
+    let mut subscriber = sub_channel
+        .basic_consume(
+            queue.name().as_str(),
+            &"",
+            BasicConsumeOptions::default(),
+            FieldTable::default(),
+        )
+        .await?;
+
+    let mut last_joined = None;
+    let mut talking_to = None;
+
+    while let Some(amqp_message) = subscriber.next().await {
+        let (_, delivery) = amqp_message.expect("error in consumer");
+        delivery.ack(BasicAckOptions::default()).await?;
+
+        match delivery.routing_key.as_str() {
+            "msg.hedgewars" => {
+                let chat_message = String::from_utf8_lossy(&delivery.data);
+                if let Some((who, _)) = chat_message.split_once('\n') {
+                    let who = Some(who.to_owned());
+                    if talking_to == who || last_joined == who {
+                        talking_to = who;
+                        pub_channel
+                            .basic_publish(
+                                "irc",
+                                "cmd.say.hedgewars",
+                                BasicPublishOptions::default(),
+                                vec![],
+                                BasicProperties::default(),
+                            )
+                            .await?;
+                    } else {
+                        last_joined = None;
+                        talking_to = None;
+                    }
+                }
+            }
+            "join.hedgewars" => {
+                last_joined = Some(String::from_utf8_lossy(&delivery.data).to_string());
+            }
+            _ => (),
+        }
+    }
+
+    Ok(())
+}