MicroCommunity/service-job/src/main/java/com/java110/job/cmd/plugin/GetPluginTokenCmd.java

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.job.adapt.hcIotNew.http.ISendIot;
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<UserDto> userDtos = userV1InnerServiceSMOImpl.queryUsers(userDto);

        Assert.listOnlyOne(userDtos, "用户未登录");

        StoreUserDto storeUserDto = new StoreUserDto();
        storeUserDto.setUserId(userId);
        List<StoreUserDto> storeUserDtos = storeUserV1InnerServiceSMOImpl.queryStoreUsers(storeUserDto);

        Assert.listOnlyOne(storeUserDtos, "未包含商户");


        StoreDto storeDto = new StoreDto();
        storeDto.setStoreId(storeUserDtos.get(0).getStoreId());
        List<StoreDto> 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<String> httpEntity = new HttpEntity<String>(staff.toJSONString(), header);

        String pluginUrl = MappingCache.getValue("PLUGIN", "PLUGIN_URL") + "/pluginToken/apply";

        ResponseEntity<String> 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;
    }
}