スタンプの削除を実装

kamecha · Oct 27, 2024 · 7bc6b69 · 7bc6b69
1 parent de30e5a
commit 7bc6b69
Show file tree

Hide file tree

Showing 3 changed files with 69 additions and 1 deletion.
diff --git a/autoload/traqvim/command.vim b/autoload/traqvim/command.vim
@@ -123,7 +123,7 @@ function traqvim#command#stamp(args) abort
 	if a:args[0] ==# 'add'
 		call denops#request('traqvim', 'messageAddStamps', [bufnr(), traqvim#message#get_message(), a:args[1:]])
 	elseif a:args[0] ==# 'remove'
-		call denops#request('traqvim', 'removeStamps', [bufnr(), traqvim#message#get_message(), a:args[1:]])
+		call denops#request('traqvim', 'messageRemoveStamps', [bufnr(), traqvim#message#get_message(), a:args[1:]])
 	endif
 endfunction
 

diff --git a/denops/traqvim/action.ts b/denops/traqvim/action.ts
@@ -10,6 +10,7 @@ import {
   deleteMessage,
   editMessage,
   getMessageStamps,
+  removeMessageStamp,
   removePin,
 } from "./model.ts";
 import { isMessage } from "./type_check.ts";
@@ -321,3 +322,50 @@ export const actionAddMessageStamp = async (
     editedTimeline.find((m) => m.id === message.id),
   );
 }
+
+export const actionRemoveMessageStamp = async (
+  denops: Denops,
+  message: Message,
+  stamp: string,
+  bufNum: number,
+): Promise<void> => {
+  try {
+    await removeMessageStamp(message.id, stamp);
+  } catch (e) {
+    console.error(e);
+    return;
+  }
+  // 既存メッセージの取得
+  const timeline = await vars.buffers.get(denops, "channelTimeline");
+  assert(timeline, is.ArrayOf(isMessage));
+  // 一旦対象メッセージを削除する
+  await vars.buffers.set(
+    denops,
+    "channelTimeline",
+    timeline.filter((m) => m.id !== message.id),
+  );
+  await denops.call("traqvim#view#draw_delete_message", bufNum, message);
+  // 更新されたメッセージにあるスタンプを取得
+  const updatedMessage = await getMessageStamps(message.id);
+  const editedTimeline = timeline.map((m) => {
+    if (m.id === message.id) {
+      return {
+        ...m,
+        stamps: updatedMessage,
+      };
+    } else {
+      return m;
+    }
+  });
+  // 編集したものをセット
+  await vars.buffers.set(
+    denops,
+    "channelTimeline",
+    editedTimeline,
+  );
+  await denops.call(
+    "traqvim#view#draw_append_message",
+    bufNum,
+    editedTimeline.find((m) => m.id === message.id),
+  );
+};
diff --git a/denops/traqvim/main.ts b/denops/traqvim/main.ts
@@ -33,6 +33,7 @@ import {
   actionForwardChannelMessage,
   actionOpenActivity,
   actionOpenChannel,
+  actionRemoveMessageStamp,
   actionRemovePin,
   actionYankMessageLink,
   actionYankMessageMarkdown,
@@ -425,6 +426,25 @@ export async function main(denops: Denops) {
       }
       return;
     },
+    async messageRemoveStamps(
+      bufNum: unknown,
+      message: unknown,
+      stampNames: unknown,
+    ): Promise<unknown> {
+      assert(bufNum, is.Number);
+      assert(message, isMessage);
+      assert(stampNames, is.ArrayOf(is.String));
+      for (const stampName of stampNames) {
+        const stampId = await getStampId(stampName);
+        if (stampId === undefined) {
+          helper.echo(denops, `Stamp not found: ${stampName}`);
+          continue;
+        }
+        const stamp = await getStamp(stampId);
+        await actionRemoveMessageStamp(denops, message, stamp.id, bufNum);
+      }
+      return;
+    },
     async createPin(
       bufNum: unknown,
       message: unknown,