package com.java110.job.cmd.plugin; import com.alibaba.fastjson.JSONObject; import com.java110.core.annotation.Java110Cmd; import com.java110.core.client.RestTemplate; import com.java110.core.context.CmdContextUtils; import com.java110.core.context.ICmdDataFlowContext; import com.java110.core.event.cmd.Cmd; import com.java110.core.event.cmd.CmdEvent; import com.java110.core.factory.AuthenticationFactory; import com.java110.dto.store.StoreDto; import com.java110.dto.store.StoreUserDto; import com.java110.dto.user.UserDto; import com.java110.intf.store.IStoreUserV1InnerServiceSMO; import com.java110.intf.store.IStoreV1InnerServiceSMO; import com.java110.intf.user.IUserV1InnerServiceSMO; import com.java110.job.adapt.hcIot.IotConstant; import com.java110.utils.cache.MappingCache; import com.java110.utils.exception.CmdException; import com.java110.utils.util.Assert; import com.java110.utils.util.DateUtil; import com.java110.vo.ResultVo; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.http.HttpEntity; import org.springframework.http.HttpHeaders; import org.springframework.http.HttpMethod; import org.springframework.http.ResponseEntity; import java.text.ParseException; import java.util.List; /** * 获取插件token */ @Java110Cmd(serviceCode = "plugin.getPluginToken") public class GetPluginTokenCmd extends Cmd { public static final String URL_DOMAIN = "URL_DOMAIN"; // 物联网域 @Autowired private IUserV1InnerServiceSMO userV1InnerServiceSMOImpl; @Autowired private IStoreUserV1InnerServiceSMO storeUserV1InnerServiceSMOImpl; @Autowired private IStoreV1InnerServiceSMO storeV1InnerServiceSMOImpl; @Autowired private RestTemplate outRestTemplate; @Override public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException, ParseException { Assert.hasKeyAndValue(reqJson, "targetUrl", "未包含targetUrl"); Assert.hasKeyAndValue(reqJson, "communityId", "未包含communityId"); String pluginSwitch = MappingCache.getValue("PLUGIN", "PLUGIN_SWITCH"); if (!"ON".equals(pluginSwitch)) { throw new CmdException("插件系统未部署"); } } @Override public void doCmd(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException, ParseException { String userId = CmdContextUtils.getUserId(context); UserDto userDto = new UserDto(); userDto.setUserId(userId); List userDtos = userV1InnerServiceSMOImpl.queryUsers(userDto); Assert.listOnlyOne(userDtos, "用户未登录"); StoreUserDto storeUserDto = new StoreUserDto(); storeUserDto.setUserId(userId); List storeUserDtos = storeUserV1InnerServiceSMOImpl.queryStoreUsers(storeUserDto); Assert.listOnlyOne(storeUserDtos, "未包含商户"); StoreDto storeDto = new StoreDto(); storeDto.setStoreId(storeUserDtos.get(0).getStoreId()); List storeDtos = storeV1InnerServiceSMOImpl.queryStores(storeDto); Assert.listOnlyOne(storeDtos, "商户不存在"); JSONObject staff = new JSONObject(); staff.put("staffId", userId); staff.put("timestamp", DateUtil.getCurrentDate().getTime()); staff.put("sign", createSign(staff)); HttpHeaders header = new HttpHeaders(); HttpEntity httpEntity = new HttpEntity(staff.toJSONString(), header); String pluginUrl = MappingCache.getValue("PLUGIN", "PLUGIN_URL") + "/pluginToken/apply"; ResponseEntity tokenRes = outRestTemplate.exchange(pluginUrl, HttpMethod.POST, httpEntity, String.class); String body = tokenRes.getBody(); JSONObject paramOut = JSONObject.parseObject(body); if (paramOut.getIntValue("code") != ResultVo.CODE_OK) { throw new CmdException(paramOut.getString("msg")); } pluginUrl = MappingCache.getValue(URL_DOMAIN, IotConstant.PLUGIN_URL); String targetUrl = pluginUrl + reqJson.getString("targetUrl"); String url = pluginUrl + "/sso.html?token=" + paramOut.getString("data") + "&communityId=" + reqJson.getString("communityId") + "&targetUrl=" + targetUrl; paramOut.put("url", url); context.setResponseEntity(ResultVo.createResponseEntity(paramOut)); } public String createSign(JSONObject staff) { String secure = MappingCache.getValue("PLUGIN", "PLUGIN_SECURE"); String sign = staff.getString("staffId") + staff.getString("timestamp") + secure; sign = AuthenticationFactory.md5(sign); return sign; } }