Add builder argument patterns for assistant thread utilities

Author: seratch

bolt-socket-mode/src/test/java/samples/AssistantSimpleApp.java

@@ -22,29 +22,31 @@ public static void main(String[] args) throws Exception {
 
         assistant.threadStarted((req, ctx) -> {
             try {
-                ctx.say(r -> r.text("Hi, how can I help you today?"));
-                ctx.setSuggestedPrompts(Collections.singletonList(
-                        SuggestedPrompt.create("What does SLACK stand for?")
-                ));
+                ctx.say("Hi, how can I help you today?");
+                ctx.setSuggestedPrompts(r -> r
+                        .title("Select one of the following:") // optional
+                        .prompts(Collections.singletonList(SuggestedPrompt.create("What does SLACK stand for?")))
+                );
             } catch (Exception e) {
                 ctx.logger.error("Failed to handle assistant thread started event: {e}", e);
             }
         });
 
         assistant.userMessage((req, ctx) -> {
             try {
+                // ctx.setStatus(r -> r.status("is typing...")); works too
                 ctx.setStatus("is typing...");
                 Thread.sleep(500L);
                 if (ctx.getThreadContext() != null) {
                     String contextChannel = ctx.getThreadContext().getChannelId();
-                    ctx.say(r -> r.text("I am ware of the channel context: <#" + contextChannel + ">"));
+                    ctx.say("I am ware of the channel context: <#" + contextChannel + ">");
                 } else {
-                    ctx.say(r -> r.text("Here you are!"));
+                    ctx.say("Here you are!");
                 }
             } catch (Exception e) {
                 ctx.logger.error("Failed to handle assistant user message event: {e}", e);
                 try {
-                    ctx.say(r -> r.text(":warning: Sorry, something went wrong during processing your request!"));
+                    ctx.say(":warning: Sorry, something went wrong during processing your request!");
                 } catch (Exception ee) {
                     ctx.logger.error("Failed to inform the error to the end-user: {ee}", ee);
                 }
@@ -57,11 +59,11 @@ public static void main(String[] args) throws Exception {
                 Thread.sleep(500L);
                 ctx.setStatus("is still checking the files...");
                 Thread.sleep(500L);
-                ctx.say(r -> r.text("Your files do not have any issues!"));
+                ctx.say("Your files do not have any issues!");
             } catch (Exception e) {
                 ctx.logger.error("Failed to handle assistant user message event: {e}", e);
                 try {
-                    ctx.say(r -> r.text(":warning: Sorry, something went wrong during processing your request!"));
+                    ctx.say(":warning: Sorry, something went wrong during processing your request!");
                 } catch (Exception ee) {
                     ctx.logger.error("Failed to inform the error to the end-user: {ee}", ee);
                 }

bolt/src/main/java/com/slack/api/bolt/context/builtin/EventContext.java

@@ -1,14 +1,19 @@
 package com.slack.api.bolt.context.builtin;
 
+import com.slack.api.RequestConfigurator;
 import com.slack.api.bolt.context.Context;
 import com.slack.api.bolt.context.FunctionUtility;
 import com.slack.api.bolt.context.SayUtility;
 import com.slack.api.bolt.service.AssistantThreadContextService;
 import com.slack.api.bolt.util.BuilderConfigurator;
 import com.slack.api.methods.SlackApiException;
+import com.slack.api.methods.request.assistant.threads.AssistantThreadsSetStatusRequest;
+import com.slack.api.methods.request.assistant.threads.AssistantThreadsSetSuggestedPromptsRequest;
+import com.slack.api.methods.request.assistant.threads.AssistantThreadsSetTitleRequest;
 import com.slack.api.methods.request.chat.ChatPostMessageRequest;
 import com.slack.api.methods.response.asssistant.threads.AssistantThreadsSetStatusResponse;
 import com.slack.api.methods.response.asssistant.threads.AssistantThreadsSetSuggestedPromptsResponse;
+import com.slack.api.methods.response.asssistant.threads.AssistantThreadsSetTitleResponse;
 import com.slack.api.methods.response.chat.ChatPostMessageResponse;
 import com.slack.api.model.Message;
 import com.slack.api.model.assistant.AssistantThreadContext;
@@ -111,6 +116,40 @@ public AssistantThreadsSetStatusResponse setStatus(String status) throws IOExcep
         }
     }
 
+    public AssistantThreadsSetStatusResponse setStatus(RequestConfigurator<AssistantThreadsSetStatusRequest.AssistantThreadsSetStatusRequestBuilder> req) throws IOException, SlackApiException {
+        if (isAssistantThreadEvent()) {
+            return this.client().assistantThreadsSetStatus(req.configure(AssistantThreadsSetStatusRequest.builder()
+                    .channelId(this.getChannelId())
+                    .threadTs(this.getThreadTs())
+            ).build());
+        } else {
+            throw new IllegalStateException("This utility is only available for Assistant feature enabled app!");
+        }
+    }
+
+    public AssistantThreadsSetTitleResponse setTitle(String title) throws IOException, SlackApiException {
+        if (isAssistantThreadEvent()) {
+            return this.client().assistantThreadsSetTitle(r -> r
+                    .channelId(this.getChannelId())
+                    .threadTs(this.getThreadTs())
+                    .title(title)
+            );
+        } else {
+            throw new IllegalStateException("This utility is only available for Assistant feature enabled app!");
+        }
+    }
+
+    public AssistantThreadsSetTitleResponse setTitle(RequestConfigurator<AssistantThreadsSetTitleRequest.AssistantThreadsSetTitleRequestBuilder> req) throws IOException, SlackApiException {
+        if (isAssistantThreadEvent()) {
+            return this.client().assistantThreadsSetTitle(req.configure(AssistantThreadsSetTitleRequest.builder()
+                    .channelId(this.getChannelId())
+                    .threadTs(this.getThreadTs())
+            ).build());
+        } else {
+            throw new IllegalStateException("This utility is only available for Assistant feature enabled app!");
+        }
+    }
+
     public AssistantThreadsSetSuggestedPromptsResponse setSuggestedPrompts(List<SuggestedPrompt> prompts) throws IOException, SlackApiException {
         if (isAssistantThreadEvent()) {
             return this.client().assistantThreadsSetSuggestedPrompts(r -> r
@@ -123,6 +162,17 @@ public AssistantThreadsSetSuggestedPromptsResponse setSuggestedPrompts(List<Sugg
         }
     }
 
+    public AssistantThreadsSetSuggestedPromptsResponse setSuggestedPrompts(RequestConfigurator<AssistantThreadsSetSuggestedPromptsRequest.AssistantThreadsSetSuggestedPromptsRequestBuilder> req) throws IOException, SlackApiException {
+        if (isAssistantThreadEvent()) {
+            return this.client().assistantThreadsSetSuggestedPrompts(req.configure(AssistantThreadsSetSuggestedPromptsRequest.builder()
+                    .channelId(this.getChannelId())
+                    .threadTs(this.getThreadTs())
+            ).build());
+        } else {
+            throw new IllegalStateException("This utility is only available for Assistant feature enabled app!");
+        }
+    }
+
     public AssistantThreadsSetSuggestedPromptsResponse setSuggestedPrompts(List<SuggestedPrompt> prompts, String title) throws IOException, SlackApiException {
         if (isAssistantThreadEvent()) {
             return this.client().assistantThreadsSetSuggestedPrompts(r -> r

bolt/src/test/java/test_locally/app/EventAssistantTest.java

@@ -16,10 +16,7 @@
 import util.AuthTestMockServer;
 import util.MockSlackApiServer;
 
-import java.util.Collections;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 import java.util.concurrent.atomic.AtomicBoolean;
 
 import static org.junit.Assert.assertEquals;
@@ -55,6 +52,15 @@ public void test() throws Exception {
         Assistant assistant = new Assistant(app.executorService());
 
         assistant.threadStarted((req, ctx) -> {
+            try {
+                ctx.setTitle("title");
+                ctx.setTitle(r -> r.title("title"));
+                ctx.setStatus("is typing...");
+                ctx.setStatus(r -> r.status("is typing..."));
+                ctx.setSuggestedPrompts(new ArrayList<>());
+                ctx.setSuggestedPrompts(r -> r.prompts(new ArrayList<>()));
+            } catch (Exception e) {
+            }
             threadStartedReceived.set(true);
         });
         assistant.threadContextChanged((req, ctx) -> {