net.deechael.khl.entity.UserEntity Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of khl Show documentation
Java sdk for Kaiheila bot development
The newest version!
package net.deechael.khl.entity;

import com.fasterxml.jackson.databind.JsonNode;
import net.deechael.khl.api.User;
import net.deechael.khl.core.KaiheilaObject;
import net.deechael.khl.gate.Gateway;
import net.deechael.khl.message.BotChannelMessage;
import net.deechael.khl.message.Message;
import net.deechael.khl.message.ReceivedMessage;
import net.deechael.khl.message.TextMessage;
import net.deechael.khl.message.kmarkdown.KMarkdownMessage;
import net.deechael.khl.restful.RestRoute;

public class UserEntity extends KaiheilaObject implements User {

    private String id;
    private String username;
    private String identifyNum;
    private boolean online;
    private boolean bot;
    private int status;
    private String banner;
    private String avatar;
    private String vipAvatar;
    private boolean mobileVerified;

    public UserEntity(Gateway gateway, JsonNode node) {
        super(gateway);
        this.setId(node.get("id").asText());
        this.setUsername(node.get("username").asText());
        this.setIdentifyNum(node.get("identify_num").asText());
        this.setOnline(node.get("online").asBoolean());
        this.setBanner(node.get("banner").asText());
        this.setBot(node.get("bot").asBoolean());
        this.setStatus(node.get("status").asInt());
        this.setAvatar(node.get("avatar").asText());
        this.setMobileVerified(node.get("mobile_verified").asBoolean());
    }

    public String getName() {
        return username + "#" + identifyNum;
    }

    /**
     * 开黑啦唯一表示符 用户 Id
     *
     * @return 用户 Id
     */
    @Override
    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    /**
     * 当前用户名，不带标识数字
     *
     * @return 用户名
     * @see User#getIdentifyNum()
     * @see User#getFullName()
     */
    @Override
    public String getUsername() {
        return username;
    }

    public void setUsername(String username) {
        this.username = username;
    }

    /**
     * 当前用户名，带标识数字
     *
     * @return 带标识数字的用户名
     * @see User#getIdentifyNum()
     * @see User#getUsername()
     */
    @Override
    public String getFullName() {
        return getName();
    }

    /**
     * 当前标识数字
     *
     * @return 用户标识数字
     * @see User#getUsername()
     * @see User#getFullName()
     */
    @Override
    public String getIdentifyNum() {
        return identifyNum;
    }

    public void setIdentifyNum(String identifyNum) {
        this.identifyNum = identifyNum;
    }

    /**
     * 当前用户是否在线
     *
     * @return true 当前用户在线，否则用户当前离线
     */
    @Override
    public boolean isOnline() {
        return online;
    }

    public void setOnline(boolean online) {
        this.online = online;
    }

    /**
     * 当前用户是否机器人
     *
     * @return true 机器人用户，否则为普通用户
     */
    @Override
    public boolean isBot() {
        return bot;
    }

    public void setBot(boolean bot) {
        this.bot = bot;
    }

    // getter / setter

    /**
     * 当前用户是否被开黑啦封禁
     *
     * @return true 用户被平台封禁，否则为正常用户
     */
    @Override
    public boolean isBanned() {
        return status == 0 || status == 1;
    }

    /**
     * 当前用户Buff的个人横幅资源路径
     *
     * @return 个人横幅资源路径
     */
    @Override
    public String getBanner() {
        return banner;
    }

    public void setBanner(String banner) {
        this.banner = banner;
    }

    /**
     * 当前用户普通头像资源路径
     *
     * @return 头像资源路径
     */
    @Override
    public String getAvatarUrl() {
        return avatar;
    }

    /**
     * 当前用户Buff头像资源路径
     *
     * @return Buff头像资源路径
     */
    @Override
    public String getVipAvatarUrl() {
        return vipAvatar;
    }

    /**
     * 当前用户是否已验证手机号码
     *
     * @return true 用户已经通过手机验证，否则为未验证用户
     */
    @Override
    public boolean isMobileVerified() {
        return mobileVerified;
    }

    public void setMobileVerified(boolean mobileVerified) {
        this.mobileVerified = mobileVerified;
    }

    public int getStatus() {
        return status;
    }

    public void setStatus(int status) {
        this.status = status;
    }

    public String getAvatar() {
        return avatar;
    }

    public void setAvatar(String avatar) {
        this.avatar = avatar;
    }

    public String getVipAvatar() {
        return vipAvatar;
    }

    public void setVipAvatar(String vipAvatar) {
        this.vipAvatar = vipAvatar;
    }

    public ReceivedMessage sendMessage(String message, boolean isKMarkdown) {
        return this.sendMessage(isKMarkdown ? KMarkdownMessage.create(message) : new TextMessage(message));
    }

    public ReceivedMessage sendMessage(Message message) {
        JsonNode data = this.getGateway().executeRequest(RestRoute.DirectMessage.SEND_DIRECT_MESSAGE.compile()
                .withQueryParam("target_id", this.getId())
                .withQueryParam("nonce", "bot-message")
                .withQueryParam("content", message.getContent())
                .withQueryParam("type", message.getType().getType())
        );
        return new BotChannelMessage(data.get("msg_id").asText(), data.get("msg_timestamp").asInt(), message, getGateway().getKaiheilaBot().getSelf(), null);
    }

    public ReceivedMessage reply(String message, String msgId, boolean isKMarkdown) {
        return this.reply(isKMarkdown ? KMarkdownMessage.create(message) : new TextMessage(message), msgId);
    }

    public ReceivedMessage reply(Message message, String msgId) {
        JsonNode data = this.getGateway().executeRequest(RestRoute.DirectMessage.SEND_DIRECT_MESSAGE.compile()
                .withQueryParam("target_id", this.getId())
                .withQueryParam("nonce", "bot-message")
                .withQueryParam("content", message.getContent())
                .withQueryParam("type", message.getType().getType())
                .withQueryParam("quote", msgId)
        );
        return new BotChannelMessage(data.get("msg_id").asText(), data.get("msg_timestamp").asInt(), message, getGateway().getKaiheilaBot().getSelf(), null);
    }

    public void updateIntimacy(int value) {
        this.getGateway().executeRequest(RestRoute.Intimacy.UPDATE_USER_INTIMACY.compile()
                .withQueryParam("user_id", this.getId())
                .withQueryParam("score", value)
        );
    }

    public int getUserIntimacy() {
        JsonNode data = this.getGateway().executeRequest(RestRoute.Intimacy.INTIMACY_LIST.compile()
                .withQueryParam("user_id", this.getId())
        );
        return data.get("score").asInt();
    }

}