package com.yami.trading.admin.controller.agent;
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.yami.trading.admin.controller.service.SysUserOperService;
|
import com.yami.trading.admin.facade.PermissionFacade;
|
import com.yami.trading.admin.model.*;
|
import com.yami.trading.bean.agent.AgentDto;
|
import com.yami.trading.bean.model.UserRecom;
|
import com.yami.trading.bean.user.dto.UserDto;
|
import com.yami.trading.common.constants.Constants;
|
import com.yami.trading.common.domain.Result;
|
import com.yami.trading.common.exception.YamiShopBindException;
|
import com.yami.trading.common.util.GoogleAuthenticator;
|
import com.yami.trading.security.common.manager.PasswordManager;
|
import com.yami.trading.service.agent.AgentService;
|
import com.yami.trading.service.syspara.SysparaService;
|
import com.yami.trading.service.system.LogService;
|
import com.yami.trading.service.user.UserRecomService;
|
import com.yami.trading.service.user.UserService;
|
import com.yami.trading.sys.service.SysUserService;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.security.crypto.password.PasswordEncoder;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.validation.Valid;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
@RestController
|
@CrossOrigin
|
@Api(tags = "代理商")
|
@RequestMapping("agent")
|
public class AgentController {
|
@Autowired
|
PasswordEncoder passwordEncoder;
|
@Autowired
|
PasswordManager passwordManager;
|
@Autowired
|
UserService userService;
|
@Autowired
|
SysparaService sysparaService;
|
@Autowired
|
AgentService agentService;
|
@Autowired
|
SysUserService sysUserService;
|
@Autowired
|
LogService logService;
|
@Autowired
|
SysUserOperService sysUserOperService;
|
|
@Autowired
|
PermissionFacade permissionFacade;
|
@Autowired
|
private UserRecomService userRecomService;
|
|
@ApiOperation(value = "列表")
|
@PostMapping("list")
|
public Result<Page<AgentDto>> list(@RequestBody @Valid UserAgentListModel model) {
|
List<String> roleNames = new ArrayList<>();
|
roleNames.add(Constants.SECURITY_ROLE_AGENT);
|
Page<AgentDto> page = new Page(model.getCurrent(), model.getSize());
|
if ("list".equals(model.getViewType())) {
|
page = agentService.listTotal(page, model.getUserName(),permissionFacade.getOwnerUserIds());
|
}else if ("child".equals(model.getViewType())) {
|
List<UserRecom> recoms = userRecomService.findRecoms(model.getUserId());
|
// 只取一级
|
List<String> childs = recoms.stream().map(UserRecom::getRecomUserId).collect(Collectors.toList());
|
page = agentService.listTotal(page, null,childs);
|
}
|
else {
|
page = agentService.pagedQueryNetwork(page);
|
}
|
String shareUrl = sysparaService.find("share_url").getSvalue();
|
shareUrl = shareUrl + "#/?usercode=";
|
Map<String, Integer> allRecommCount = userRecomService.findAllRecommCount();
|
for (AgentDto agentDto : page.getRecords()) {
|
String userId = agentDto.getUserId();
|
agentDto.setShareUrl(shareUrl + agentDto.getUserCode());
|
agentDto.setLoginAuthority(agentDto.getStatus() == 1);
|
agentDto.setOperaAuthority(Constants.SECURITY_ROLE_AGENT.equals(agentDto.getRoleName()));
|
agentDto.setUserName(agentDto.getUserName() +"(网络代理数: "+allRecommCount.getOrDefault(userId, 0)+")");
|
}
|
|
return Result.ok(page);
|
}
|
|
@RequestMapping("getReconNumNet.action")
|
public Result<List<Map<String,Integer>>> getReconNumNet(String userId) {
|
List<Integer> recoNumNetList = userRecomService.getRecoNumNetList(userId);
|
List<Map<String,Integer>> result = new ArrayList<>();
|
for(int i=1;i<=recoNumNetList.size();i++){
|
Map<String,Integer> data = new HashMap<>();
|
data.put("level", i);
|
data.put("userCount", recoNumNetList.get(i-1));
|
result.add(data);
|
}
|
return Result.ok(result);
|
}
|
|
|
@ApiOperation(value = "获取下级代理")
|
@PostMapping("child")
|
public Result<Page<AgentDto>> child(@RequestBody @Valid UserAgentListModel model) {
|
Page<UserDto> page = new Page(model.getCurrent(), model.getSize());
|
|
List<UserRecom> recoms = userRecomService.findRecoms(model.getUserId());
|
Map<String, String> selfAndParentMap = new HashMap<>();
|
for(UserRecom recom: recoms){
|
selfAndParentMap.put(recom.getRecomUserId(), recom.getUserId());
|
}
|
// 只取一级
|
List<String> childs = recoms.stream().map(UserRecom::getRecomUserId).collect(Collectors.toList());
|
|
String shareUrl = sysparaService.find("share_url").getSvalue();
|
shareUrl = shareUrl + "#/?usercode=";
|
Map<String, Integer> allRecommCount = userRecomService.findAllRecommCount();
|
List<String> roleNames = new ArrayList<>();
|
roleNames.add(Constants.SECURITY_ROLE_GUEST);
|
roleNames.add(Constants.SECURITY_ROLE_MEMBER);
|
roleNames.add(Constants.SECURITY_ROLE_TEST);
|
|
userService.listUser(page, roleNames, null, null, null,null, childs);
|
List<AgentDto> agentDtos = new ArrayList<>();
|
for (UserDto userDto : page.getRecords()) {
|
AgentDto agentDto = new AgentDto();
|
String userId =userDto .getUserId();
|
agentDto.setId(userId);
|
agentDto.setUserId(userId);
|
agentDto.setShareUrl(shareUrl + agentDto.getUserCode());
|
agentDto.setLoginAuthority(agentDto.getStatus() == 1);
|
agentDto.setOperaAuthority(Constants.SECURITY_ROLE_AGENT.equals(agentDto.getRoleName()));
|
agentDto.setRoleName(userDto.getRolename());
|
agentDto.setUserCode(userDto.getUserCode());
|
agentDto.setUserName(userDto.getUserName() +"(网络代理数: "+allRecommCount.getOrDefault(userId, 0)+")");
|
agentDto.setUserNameParent(userService.getById(selfAndParentMap.get(userId)).getUserName());
|
agentDtos.add(agentDto);
|
}
|
Page<AgentDto> pageResult = new Page(model.getCurrent(), model.getSize());
|
pageResult.setTotal(page.getTotal());
|
pageResult.setRecords(agentDtos);
|
return Result.ok(pageResult);
|
}
|
|
|
|
@ApiOperation(value = "新增")
|
@PostMapping("add")
|
public Result<?> add(@RequestBody @Valid AgentAndModel model) {
|
//sysUserService.checkSafeWord(model.getSafeword());
|
String roleName = model.isOperaAuthority() ? Constants.SECURITY_ROLE_AGENT : Constants.SECURITY_ROLE_AGENTLOW;
|
sysUserOperService.addAgent(model.getUserName(),
|
passwordManager.decryptPassword(model.getPassword()), "000000", roleName, model.getRemarks(), model.getParentsUseCode(),
|
model.isLoginAuthority(), model.isOperaAuthority());
|
return Result.ok(null);
|
}
|
|
@ApiOperation(value = "修改代理商")
|
@PostMapping("update")
|
public Result<?> update(@RequestBody @Valid UpdateAgentModel model) {
|
agentService.updateAgent(model.getId(), model.isLoginAuthority(), model.isOperaAuthority(), model.getRemarks());
|
return Result.ok(null);
|
}
|
|
@ApiOperation(value = "重置登录密码")
|
@PostMapping("restPassword")
|
public Result<?> add(@RequestBody @Valid RestPasswordModel model) {
|
//sysUserService.checkSafeWord(model.getSafeword());
|
sysUserOperService.restPassword(passwordManager.decryptPassword(model.getPassword()),model.getId());
|
return Result.ok(null);
|
}
|
|
@ApiOperation(value = "绑定谷歌验证码")
|
@PostMapping("bindGoogleAuthCode")
|
public Result<?> bindGoogleAuthCode(@RequestBody @Valid AgentBindGoogleModel model) {
|
sysUserService.checkSuperGoogleAuthCode(model.getRootGoogleCode());
|
long t = System.currentTimeMillis();
|
GoogleAuthenticator ga = new GoogleAuthenticator();
|
ga.setWindowSize(5); // should give 5 * 30 seconds of grace...
|
boolean checkCode = ga.check_code(model.getGoogleAuthSecret(),
|
Long.valueOf(model.getGoogleAuthCode()), t);
|
if (!checkCode) {
|
throw new YamiShopBindException("谷歌验证码错误");
|
}
|
sysUserOperService.bindGoogleAuthCode(model.getGoogleAuthSecret(), model.getId());
|
return Result.ok(null);
|
}
|
|
@ApiOperation(value = "解除绑定谷歌验证码")
|
@PostMapping("unbindGoogleAuthCode")
|
public Result<?> unbindGoogleAuthCode(@RequestBody @Valid IdModel model) {
|
// sysUserService.checkSuperGoogleAuthCode(model.getRootGoogleCode());
|
sysUserOperService.unbindGoogleAuthCode( model.getId());
|
return Result.ok(null);
|
}
|
}
|
