package cn.gistack.alerts.push.service.impl;

import cn.gistack.alerts.push.service.IPushService;
import cn.gistack.alerts.push.utils.GeTuiUtils;
import cn.gistack.alerts.push.vo.PushVO;
import cn.gistack.alerts.push.vo.UniCloudReturnRes;
import cn.gistack.sm.message.entity.MessageRecord;
import cn.gistack.sm.message.feign.IMessageClient;
import lombok.AllArgsConstructor;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.redis.cache.BladeRedis;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.utils.Func;
import org.springframework.stereotype.Service;
import org.springframework.web.client.RestTemplate;

import java.util.*;
import java.util.concurrent.atomic.AtomicReference;

@Service
@AllArgsConstructor
public class PushServiceImpl implements IPushService {
	private static final String PUSH_CLIENT_ID = "push:client:id:";
	private static final String UNI_CLOUD_PUSH_URL = "https://fc-mp-ff11cc4d-ca8e-4b47-a57f-e26804379530.next.bspapp.com/push";

	private BladeRedis bladeRedis;
	private IMessageClient messageClient;
	private GeTuiUtils geTuiUtils;

	@Override
	public Boolean saveCId(String userId, String cId) {
		bladeRedis.set(PUSH_CLIENT_ID + userId, cId);
		return Boolean.TRUE;
	}

	@Override
	public Boolean pushMessageList(List<MessageRecord> messageRecordList) {
		// 加入测试人员
//		addSendMsgTest(messageRecordList);
		List<String> cIdList = new ArrayList<>();
		AtomicReference<Boolean> push = new AtomicReference<>(false);
		//相同的消息推送给多用户
		messageRecordList.forEach(messageRecord -> {
			//获取用户cId
			String clientId = getClientIdByUserId(messageRecord.getRecipient());
			if (null!=clientId && !clientId.equals("")) {
				cIdList.add(clientId);
				PushVO pushVO = new PushVO();
				pushVO.setCIds(cIdList);
				pushVO.setTitle(messageRecord.getTheme());
				pushVO.setContent(messageRecord.getContent());

				HashMap<String, Object> payload = new HashMap<>();
				payload.put("title",messageRecord.getTheme());
				payload.put("content", messageRecord.getContent());
				payload.put("payload", "test");
				pushVO.setPayload(payload);
				push.set(pushMessage(pushVO));
			}
		});
		return push.get();
	}

	/**
	 * 新增测试信息
	 *
	 * @param messageRecordList
	 */
	private void addSendMsgTest(List<MessageRecord> messageRecordList) {
		MessageRecord messageRecord = new MessageRecord();
		messageRecord.setRecipient("1654164141753593858");
		messageRecord.setContent(messageRecordList.get(0).getContent());
		messageRecord.setTheme(messageRecordList.get(0).getTheme());
		messageRecord.setSender(AuthUtil.getUserId().toString());
		messageRecord.setSource("智能告警");
		messageRecord.setStatus(0);
		// 加入集合
		messageRecordList.add(messageRecord);
	}

	@Override
	public Boolean clearCId(String userId) {
		Boolean del = bladeRedis.del(PUSH_CLIENT_ID + userId);
		return del;
	}

	@Override
	public Boolean pushMessageOnLaunch(String userId) {
		//获取未读的消息列表
		List<MessageRecord> notReadList = messageClient.getNotReadList(userId);

		PushVO pushVO = new PushVO();
		List<String> cIdList = new ArrayList<>();

		//设置cId
		cIdList.add(getClientIdByUserId(userId));
		pushVO.setCIds(cIdList);
		//设置标题和内容
		notReadList.forEach(messageRecord -> {
			pushVO.setTitle(messageRecord.getTheme());
			pushVO.setContent(messageRecord.getContent());
			//推送消息
			pushMessage(pushVO);
		});

		return true;
	}

	@Override
	public Boolean pushMessageByUserId(PushVO pushVO) {

		if (Func.isEmpty(pushVO.getUserId())) {
			throw new ServiceException("请填写用户id!");
		}

		String clientIdByUserId = getClientIdByUserId(pushVO.getUserId());
		List<String> cIdList = new ArrayList<>();
		cIdList.add(clientIdByUserId);
		pushVO.setCIds(cIdList);

		HashMap<String, Object> payload = new HashMap<>();
		payload.put("title", pushVO.getTitle());
		payload.put("content", pushVO.getContent());
		payload.put("payload", "test");
		pushVO.setPayload(payload);


		return pushMessage(pushVO);
	}

	@Override
	public Boolean geTuiPushMessage(PushVO pushVO) {
		List<String> cIdList = new ArrayList<>();
		cIdList.add(getClientIdByUserId(pushVO.getUserId()));
//		return geTuiUtils.pushMsg(cIdList, pushVO.getTitle(), pushVO.getContent());
		return geTuiUtils.myPushMessage(cIdList, pushVO.getTitle(), pushVO.getContent());
	}

	@Override
	public Boolean pushMessageByUserList(PushVO pushVO) {

		if (Func.isEmpty(pushVO.getUserId())) {
			throw new ServiceException("请填写用户id!");
		}

		//获取userIdList
		List<String> userIdList = Arrays.asList(pushVO.getUserId().split(","));
		List<MessageRecord> messageRecordList = new ArrayList<>();

		userIdList.forEach(userId->{
			MessageRecord messageRecord = new MessageRecord();

			messageRecord.setContent(pushVO.getContent());
			messageRecord.setTheme(pushVO.getTitle());
			messageRecord.setRecipient(userId);

			messageRecordList.add(messageRecord);
		});

		Boolean aBoolean = pushMessageList(messageRecordList);

		return aBoolean;
	}

	/**
	 * 根据用户id从redis中获取cid
	 *
	 * @param userId
	 * @return
	 */
	public String getClientIdByUserId(String userId) {
		if (Func.isEmpty(userId)) {
			throw new ServiceException("请填写用户id!");
		}
		return bladeRedis.get(PUSH_CLIENT_ID + userId);
	}

	/**
	 * 2.0云函数消息推送
	 *
	 * @param pushVO
	 * @return
	 */
	public Boolean pushMessage(PushVO pushVO) {
		RestTemplate template = new RestTemplate();
		Map<String, Object> params = new LinkedHashMap<>();
		//用户cId，可数组，可字符串
		params.put("pushClientid", pushVO.getCIds());
		//标题
		params.put("title", pushVO.getTitle());
		//内容
		params.put("content", pushVO.getContent());

		params.put("requestId", UUID.randomUUID());

		params.put("payload", pushVO.getPayload());
		UniCloudReturnRes uniCloudReturnRes = template.postForObject(UNI_CLOUD_PUSH_URL, params, UniCloudReturnRes.class);
		return uniCloudReturnRes.getErrCode().equals("0") ? true : false;

	}

	/**
	 * 对集合按指定数量进行分组
	 *
	 * @param list     分组的集合
	 * @param quantity 数量
	 * @param <T>
	 * @return
	 */
	public <T> List<List<T>> groupListByQuantity(List list, int quantity) {
		if (list == null || list.size() == 0) {
			return list;
		}

		if (quantity <= 0) {
			new IllegalArgumentException("Wrong quantity.");
		}

		List wrapList = new ArrayList();
		int count = 0;
		while (count < list.size()) {
			wrapList.add(list.subList(count, (count + quantity) > list.size() ? list.size() : count + quantity));
			count += quantity;
		}

		return wrapList;
	}
}