ContentGrabbingJi/src/main/java/net/lamgc/cgj/bot/event/SpringCQMessageEvent.java

package net.lamgc.cgj.bot.event;

import net.lamgc.cgj.bot.BotCode;
import net.lz1998.cq.event.message.CQDiscussMessageEvent;
import net.lz1998.cq.event.message.CQGroupMessageEvent;
import net.lz1998.cq.event.message.CQMessageEvent;
import net.lz1998.cq.robot.CoolQ;

import java.net.InetSocketAddress;
import java.util.Objects;

public class SpringCQMessageEvent extends MessageEvent {

    private final static int TYPE_PRIVATE = 0;
    private final static int TYPE_GROUP = 1;
    private final static int TYPE_DISCUSS = 2;

    private final CoolQ cq;
    private final int type;
    private final CQMessageEvent messageEvent;

    public SpringCQMessageEvent(CoolQ cq, CQMessageEvent messageEvent) {
        super(messageEvent instanceof CQGroupMessageEvent ? (
                (CQGroupMessageEvent) messageEvent).getGroupId() :
              messageEvent instanceof CQDiscussMessageEvent ?
                ((CQDiscussMessageEvent) messageEvent).getDiscussId() : 0,
              messageEvent.getUserId(), messageEvent.getMessage());
        this.cq = Objects.requireNonNull(cq);
        if(messageEvent instanceof CQGroupMessageEvent) {
            type = TYPE_GROUP;
        } else if (messageEvent instanceof CQDiscussMessageEvent) {
            type = TYPE_DISCUSS;
        } else {
            type = TYPE_PRIVATE;
        }
        this.messageEvent = messageEvent;
    }

    @Override
    public int sendMessage(final String message) {
        switch(type) {
            case TYPE_PRIVATE:
                return cq.sendPrivateMsg(getFromQQ(), message, false).getData().getMessageId();
            case TYPE_GROUP:
            case TYPE_DISCUSS:
                return cq.sendGroupMsg(getFromGroup(), message, false).getData().getMessageId();
            default:
                return -1;
        }
    }

    /**
     * 通过CQ码获取图片下载链接.
     * @param imageFileName 图片完整CQ码
     * @return 图片下载链接
     */
    @Override
    public String getImageUrl(String imageFileName) {
        BotCode code;
        if(imageFileName.startsWith("[CQ:") && imageFileName.endsWith("]")) {
            code = BotCode.parse(imageFileName);
            return code.getParameter("url");
        } else {
            InetSocketAddress remoteAddress = cq.getBotSession().getRemoteAddress();
            if(remoteAddress == null) {
                throw new IllegalStateException("remoteAddress failed to get");
            }
            String file = cq.getImage(imageFileName).getData().getFile().replaceAll("\\\\", "/");
            return "http://" + remoteAddress.getHostString() + ":5700/data" + file.substring(file.lastIndexOf("/data") + 5);
        }
    }

    @Override
    public Object getRawMessage() {
        return messageEvent;
    }
}