package com.infoepoch.pms.dispatchassistant.common.component;


import com.infoepoch.pms.dispatchassistant.common.utils.JsonUtils;

import java.util.HashMap;
import java.util.Map;

/**
 * Json返回值定义
 */
public class Result {

    private int code;           //返回码
    private String message;     //返回消息
    private Object data;        //数据

    public Result()
    {}

    /**
     * 构造函数
     */
    private Result(int code, String message) {
        this.code = code;
        this.message = message;
    }

//    /**
//     * 构造函数
//     */
//    private Result(int code, String message, Object data) {
//        this.code = code;
//        this.message = message;
//        this.data = data;
//    }

    public int getCode() {
        return code;
    }

    public String getMessage() {
        return message;
    }

    public Object getData() {
        return data;
    }

    /**
     * 成功的返回码为0
     */
    private static final int CODE_SUCCESS = 0;

    /**
     * 是否成功
     */
    public boolean isSuccess() {
        return code == CODE_SUCCESS;
    }

    /**
     * 成功，成功时返回码为0
     */
    public static Result success() {
        return new Result(CODE_SUCCESS, "");
    }

    /**
     * 成功，成功时返回码为0
     */
    public static Result success(String message) {
        return new Result(CODE_SUCCESS, message);
    }

    /**
     * 成功，成功时返回码为0
     */
    public static Result success(String message, Object data) {
        return new Result(CODE_SUCCESS, message, data);
    }

    /**
     * 成功，成功时返回码为0
     */
    public static Result successData(Object data) {
        return new Result(CODE_SUCCESS, "", data == null ? "" : data);
    }

    /**
     * 失败
     *
     * @param code 返回码，由于默认0为成功时的返回码，如此参数传入0将自动替换为-1
     * @return
     */
    public static Result error(int code) {
        if (code == CODE_SUCCESS) {
            code = -1;
        }
        return new Result(code, "");
    }

    /**
     * 失败
     *
     * @param code    返回码，由于默认0为成功时的返回码，如此参数传入0将自动替换为-1
     * @param message
     * @return
     */
    public static Result error(int code, String message) {
        if (code == CODE_SUCCESS) {
            code = -1;
        }
        return new Result(code, message);
    }

    /**
     * 失败
     *
     * @param code    返回码，由于默认0为成功时的返回码，如此参数传入0将自动替换为-1
     * @param message
     * @param data
     * @return
     */
    public static Result error(int code, String message, Object data) {
        if (code == CODE_SUCCESS) {
            code = -1;
        }
        return new Result(code, message, data);
    }

    /**
     * 转换为map，便于动态添加属性
     */
    public Map<String, Object> toMap() {
        Map<String, Object> map = new HashMap<>();
        map.put("code", code);
        map.put("message", message);
        map.put("data", data);
        map.put("success", isSuccess());
        return map;
    }
    /**
     * 构造函数
     */
    private Result(int code, String message, Object data) {
        this.code = code;
        this.message = message;
        this.data =  JsonUtils.jsonToObject(Result.filterXssResponse(JsonUtils.objectToJson(data))
                ,Object.class);
    }
    /**
     * 过滤XSS攻击
     * @param value 需要过滤的字符串
     * @return 过滤后的字符串
     */
    private static String filterXssResponse(String value) {
        if (value == null) {
            return null;
        }

        // 清理空白字符
        String cleanValue = value.trim();

        // 过滤常见的XSS攻击字符
        //.replaceAll("[\\\"\\\'][\\s]*javascript:(.*)[\\\"\\\']", "\"\"")
        cleanValue = cleanValue.replaceAll("<", "&lt;")
                .replaceAll(">", "&gt;")
                .replaceAll("\\(", "&#40;")
                .replaceAll("\\)", "&#41;")
                .replaceAll("'", "&#39;")
                .replaceAll("eval\\((.*)\\)", "")
                .replaceAll("alert\\((.*)\\)", "")      // 添加：alert
                .replaceAll("confirm\\((.*)\\)", "")    // 添加：confirm
                .replaceAll("prompt\\((.*)\\)", "")     // 添加：prompt
                .replaceAll("open\\((.*)\\)", "")       // 添加：open
                .replaceAll("write\\((.*)\\)", "")       // 添加：write
                .replaceAll("writeln\\((.*)\\)", "")    // 添加：writeln

                .replaceAll("(?i)script", "")
                .replaceAll("(?i)onload", "")
                .replaceAll("(?i)onerror", "")
                .replaceAll("(?i)onclick", "")
                .replaceAll("(?i)onmouseover", "")
                .replaceAll("(?i)onfocus", "")
                .replaceAll("(?i)onblur", "");

        return cleanValue;
    }


}

