package com.terra.system.controller.all;

import com.terra.common.controller.all.BaseController;
import com.terra.common.entity.all.ResponseMsg;
import com.terra.common.entity.all.StaticData;
import com.terra.common.annotation.SysLog;
import com.terra.system.entity.all.*;
import com.terra.system.entity.sys.LayerEntity;
import com.terra.system.entity.sys.MenuEntity;
import com.terra.system.entity.sys.ResEntity;
import com.terra.system.entity.sys.UserEntity;
import com.terra.system.service.all.PermsService;
import com.terra.system.service.sys.LayerService;
import com.terra.system.service.sys.MenuService;
import com.terra.system.service.sys.TokenService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.Parameters;
import javax.annotation.Resource;

import io.swagger.v3.oas.annotations.enums.ParameterIn;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletRequest;
import java.util.List;

/**
 * 授权控制器
 * @author WWW
 */
@Tag(name = "运维管理\\授权管理")
@RestController
@RequestMapping("/perms")
public class PermsController extends BaseController {
    @Resource
    PermsService permsService;

    @Resource
    TokenService tokenService;

    @Resource
    MenuService menuService;

    @Resource
    LayerService layerService;

    @SysLog()
    @Operation(summary = "查询当前用户的图层授权")
    @GetMapping(value = "/selectLayers")
    public ResponseMsg<Object> selectLayers(HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue == null) {
                return fail("用户未登录", null);
            }

            String uid = StaticData.ADMIN.equals(ue.getUid()) ? null : ue.getUid();
            List<LayerEntity> rs = layerService.selectLayers(uid);

            return success(null == rs ? 0 : rs.size(), rs);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询当前用户的投影图层授权")
    @GetMapping(value = "/selectProjectLayers")
    public ResponseMsg<Object> selectProjectLayers(HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue == null) {
                return fail("用户未登录", null);
            }

            String uid = StaticData.ADMIN.equals(ue.getUid()) ? null : ue.getUid();
            List<LayerEntity> rs = layerService.selectProjectLayers(uid);

            return success(null == rs ? 0 : rs.size(), rs);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询当前用户的资源授权")
    @GetMapping(value = "/selectRes")
    public ResponseMsg<Object> selectRes(HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue == null) {
                return fail("用户未登录", null);
            }

            String uid = StaticData.ADMIN.equals(ue.getUid()) ? null : ue.getUid();
            List<ResEntity> rs = permsService.selectRes(uid);

            return success(null == rs ? 0 : rs.size(), rs);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询当前用户的菜单授权")
    @GetMapping(value = "/selectMenus")
    public ResponseMsg<List<MenusAuthEntity>> selectMenus(HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue == null) {
                return fail("用户未登录", null);
            }

            String uid = StaticData.ADMIN.equals(ue.getUid()) ? null : ue.getUid();
            List<MenusAuthEntity> rs = permsService.selectMenus(uid);

            return success(rs);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询当前用户的权限授权")
    @GetMapping(value = "/selectPerms")
    public ResponseMsg<List<String>> selectPerms(HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue == null) {
                return fail("用户未登录", null);
            }

            String uid = StaticData.ADMIN.equals(ue.getUid()) ? null : ue.getUid();
            List<String> rs = permsService.selectPerms(uid);

            return success(rs);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询当前用户的权限授权实体集合")
    @GetMapping(value = "/selectPermsEntity")
    public ResponseMsg<List<PermsAuthEntity>> selectPermsEntity(HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue == null) {
                return fail("用户未登录", null);
            }

            String uid = StaticData.ADMIN.equals(ue.getUid()) ? null : ue.getUid();
            List<PermsAuthEntity> rs = permsService.selectPermsEntity(uid);

            return success(rs);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "清空授权缓存")
    @GetMapping(value = "/deletePermsCache")
    public ResponseMsg<Boolean> deletePermsCache() {
        try {
            permsService.clearPermsCache();

            return success(true);
        } catch (Exception ex) {
            return fail(ex, false);
        }
    }

    @SysLog()
    @Operation(summary = "清空所有缓存")
    @GetMapping(value = "/deleteAllCache")
    public ResponseMsg<Boolean> deleteAllCache() {
        try {
            permsService.clearAllCache();

            return success(true);
        } catch (Exception ex) {
            return fail(ex, false);
        }
    }

    @SysLog()
    @Operation(summary = "查询用户角色")
    @GetMapping("/selectRoles")
    public ResponseMsg<Object> selectRoles(HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue == null) {
                return fail("用户未登录", null);
            }

            List<Integer> list = permsService.selectRoles(ue.getUid());

            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "递归查询")
    @Parameters({
            @Parameter(name = "id", description = "菜单ID", in = ParameterIn.QUERY, example = "14")
    })
    @GetMapping(value = "/selectMenuRecursive")
    public ResponseMsg<Object> selectMenuRecursive(int id, HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue == null) {
                return fail("用户未登录", null);
            }

            String uid = StaticData.ADMIN.equals(ue.getUid()) ? null : ue.getUid();
            List<MenuEntity> list = permsService.selectMenuRecursive(id, uid);

            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }
}