<template>
  <div class="app-container">
    <el-form :model="queryParams" ref="queryForm" size="small" :inline="true" v-show="showSearch">
      <el-form-item label="节点名称" prop="nodeName">
        <el-input
          v-model="queryParams.nodeName"
          placeholder="请输入节点名称"
          clearable
          @keyup.enter.native="handleQuery"
        />
      </el-form-item>
      <el-form-item label="状态" prop="status">
        <el-select v-model="queryParams.status" placeholder="节点状态" clearable>
          <el-option
            v-for="dict in dict.type.sys_normal_disable"
            :key="dict.value"
            :label="dict.label"
            :value="dict.value"
          />
        </el-select>
      </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="handleAddRoot"
          v-hasPermi="['generator:structure:add']"
        >新增</el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="info"
          plain
          icon="el-icon-sort"
          size="mini"
          @click="toggleExpandAll"
        >展开/折叠</el-button>
      </el-col>
      <right-toolbar :showSearch.sync="showSearch" @queryTable="getList"></right-toolbar>
    </el-row>

    <el-table
      v-if="refreshTable"
      v-loading="loading"
      :data="structureData"
      row-key="nodeId"
      :default-expand-all="isExpandAll"
      :tree-props="{children: 'children', hasChildren: 'hasChildren'}"
    >
      <el-table-column prop="nodeName" label="节点名称" width="350"></el-table-column>
      <el-table-column prop="nodeType" label="节点类型" width="100">
        <template slot-scope="scope">
          <dict-tag :options="dict.type.sys_node_type" :value="scope.row.nodeType"/>
        </template>
      </el-table-column>
      <el-table-column prop="module" label="功能模块" width="120"></el-table-column>
      <el-table-column prop="tableId" label="表ID" width="120">
        <template slot-scope="scope">
          <span v-if="scope.row.nodeType === 'file'">{{ scope.row.tableId }}</span>
        </template>
      </el-table-column>
      <el-table-column prop="category" label="分类" width="120">
        <template slot-scope="scope">
          <span v-if="scope.row.nodeType === 'file'">{{ scope.row.category }}</span>
        </template>
      </el-table-column>
      <el-table-column prop="sortOrder" label="排序" width="100"></el-table-column>
      <el-table-column prop="status" label="状态" width="100">
        <template slot-scope="scope">
          <dict-tag :options="dict.type.sys_normal_disable" :value="scope.row.status"/>
        </template>
      </el-table-column>
<!--      <el-table-column label="创建时间" align="center" prop="createTime" width="200">-->
<!--        <template slot-scope="scope">-->
<!--          <span>{{ parseTime(scope.row.createTime) }}</span>-->
<!--        </template>-->
<!--      </el-table-column>-->
      <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="handleEdit(scope.row)"
            v-hasPermi="['generator:structure:edit']"
          >修改</el-button>
          <el-button
            v-if="scope.row.nodeType !== 'file'"
            size="mini"
            type="text"
            icon="el-icon-plus"
            @click="handleAdd(scope.row)"
            v-hasPermi="['generator:structure:add']"
          >新增</el-button>
          <el-button
            v-if="scope.row.parentId != 0"
            size="mini"
            type="text"
            icon="el-icon-delete"
            @click="handleDelete(scope.row)"
            v-hasPermi="['generator:structure:remove']"
          >删除</el-button>
        </template>
      </el-table-column>
    </el-table>

    <!-- 添加或修改节点对话框 -->
    <el-dialog :title="title" :visible.sync="open" width="600px" append-to-body>
      <el-form ref="form" :model="form" :rules="rules" label-width="80px">
        <el-row>
          <el-col :span="24" v-if="form.parentId !== 0">
            <el-form-item label="上级节点" prop="parentId">
              <treeselect v-model="form.parentId" :options="structureOptions" :normalizer="normalizer" placeholder="选择上级节点" />
            </el-form-item>
          </el-col>
        </el-row>
        <el-row>
          <el-col :span="12">
            <el-form-item label="节点名称" prop="nodeName">
              <el-input v-model="form.nodeName" placeholder="请输入节点名称" />
            </el-form-item>
          </el-col>
          <el-col :span="12">
            <el-form-item label="节点类型" prop="nodeType">
              <el-select v-model="form.nodeType" placeholder="请选择节点类型">
                <el-option label="文件夹" value="folder" />
                <el-option label="文件" value="file" />
              </el-select>
            </el-form-item>
          </el-col>
        </el-row>
        <el-row>
          <el-col :span="12">
            <el-form-item label="显示排序" prop="sortOrder">
              <el-input-number v-model="form.sortOrder" controls-position="right" :min="0" />
            </el-form-item>
          </el-col>
          <el-col :span="12">
            <el-form-item label="节点状态">
              <el-radio-group v-model="form.status">
                <el-radio
                  v-for="dict in dict.type.sys_normal_disable"
                  :key="dict.value"
                  :label="dict.value"
                >{{dict.label}}</el-radio>
              </el-radio-group>
            </el-form-item>
          </el-col>
        </el-row>
        <el-row>
          <el-col :span="24">
            <el-form-item label="功能模块" prop="module">
              <el-input v-model="form.module" placeholder="请输入功能模块" />
            </el-form-item>
          </el-col>
        </el-row>
        <el-row v-if="form.nodeType === 'file'">
          <el-col :span="12">
            <el-form-item label="表ID" prop="tableId">
              <el-input v-model="form.tableId" placeholder="请输入表ID" />
            </el-form-item>
          </el-col>
          <el-col :span="12">
            <el-form-item label="分类" prop="category">
              <el-input v-model="form.category" placeholder="请输入分类" />
            </el-form-item>
          </el-col>
        </el-row>
      </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>
  </div>
</template>

<script>
import { listStructure, getStructure, delStructure, addStructure, updateStructure, listStructureExcludeChild, listAllStructure } from '@/api/generator/structure'
import Treeselect from '@riophae/vue-treeselect'
import '@riophae/vue-treeselect/dist/vue-treeselect.css'
import { handleTree } from '@/utils/ruoyi'

export default {
  name: 'Structure',
  dicts: ['sys_normal_disable', 'sys_node_type'],
  components: { Treeselect },
  props: {
    templateId: {
      type: [Number, String],
      default: null
    }
  },
  data() {
    return {
      // 遮罩层
      loading: true,
      // 显示搜索条件
      showSearch: true,
      // 表格树数据
      structureData: [],
      // 节点树选项
      structureOptions: [],
      // 弹出层标题
      title: '',
      // 是否显示弹出层
      open: false,
      // 是否展开，默认全部展开
      isExpandAll: true,
      // 重新渲染表格状态
      refreshTable: true,
      // 查询参数
      queryParams: {
        nodeName: undefined,
        status: undefined,
        templateId: undefined
      },
      // 表单参数
      form: {},
      // 表单校验
      rules: {
        parentId: [
          { required: true, message: '上级节点不能为空', trigger: 'blur' }
        ],
        nodeName: [
          { required: true, message: '节点名称不能为空', trigger: 'blur' }
        ],
        nodeType: [
          { required: true, message: '节点类型不能为空', trigger: 'change' }
        ],
        sortOrder: [
          { required: true, message: '显示排序不能为空', trigger: 'blur' }
        ],
        tableId: [
          { required: true, message: '表ID不能为空', trigger: 'blur' }
        ],
        category: [
          { required: true, message: '分类不能为空', trigger: 'blur' }
        ]
      }
    }
  },
  created() {
    this.queryParams.templateId = this.templateId
    this.getList()
  },
  watch: {
    templateId: {
      handler(newVal) {
        this.queryParams.templateId = newVal
        this.getList()
      },
      immediate: true
    }
  },
  methods: {
    /** 查询节点列表 */
    getList() {
      this.loading = true
      listAllStructure(this.queryParams).then(response => {
        if (response.code === 200 && response.data) {
          console.log('Response:', response)
          this.structureData = handleTree(response.data, 'nodeId', 'parentId', 'children')
          console.log('Tree structure:', JSON.stringify(this.structureData, null, 2))
        } else {
          this.structureData = []
          console.warn('Invalid response format:', response)
        }
        this.loading = false
      }).catch(error => {
        console.error('Failed to fetch structure data:', error)
        this.structureData = []
        this.loading = false
      })
    },
    /** 转换节点数据结构 */
    normalizer(node) {
      if (node.children && !node.children.length) {
        delete node.children
      }
      return {
        id: node.nodeId,
        label: node.nodeName,
        children: node.children
      }
    },
    // 取消按钮
    cancel() {
      this.open = false
      this.reset()
    },
    // 表单重置
    reset() {
      this.form = {
        nodeId: undefined,
        parentId: undefined,
        nodeName: undefined,
        nodeType: undefined,
        module: undefined,
        sortOrder: 0,
        status: '0',
        templateId: this.templateId,
        tableId: undefined,
        category: undefined
      }
      this.resetForm('form')
    },
    /** 搜索按钮操作 */
    handleQuery() {
      this.getList()
    },
    /** 重置按钮操作 */
    resetQuery() {
      this.resetForm('queryForm')
      this.handleQuery()
    },
    /** 新增按钮操作 */
    handleAddRoot() {
      this.reset()
      this.form.parentId = 0
      this.open = true
      this.title = '添加根节点'
      listStructure().then(response => {
        if (response.code === 200 && response.rows) {
          this.structureOptions = handleTree(response.rows, 'nodeId', 'parentId', 'children')
        }
      })
    },
    /** 新增子节点操作 */
    handleAdd(row) {
      this.reset()
      this.form.parentId = row.nodeId
      this.open = true
      this.title = '添加子节点'
      listStructure().then(response => {
        if (response.code === 200 && response.rows) {
          this.structureOptions = handleTree(response.rows, 'nodeId', 'parentId', 'children')
        }
      })
    },
    /** 展开/折叠操作 */
    toggleExpandAll() {
      this.refreshTable = false
      this.isExpandAll = !this.isExpandAll
      this.$nextTick(() => {
        this.refreshTable = true
      })
    },
    /** 修改按钮操作 */
    handleEdit(row) {
      this.reset()
      getStructure(row.nodeId).then(response => {
        this.form = response.data
        this.open = true
        this.title = '修改节点'
        listStructureExcludeChild(row.nodeId).then(response => {
          if (response.code === 200 && response.rows) {
            this.structureOptions = handleTree(response.rows, 'nodeId', 'parentId', 'children')
            if (this.structureOptions.length === 0) {
              const noResultsOptions = { nodeId: this.form.parentId, nodeName: this.form.parentName, children: [] }
              this.structureOptions.push(noResultsOptions)
            }
          }
        })
      })
    },
    /** 提交按钮 */
    submitForm() {
      this.$refs['form'].validate(valid => {
        if (valid) {
          if (this.form.nodeId !== undefined) {
            updateStructure(this.form).then(response => {
              this.$modal.msgSuccess('修改成功')
              this.open = false
              this.getList()
            })
          } else {
            addStructure(this.form).then(response => {
              this.$modal.msgSuccess('新增成功')
              this.open = false
              this.getList()
            })
          }
        }
      })
    },
    /** 删除按钮操作 */
    handleDelete(row) {
      this.$modal.confirm('是否确认删除名称为"' + row.nodeName + '"的节点？').then(() => {
        return delStructure(row.nodeId)
      }).then(() => {
        this.getList()
        this.$modal.msgSuccess('删除成功')
      }).catch(() => {})
    }
  }
}
</script>

<style scoped>
.custom-tree-node {
  flex: 1;
  display: flex;
  align-items: center;
  justify-content: space-between;
  font-size: 14px;
  padding-right: 8px;
}
</style>
