package com.tiesheng.util.pojos;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.extra.spring.SpringUtil;
import com.tiesheng.util.config.TsTokenConfig;
import com.tiesheng.util.jwt.JWT;

/**
 * @author hao
 */
public class TokenBean {

    private String id;
    private String roleId;
    private String extra;

    public TokenBean() {
    }

    public TokenBean(String id, String roleId) {
        this.id = id;
        this.roleId = roleId;
        this.extra = "";
    }


    /**
     * 设置token
     */
    public String toToken() {
        TsTokenConfig tsTokenConfig = SpringUtil.getBean(TsTokenConfig.class);
        return JWT.create()
                .setExpiresAt(DateUtil.offsetHour(DateUtil.date(), tsTokenConfig.getExpireHours()))
                .setPayload("id", getId())
                .setPayload("roleId", StrUtil.emptyToDefault(getRoleId(), ""))
                .setPayload("extra", StrUtil.emptyToDefault(getExtra(), ""))
                .setKey(tsTokenConfig.getEncryptKey().getBytes())
                .sign();
    }

    ///////////////////////////////////////////////////////////////////////////
    // setter\getter
    ///////////////////////////////////////////////////////////////////////////

    public String getId() {
        return id;
    }

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

    public String getExtra() {
        return extra;
    }

    public void setExtra(String extra) {
        this.extra = extra;
    }

    public String getRoleId() {
        return roleId;
    }

    public void setRoleId(String roleId) {
        this.roleId = roleId;
    }
}