package com.vincent.rsf.server.system.entity;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import com.fasterxml.jackson.annotation.JsonFormat;
import com.vincent.rsf.framework.common.Cools;
import com.vincent.rsf.framework.common.SpringUtils;
import com.vincent.rsf.server.system.service.TenantService;
import com.vincent.rsf.server.system.service.UserService;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.experimental.Accessors;
import org.springframework.format.annotation.DateTimeFormat;

import java.io.Serializable;
import java.text.SimpleDateFormat;
import java.util.Date;

@Data
@Accessors(chain = true)
@TableName("sys_ai_param")
public class AiParam implements Serializable {

    private static final long serialVersionUID = 1L;

    @ApiModelProperty(value= "ID")
    @TableId(value = "id", type = IdType.AUTO)
    private Long id;

    @ApiModelProperty(value= "编号")
    private String uuid;

    @ApiModelProperty(value= "名称")
    private String name;

    @ApiModelProperty(value= "模型编码")
    private String modelCode;

    @ApiModelProperty(value= "供应商")
    private String provider;

    @ApiModelProperty(value= "聊天地址")
    private String chatUrl;

    @ApiModelProperty(value= "API密钥")
    private String apiKey;

    @ApiModelProperty(value= "模型名称")
    private String modelName;

    @ApiModelProperty(value= "系统提示词")
    private String systemPrompt;

    @ApiModelProperty(value= "上下文轮数")
    private Integer maxContextMessages;

    @ApiModelProperty(value= "默认模型 1: 是  0: 否")
    private Integer defaultFlag;

    @ApiModelProperty(value= "排序")
    private Integer sort;

    @ApiModelProperty(value= "状态 1: 正常  0: 冻结  ")
    private Integer status;

    @ApiModelProperty(value= "是否删除 1: 是  0: 否  ")
    private Integer deleted;

    @ApiModelProperty(value= "租户")
    private Long tenantId;

    @ApiModelProperty(value= "添加人员")
    private Long createBy;

    @ApiModelProperty(value= "添加时间")
    @DateTimeFormat(pattern="yyyy-MM-dd HH:mm:ss")
    @JsonFormat(pattern="yyyy-MM-dd HH:mm:ss",timezone = "GMT+8")
    private Date createTime;

    @ApiModelProperty(value= "修改人员")
    private Long updateBy;

    @ApiModelProperty(value= "修改时间")
    @DateTimeFormat(pattern="yyyy-MM-dd HH:mm:ss")
    @JsonFormat(pattern="yyyy-MM-dd HH:mm:ss",timezone = "GMT+8")
    private Date updateTime;

    @ApiModelProperty(value= "备注")
    private String memo;

    public String getTenantId$(){
        TenantService service = SpringUtils.getBean(TenantService.class);
        Tenant tenant = service.getById(this.tenantId);
        if (!Cools.isEmpty(tenant)){
            return String.valueOf(tenant.getName());
        }
        return null;
    }

    public String getCreateBy$(){
        UserService service = SpringUtils.getBean(UserService.class);
        User user = service.getById(this.createBy);
        if (!Cools.isEmpty(user)){
            return String.valueOf(user.getNickname());
        }
        return null;
    }

    public String getCreateTime$(){
        if (Cools.isEmpty(this.createTime)){
            return "";
        }
        return new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(this.createTime);
    }

    public String getUpdateBy$(){
        UserService service = SpringUtils.getBean(UserService.class);
        User user = service.getById(this.updateBy);
        if (!Cools.isEmpty(user)){
            return String.valueOf(user.getNickname());
        }
        return null;
    }

    public String getUpdateTime$(){
        if (Cools.isEmpty(this.updateTime)){
            return "";
        }
        return new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(this.updateTime);
    }

    public Boolean getStatusBool(){
        if (null == this.status){ return null; }
        switch (this.status){
            case 1:
                return true;
            case 0:
                return false;
            default:
                return null;
        }
    }

    public Boolean getDefaultFlagBool(){
        if (null == this.defaultFlag){ return null; }
        switch (this.defaultFlag){
            case 1:
                return true;
            case 0:
                return false;
            default:
                return null;
        }
    }

    public String getDefaultFlag$(){
        if (null == this.defaultFlag){ return null; }
        switch (this.defaultFlag){
            case 1:
                return "是";
            case 0:
                return "否";
            default:
                return String.valueOf(this.defaultFlag);
        }
    }

}