机器人api

2021-01-16 10:29:03 +08:00
parent 46423a341d
commit a725b507c7
12 changed files with 365 additions and 10 deletions
@@ -0,0 +1,41 @@
+/*
+ * This file is part of the Wildfire Chat package.
+ * (c) Heavyrain2012 <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+package com.xiaoleilu.loServer.action.robot;
+
+import cn.wildfirechat.common.APIPath;
+import cn.wildfirechat.common.ErrorCode;
+import cn.wildfirechat.pojos.RobotCallbackPojo;
+import com.google.gson.Gson;
+import com.xiaoleilu.loServer.RestResult;
+import com.xiaoleilu.loServer.annotation.HttpMethod;
+import com.xiaoleilu.loServer.annotation.Route;
+import com.xiaoleilu.loServer.handler.Request;
+import com.xiaoleilu.loServer.handler.Response;
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.HttpResponseStatus;
+import io.netty.util.internal.StringUtil;
+
+@Route(APIPath.Robot_Delete_Callback)
+@HttpMethod("POST")
+public class DeleteCallbackAction extends RobotAction {
+
+    @Override
+    public boolean action(Request request, Response response) {
+        if (request.getNettyRequest() instanceof FullHttpRequest) {
+            if(!StringUtil.isNullOrEmpty(robot.getCallback())) {
+                robot = robot.toBuilder().clearCallback().setState(0).build();
+                messagesStore.addRobot(robot);
+            }
+            response.setStatus(HttpResponseStatus.OK);
+            RestResult result = RestResult.ok();
+            response.setContent(new Gson().toJson(result));
+        }
+        return true;
+    }
+}
@@ -0,0 +1,42 @@
+/*
+ * This file is part of the Wildfire Chat package.
+ * (c) Heavyrain2012 <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+package com.xiaoleilu.loServer.action.robot;
+
+import cn.wildfirechat.common.APIPath;
+import cn.wildfirechat.common.ErrorCode;
+import cn.wildfirechat.pojos.InputGetUserInfo;
+import cn.wildfirechat.pojos.InputOutputUserInfo;
+import cn.wildfirechat.pojos.RobotCallbackPojo;
+import cn.wildfirechat.proto.WFCMessage;
+import com.google.gson.Gson;
+import com.xiaoleilu.loServer.RestResult;
+import com.xiaoleilu.loServer.annotation.HttpMethod;
+import com.xiaoleilu.loServer.annotation.Route;
+import com.xiaoleilu.loServer.handler.Request;
+import com.xiaoleilu.loServer.handler.Response;
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.HttpResponseStatus;
+import io.netty.util.internal.StringUtil;
+
+@Route(APIPath.Robot_Get_Callback)
+@HttpMethod("POST")
+public class GetCallbackAction extends RobotAction {
+
+    @Override
+    public boolean action(Request request, Response response) {
+        if (request.getNettyRequest() instanceof FullHttpRequest) {
+            RobotCallbackPojo out = new RobotCallbackPojo();
+            out.setUrl(robot.getCallback());
+            response.setStatus(HttpResponseStatus.OK);
+            RestResult result = RestResult.ok(out);
+            response.setContent(new Gson().toJson(result));
+        }
+        return true;
+    }
+}
@@ -0,0 +1,51 @@
+/*
+ * This file is part of the Wildfire Chat package.
+ * (c) Heavyrain2012 <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+package com.xiaoleilu.loServer.action.robot;
+
+import cn.wildfirechat.common.APIPath;
+import cn.wildfirechat.common.ErrorCode;
+import cn.wildfirechat.pojos.InputGetGroup;
+import cn.wildfirechat.pojos.RobotCallbackPojo;
+import com.google.gson.Gson;
+import com.xiaoleilu.loServer.RestResult;
+import com.xiaoleilu.loServer.annotation.HttpMethod;
+import com.xiaoleilu.loServer.annotation.Route;
+import com.xiaoleilu.loServer.handler.Request;
+import com.xiaoleilu.loServer.handler.Response;
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.HttpResponseStatus;
+import io.netty.util.internal.StringUtil;
+
+@Route(APIPath.Robot_Set_Callback)
+@HttpMethod("POST")
+public class SetCallbackAction extends RobotAction {
+
+    @Override
+    public boolean action(Request request, Response response) {
+        if (request.getNettyRequest() instanceof FullHttpRequest) {
+            RobotCallbackPojo input = getRequestBody(request.getNettyRequest(), RobotCallbackPojo.class);
+            if(!StringUtil.isNullOrEmpty(input.getUrl())) {
+                if(!input.getUrl().equals(robot.getCallback()) && robot.getState() != 1) {
+                    robot = robot.toBuilder().setCallback(input.getUrl()).setState(1).build();
+                    messagesStore.addRobot(robot);
+                }
+
+                response.setStatus(HttpResponseStatus.OK);
+                RestResult result = RestResult.ok();
+                response.setContent(new Gson().toJson(result));
+            } else {
+                response.setStatus(HttpResponseStatus.OK);
+                RestResult result = RestResult.resultOf(ErrorCode.INVALID_PARAMETER);
+                response.setContent(new Gson().toJson(result));
+            }
+
+        }
+        return true;
+    }
+}
@@ -0,0 +1,94 @@
+/*
+ * This file is part of the Wildfire Chat package.
+ * (c) Heavyrain2012 <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+package com.xiaoleilu.loServer.action.robot;
+
+import cn.wildfirechat.common.APIPath;
+import cn.wildfirechat.common.ErrorCode;
+import cn.wildfirechat.pojos.IntStringPairPojo;
+import cn.wildfirechat.proto.WFCMessage;
+import com.google.gson.Gson;
+import com.xiaoleilu.loServer.RestResult;
+import com.xiaoleilu.loServer.annotation.HttpMethod;
+import com.xiaoleilu.loServer.annotation.Route;
+import com.xiaoleilu.loServer.handler.Request;
+import com.xiaoleilu.loServer.handler.Response;
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.HttpResponseStatus;
+
+import static cn.wildfirechat.pojos.MyInfoType.*;
+
+@Route(APIPath.Robot_Update_Profile)
+@HttpMethod("POST")
+public class UpdateProfileAction extends RobotAction {
+
+    @Override
+    public boolean action(Request request, Response response) {
+        if (request.getNettyRequest() instanceof FullHttpRequest) {
+            IntStringPairPojo input = getRequestBody(request.getNettyRequest(), IntStringPairPojo.class);
+
+            WFCMessage.User.Builder builder = messagesStore.getUserInfo(robot.getUid()).toBuilder();
+            boolean modified = false;
+            switch (input.getIntValue()) {
+                case Modify_DisplayName:
+                    builder.setDisplayName(input.getStrValue());
+                    modified = true;
+                    break;
+                case Modify_Gender:
+                    builder.setGender(Integer.parseInt(input.getStrValue()));
+                    modified = true;
+                    break;
+                case Modify_Portrait:
+                    builder.setPortrait(input.getStrValue());
+                    modified = true;
+                    break;
+                //不允许修改电话号码，如果修改电话号码必须通过admin进行修改
+//              case Modify_Mobile:
+//                    builder.setMobile(input.getStrValue());
+//                    modified = true;
+//                    break;
+                case Modify_Email:
+                    builder.setEmail(input.getStrValue());
+                    modified = true;
+                    break;
+                case Modify_Address:
+                    builder.setAddress(input.getStrValue());
+                    modified = true;
+                    break;
+                case Modify_Company:
+                    builder.setCompany(input.getStrValue());
+                    modified = true;
+                    break;
+                case Modify_Social:
+                    builder.setSocial(input.getStrValue());
+                    modified = true;
+                    break;
+                case Modify_Extra:
+                    builder.setExtra(input.getStrValue());
+                    modified = true;
+                    break;
+                default:
+                    break;
+            }
+
+            response.setStatus(HttpResponseStatus.OK);
+            RestResult result = RestResult.ok();
+            if(modified) {
+                try {
+                    messagesStore.updateUserInfo(builder.build());
+                } catch (Exception e) {
+                    e.printStackTrace();
+                    result.setErrorCode(ErrorCode.ERROR_CODE_SERVER_ERROR);
+                }
+            }
+
+            response.setContent(new Gson().toJson(result));
+        }
+        return true;
+    }
+}