Pu Zhibing
2024-12-12 7c0967daf92243b86f2c0a2435b127287358d9c4
ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/PartnerController.java
@@ -4,11 +4,15 @@
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.model.TPartnerSite;
import com.ruoyi.chargingPile.api.model.TSiteMenu;
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.chargingPile.service.TPartnerSiteService;
import com.ruoyi.chargingPile.service.TSiteMenuService;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.web.domain.AjaxResult;
import com.ruoyi.common.core.web.page.PageInfo;
@@ -27,6 +31,7 @@
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;
@@ -43,8 +48,11 @@
   
   @Resource
   private SysUserClient sysUserClient;
   @Resource
   private TSiteMenuService siteMenuService;
   @Resource
   private TPartnerSiteService partnerSiteService;
   
   
   
@@ -204,4 +212,86 @@
   public Boolean authMenu(@PathVariable("partnerId") Integer partnerId, @RequestParam("siteId") Integer siteId, @RequestParam("siteMenu") String siteMenu){
      return partnerService.authMenu(partnerId, siteId, SiteMenu.getSiteMenu(siteMenu));
   }
   // 充电算帐单按钮权限
   @PostMapping("/getChargingBillMenu/{partnerId}/{siteId}")
   R<Boolean> getChargingBillMenu(@PathVariable("partnerId")Integer partnerId,@PathVariable("siteId") Integer siteId){
      TSiteMenu one = siteMenuService.getOne(new LambdaQueryWrapper<TSiteMenu>().eq(TSiteMenu::getPath, "/data/charge_bill_order/download").eq(TSiteMenu::getStatus, 0));
      TPartnerSite partnerSite = partnerSiteService.getOne(new LambdaQueryWrapper<TPartnerSite>()
            .eq(TPartnerSite::getPartnerId, partnerId)
            .eq(TPartnerSite::getSiteMenuId, one.getMenuId())
            .eq(TPartnerSite::getSiteId, siteId));
      return partnerSite!=null?R.ok(true):R.ok(false);
   }
   // 充电时段统计按钮权限
   @PostMapping("/getChargingListMenu/{partnerId}/{siteId}")
   R<List<Boolean>> getChargingListMenu(@PathVariable("partnerId")Integer partnerId,@PathVariable("siteId") Integer siteId){
      List<Boolean> booleans = new ArrayList<>();
      TSiteMenu one = siteMenuService.getOne(new LambdaQueryWrapper<TSiteMenu>().eq(TSiteMenu::getPath, "/data/charging_period_statistics/order_info").eq(TSiteMenu::getStatus, 0));
      TPartnerSite partnerSite = partnerSiteService.getOne(new LambdaQueryWrapper<TPartnerSite>()
            .eq(TPartnerSite::getPartnerId, partnerId)
            .eq(TPartnerSite::getSiteMenuId, one.getMenuId())
            .eq(TPartnerSite::getSiteId, siteId));
      TSiteMenu two = siteMenuService.getOne(new LambdaQueryWrapper<TSiteMenu>().eq(TSiteMenu::getPath, "/data/charging_period_statistics/charging_curve").eq(TSiteMenu::getStatus, 0));
      TPartnerSite partnerSite1 = partnerSiteService.getOne(new LambdaQueryWrapper<TPartnerSite>()
            .eq(TPartnerSite::getPartnerId, partnerId)
            .eq(TPartnerSite::getSiteMenuId, two.getMenuId())
            .eq(TPartnerSite::getSiteId, siteId));
      if (partnerSite!=null){
         booleans.add(true);
      }else{
         booleans.add(false);
      }
      if (partnerSite1!=null){
         booleans.add(true);
      }else{
         booleans.add(false);
      }
      return R.ok(booleans);
   }
   // 账户结算帐单按钮权限
   @PostMapping("/getAccountMenu/{partnerId}/{siteId}")
   R<Boolean> getAccountMenu(@PathVariable("partnerId")Integer partnerId,@PathVariable("siteId") Integer siteId){
      TSiteMenu one = siteMenuService.getOne(new LambdaQueryWrapper<TSiteMenu>().eq(TSiteMenu::getPath, "/data/account_settlement/download").eq(TSiteMenu::getStatus, 0));
      TPartnerSite partnerSite = partnerSiteService.getOne(new LambdaQueryWrapper<TPartnerSite>()
            .eq(TPartnerSite::getPartnerId, partnerId)
            .eq(TPartnerSite::getSiteMenuId, one.getMenuId())
            .eq(TPartnerSite::getSiteId, siteId));
      return partnerSite!=null?R.ok(true):R.ok(false);
   }
   // 结算表记录按钮权限
   @PostMapping("/getSettlementListMenu/{partnerId}/{siteId}")
   R<List<Boolean>> getSettlementListMenu(@PathVariable("partnerId")Integer partnerId,@PathVariable("siteId") Integer siteId){
      List<Boolean> booleans = new ArrayList<>();
      TSiteMenu one = siteMenuService.getOne(new LambdaQueryWrapper<TSiteMenu>().eq(TSiteMenu::getPath, "/data/balance_sheet_record/download").eq(TSiteMenu::getStatus, 0));
      TPartnerSite partnerSite = partnerSiteService.getOne(new LambdaQueryWrapper<TPartnerSite>()
            .eq(TPartnerSite::getPartnerId, partnerId)
            .eq(TPartnerSite::getSiteMenuId, one.getMenuId())
            .eq(TPartnerSite::getSiteId, siteId));
      TSiteMenu two = siteMenuService.getOne(new LambdaQueryWrapper<TSiteMenu>().eq(TSiteMenu::getPath, "/data/balance_sheet_record/del").eq(TSiteMenu::getStatus, 0));
      TPartnerSite partnerSite1 = partnerSiteService.getOne(new LambdaQueryWrapper<TPartnerSite>()
            .eq(TPartnerSite::getPartnerId, partnerId)
            .eq(TPartnerSite::getSiteMenuId, two.getMenuId())
            .eq(TPartnerSite::getSiteId, siteId));
      if (partnerSite!=null){
         booleans.add(true);
      }else{
         booleans.add(false);
      }
      if (partnerSite1!=null){
         booleans.add(true);
      }else{
         booleans.add(false);
      }
      return R.ok(booleans);
   }
   // 充电桩订单按钮权限
   @PostMapping("/getChargingOrderMenu/{partnerId}/{siteId}")
   R<Boolean> getChargingOrderMenu(@PathVariable("partnerId")Integer partnerId,@PathVariable("siteId") Integer siteId){
      TSiteMenu one = siteMenuService.getOne(new LambdaQueryWrapper<TSiteMenu>().eq(TSiteMenu::getPath, "/data/charging_pile_order/record").eq(TSiteMenu::getStatus, 0));
      TPartnerSite partnerSite = partnerSiteService.getOne(new LambdaQueryWrapper<TPartnerSite>()
            .eq(TPartnerSite::getPartnerId, partnerId)
            .eq(TPartnerSite::getSiteMenuId, one.getMenuId())
            .eq(TPartnerSite::getSiteId, siteId));
      return partnerSite!=null?R.ok(true):R.ok(false);
   }
}