package com.ruoyi.common.core.domain;

import com.ruoyi.common.constant.HttpStatus;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 响应信息主体
 *
 * @author Lion Li
 */
@Data
@NoArgsConstructor
public class R<T> implements Serializable {
    private static final long serialVersionUID = 1L;

    /**
     * 成功
     */
    public static final int SUCCESS = 200;

    /**
     * 失败
     */
    public static final int FAIL = 500;

    private int code;

    private String msg;

    private T data;

    public R<T> setCode(int code){
        this.code = code;
        return this;
    }

    public R<T> setMsg(String msg){
        this.msg = msg;
        return this;
    }

    public R<T> setData(T data) {
        this.data = data;
        return this;
    }

    public static <T> R<T> ok() {
        return restResult(null, SUCCESS, "操作成功");
    }

    public static <T> R<T> ok(T data) {
        return restResult(data, SUCCESS, "操作成功");
    }

    public static <T> R<T> ok(String msg) {
        return restResult(null, SUCCESS, msg);
    }

    public static <T> R<T> ok(String msg, T data) {
        return restResult(data, SUCCESS, msg);
    }

    public static <T> R<T> fail() {
        return restResult(null, FAIL, "操作失败");
    }

    public static <T> R<T> fail(String msg) {
        return restResult(null, FAIL, msg);
    }

    public static <T> R<T> fail(T data) {
        return restResult(data, FAIL, "操作失败");
    }

    public static <T> R<T> fail(String msg, T data) {
        return restResult(data, FAIL, msg);
    }

    public static <T> R<T> fail(int code, String msg) {
        return restResult(null, code, msg);
    }

    /**
     * 返回警告消息
     *
     * @param msg 返回内容
     * @return 警告消息
     */
    public static <T> R<T> warn(String msg) {
        return restResult(null, HttpStatus.WARN, msg);
    }

    /**
     * 返回警告消息
     *
     * @param msg 返回内容
     * @param data 数据对象
     * @return 警告消息
     */
    public static <T> R<T> warn(String msg, T data) {
        return restResult(data, HttpStatus.WARN, msg);
    }

    private static <T> R<T> restResult(T data, int code, String msg) {
        R<T> r = new R<>();
        r.setCode(code);
        r.setData(data);
        r.setMsg(msg);
        return r;
    }

    public static <T> Boolean isError(R<T> ret) {
        return !isSuccess(ret);
    }

    public static <T> Boolean isSuccess(R<T> ret) {
        return R.SUCCESS == ret.getCode();
    }

    public static R<Map<String, ?>> map(){
        R<Map<String, ?>> r = new R<>();
        r.setCode(SUCCESS);
        r.setMsg("操作成功");
        r.setData(new HashMap<>(10));
        return r;
    }

    @SuppressWarnings("unchecked")
    public R<Map<String, ?>> put(String key,Object value){
        if(this.data instanceof Map) {
            ((Map<String,Object>)this.data).put(key, value);
            return (R<Map<String, ?>>) this;
        }else {
            throw new RuntimeException("data does not belong to Map");
        }
    }

    public static R<List<?>> list(){
        R<List<?>> r = new R<>();
        r.setData(new ArrayList<>(10));
        r.setCode(SUCCESS);
        r.setMsg("操作成功");
        return r;
    }

    @SuppressWarnings("unchecked")
    public R<List<?>> add(Object value){
        if(this.data instanceof List) {
            ((List<Object>)this.data).add(value);
            return (R<List<?>>) this;
        }else {
            throw new RuntimeException("data does not belong to List");
        }
    }

}
