詹子聪 il y a 5 ans
Parent
commit
710ba2812d

+ 36 - 0
src/main/java/com/miekir/shibei/controller/api/UserController.java

@@ -3,9 +3,11 @@ package com.miekir.shibei.controller.api;
 import com.alibaba.fastjson.JSON;
 import com.miekir.shibei.bean.ResponseResult;
 import com.miekir.shibei.bean.ResultCode;
+import com.miekir.shibei.bean.db.SystemBean;
 import com.miekir.shibei.bean.db.User;
 import com.miekir.shibei.repository.UserRepository;
 import com.miekir.shibei.tool.CodeGenerator;
+import com.miekir.shibei.tool.RequestTool;
 import com.miekir.shibei.tool.TextUtils;
 import com.miekir.shibei.tool.TokenGenerator;
 import com.miekir.shibei.tool.email.EmailTool;
@@ -233,6 +235,40 @@ public class UserController {
         }
     }
 
+    /**
+     * 设置为VIP
+     */
+    @RequestMapping(value = "/api/setVip", method = RequestMethod.POST, produces = "application/json; charset=utf-8")
+    @ResponseBody
+    public String setApiConfig(@RequestHeader HttpHeaders header, String email) {
+        ResponseResult<String> responseResult = new ResponseResult<String>();
+        responseResult.setMessage("设置失败");
+        if (!RequestTool.isRequestAdminValid(header, userRepository)) {
+            responseResult.setMessage("登录过期或没有权限");
+            return JSON.toJSONString(responseResult);
+        }
+
+        User user = userRepository.findUserByEmail(email);
+        if (user == null) {
+            responseResult.setMessage("没有此用户");
+            return JSON.toJSONString(responseResult);
+        }
+
+        user.setVip(true);
+        userRepository.save(user);
+
+        responseResult.setCode(ResultCode.SUCCESS);
+        responseResult.setMessage("操作成功");
+        return JSON.toJSONString(responseResult);
+    }
+
+
+
+
+
+
+
+
     /**
      * 登录:email+token
      */