growth-manager/api-module/src/main/java/com/tzld/piaoquan/api/component/AigcApiService.java

package com.tzld.piaoquan.api.component;

import com.alibaba.fastjson.JSONObject;
import com.tzld.piaoquan.api.common.exception.CommonException;
import com.tzld.piaoquan.api.model.vo.WxAccountDatastatVO;
import com.tzld.piaoquan.growth.common.component.HttpPoolClient;
import com.tzld.piaoquan.growth.common.utils.DateUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.Objects;

@Slf4j
@Component
public class AigcApiService {

    @Autowired
    private HttpPoolClient httpPoolClient;

    @Value("${aigc.api.host:http://aigc-api.cybertogether.net/aigc}")
    private String aigcApiHost;

    @Value("${aigc.api.token:9ebfcb397e954c41986971f183eb1707}")
    private String aigcApiToken;

    private String getAigcPostParam(JSONObject params) {
        JSONObject baseInfo = JSONObject.parseObject("{\n" +
                "    \"token\": \"" + aigcApiToken + "\",\n" +
                "    \"appType\": 9,\n" +
                "    \"platform\": \"pc\",\n" +
                "    \"appVersionCode\": 1000,\n" +
                "    \"clientTimestamp\": 1,\n" +
                "    \"fid\": 1,\n" +
                "    \"loginUid\": 1,\n" +
                "    \"pageSource\": 1,\n" +
                "    \"requestId\": 1,\n" +
                "    \"rid\": 1,\n" +
                "    \"uid\": 1\n" +
                "}");
        JSONObject param = new JSONObject();
        param.put("params", params);
        param.put("baseInfo", baseInfo);
        return param.toJSONString();
    }

    public String generateQrcode(String uuid) {
        String url = aigcApiHost + "/publish/account/generateQrcode";
        JSONObject params = new JSONObject();
        params.put("uuid", uuid);
        try {
            String post = httpPoolClient.post(url, getAigcPostParam(params));
            JSONObject res = JSONObject.parseObject(post);
            JSONObject data = res.getJSONObject("data");
            return data.getString("qrcodeStr");
        } catch (Exception e) {
            log.error("generateQrcode error", e);
        }
        return null;
    }

    public String getAuthResult(String uuid) {
        String url = aigcApiHost + "/publish/account/authResult";
        JSONObject params = new JSONObject();
        params.put("uuid", uuid);
        try {
            String post = httpPoolClient.post(url, getAigcPostParam(params));
            JSONObject res = JSONObject.parseObject(post);
            if (Objects.isNull(res)) {
                return null;
            }
            JSONObject data = res.getJSONObject("data");
            return data.getString("id");
        } catch (Exception e) {
            log.error("getAuthResult error", e);
        }
        return null;
    }

    public JSONObject getAccountDetail(String publishAccountId) {
        String url = aigcApiHost + "/publish/account/detail";
        JSONObject params = new JSONObject();
        params.put("id", publishAccountId);
        try {
            String post = httpPoolClient.post(url, getAigcPostParam(params));
            JSONObject res = JSONObject.parseObject(post);
            return res.getJSONObject("data");
        } catch (Exception e) {
            log.error("getAccountDetail error", e);
        }
        return null;
    }

    public String getDetailByGhId(String ghId) {
        String url = aigcApiHost + "/publish/account/getDetailByGhId";
        JSONObject params = new JSONObject();
        params.put("id", ghId);
        try {
            String post = httpPoolClient.post(url, getAigcPostParam(params));
            JSONObject res = JSONObject.parseObject(post);
            JSONObject data = res.getJSONObject("data");
            return data.getString("id");
        } catch (Exception e) {
            log.error("getAccountDetail error", e);
        }
        return null;
    }

    public String createPublishPlan(String publishAccountId, String name) {
        String url = aigcApiHost + "/publish/plan/save";
        JSONObject params = getPublishPlanAddParam(publishAccountId, name);
        JSONObject res = null;
        try {
            String post = httpPoolClient.post(url, getAigcPostParam(params));
            res = JSONObject.parseObject(post);
        } catch (Exception e) {
            log.error("createPublishPlan error", e);
        }
        if (Objects.isNull(res)) {
            throw new CommonException(3000, "创建计划失败，请稍后重试");
        }
        Integer code = res.getInteger("code");
        if (code != 0) {
            throw new CommonException(3000, res.getString("msg"));
        }
        JSONObject data = res.getJSONObject("data");
        return data.getString("id");
    }

    private JSONObject getPublishPlanAddParam(String publishAccountId, String name) {
        String dateStr = DateUtil.getCurrentDateStr("yyyyMMdd");
        JSONObject params = JSONObject.parseObject("{\n" +
                "    \"accountIds\": [\n" +
                "        \"" + publishAccountId + "\"\n" +
                "    ],\n" +
                "    \"activeManualReview\": 0,\n" +
                "    \"channel\": 5,\n" +
                "    \"contentAllocationRules\": 0,\n" +
                "    \"contentModal\": 7,\n" +
                "    \"douyinPublishAccoutSetting\": [],\n" +
                "    \"filterMatchMode\": 1,\n" +
                "    \"name\": \"自动回复-" + dateStr + "-" + name + "\",\n" +
                "    \"publishAccoutJson\": \"\",\n" +
                "    \"publishBgmType\": 0,\n" +
                "    \"publishDate\": \"\",\n" +
                "    \"publishLocation\": [],\n" +
                "    \"publishNum\": \"\",\n" +
                "    \"publishPushTime\": \"\",\n" +
                "    \"publishRate\": 0,\n" +
                "    \"publishTimeInterval\": 240,\n" +
                "    \"wxContentInsert\": [\n" +
                "        {\n" +
                "            \"accountId\": \"" + publishAccountId + "\",\n" +
                "            \"name\": \"" + name + "\",\n" +
                "            \"bottomInsertFlag\": 0,\n" +
                "            \"topInsertFlag\": 0,\n" +
                "            \"miniprogramInsertFlag\": 0,\n" +
                "            \"gzhArticleSortFlag\": 0,\n" +
                "            \"publishOpenFlag\": 1\n" +
                "        }\n" +
                "    ],\n" +
                "    \"wxVideoPublishAccountSetting\": [],\n" +
                "    \"scoreJudgeFlag\": 0,\n" +
                "    \"scoreJudgeTasks\": [],\n" +
                "    \"inputGroups\": [],\n" +
                "    \"machineReviewMatchMode\": 1,\n" +
                "    \"gzhOnlyMiniprogramFlag\": 0,\n" +
                "    \"planType\": 1,\n" +
                "    \"pushType\": 6,\n" +
                "    \"inputFilters\": [],\n" +
                "    \"inputSources\": [],\n" +
                "    \"triggerEvent\": 1,\n" +
                "    \"gzhTriggerSyncFrequency\": 1,\n" +
                "    \"gzhTriggerSendContentType\": 1,\n" +
                "    \"gzhAutoReplyDelayFlag\": 1,\n" +
                "    \"gzhAutoReplyDelayMillisecond\": 500,\n" +
                "    \"pushContentSortingRules\": 1\n" +
                "}");
        return params;
    }

    public void closePublishPlan(String externalId) {
        String url = aigcApiHost + "/publish/plan/updatePlanStatus";
        JSONObject params = new JSONObject();
        params.put("id", externalId);
        params.put("status", 0);
        try {
            String post = httpPoolClient.post(url, getAigcPostParam(params));
            JSONObject res = JSONObject.parseObject(post);
        } catch (Exception e) {
            log.error("closePublishPlan error", e);
        }
    }

    public void refreshGzhAutoReplyMsgData(String ghId) {
        String url = aigcApiHost + "/publish/api/refreshGzhAutoReplyMsgData?ghId=" + ghId;
        try {
            httpPoolClient.get(url);
        } catch (Exception e) {
            log.error("refreshGzhAutoReplyMsgData error", e);
        }
    }

    public List<WxAccountDatastatVO> getWxAccountDatastat(String dateStr, List<String> accountIds) {
        String url = aigcApiHost + "/dataStat/getWxAccountDatastat";
        JSONObject params = new JSONObject();
        params.put("dateStr", dateStr);
        params.put("accountIds", accountIds);
        try {
            String post = httpPoolClient.post(url, getAigcPostParam(params));
            JSONObject res = JSONObject.parseObject(post);
            return res.getJSONArray("data").toJavaList(WxAccountDatastatVO.class);
        } catch (Exception e) {
            log.error("closePublishPlan error", e);
        }
        return null;
    }
}