From cc6ef4d39b154077445fa1e4e84c1b0ca9674b60 Mon Sep 17 00:00:00 2001
From: Pu Zhibing <393733352@qq.com>
Date: 星期四, 12 十二月 2024 16:38:30 +0800
Subject: [PATCH] 修改合作商权限功能

---
 ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/PartnerController.java |   56 ++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 54 insertions(+), 2 deletions(-)

diff --git a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/PartnerController.java b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/PartnerController.java
index 46ae69f..64d447b 100644
--- a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/PartnerController.java
+++ b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/PartnerController.java
@@ -1,12 +1,12 @@
 package com.ruoyi.chargingPile.controller;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
-import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
 import com.ruoyi.chargingPile.api.dto.GetPermissionConfigurationDTO;
 import com.ruoyi.chargingPile.api.dto.PartnerListDTO;
 import com.ruoyi.chargingPile.api.model.Partner;
 import com.ruoyi.chargingPile.api.query.GetPartnerList;
 import com.ruoyi.chargingPile.api.query.SetPermissionConfiguration;
+import com.ruoyi.chargingPile.api.domain.SiteMenu;
 import com.ruoyi.chargingPile.dto.ResetPassword;
 import com.ruoyi.chargingPile.service.IPartnerService;
 import com.ruoyi.common.core.domain.R;
@@ -14,6 +14,8 @@
 import com.ruoyi.common.core.web.page.PageInfo;
 import com.ruoyi.common.log.annotation.Log;
 import com.ruoyi.common.log.enums.BusinessType;
+import com.ruoyi.common.security.annotation.Logical;
+import com.ruoyi.common.security.annotation.RequiresPermissions;
 import com.ruoyi.common.security.utils.SecurityUtils;
 import com.ruoyi.system.api.domain.SysUser;
 import com.ruoyi.system.api.feignClient.SysUserClient;
@@ -24,6 +26,7 @@
 
 import javax.annotation.Resource;
 import java.util.List;
+import java.util.Set;
 
 /**
  * @author zhibing.pu
@@ -39,10 +42,11 @@
 	@Resource
 	private SysUserClient sysUserClient;
 	
-
 	
 	
 	
+	
+	@RequiresPermissions(value = {"/balanceSheetRecord", "/partners", "/chargingPile/add", "/chargingPile/select", "/chargingPile/update"}, logical = Logical.OR)
 	@ResponseBody
 	@GetMapping("/getPartnerList")
 	@ApiOperation(value = "获取合作商列表", tags = {"管理后台-合作商管理"})
@@ -52,6 +56,8 @@
 	}
 	
 	
+	
+	@RequiresPermissions(value = {"/partners/add"}, logical = Logical.OR)
 	@ResponseBody
 	@PostMapping("/addPartner")
 	@ApiOperation(value = "添加合作商", tags = {"管理后台-合作商管理"})
@@ -62,6 +68,7 @@
 	
 	
 	
+	@RequiresPermissions(value = {"/partners/seleete", "/partners/update"}, logical = Logical.OR)
 	@ResponseBody
 	@GetMapping("/getPartner/{id}")
 	@ApiOperation(value = "获取合作商详情", tags = {"管理后台-合作商管理", "管理后台-充电桩信息"})
@@ -72,6 +79,9 @@
 		Partner partner = partnerService.getPartner(id);
 		return AjaxResult.success(partner);
 	}
+	
+	
+	
 	@ResponseBody
 	@GetMapping("/getPartnerR/{id}")
 	public R<Partner> getPartnerR(@PathVariable("id") Integer id){
@@ -81,6 +91,7 @@
 	
 	
 	
+	@RequiresPermissions(value = {"/partners/update"}, logical = Logical.OR)
 	@ResponseBody
 	@PostMapping("/editPartner")
 	@ApiOperation(value = "编辑合作商", tags = {"管理后台-合作商管理"})
@@ -91,6 +102,7 @@
 	
 	
 	
+	@RequiresPermissions(value = {"/partners/del"}, logical = Logical.OR)
 	@ResponseBody
 	@DeleteMapping("/delPartner")
 	@ApiOperation(value = "删除合作商", tags = {"管理后台-合作商管理"})
@@ -103,6 +115,7 @@
 	}
 	
 	
+	@RequiresPermissions(value = {"/partners/resetPassword"}, logical = Logical.OR)
 	@ResponseBody
 	@PostMapping("/resetPassword")
 	@ApiOperation(value = "合作商重置密码", tags = {"管理后台-合作商管理"})
@@ -112,6 +125,8 @@
 	}
 	
 	
+	
+	@RequiresPermissions(value = {"/site/add", "/site/update"}, logical = Logical.OR)
 	@ResponseBody
 	@GetMapping("/getPartner")
 	@ApiOperation(value = "获取所有合作商", tags = {"管理后台-站点管理"})
@@ -129,6 +144,8 @@
 	
 	
 	
+	
+	@RequiresPermissions(value = {"/site/partner_authority"}, logical = Logical.OR)
 	@ResponseBody
 	@GetMapping("/getPermissionConfiguration/{siteId}")
 	@ApiOperation(value = "获取合作商权限设置", tags = {"管理后台-站点管理"})
@@ -138,6 +155,7 @@
 	}
 	
 	
+	@RequiresPermissions(value = {"/site/partner_authority"}, logical = Logical.OR)
 	@ResponseBody
 	@PostMapping("/setPermissionConfiguration/{siteId}")
 	@ApiOperation(value = "设置合作商权限", tags = {"管理后台-站点管理"})
@@ -148,4 +166,38 @@
 	}
 	
 	
+	/**
+	 * 根据id获取合作商信息
+	 * @param id
+	 * @return
+	 */
+	@PostMapping("/getPartnerById/{id}")
+	public R<Partner> getPartnerById(@PathVariable("id") Integer id){
+		Partner partner = partnerService.getPartner(id);
+		return R.ok(partner);
+	}
+	
+	/**
+	 * 获取合作商授权的站点数据
+	 * @param partnerId
+	 * @param siteMenu
+	 * @return
+	 */
+	@PostMapping("/authSite/{partnerId}")
+	public R<Set<Integer>> authSite(@PathVariable("partnerId") Integer partnerId, @RequestParam("siteMenu") String siteMenu){
+		Set<Integer> integers = partnerService.authSite(partnerId, SiteMenu.getSiteMenu(siteMenu));
+		return R.ok(integers);
+	}
+	
+	/**
+	 * 判断合作商权限
+	 * @param partnerId
+	 * @param siteId
+	 * @param siteMenu
+	 * @return
+	 */
+	@PostMapping("/authMenu/{partnerId}")
+	public Boolean authMenu(@PathVariable("partnerId") Integer partnerId, @RequestParam("siteId") Integer siteId, @RequestParam("siteMenu") String siteMenu){
+		return partnerService.authMenu(partnerId, siteId, SiteMenu.getSiteMenu(siteMenu));
+	}
 }

--
Gitblit v1.7.1