package org.jeecg.modules.doc.controller;

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.system.query.QueryGenerator;
import org.jeecg.common.util.oConvertUtils;
import org.jeecg.modules.doc.api.JeecgSystemClient;
import org.jeecg.modules.doc.entity.PathPermission;
import org.jeecg.modules.doc.entity.UserStorageFlow;
import org.jeecg.modules.doc.service.IUserStorageFlowService;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.extern.slf4j.Slf4j;

import org.jeecg.modules.doc.vo.PathPermissionVO;
import org.jeecgframework.poi.excel.ExcelImportUtil;
import org.jeecgframework.poi.excel.def.NormalExcelConstants;
import org.jeecgframework.poi.excel.entity.ExportParams;
import org.jeecgframework.poi.excel.entity.ImportParams;
import org.jeecgframework.poi.excel.view.JeecgEntityExcelView;
import org.jeecg.common.system.base.controller.JeecgController;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.servlet.ModelAndView;
import com.alibaba.fastjson.JSON;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.jeecg.common.aspect.annotation.AutoLog;

 /**
 * @Description: user_storage_flow
 * @Author: jeecg-boot
 * @Date:   2022-08-04
 * @Version: V1.0
 */
@Api(tags="user_storage_flow")
@RestController
@RequestMapping("/userStorageFlow")
@Slf4j
public class UserStorageFlowController extends JeecgController<UserStorageFlow, IUserStorageFlowService> {
	@Autowired
	private IUserStorageFlowService userStorageFlowService;

	@Autowired
	private JeecgSystemClient jeecgSystemClient;

	/**
	 * 分页列表查询
	 *
	 * @param userStorageFlow
	 * @param pageNo
	 * @param pageSize
	 * @param req
	 * @return
	 */
	@AutoLog(value = "user_storage_flow-分页列表查询")
	@ApiOperation(value="user_storage_flow-分页列表查询", notes="user_storage_flow-分页列表查询")
	@GetMapping(value = "/list")
	public Result<?> queryPageList(UserStorageFlow userStorageFlow,
								   @RequestParam(name="pageNo", defaultValue="1") Integer pageNo,
								   @RequestParam(name="pageSize", defaultValue="10") Integer pageSize,
								   HttpServletRequest req) {
		QueryWrapper<UserStorageFlow> queryWrapper = QueryGenerator.initQueryWrapper(userStorageFlow, req.getParameterMap());
		Page<UserStorageFlow> page = new Page<UserStorageFlow>(pageNo, pageSize);
		IPage<UserStorageFlow> pageList = userStorageFlowService.page(page, queryWrapper);
		return Result.OK(pageList);
	}

	/**
	 *   添加
	 *
	 * @param userStorageFlow
	 * @return
	 */
	@AutoLog(value = "user_storage_flow-添加")
	@ApiOperation(value="user_storage_flow-添加", notes="user_storage_flow-添加")
	@PostMapping(value = "/add")
	public Result<?> add(@RequestBody UserStorageFlow userStorageFlow) {
		userStorageFlowService.save(userStorageFlow);
		return Result.OK("添加成功！");
	}

	/**
	 *  编辑
	 *
	 * @param userStorageFlow
	 * @return
	 */
	@AutoLog(value = "user_storage_flow-编辑")
	@ApiOperation(value="user_storage_flow-编辑", notes="user_storage_flow-编辑")
	@PutMapping(value = "/edit")
	public Result<?> edit(@RequestBody UserStorageFlow userStorageFlow) {
		userStorageFlowService.updateById(userStorageFlow);
		return Result.OK("编辑成功!");
	}

	/**
	 *   通过id删除
	 *
	 * @param id
	 * @return
	 */
	@AutoLog(value = "user_storage_flow-通过id删除")
	@ApiOperation(value="user_storage_flow-通过id删除", notes="user_storage_flow-通过id删除")
	@DeleteMapping(value = "/delete")
	public Result<?> delete(@RequestParam(name="id",required=true) String id) {
		userStorageFlowService.removeById(id);
		return Result.OK("删除成功!");
	}

	/**
	 *  批量删除
	 *
	 * @param ids
	 * @return
	 */
	@AutoLog(value = "user_storage_flow-批量删除")
	@ApiOperation(value="user_storage_flow-批量删除", notes="user_storage_flow-批量删除")
	@DeleteMapping(value = "/deleteBatch")
	public Result<?> deleteBatch(@RequestParam(name="ids",required=true) String ids) {
		this.userStorageFlowService.removeByIds(Arrays.asList(ids.split(",")));
		return Result.OK("批量删除成功!");
	}

	/**
	 * 通过id查询
	 *
	 * @param id
	 * @return
	 */
	@AutoLog(value = "user_storage_flow-通过id查询")
	@ApiOperation(value="user_storage_flow-通过id查询", notes="user_storage_flow-通过id查询")
	@GetMapping(value = "/queryById")
	public Result<?> queryById(@RequestParam(name="id",required=true) String id) {
		UserStorageFlow userStorageFlow = userStorageFlowService.getById(id);
		if(userStorageFlow==null) {
			return Result.error("未找到对应数据");
		}
		return Result.OK(userStorageFlow);
	}

    /**
    * 导出excel
    *
    * @param request
    * @param userStorageFlow
    */
    @RequestMapping(value = "/exportXls")
    public ModelAndView exportXls(HttpServletRequest request, UserStorageFlow userStorageFlow) {
        return super.exportXls(request, userStorageFlow, UserStorageFlow.class, "user_storage_flow");
    }

    /**
      * 通过excel导入数据
    *
    * @param request
    * @param response
    * @return
    */
    @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
    public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
        return super.importExcel(request, response, UserStorageFlow.class);
    }



	 @ApiOperation(value = "保存用户下载限额分配")
	 @PostMapping("/save")
	 public Result<?> savePathPerm(@RequestBody UserStorageFlow storageFlow) {
		 try{
			 storageFlow.setFlow(storageFlow.getFlow()*1024*1024);
			 userStorageFlowService.saveOrUpdate(storageFlow);
			 return Result.OK();
		 }catch (Exception e) {
			 return Result.error("权限设置失败！");
		 }

	 }


	 /**
	  * 查询所有角色和用户的下载限额
	  * 先查询出所有角色及角色对应的用户列表，将下载限额设置情况合并到列表，返回给前端展示。
	  * @return  Result<List<PathPermissionVO>>
	  */
	 @ApiOperation(value="查询角色用户下载限额", notes="查询角色用户下载限额")
	 @GetMapping("/queryAll")
	 public Result<List<PathPermissionVO>> queryAllPathPermission() {
		 // 查询所有已设置下载限额
		 LambdaQueryWrapper<UserStorageFlow> queryWrapper = new LambdaQueryWrapper<>();

		 List<UserStorageFlow> list = userStorageFlowService.list(queryWrapper);
		 // 角色/用户对应的下载限额转换成MAP,方便查询
		 Map<String, UserStorageFlow> permMap = list.stream()
				 .collect(Collectors.toMap(path -> path.getRoleId()==null?path.getUserId():path.getRoleId(), pp -> pp));
		 // 调用system的接口查询角色和用户
		 List<PathPermissionVO> vos = jeecgSystemClient.queryRoleUsers();
		 // 将查到的文件权限绑定到对应角色，给前台展示
		 if (vos != null && vos.size() > 0) {
			 vos.stream().forEach( r -> {

				 convertVo(r, r.getRoleId(), permMap);
			 });
		 }
		 return Result.OK(vos);
	 }


	 /**
	  * 将权限实体转换成vo对象
	  * @param r
	  * @param permMap
	  */
	 private void convertVo(PathPermissionVO r,String key, Map<String, UserStorageFlow> permMap) {
		 UserStorageFlow p = permMap.get(key);
		 if (p != null) {
			 r.setId(p.getId());
			 if (p.getFlow() != null) {
				 r.setFlow(p.getFlow() / 1024 / 1024);
			 }

		 }
		 r.setKey(key);
		 if (r.getChildren() != null) {
			 r.getChildren().stream().forEach( u -> {

				 u.setChildren(null);
				 convertVo(u, u.getUserId(), permMap);
			 });
		 }

	 }
 }