ADD file via upload

flower_back/src/main/java/com/example/flower/controller/StaffController.java

@@ -0,0 +1,381 @@
+package com.example.flower.controller;
+
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
+import com.example.flower.entity.Staff;
+import com.example.flower.service.StaffService;
+import com.example.flower.unit.JWTUtil;
+import com.example.flower.unit.MenuList;
+import io.jsonwebtoken.Claims;
+import org.springframework.web.bind.annotation.*;
+import jakarta.annotation.Resource;
+
+import java.util.List;
+
+@RestController
+@RequestMapping("/staff")
+public class StaffController {
+    @Resource
+    private StaffService staffService;
+
+    @PostMapping("/getByToken") //通过token获取账号信息
+    public JSONObject getStaffByToken(@RequestHeader String Authorization){
+        JSONObject jsonObject = new JSONObject();
+
+        if(!JWTUtil.checkToken(Authorization)){ //token认证失败
+            JWTUtil.checkTokenFailed(jsonObject);
+            return jsonObject;
+        }
+        Claims claims = JWTUtil.getBodyByToken(Authorization);
+        int staff_id = (int) claims.get("id");
+
+        Staff staff = staffService.staffInfo(staff_id);
+
+        if(staff!=null){
+            JSONObject jsonObject1 = new JSONObject();
+            jsonObject1.put("staff_id",staff.getStaff_id());
+            jsonObject1.put("staff_name",staff.getStaff_name());
+            jsonObject1.put("staff_role",staff.getStaff_role());
+            jsonObject1.put("staff_pic",staff.getStaff_pic());
+
+            jsonObject.put("code",200);
+            jsonObject.put("msg","成功");
+            jsonObject.put("data",jsonObject1);
+
+        }
+        else{
+            jsonObject.put("code",403);
+            jsonObject.put("msg","获取员工信息失败");
+            jsonObject.put("data",null);
+        }
+
+        return jsonObject;
+    }
+
+    @PostMapping("/login") //staffLogin员工后台登录
+    public JSONObject staffLogin(@RequestBody JSONObject param){
+        JSONObject jsonObject = new JSONObject();
+
+        Staff staff = staffService.staffLogin(param.getString("staff_name"));
+        //根据用户名没有查到用户
+        if(staff==null){
+            jsonObject.put("code",403 );
+            jsonObject.put("msg"," 用户名不存在" );
+            jsonObject.put("token",null);
+            jsonObject.put("data",null); //菜单信息
+        }
+        else if (staff.getStaff_state()==0) { //账号为禁用状态
+            jsonObject.put("code",403);
+            jsonObject.put("msg","账号禁用，请联系管理员" );
+            jsonObject.put("token",null);
+            jsonObject.put("data",null); //菜单信息
+        }
+        //密码验证,创建token
+        else if(staff.getStaff_pwd().equals(param.getString("staff_pwd"))){
+            String token = JWTUtil.createToken(staff.getStaff_id(),staff.getStaff_role());
+
+            jsonObject.put("code",200 );
+            jsonObject.put("msg","登录成功" );
+            jsonObject.put("token",token);
+
+            JSONObject jsonObject1 = new JSONObject(); //放menu
+            JSONObject[] menuList = new JSONObject[0];
+
+            if(staff.getStaff_role()==1){ //店长可以获得所有菜单的权限
+                menuList = new JSONObject[8];
+                MenuList.setMenu(menuList, 1);
+            }
+            else if(staff.getStaff_role()==0){ //员工菜单
+                menuList = new JSONObject[7];
+                MenuList.setMenu(menuList, 0);
+            }
+            jsonObject1.put("menu",menuList);
+            jsonObject.put("data",jsonObject1);
+
+        }
+        else{
+            jsonObject.put("code",403 );
+            jsonObject.put("msg","  密码错误" );
+            jsonObject.put("token",null);
+            jsonObject.put("data",null); //菜单信息
+        }
+
+        return jsonObject;
+    }
+
+    @PostMapping("/list") //staffList获取所有用户信息列表
+    public JSONObject staffList(@RequestHeader String Authorization, @RequestBody JSONObject param){
+        JSONObject jsonObject = new JSONObject();
+
+        if(!JWTUtil.checkToken(Authorization)){ //token认证失败
+            JWTUtil.checkTokenFailed(jsonObject);
+            return jsonObject;
+        }
+
+        JSONObject jsonObject1 = new JSONObject();
+        int page=param.getIntValue("page");
+        int page_size=param.getIntValue("page_size");
+
+        List<Staff> staffList = staffService.staffList(page,page_size).getList();
+
+        jsonObject1.put("page_number",staffService.staffList(page,page_size).getPages());
+        jsonObject1.put("total",staffService.staffList(page,page_size).getTotal());
+
+        int size=staffList.size();
+
+        JSONObject[] objects = new JSONObject[size];
+        for(int i = 0;i<size;i++){
+            objects[i]=new JSONObject();
+            objects[i].put("staff_id", staffList.get(i).getStaff_id());
+            objects[i].put("staff_name", staffList.get(i).getStaff_name());
+            objects[i].put("staff_pwd", staffList.get(i).getStaff_pwd());
+            objects[i].put("staff_full_name",staffList.get(i).getStaff_full_name());
+            objects[i].put("staff_phone", staffList.get(i).getStaff_phone());
+            objects[i].put("staff_state", staffList.get(i).getStaff_state());
+            objects[i].put("staff_role", staffList.get(i).getStaff_role());
+            objects[i].put("staff_pic",staffList.get(i).getStaff_pic());
+
+        }
+        jsonObject1.put("staffs",objects);
+
+        jsonObject.put("code",200 );
+        jsonObject.put("msg","token认证成功!" );
+        jsonObject.put("data",jsonObject1);
+        return jsonObject;
+    }
+
+    @PostMapping("/query") //staffListByQuery查询用户信息列表
+    public JSONObject staffListByQuery(@RequestHeader String Authorization, @RequestBody JSONObject param){
+        JSONObject jsonObject = new JSONObject();
+
+        if(!JWTUtil.checkToken(Authorization)){ //token认证失败
+            JWTUtil.checkTokenFailed(jsonObject);
+            return jsonObject;
+        }
+
+        JSONObject jsonObject1 = new JSONObject();
+        int page=param.getIntValue("page");
+        int page_size=param.getIntValue("page_size");
+        String staff_name = param.getString("staff_name");
+
+        List<Staff> staffList = staffService.staffListByQuery(page,page_size,staff_name).getList();
+
+        jsonObject1.put("page_number",staffService.staffListByQuery(page,page_size,staff_name).getPages());
+        jsonObject1.put("total",staffService.staffListByQuery(page,page_size,staff_name).getTotal());
+
+        int size=staffList.size();
+
+        JSONObject[] objects = new JSONObject[size];
+        for(int i = 0;i<size;i++){
+            objects[i]=new JSONObject();
+            objects[i].put("staff_id", staffList.get(i).getStaff_id());
+            objects[i].put("staff_name", staffList.get(i).getStaff_name());
+            objects[i].put("staff_pwd", staffList.get(i).getStaff_pwd());
+            objects[i].put("staff_full_name",staffList.get(i).getStaff_full_name());
+            objects[i].put("staff_phone", staffList.get(i).getStaff_phone());
+            objects[i].put("staff_state", staffList.get(i).getStaff_state());
+            objects[i].put("staff_role", staffList.get(i).getStaff_role());
+            objects[i].put("staff_pic",staffList.get(i).getStaff_pic());
+
+        }
+        jsonObject1.put("staffs",objects);
+
+        jsonObject.put("code",200 );
+        jsonObject.put("msg","token认证成功!" );
+        jsonObject.put("data",jsonObject1);
+        return jsonObject;
+    }
+
+    @PostMapping("/add") //staffAdd添加新员工
+    public JSONObject staffAdd(@RequestHeader String Authorization, @RequestBody JSONObject param){
+        JSONObject jsonObject = new JSONObject();
+
+        if(!JWTUtil.checkToken(Authorization)){ //token认证失败
+            JWTUtil.checkTokenFailed(jsonObject);
+            return jsonObject;
+        }
+
+        Staff staff = new Staff();
+        staff.setStaff_name(param.getString("staff_name"));
+        staff.setStaff_pwd(param.getString("staff_pwd"));
+        staff.setStaff_full_name(param.getString("staff_full_name"));
+        staff.setStaff_phone(param.getString("staff_phone"));
+        staff.setStaff_role(param.getIntValue("staff_role"));
+        staff.setStaff_state(param.getIntValue("staff_state"));
+        staff.setStaff_pic(param.getString("staff_pic"));
+
+        staffService.staffAdd(staff);
+
+        jsonObject.put("code",200 );
+        jsonObject.put("msg","成功" );
+        jsonObject.put("staff_id",staff.getStaff_id());
+
+        return jsonObject;
+    }
+
+    @PostMapping("/deletePer") //staffDeletePer删除单个员工
+    public JSONObject staffDeletePer(@RequestHeader String Authorization, @RequestBody JSONObject param){
+        JSONObject jsonObject = new JSONObject();
+
+        if(!JWTUtil.checkToken(Authorization)){ //token认证失败
+            JWTUtil.checkTokenFailed(jsonObject);
+            return jsonObject;
+        }
+
+        int flag = staffService.staffDeletePer(param.getIntValue("staff_id"));
+        if(flag==1){
+            jsonObject.put("code",200 );
+            jsonObject.put("msg","成功" );
+        }
+        else {
+            jsonObject.put("code",403 );
+            jsonObject.put("msg","删除失败" );
+        }
+
+        return jsonObject;
+    }
+
+    @PostMapping("/deleteMul") //staffDeleteMul批量删除员工
+    public JSONObject staffDeleteMul(@RequestHeader String Authorization, @RequestBody JSONObject param){
+        JSONObject jsonObject = new JSONObject();
+
+        if(!JWTUtil.checkToken(Authorization)){ //token认证失败
+            JWTUtil.checkTokenFailed(jsonObject);
+            return jsonObject;
+        }
+
+        //获取请求参数
+        JSONArray staff_ids = param.getJSONArray("staff_ids");
+
+        int [] ids = new int[staff_ids.size()];
+
+        //把userid中的数据进行类型转换并存入到int数组中
+        for (int i = 0; i < staff_ids.size(); i++) {
+            ids[i] = Integer.parseInt(staff_ids.getString(i));
+        }
+
+        int result = staffService.staffDeleteMul(ids);
+        if(result == ids.length){
+            jsonObject.put("code",200);
+            jsonObject.put("msg","批量删除成功,删除了"+result+"条员工数据");
+        }else{
+            jsonObject.put("code",403);
+            jsonObject.put("msg","批量删除失败,删除了"+result+"条员工数据");
+        }
+        return jsonObject;
+
+    }
+
+    @PostMapping("/info") //staffInfo获取员工信息
+    public JSONObject staffInfo(@RequestHeader String Authorization, @RequestBody JSONObject param){
+        JSONObject jsonObject = new JSONObject();
+
+        if(!JWTUtil.checkToken(Authorization)){ //token认证失败
+            JWTUtil.checkTokenFailed(jsonObject);
+            return jsonObject;
+        }
+
+        int staff_id = param.getIntValue("staff_id");
+        Staff staff = staffService.staffInfo(staff_id);
+
+        if(staff!=null){
+            JSONObject jsonObject1 = new JSONObject();
+            jsonObject1.put("staff_id",staff.getStaff_id());
+            jsonObject1.put("staff_name",staff.getStaff_name());
+            jsonObject1.put("staff_pwd",staff.getStaff_pwd());
+            jsonObject1.put("staff_full_name",staff.getStaff_full_name());
+            jsonObject1.put("staff_phone",staff.getStaff_phone());
+            jsonObject1.put("staff_state",staff.getStaff_state());
+            jsonObject1.put("staff_role",staff.getStaff_role());
+            jsonObject1.put("staff_pic",staff.getStaff_pic());
+
+            jsonObject.put("code",200);
+            jsonObject.put("msg","成功");
+            jsonObject.put("data",jsonObject1);
+
+        }
+        else{
+            jsonObject.put("code",403);
+            jsonObject.put("msg","获取员工信息失败");
+            jsonObject.put("data",null);
+        }
+
+        return jsonObject;
+    }
+
+    @PostMapping("/modify") //staffModify修改员工信息
+    public JSONObject staffModify(@RequestHeader String Authorization, @RequestBody JSONObject param){
+        JSONObject jsonObject = new JSONObject();
+
+        if(!JWTUtil.checkToken(Authorization)){ //token认证失败
+            JWTUtil.checkTokenFailed(jsonObject);
+            return jsonObject;
+        }
+
+        Staff staff = new Staff();
+        staff.setStaff_id(param.getIntValue("staff_id"));
+        staff.setStaff_name(param.getString("staff_name"));
+        staff.setStaff_pwd(param.getString("staff_pwd"));
+        staff.setStaff_full_name(param.getString("staff_full_name"));
+        staff.setStaff_phone(param.getString("staff_phone"));
+        staff.setStaff_role(param.getIntValue("staff_role"));
+        staff.setStaff_state(param.getIntValue("staff_state"));
+        staff.setStaff_pic(param.getString("staff_pic"));
+
+        int flag = staffService.staffModify(staff);
+        if(flag==1){
+            jsonObject.put("code",200 );
+            jsonObject.put("msg","修改成功" );
+        }
+        else{
+            jsonObject.put("code",403 );
+            jsonObject.put("msg","修改失败" );
+        }
+
+        return jsonObject;
+    }
+
+    @PostMapping("/modifyState") //staffModifyState修改单个员工信息
+    public JSONObject staffModifyState(@RequestHeader String Authorization, @RequestBody JSONObject param){
+        JSONObject jsonObject = new JSONObject();
+
+        if(!JWTUtil.checkToken(Authorization)){ //token认证失败
+            JWTUtil.checkTokenFailed(jsonObject);
+            return jsonObject;
+        }
+
+        int staff_id = param.getIntValue("staff_id");
+
+        Staff staff = staffService.staffInfo(staff_id);
+        if(staff==null){
+            jsonObject.put("code",403 );
+            jsonObject.put("msg","员工不存在" );
+            return jsonObject;
+        }
+        int staff_state = staff.getStaff_state();
+        int state_modify = -1;
+
+        if(staff_state==0) state_modify=1;
+        else if (staff_state==1) state_modify=0;
+
+        if(state_modify==-1){
+            jsonObject.put("code",403 );
+            jsonObject.put("msg","员工状态错误" );
+            return jsonObject;
+        }
+
+        int flag = staffService.staffModifyState(staff_id,state_modify);
+
+        if(flag==1){
+            jsonObject.put("code",200 );
+            jsonObject.put("msg","状态修改成功" );
+        }
+        else{
+            jsonObject.put("code",403 );
+            jsonObject.put("msg","状态修改失败" );
+        }
+
+        return jsonObject;
+    }
+
+}