钉钉

Author: wxywxq

src/main/java/com/cyl/ctrbt/controller/DingTalkController.java

@@ -1,15 +1,53 @@
 package com.cyl.ctrbt.controller;
 
+import cn.hutool.json.JSONObject;
+import cn.hutool.json.JSONUtil;
+import com.cyl.ctrbt.openai.OpenAiUtil;
+import com.dingtalk.api.DefaultDingTalkClient;
+import com.dingtalk.api.DingTalkClient;
+import com.dingtalk.api.request.OapiRobotSendRequest;
+import com.dingtalk.api.response.OapiRobotSendResponse;
+import com.taobao.api.ApiException;
+import com.theokanning.openai.completion.CompletionChoice;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.ResponseEntity;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
+
+import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
 
 @RequestMapping("/ding-talk")
 @RestController
 public class DingTalkController {
-    @GetMapping("/hello-world")
-    public ResponseEntity<String> helloWorld() {
-        return ResponseEntity.ok("hello world");
+
+  @Autowired
+  private OpenAiUtil openAiUtil;
+
+  @PostMapping("/receive")
+  public String helloRobots(@RequestBody(required = false) JSONObject json) {
+    System.out.println(JSONUtil.toJsonStr(json));
+    String content = json.getJSONObject("text").get("content").toString().replaceAll(" ", "");
+    String sessionWebhook = json.getStr("sessionWebhook");
+    DingTalkClient client = new DefaultDingTalkClient(sessionWebhook);
+    if ("text".equals(json.getStr("msgtype"))) {
+      text(client,content);
+    }
+    return null;
+  }
+
+  private void text(DingTalkClient client,String content) {
+    try {
+      OapiRobotSendRequest request = new OapiRobotSendRequest();
+      request.setMsgtype("text");
+      OapiRobotSendRequest.Text text = new OapiRobotSendRequest.Text();
+      List<CompletionChoice> completionChoices = openAiUtil.sendComplete(content);
+      text.setContent(completionChoices.stream().map(it->it.getText()).collect(Collectors.joining(";")));
+      request.setText(text);
+      OapiRobotSendResponse response = client.execute(request);
+      System.out.println(response.getBody());
+    } catch (ApiException e) {
+      e.printStackTrace();
     }
+  }
 }