package com.casic.missiles.es;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.ObjectUtil;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * ES查询请求体
 *
 * @author zhangyingjie123
 * @since 2020-11-30
 */
public class ElasticSearchQuery {
    private JSONObject body;
    private JSONObject query;
    private JSONObject bool;
    private JSONArray boolMust;
    private JSONObject filter;
    private JSONObject filterBool;
    private JSONArray filterBoolMust;
    private JSONArray filterBoolMustnot;

    public static final String MUST_FLAG = "1";
    public static final String MUST_NOT_FLAG = "0";

    public static final String QUERY = "query";
    public static final String BOOL = "bool";
    public static final String MUST = "must";
    public static final String MUST_NOT = "must_not";
    public static final String FILTER = "filter";


    /**
     * 生成查询请求体基础结构：
     * {
     * "query": {
     * "bool": {
     * "must" [],
     * "filter": {
     * "bool": {
     * "must": [],
     * "must_not": []
     * }
     * }
     * }
     * }
     * }
     */
    public ElasticSearchQuery() {
        this.body = new JSONObject();
        this.query = this.query();
        this.bool = this.bool();
        this.boolMust = this.boolMust();
        this.filter = this.filter();
        this.filterBool = this.filterBool();
        this.filterBoolMust = this.filterBoolMust();
        this.filterBoolMustnot = this.filterBoolMustnot();
    }

    public ElasticSearchQuery(JSONObject body) {
        this.body = body;
    }

    public JSONObject getBody() {
        return body;
    }

    private JSONObject query() {
        if (ObjectUtil.isEmpty(this.body.get(QUERY))) {
            this.body.put("query", new JSONObject());
        }
        return this.body.getJSONObject(QUERY);

    }

    private JSONObject bool() {
        JSONObject query = this.query;
        if (ObjectUtil.isEmpty(query.get(BOOL))) {
            query.put(BOOL, new JSONObject());
        }
        return query.getJSONObject(BOOL);
    }

    private JSONArray boolMust() {
        JSONObject bool = this.bool;
        if (ObjectUtil.isEmpty(bool.get(MUST))) {
            bool.put(MUST, new JSONArray());
        }
        return bool.getJSONArray(MUST);
    }

    private JSONObject filter() {
        JSONObject bool = this.bool;
        if (ObjectUtil.isEmpty(bool.get(FILTER))) {
            bool.put(FILTER, new JSONObject());
        }
        return bool.getJSONObject(FILTER);
    }

    private JSONObject filterBool() {
        JSONObject filter = this.filter;
        if (ObjectUtil.isEmpty(filter.get(BOOL))) {
            filter.put(BOOL, new JSONObject());
        }
        return filter.getJSONObject(BOOL);
    }

    private JSONArray filterBoolMust() {
        JSONObject bool = this.filterBool;
        if (ObjectUtil.isEmpty(bool.get(MUST))) {
            bool.put(MUST, new JSONArray());
        }
        return bool.getJSONArray(MUST);
    }

    private JSONArray filterBoolMustnot() {
        JSONObject bool = this.filterBool;
        if (ObjectUtil.isEmpty(bool.get(MUST_NOT))) {
            bool.put(MUST_NOT, new JSONArray());
        }
        return bool.getJSONArray(MUST_NOT);
    }

    public void match(String filed, Serializable value) {
        JSONArray boolMust = this.boolMust;
        JSONObject match = new JSONObject();
        JSONObject filedJson = new JSONObject();
        filedJson.put(filed, value);
        match.put("match", filedJson);
        boolMust.add(match);
    }

    public void matchPhrase(String filed, Serializable value) {
        JSONArray boolMust = this.boolMust;
        JSONObject matchPhrase = new JSONObject();
        JSONObject filedJson = new JSONObject();
        filedJson.put(filed, value);
        matchPhrase.put("match_phrase", filedJson);
        boolMust.add(matchPhrase);
    }

    public void term(String filed, Serializable value) {
        this.term(filed, value, MUST_FLAG);
    }

    public void term(String filed, Serializable value, String mustFlag) {
        JSONObject filedJson = new JSONObject();
        filedJson.put(filed, value);
        JSONObject term = new JSONObject();
        term.put("term", filedJson);
        if (MUST_FLAG.equals(mustFlag)) {
            this.filterBoolMust.add(term);
        }
        if (MUST_NOT_FLAG.equals(mustFlag)) {
            this.filterBoolMustnot.add(term);
        }
    }

    public void terms(String filed, List value) {
        this.terms(filed, value, MUST_FLAG);
    }

    public void terms(String filed, List value, String mustFlag) {
        JSONObject filedJson = new JSONObject();
        filedJson.put(filed, value);
        JSONObject terms = new JSONObject();
        terms.put("terms", filedJson);
        if (MUST_FLAG.equals(mustFlag)) {
            this.filterBoolMust.add(terms);
        }
        if (MUST_NOT_FLAG.equals(mustFlag)) {
            this.filterBoolMustnot.add(terms);
        }
    }

    public void range(String filed, Serializable from, Serializable to) {
        this.range(filed, from, to, MUST_FLAG);
    }

    /**
     * 范围条件是否必须
     */
    public void range(String filed, Serializable from, Serializable to, String mustFlag) {
        JSONObject rangeJson = new JSONObject();
        if (ObjectUtil.isNotEmpty(from)) {
            rangeJson.put("from", from);
        }
        if (ObjectUtil.isNotEmpty(to)) {
            rangeJson.put("to", to);
        }
        JSONObject filedJson = new JSONObject();
        filedJson.put(filed, rangeJson);
        JSONObject range = new JSONObject();
        range.put("range", filedJson);
        if (MUST_FLAG.equals(mustFlag)) {
            this.filterBoolMust.add(range);
        }
        if (MUST_NOT_FLAG.equals(mustFlag)) {
            this.filterBoolMustnot.add(range);
        }
    }

    public void size(int size) {
        this.body.put("size", size);
    }

    public void from(int from) {
        this.body.put("from", from);
    }

    public void sort(String sort, String order) {
        JSONObject sortJson = new JSONObject();
        JSONObject orderJson = new JSONObject();
        orderJson.put("order", order);
        sortJson.put(sort, orderJson);
        this.body.put("sort", sortJson);
    }

    public void aggs(String aggsName, String filed) {
        JSONObject terms = new JSONObject();
        terms.put("field", filed);
        JSONObject aggsTerms = new JSONObject();
        aggsTerms.put("terms", terms);
        JSONObject aggsBody = new JSONObject();
        aggsBody.put(aggsName, aggsTerms);
        this.body.put("aggs", aggsBody);
    }

    public void aggs(JSONObject aggBody) {
        this.body.put("aggs", aggBody);
    }


    public static void main(String[] args) {
        ElasticSearchQuery query = new ElasticSearchQuery();
        query.matchPhrase("name", "张三");
        query.term("sex", "2");
        List<Long> deptIds = new ArrayList<>();
        deptIds.add(0L);
        deptIds.add(24L);
        query.terms("deptId", deptIds);
        Date date1 = DateUtil.parse("2020-01-01", "yyyy-MM-dd");
        Date date2 = DateUtil.parse("2020-04-01", "yyyy-MM-dd");
        query.range("collTime", date1.getTime(), date2.getTime());
        query.term("collState", "3", MUST_NOT_FLAG);
        query.size(10);
        query.from(0);
        query.sort("irId", "asc");
        System.out.println(query.getBody().toJSONString());
    }


}