From 60770f65b6583c43de74d3e24bf03834dd1cbacf Mon Sep 17 00:00:00 2001
From: mii <kurochian2@gmail.com>
Date: Thu, 8 Dec 2022 08:25:00 +0000
Subject: [PATCH] add skip command

---
 src/commands/mod.rs  |  1 +
 src/commands/skip.rs | 90 ++++++++++++++++++++++++++++++++++++++++++++
 src/event_handler.rs |  7 +++-
 src/events/ready.rs  |  3 ++
 src/tts/instance.rs  |  8 ++++
 5 files changed, 107 insertions(+), 2 deletions(-)
 create mode 100644 src/commands/skip.rs

diff --git a/src/commands/mod.rs b/src/commands/mod.rs
index ab78f39..1aac861 100644
--- a/src/commands/mod.rs
+++ b/src/commands/mod.rs
@@ -1,3 +1,4 @@
 pub mod config;
 pub mod setup;
+pub mod skip;
 pub mod stop;
diff --git a/src/commands/skip.rs b/src/commands/skip.rs
new file mode 100644
index 0000000..7ce2ce4
--- /dev/null
+++ b/src/commands/skip.rs
@@ -0,0 +1,90 @@
+use serenity::{
+    model::prelude::{
+        interaction::{application_command::ApplicationCommandInteraction, MessageFlags},
+        UserId,
+    },
+    prelude::Context,
+};
+
+use crate::data::TTSData;
+
+pub async fn skip_command(
+    ctx: &Context,
+    command: &ApplicationCommandInteraction,
+) -> Result<(), Box<dyn std::error::Error>> {
+    if let None = command.guild_id {
+        command
+            .create_interaction_response(&ctx.http, |f| {
+                f.interaction_response_data(|d| {
+                    d.content("このコマンドはサーバーでのみ使用可能です．")
+                        .flags(MessageFlags::EPHEMERAL)
+                })
+            })
+            .await?;
+        return Ok(());
+    }
+
+    let guild = command.guild_id.unwrap().to_guild_cached(&ctx.cache);
+    if let None = guild {
+        command
+            .create_interaction_response(&ctx.http, |f| {
+                f.interaction_response_data(|d| {
+                    d.content("ギルドキャッシュを取得できませんでした．")
+                        .flags(MessageFlags::EPHEMERAL)
+                })
+            })
+            .await?;
+        return Ok(());
+    }
+    let guild = guild.unwrap();
+
+    let channel_id = guild
+        .voice_states
+        .get(&UserId(command.user.id.0))
+        .and_then(|state| state.channel_id);
+
+    if let None = channel_id {
+        command
+            .create_interaction_response(&ctx.http, |f| {
+                f.interaction_response_data(|d| {
+                    d.content("ボイスチャンネルに参加してから実行してください．")
+                        .flags(MessageFlags::EPHEMERAL)
+                })
+            })
+            .await?;
+        return Ok(());
+    }
+
+    let storage_lock = {
+        let data_read = ctx.data.read().await;
+        data_read
+            .get::<TTSData>()
+            .expect("Cannot get TTSStorage")
+            .clone()
+    };
+
+    {
+        let mut storage = storage_lock.write().await;
+        if !storage.contains_key(&guild.id) {
+            command
+                .create_interaction_response(&ctx.http, |f| {
+                    f.interaction_response_data(|d| {
+                        d.content("読み上げしていません")
+                            .flags(MessageFlags::EPHEMERAL)
+                    })
+                })
+                .await?;
+            return Ok(());
+        }
+
+        storage.get_mut(&guild.id).unwrap().skip(&ctx).await;
+    }
+
+    command
+        .create_interaction_response(&ctx.http, |f| {
+            f.interaction_response_data(|d| d.content("スキップしました"))
+        })
+        .await?;
+
+    Ok(())
+}
diff --git a/src/event_handler.rs b/src/event_handler.rs
index ead0f27..88c3ca3 100644
--- a/src/event_handler.rs
+++ b/src/event_handler.rs
@@ -1,5 +1,7 @@
 use crate::{
-    commands::{config::config_command, setup::setup_command, stop::stop_command},
+    commands::{
+        config::config_command, setup::setup_command, skip::skip_command, stop::stop_command,
+    },
     data::DatabaseClientData,
     database::dictionary::Rule,
     events,
@@ -12,7 +14,7 @@ use serenity::{
         channel::Message,
         gateway::Ready,
         prelude::{
-            component::{ActionRowComponent, ButtonStyle, InputText, InputTextStyle},
+            component::{ActionRowComponent, ButtonStyle, InputTextStyle},
             interaction::{Interaction, InteractionResponseType, MessageFlags},
         },
         voice::VoiceState,
@@ -38,6 +40,7 @@ impl EventHandler for Handler {
                 "setup" => setup_command(&ctx, &command).await.unwrap(),
                 "stop" => stop_command(&ctx, &command).await.unwrap(),
                 "config" => config_command(&ctx, &command).await.unwrap(),
+                "skip" => skip_command(&ctx, &command).await.unwrap(),
                 _ => {}
             }
         }
diff --git a/src/events/ready.rs b/src/events/ready.rs
index 2de61e9..7ba1cd6 100644
--- a/src/events/ready.rs
+++ b/src/events/ready.rs
@@ -24,6 +24,9 @@ pub async fn ready(ctx: Context, ready: Ready) {
                     })
             })
             .create_application_command(|command| command.name("config").description("Config"))
+            .create_application_command(|command| {
+                command.name("skip").description("skip tts message")
+            })
     })
     .await;
 }
diff --git a/src/tts/instance.rs b/src/tts/instance.rs
index 1968991..a1ef5c0 100644
--- a/src/tts/instance.rs
+++ b/src/tts/instance.rs
@@ -38,4 +38,12 @@ impl TTSInstance {
             call.enqueue_source(input);
         }
     }
+
+    pub async fn skip(&mut self, ctx: &Context) {
+        let manager = songbird::get(&ctx).await.unwrap();
+        let call = manager.get(self.guild).unwrap();
+        let call = call.lock().await;
+        let queue = call.queue();
+        let _ = queue.skip();
+    }
 }