<template>
  <div class="app-container">
    <el-form :model="queryParams" ref="queryForm" size="small" :inline="true" v-show="showSearch" label-width="100px">
      <el-form-item label="实体关系名称" prop="name">
        <el-input
          v-model="queryParams.name"
          placeholder="请输入实体关系名称"
          clearable
          @keyup.enter.native="handleQuery"
        />
      </el-form-item>
      <el-form-item>
        <el-button type="primary" icon="el-icon-search" size="mini" @click="handleQuery">搜索</el-button>
        <el-button icon="el-icon-refresh" size="mini" @click="resetQuery">重置</el-button>
      </el-form-item>
    </el-form>

    <el-row :gutter="10" class="mb8">
      <el-col :span="1.5">
        <el-button
          type="primary"
          plain
          icon="el-icon-plus"
          size="mini"
          @click="handleAdd"
          v-hasPermi="['knowledge:entityRelation:add']"
        >新增</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="primary"
          plain
          icon="el-icon-plus"
          size="mini"
          @click="handleAddByClass"
          v-hasPermi="['knowledge:entityRelation:add']"
        >根据类新增</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="success"
          plain
          icon="el-icon-edit"
          size="mini"
          :disabled="single"
          @click="handleUpdate"
          v-hasPermi="['knowledge:entityRelation:edit']"
        >修改</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="danger"
          plain
          icon="el-icon-delete"
          size="mini"
          :disabled="multiple"
          @click="handleDelete"
          v-hasPermi="['knowledge:entityRelation:remove']"
        >删除</el-button>
      </el-col>
      <!-- <el-col :span="1.5">
        <el-button
          type="warning"
          plain
          icon="el-icon-download"
          size="mini"
          @click="handleExport"
          v-hasPermi="['knowledge:entityRelation:export']"
        >导出</el-button>
      </el-col> -->
      <right-toolbar :showSearch.sync="showSearch" @queryTable="getList"></right-toolbar>
    </el-row>

    <el-table v-loading="loading" :data="entityRelationList" @selection-change="handleSelectionChange">
      <el-table-column type="selection" width="55" align="center" />
      <el-table-column label="实体关系名称" align="center" prop="name" />
      <el-table-column label="起点实体" align="center" prop="startNode.name" />
      <el-table-column label="终点实体" align="center" prop="endNode.name" />
      <el-table-column label="操作" align="center" class-name="small-padding fixed-width">
        <template slot-scope="scope">
          <el-button
            size="mini"
            type="text"
            icon="el-icon-edit"
            @click="handleUpdate(scope.row)"
            v-hasPermi="['knowledge:entityRelation:edit']"
          >修改</el-button>
          <el-button
            size="mini"
            type="text"
            icon="el-icon-delete"
            @click="handleDelete(scope.row)"
            v-hasPermi="['knowledge:entityRelation:remove']"
          >删除</el-button>
        </template>
      </el-table-column>
    </el-table>

    <pagination
      v-show="total>0"
      :total="total"
      :page.sync="queryParams.pageNum"
      :limit.sync="queryParams.pageSize"
      @pagination="getList"
    />

    <!-- 添加或修改实体关系管理对话框 -->
    <el-dialog :title="title" :visible.sync="open" width="500px" append-to-body>
      <el-form ref="form" :model="form" :rules="rules" label-width="100px">
        <el-form-item label="实体关系名称" prop="name">
          <el-input v-model="form.name" placeholder="请输入实体关系名称" />
        </el-form-item>
        <el-form-item label="起点实体" prop="startNode">
          <el-select v-model="form.startNode" placeholder="请选择起点实体" value-key="id" filterable >
            <el-option
              v-for="item in entityList"
              :key="item.id"
              :label="item.name"
              :value="item">
            </el-option>
          </el-select>
        </el-form-item>
        <el-form-item label="终点实体" prop="endNode">
          <el-select v-model="form.endNode" placeholder="请选择终点实体" value-key="id" filterable >
            <el-option
              v-for="item in entityList"
              :key="item.id"
              :label="item.name"
              :value="item">
            </el-option>
          </el-select>
        </el-form-item>
      </el-form>
      <div slot="footer" class="dialog-footer">
        <el-button type="primary" @click="submitForm">确 定</el-button>
        <el-button @click="cancel">取 消</el-button>
      </div>
    </el-dialog>

    <el-dialog :title="title" :visible.sync="openByClass" width="500px" append-to-body>
      <el-form ref="formClass" :model="formByClass" :rules="rulesByClass" label-width="100px">
        <el-form-item label="起点实体类" prop="subEntityClassId">
          <el-select v-model="formByClass.subEntityClassId" placeholder="请选择起点实体" value-key="id" filterable @change="getEntityClassRelationOption">
            <el-option
              v-for="item in entityClassOption"
              :key="item.id"
              :label="item.name"
              :value="item.id">
            </el-option>
          </el-select>
        </el-form-item>
        <el-form-item label="终点实体类" prop="objEntityClassId">
          <el-select v-model="formByClass.objEntityClassId" placeholder="请选择终点实体" value-key="id" filterable @change="getEntityClassRelationOption">
            <el-option
              v-for="item in entityClassOption"
              :key="item.id"
              :label="item.name"
              :value="item.id">
            </el-option>
          </el-select>
        </el-form-item>
        <el-form-item label="实体关系类" prop="entityClassRelId">
          <el-select v-model="formByClass.entityClassRelId" placeholder="请选实体关系类" value-key="id" filterable clearable >
            <el-option
              v-for="item in entityClassRelationOption"
              :key="item.id"
              :label="item.name"
              :value="item.id">
            </el-option>
          </el-select>
        </el-form-item>
        <el-form-item label="起点实体" prop="subEntity">
          <el-input type="textarea" :rows="2" v-model="formByClass.subEntity" placeholder="请输入起点实体" />
        </el-form-item>
        <el-form-item label="终点实体" prop="objEntity">
          <el-input type="textarea" :rows="2"  v-model="formByClass.objEntity" placeholder="请输入终点实体" />
        </el-form-item>
      </el-form>
      <div slot="footer" class="dialog-footer">
        <el-button type="primary" @click="submitFormByClass(1)">提交并继续添加</el-button>
        <el-button type="primary" @click="submitFormByClass(0)">提交并关闭</el-button>
        <el-button @click="cancelByClass">取 消</el-button>
      </div>
    </el-dialog>
  </div>
</template>

<script>
import { listEntityRelation, getEntityRelation, delEntityRelation, addEntityRelation, updateEntityRelation } from "@/api/knowledge/entityRelation";
import { findAll } from "@/api/knowledge/entity";
import { getEntityClassOption } from "@/api/neo4j/class"
import { getClassRelationOptionByClassId, addEntityAndRelation } from "@/api/neo4j/class_relation"
import { bit2Info } from '@/utils/validate'

export default {
  name: "EntityRelation",
  data() {
    return {
      // 遮罩层
      loading: true,
      // 选中数组
      ids: [],
      // 非单个禁用
      single: true,
      // 非多个禁用
      multiple: true,
      // 显示搜索条件
      showSearch: true,
      // 总条数
      total: 0,
      // 实体关系管理表格数据
      entityRelationList: [],
      // 弹出层标题
      title: "",
      // 是否显示弹出层
      open: false,
      // 查询参数
      queryParams: {
        pageNum: 1,
        pageSize: 10,
        name: null,
      },
      // 表单参数
      form: {},
      formByClass: {},
      // 表单校验
      rules: {
        name: [
          {required: true, message: '实体关系名称不能为空', trigger: 'blur'},
        ],
        startNode: [
          {required: true, message: '起点实体不能为空', trigger: 'blur'},
        ],
        endNode: [
          {required: true, message: '终点实体不能为空', trigger: 'blur'},
        ],
      },
      rulesByClass: {
        subEntityClassId: [
          {required: true, message: '起点实体类不能为空', trigger: 'blur'},
        ],
        objEntityClassId: [
          {required: true, message: '终点实体类不能为空', trigger: 'blur'},
        ],
        entityClassRelId: [
          {required: true, message: '实体关系类不能为空', trigger: 'blur'},
        ],
        subEntity: [
          {required: true, message: '起点实体不能为空', trigger: 'blur'},
        ],
        objEntity: [
          {required: true, message: '终点实体不能为空', trigger: 'blur'},
        ],
      },
      entityList: [],
      entityClassOption: [],
      entityClassRelationOption: [],
      openByClass: false,
    };
  },
  created() {
    this.getList();
    this.getOption();
  },
  activated(){
    this.getOption();
  },
  methods: {
    /** 查询实体关系管理列表 */
    bit2Info,
    getList() {
      this.loading = true;
      listEntityRelation(this.queryParams).then(response => {
        this.entityRelationList = response.content;
        this.total = response.totalElements;
        this.loading = false;
      });
    },
    // 取消按钮
    cancel() {
      this.open = false;
      this.reset();
    },
    // 表单重置
    reset() {
      this.form = {
        name: null,
        content: null,
        startNode: null,
        endNode: null,
      };
      this.resetForm("form");
    },
    /** 搜索按钮操作 */
    handleQuery() {
      this.queryParams.pageNum = 1;
      this.getList();
    },
    /** 重置按钮操作 */
    resetQuery() {
      this.resetForm("queryForm");
      this.handleQuery();
    },
    // 多选框选中数据
    handleSelectionChange(selection) {
      this.ids = selection.map(item => item.id)
      this.single = selection.length!==1
      this.multiple = !selection.length
    },
    /** 新增按钮操作 */
    handleAdd() {
      this.reset();
      this.open = true;
      this.title = "添加实体关系管理";
    },
    /** 修改按钮操作 */
    handleUpdate(row) {
      this.reset();
      const relId = row.id || this.ids
      getEntityRelation(relId).then(response => {
        this.form = response.data;
        this.open = true;
        this.title = "修改实体关系管理";
      });
    },
    /** 提交按钮 */
    submitForm() {
      this.$refs["form"].validate(valid => {
        if (valid) {
          if (this.form.id != null) {
            updateEntityRelation(this.form).then(response => {
              this.$modal.msgSuccess("修改成功");
              this.open = false;
              this.getList();
            });
          } else {
            addEntityRelation(this.form).then(response => {
              this.$modal.msgSuccess("新增成功");
              this.open = false;
              this.getList();
            });
          }
        }
      });
    },
    /** 删除按钮操作 */
    handleDelete(row) {
      const relIds = row.id || this.ids;
      this.$modal.confirm('是否确认删除实体关系管理编号为"' + relIds + '"的数据项？').then(function() {
        return delEntityRelation(relIds);
      }).then(() => {
        this.getList();
        this.$modal.msgSuccess("删除成功");
      }).catch(() => {});
    },
    /** 导出按钮操作 */
    handleExport() {
      this.download('knowledge/entityRelation/export', {
        ...this.queryParams
      }, `entityRelation_${new Date().getTime()}.xlsx`)
    },
    getOption(){
      findAll().then(resp => {
        this.entityList = resp
      })
      getEntityClassOption().then(resp => {
        this.entityClassOption = resp.data
      })
    },
    getEntityClassRelationOption(){
      delete this.formByClass.entityClassRelId;
      if(this.formByClass.subEntityClassId && this.formByClass.objEntityClassId) {
        getClassRelationOptionByClassId(this.formByClass.subEntityClassId, this.formByClass.objEntityClassId).then(resp => {
          this.entityClassRelationOption = resp.data;
        })
      } else {
        this.entityClassRelationOption = []
      }
    },
    handleAddByClass(){
      this.resetClass();
      this.openByClass = true;
      this.title = "添加实体及实体关系";
    },
    resetClass(){
      this.formByClass = {
        // name: null,
        // content: null,
        // startNode: null,
        // endNode: null,
      };
      this.resetForm("formClass");
    },
    cancelByClass(){
      this.openByClass = false;
      this.resetClass();
    },
    submitFormByClass(type){
      this.$refs["formClass"].validate(valid => {
        if (valid) {
          addEntityAndRelation(this.formByClass).then(reps => {
            this.$modal.msgSuccess("提交成功");
            if(type) {
              this.handleAddByClass()
            } else {
              this.openByClass = false;
              this.getList();
            }
            
          })
        }
      })
    },
  }
};
</script>
