From 44230891aadae19c959d368203dad4ae76a556a9 Mon Sep 17 00:00:00 2001
From: yanglin <yanglin@test.foo>
Date: Fri, 10 May 2024 10:30:45 +0800
Subject: [PATCH] =?UTF-8?q?REQ-2303:=20=E5=A4=84=E7=90=86=E6=B5=81?=
 =?UTF-8?q?=E7=A8=8B=E8=87=AA=E5=AE=9A=E4=B9=89=E6=8C=89=E9=92=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/enums/RouterButtonSourceEnum.java   |  7 ++++++-
 .../response/PendingMessageResponse.java        |  2 +-
 .../pending/response/TodoButtonProvider.java    | 17 +++++++++++++++++
 3 files changed, 24 insertions(+), 2 deletions(-)
 create mode 100644 msg-center-api/src/main/java/cn/axzo/msg/center/service/pending/response/TodoButtonProvider.java

diff --git a/msg-center-api/src/main/java/cn/axzo/msg/center/service/enums/RouterButtonSourceEnum.java b/msg-center-api/src/main/java/cn/axzo/msg/center/service/enums/RouterButtonSourceEnum.java
index 96f6c72d..7daa0473 100644
--- a/msg-center-api/src/main/java/cn/axzo/msg/center/service/enums/RouterButtonSourceEnum.java
+++ b/msg-center-api/src/main/java/cn/axzo/msg/center/service/enums/RouterButtonSourceEnum.java
@@ -11,7 +11,7 @@ import lombok.Getter;
  */
 @Getter
 @AllArgsConstructor(access = AccessLevel.PRIVATE)
-public enum RouterButtonSourceEnum {
+public enum RouterButtonSourceEnum implements CodeDefinition<String> {
 
     /**
      * 系统
@@ -21,4 +21,9 @@ public enum RouterButtonSourceEnum {
      * 自定义
      */
     CUSTOM;
+
+    @Override
+    public String getCode() {
+        return name();
+    }
 }
diff --git a/msg-center-api/src/main/java/cn/axzo/msg/center/service/pending/response/PendingMessageResponse.java b/msg-center-api/src/main/java/cn/axzo/msg/center/service/pending/response/PendingMessageResponse.java
index d1e3c941..5532cddf 100644
--- a/msg-center-api/src/main/java/cn/axzo/msg/center/service/pending/response/PendingMessageResponse.java
+++ b/msg-center-api/src/main/java/cn/axzo/msg/center/service/pending/response/PendingMessageResponse.java
@@ -38,7 +38,7 @@ import java.util.List;
 @Builder
 @NoArgsConstructor
 @AllArgsConstructor
-public class PendingMessageResponse implements Serializable {
+public class PendingMessageResponse implements Serializable, TodoButtonProvider {
 
     private static final long serialVersionUID = -4133864010679470398L;
 
diff --git a/msg-center-api/src/main/java/cn/axzo/msg/center/service/pending/response/TodoButtonProvider.java b/msg-center-api/src/main/java/cn/axzo/msg/center/service/pending/response/TodoButtonProvider.java
new file mode 100644
index 00000000..33e9922e
--- /dev/null
+++ b/msg-center-api/src/main/java/cn/axzo/msg/center/service/pending/response/TodoButtonProvider.java
@@ -0,0 +1,17 @@
+package cn.axzo.msg.center.service.pending.response;
+
+import cn.axzo.msg.center.api.custombutton.ProposedButton;
+import cn.axzo.msg.center.service.dto.ButtonRouterDTO;
+
+import java.util.List;
+
+/**
+ * @author yanglin
+ */
+public interface TodoButtonProvider {
+
+    List<ButtonRouterDTO> getButtonRouters();
+
+    List<ProposedButton> getProposedButtons();
+
+}