代码初始化

vetti-common/src/main/java/com/vetti/common/ai/WapiAiClient.java

@@ -0,0 +1,159 @@
+package com.vetti.common.ai;
+
+import org.apache.http.HttpEntity;
+import org.apache.http.client.methods.CloseableHttpResponse;
+import org.apache.http.client.methods.HttpGet;
+import org.apache.http.client.methods.HttpPost;
+import org.apache.http.entity.ContentType;
+import org.apache.http.entity.StringEntity;
+import org.apache.http.impl.client.CloseableHttpClient;
+import org.apache.http.impl.client.HttpClients;
+import org.apache.http.util.EntityUtils;
+import com.google.gson.Gson;
+
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * 发送文本消息到指定的WhatsApp
+ */
+public class WapiAiClient {
+
+    private static final String BASE_URL = "https://wapi.ai/api";
+    private final String apiKey;
+    private final CloseableHttpClient httpClient;
+    private final Gson gson;
+
+    public WapiAiClient(String apiKey) {
+        this.apiKey = apiKey;
+        this.httpClient = HttpClients.createDefault();
+        this.gson = new Gson();
+    }
+
+    /**
+     * 发送文本消息到指定的WhatsApp号码
+     * @param phoneNumber 目标电话号码(带国家代码，如：+1234567890)
+     * @param message 要发送的消息内容
+     * @return  API响应
+     * @throws IOException 网络请求异常
+     */
+    public WapiResponse sendTextMessage(String phoneNumber, String message) throws IOException {
+        HttpPost httpPost = new HttpPost(BASE_URL + "/sendMessage");
+        httpPost.setHeader("Authorization", "Bearer " + apiKey);
+        httpPost.setHeader("Content-Type", "application/json");
+
+        Map<String, Object> payload = new HashMap<>();
+        payload.put("phone", phoneNumber);
+        payload.put("body", message);
+
+        StringEntity entity = new StringEntity(gson.toJson(payload), ContentType.APPLICATION_JSON);
+        httpPost.setEntity(entity);
+
+        try (CloseableHttpResponse response = httpClient.execute(httpPost)) {
+            HttpEntity responseEntity = response.getEntity();
+            String responseBody = EntityUtils.toString(responseEntity);
+            return gson.fromJson(responseBody, WapiResponse.class);
+        }
+    }
+
+    /**
+     * 获取最近的消息
+     * @param limit 消息数量限制
+     * @return 消息列表
+     * @throws IOException 网络请求异常
+     */
+    public WapiMessagesResponse getRecentMessages(int limit) throws IOException {
+        HttpGet httpGet = new HttpGet(BASE_URL + "/messages?limit=" + limit);
+        httpGet.setHeader("Authorization", "Bearer " + apiKey);
+
+        try (CloseableHttpResponse response = httpClient.execute(httpGet)) {
+            HttpEntity responseEntity = response.getEntity();
+            String responseBody = EntityUtils.toString(responseEntity);
+            return gson.fromJson(responseBody, WapiMessagesResponse.class);
+        }
+    }
+
+    // 关闭HTTP客户端
+    public void close() throws IOException {
+        httpClient.close();
+    }
+
+    // 响应模型类
+    public static class WapiResponse {
+
+        private boolean success;
+        private String message;
+        private Object data;
+
+        // getter和setter方法
+        public boolean isSuccess() { return success; }
+        public void setSuccess(boolean success) { this.success = success; }
+        public String getMessage() { return message; }
+        public void setMessage(String message) { this.message = message; }
+        public Object getData() { return data; }
+        public void setData(Object data) { this.data = data; }
+    }
+
+    // 消息响应模型类
+    public static class WapiMessagesResponse {
+        private boolean success;
+        private Message[] messages;
+
+        // getter和setter方法
+        public boolean isSuccess() { return success; }
+        public void setSuccess(boolean success) { this.success = success; }
+        public Message[] getMessages() { return messages; }
+        public void setMessages(Message[] messages) { this.messages = messages; }
+
+        public static class Message {
+            private String id;
+            private String phone;
+            private String body;
+            private String type;
+            private String timestamp;
+
+            // getter和setter方法
+            public String getId() { return id; }
+            public void setId(String id) { this.id = id; }
+            public String getPhone() { return phone; }
+            public void setPhone(String phone) { this.phone = phone; }
+            public String getBody() { return body; }
+            public void setBody(String body) { this.body = body; }
+            public String getType() { return type; }
+            public void setType(String type) { this.type = type; }
+            public String getTimestamp() { return timestamp; }
+            public void setTimestamp(String timestamp) { this.timestamp = timestamp; }
+        }
+    }
+
+    // 使用示例
+    public static void main(String[] args) {
+        String apiKey = "your_wapi_ai_api_key";
+        WapiAiClient client = new WapiAiClient(apiKey);
+
+        try {
+            // 发送消息
+            WapiResponse response = client.sendTextMessage("+1234567890", "Hello from Wapi.ai Java Client!");
+            System.out.println("Message sent: " + response.isSuccess());
+
+            // 获取最近消息
+            WapiMessagesResponse messagesResponse = client.getRecentMessages(5);
+            if (messagesResponse.isSuccess() && messagesResponse.getMessages() != null) {
+                System.out.println("Recent messages:");
+                for (WapiMessagesResponse.Message msg : messagesResponse.getMessages()) {
+                    System.out.println(msg.getPhone() + ": " + msg.getBody());
+                }
+            }
+        } catch (IOException e) {
+            e.printStackTrace();
+        } finally {
+            try {
+                client.close();
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+}
+