twiq-light: Add queue remove command

bouzuya · Dec 24, 2022 · cc2a8a6 · cc2a8a6
1 parent 5349ade
commit cc2a8a6
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/twiq-light/src/main.rs b/twiq-light/src/main.rs
@@ -8,6 +8,7 @@ mod fetch;
 mod google;
 mod import;
 mod list_queue;
+mod remove;
 mod reorder;
 mod search;
 mod store;
@@ -57,6 +58,11 @@ enum QueueSubcommand {
         #[arg(long, env = "TWIQ_LIGHT_GOOGLE_APPLICATION_CREDENTIALS")]
         google_application_credentials: Option<String>,
     },
+    Remove {
+        index: usize,
+        #[arg(long, env = "TWIQ_LIGHT_GOOGLE_APPLICATION_CREDENTIALS")]
+        google_application_credentials: Option<String>,
+    },
     Reorder {
         src: usize,
         dst: usize,
@@ -118,6 +124,10 @@ async fn main() -> anyhow::Result<()> {
             QueueSubcommand::List {
                 google_application_credentials,
             } => list_queue::run(TweetQueueStore::new(google_application_credentials)).await,
+            QueueSubcommand::Remove {
+                index,
+                google_application_credentials,
+            } => remove::run(TweetQueueStore::new(google_application_credentials), index).await,
             QueueSubcommand::Reorder {
                 src,
                 dst,

diff --git a/twiq-light/src/remove.rs b/twiq-light/src/remove.rs
@@ -0,0 +1,11 @@
+use tracing::instrument;
+
+use crate::store::TweetQueueStore;
+
+#[instrument(skip_all)]
+pub async fn run(store: TweetQueueStore, index: usize) -> anyhow::Result<()> {
+    let mut queue = store.read_all().await?;
+    queue.remove(index);
+    store.write_all(&queue).await?;
+    Ok(())
+}