wildfire chat server source initial submit

2019-03-16 17:04:23 +08:00
parent e59217bd10
commit cab3ca0916
1466 changed files with 528862 additions and 15919 deletions
@@ -0,0 +1,63 @@
+/*
+ * 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.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 com.xiaoleilu.loServer.pojos.InputCreateUser;
+import com.xiaoleilu.loServer.pojos.InputGetUserInfo;
+import com.xiaoleilu.loServer.pojos.OutputCreateUser;
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.HttpResponseStatus;
+import io.netty.util.internal.StringUtil;
+import win.liyufan.im.ErrorCode;
+
+@Route("admin/robot/user_info")
+@HttpMethod("POST")
+public class GetUserAction extends RobotAction {
+
+    @Override
+    public void action(Request request, Response response) {
+        if (request.getNettyRequest() instanceof FullHttpRequest) {
+            InputGetUserInfo inputUserId = getRequestBody(request.getNettyRequest(), InputGetUserInfo.class);
+            if (inputUserId != null
+                && (!StringUtil.isNullOrEmpty(inputUserId.getUserId()) || !StringUtil.isNullOrEmpty(inputUserId.getName()))) {
+
+                WFCMessage.User user = null;
+                if(!StringUtil.isNullOrEmpty(inputUserId.getUserId())) {
+                    user = messagesStore.getUserInfo(inputUserId.getUserId());
+                } else if(!StringUtil.isNullOrEmpty(inputUserId.getName())) {
+                    user = messagesStore.getUserInfoByName(inputUserId.getName());
+                } else if(!StringUtil.isNullOrEmpty(inputUserId.getMobile())) {
+                    user = messagesStore.getUserInfoByMobile(inputUserId.getMobile());
+                }
+
+                response.setStatus(HttpResponseStatus.OK);
+                RestResult result;
+                if (user == null) {
+                    result = RestResult.resultOf(ErrorCode.ERROR_CODE_NOT_EXIST);
+                } else {
+                    result = RestResult.ok(InputCreateUser.fromPbUser(user));
+                }
+
+                response.setContent(new Gson().toJson(result));
+            } else {
+                response.setStatus(HttpResponseStatus.OK);
+                RestResult result = RestResult.resultOf(ErrorCode.INVALID_PARAMETER);
+                response.setContent(new Gson().toJson(result));
+            }
+
+        }
+    }
+}
@@ -0,0 +1,90 @@
+/*
+ * 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.proto.WFCMessage;
+import com.google.gson.Gson;
+import com.xiaoleilu.loServer.RestResult;
+import com.xiaoleilu.loServer.action.Action;
+import com.xiaoleilu.loServer.handler.Request;
+import com.xiaoleilu.loServer.handler.Response;
+import io.moquette.persistence.RPCCenter;
+import io.moquette.spi.impl.Utils;
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.HttpRequest;
+import io.netty.handler.codec.http.HttpResponseStatus;
+import io.netty.util.internal.StringUtil;
+import org.apache.commons.codec.digest.DigestUtils;
+import win.liyufan.im.ErrorCode;
+import win.liyufan.im.RateLimiter;
+
+abstract public class RobotAction extends Action {
+    private final RateLimiter mLimitCounter = new RateLimiter(10, 1000);
+
+    protected WFCMessage.Robot robot;
+
+    @Override
+    public ErrorCode preAction(Request request, Response response) {
+        String nonce = request.getHeader("nonce");
+        String timestamp = request.getHeader("timestamp");
+        String sign = request.getHeader("sign");
+        String rid = request.getHeader("rid");
+        if (StringUtil.isNullOrEmpty(nonce) || StringUtil.isNullOrEmpty(timestamp) || StringUtil.isNullOrEmpty(sign) || StringUtil.isNullOrEmpty(rid)) {
+            return ErrorCode.INVALID_PARAMETER;
+        }
+
+        if (mLimitCounter.isGranted(rid)) {
+            return ErrorCode.ERROR_CODE_OVER_FREQUENCY;
+        }
+
+        Long ts;
+        try {
+            ts = Long.parseLong(timestamp);
+        } catch (NumberFormatException e) {
+            e.printStackTrace();
+            return ErrorCode.INVALID_PARAMETER;
+        }
+
+        if (System.currentTimeMillis() - ts > 2 * 60 * 60 * 1000) {
+            return ErrorCode.ERROR_CODE_SIGN_EXPIRED;
+        }
+
+        robot = messagesStore.getRobot(rid);
+        if (robot == null) {
+            return ErrorCode.ERROR_CODE_NOT_EXIST;
+        }
+
+        String str = nonce + "|" + robot.getSecret() + "|" + timestamp;
+        String localSign = DigestUtils.sha1Hex(str);
+        return localSign.equals(sign) ? ErrorCode.ERROR_CODE_SUCCESS : ErrorCode.ERROR_CODE_AUTH_FAILURE;
+    }
+
+    protected void sendResponse(Response response, ErrorCode errorCode, Object data) {
+        response.setStatus(HttpResponseStatus.OK);
+        if (errorCode == null) {
+            errorCode = ErrorCode.ERROR_CODE_SUCCESS;
+        }
+
+        RestResult result = RestResult.resultOf(errorCode, errorCode.getMsg(), data);
+        response.setContent(new Gson().toJson(result));
+        response.send();
+    }
+
+    protected <T> T getRequestBody(HttpRequest request, Class<T> cls) {
+        if (request instanceof FullHttpRequest) {
+            FullHttpRequest fullHttpRequest = (FullHttpRequest) request;
+            byte[] bytes = Utils.readBytesAndRewind(fullHttpRequest.content());
+            String content = new String(bytes);
+            Gson gson = new Gson();
+            T t = gson.fromJson(content, cls);
+            return t;
+        }
+        return null;
+    }
+}
@@ -0,0 +1,87 @@
+/*
+ * 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.proto.ProtoConstants;
+import com.google.gson.Gson;
+import com.xiaoleilu.loServer.RestResult;
+import com.xiaoleilu.loServer.action.admin.AdminAction;
+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 com.xiaoleilu.loServer.pojos.SendMessageData;
+import com.xiaoleilu.loServer.pojos.SendMessageResult;
+import io.moquette.persistence.RPCCenter;
+import io.moquette.persistence.TargetEntry;
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.Unpooled;
+import io.netty.handler.codec.http.FullHttpRequest;
+import io.netty.handler.codec.http.HttpResponseStatus;
+import win.liyufan.im.ErrorCode;
+import win.liyufan.im.IMTopic;
+
+import java.util.concurrent.Executor;
+
+@Route("admin/robot/send")
+@HttpMethod("POST")
+public class SendMessageAction extends RobotAction {
+
+    @Override
+    public boolean isTransactionAction() {
+        return true;
+    }
+
+    @Override
+    public void action(Request request, Response response) {
+        if (request.getNettyRequest() instanceof FullHttpRequest) {
+            SendMessageData sendMessageData = getRequestBody(request.getNettyRequest(), SendMessageData.class);
+            sendMessageData.setSender(robot.getUid());
+            if (SendMessageData.isValide(sendMessageData)) {
+                RPCCenter.getInstance().sendRequest(sendMessageData.getSender(), null, IMTopic.SendMessageTopic, sendMessageData.toProtoMessage().toByteArray(), sendMessageData.getSender(), TargetEntry.Type.TARGET_TYPE_USER, new RPCCenter.Callback() {
+                    @Override
+                    public void onSuccess(byte[] result) {
+                        ByteBuf byteBuf = Unpooled.buffer();
+                        byteBuf.writeBytes(result);
+                        ErrorCode errorCode = ErrorCode.fromCode(byteBuf.readByte());
+                        if (errorCode == ErrorCode.ERROR_CODE_SUCCESS) {
+                            long messageId = byteBuf.readLong();
+                            long timestamp = byteBuf.readLong();
+                            sendResponse(response, null, new SendMessageResult(messageId, timestamp));
+                        } else {
+                            sendResponse(response, errorCode, null);
+                        }
+                    }
+
+                    @Override
+                    public void onError(ErrorCode errorCode) {
+                        sendResponse(response, errorCode, null);
+                    }
+
+                    @Override
+                    public void onTimeout() {
+                        sendResponse(response, ErrorCode.ERROR_CODE_TIMEOUT, null);
+                    }
+
+                    @Override
+                    public Executor getResponseExecutor() {
+                        return command -> {
+                            ctx.executor().execute(command);
+                        };
+                    }
+                });
+            } else {
+                response.setStatus(HttpResponseStatus.OK);
+                RestResult result = RestResult.resultOf(ErrorCode.INVALID_PARAMETER);
+                response.setContent(new Gson().toJson(result));
+            }
+        }
+    }
+}