<template>
  <a-page-header :title="getPageTitle()" @back="() => $router.go(-1)">
    <template #extra>
      <Extra />
    </template>
    <a-card :bordered="false">
      <!-- 表格 -->
      <ele-pro-table
        ref="tableRef"
        row-key="userId"
        :columns="columns"
        :datasource="datasource"
        v-model:selection="selection"
        :scroll="{ x: 1300 }"
        :where="defaultWhere"
        :customRow="customRow"
        cache-key="proSystemUserTable"
      >
        <template #toolbar>
          <a-space>
            <a-button type="primary" class="ele-btn-icon" @click="openEdit()">
              <template #icon>
                <plus-outlined />
              </template>
              <span>新建</span>
            </a-button>
            <a-button
              type="primary"
              class="ele-btn-icon"
              @click="openEdit(selection[0])"
              :disabled="selection.length === 0"
            >
              <template #icon>
                <EditOutlined />
              </template>
              <span>修改</span>
            </a-button>
            <a-button
              type="primary"
              danger
              @click="resetPsw(selection[0])"
              :disabled="selection.length === 0"
              >重置密码
            </a-button>
            <a-button
              danger
              type="primary"
              class="ele-btn-icon"
              @click="removeBatch"
            >
              <template #icon>
                <delete-outlined />
              </template>
              <span>批量删除</span>
            </a-button>
            <template v-if="hasRole('superAdmin')">
              <a-button type="dashed" class="ele-btn-icon" @click="openImport">
                <template #icon>
                  <upload-outlined />
                </template>
                <span>导入</span>
              </a-button>
              <a-button type="dashed" @click="handleExport">导出xls</a-button>
            </template>
            <a-input-search
              allow-clear
              v-model:value="searchText"
              placeholder="请输入关键词"
              @search="handleSearch"
              @pressEnter="handleSearch"
            />
          </a-space>
        </template>
        <template #bodyCell="{ column, record }">
          <template v-if="column.key === 'avatar'">
            <a-avatar
              :size="30"
              :src="`${record.avatar}`"
              style="margin-right: 4px"
            >
              <template #icon>
                <UserOutlined />
              </template>
            </a-avatar>
          </template>
          <template v-if="column.key === 'nickname'">
            <span>{{ record.nickname }}</span>
          </template>
          <template v-if="column.key === 'mobile'">
            <span v-if="hasRole('superAdmin')">{{ record.phone }}</span>
            <span v-else>{{ record.mobile }}</span>
          </template>
          <template v-if="column.key === 'roles'">
            <a-tag v-for="item in record.roles" :key="item.roleId" color="blue">
              {{ item.roleName }}
            </a-tag>
          </template>
          <template v-if="column.key === 'platform'">
            <WechatOutlined v-if="record.platform === 'MP-WEIXIN'" />
            <Html5Outlined v-if="record.platform === 'H5'" />
            <ChromeOutlined v-if="record.platform === 'WEB'" />
          </template>
          <template v-if="column.key === 'balance'">
            <span class="ele-text-success">
              ￥{{ formatNumber(record.balance) }}
            </span>
          </template>
          <template v-if="column.key === 'expendMoney'">
            <span class="ele-text-warning">
              ￥{{ formatNumber(record.expendMoney) }}
            </span>
          </template>
          <template v-if="column.key === 'isAdmin'">
            <a-tag v-if="record.isAdmin">是</a-tag>
            <a-tag v-else>否</a-tag>
            <!--            <a-switch-->
            <!--              :checked="record.isAdmin == 1"-->
            <!--              @change="updateIsAdmin(record)"-->
            <!--            />-->
          </template>
          <template v-if="column.key === 'action'">
            <div>
              <a @click="openEdit(record)">修改</a>
              <a-divider type="vertical" />
              <!--              <a @click="resetPsw(record)">重置</a>-->
              <!--              <a-divider type="vertical" />-->
              <a-popconfirm
                placement="topRight"
                title="确定要删除此用户吗？"
                @confirm="remove(record)"
              >
                <a class="ele-text-danger">删除</a>
              </a-popconfirm>
            </div>
          </template>
        </template>
      </ele-pro-table>
    </a-card>
    <!-- 编辑弹窗 -->
    <user-edit
      v-model:visible="showEdit"
      :data="current"
      :organization-list="data"
      @done="reload"
    />
    <!-- 导入弹窗 -->
    <user-import v-model:visible="showImport" @done="reload" />
    <!-- 用户详情 -->
    <user-info v-model:visible="showInfo" :data="current" @done="reload" />
  </a-page-header>
</template>

<script lang="ts" setup>
  import { createVNode, ref, computed, reactive } from 'vue';
  import { message, Modal } from 'ant-design-vue/es';
  import {
    PlusOutlined,
    DeleteOutlined,
    UploadOutlined,
    EditOutlined,
    UserOutlined,
    Html5Outlined,
    ChromeOutlined,
    WechatOutlined,
    ExclamationCircleOutlined
  } from '@ant-design/icons-vue';
  import type { EleProTable } from 'ele-admin-pro/es';
  import type {
    DatasourceFunction,
    ColumnItem
  } from 'ele-admin-pro/es/ele-pro-table/types';
  import { messageLoading, formatNumber } from 'ele-admin-pro/es';
  import UserEdit from './components/user-edit.vue';
  import UserImport from './components/user-import.vue';
  import UserInfo from './components/user-info.vue';
  import { toDateString } from 'ele-admin-pro';
  import {
    pageUsers,
    removeUser,
    removeUsers,
    updateUserPassword,
    updateUser,
    listUsers
  } from '@/api/system/user';
  import type { User, UserParam } from '@/api/system/user/model';
  import { toTreeData, uuid } from 'ele-admin-pro';
  import { utils, writeFile } from 'xlsx';
  import dayjs from 'dayjs';
  import { listRoles } from '@/api/system/role';
  import { listOrganizations } from '@/api/system/organization';
  import { Organization } from '@/api/system/organization/model';
  import { hasRole } from '@/utils/permission';
  import { getPageTitle } from '@/utils/common';
  import Extra from './components/Extra.vue';
  import { useUserStore } from '@/store/modules/user';

  // 加载状态
  const loading = ref(true);
  // 树形数据
  const data = ref<Organization[]>([]);
  // 树展开的key
  const expandedRowKeys = ref<number[]>([]);
  // 树选中的key
  const selectedRowKeys = ref<number[]>([]);
  // 表格选中数据
  const selection = ref<User[]>([]);
  // 当前编辑数据
  const current = ref<User | null>(null);
  // 是否显示编辑弹窗
  const showEdit = ref(false);
  // 是否显示用户详情
  const showInfo = ref(false);
  // 是否显示用户导入弹窗
  const showImport = ref(false);
  const userType = ref<number>();
  const searchText = ref('');
  const userStore = useUserStore();
  // 当前用户信息
  const loginUser = computed(() => userStore.info ?? {});
  // 加载角色
  const roles = ref<any[]>([]);
  const filters = () => {
    listRoles().then((result) => {
      result.map((d) => {
        roles.value.push({
          text: d.roleName,
          value: d.roleId
        });
      });
    });
  };
  filters();
  // 加载机构
  listOrganizations()
    .then((list) => {
      loading.value = false;
      const eks: number[] = [];
      list.forEach((d) => {
        d.key = d.organizationId;
        d.value = d.organizationId;
        d.title = d.organizationName;
        if (typeof d.key === 'number') {
          eks.push(d.key);
        }
      });
      expandedRowKeys.value = eks;
      data.value = toTreeData({
        data: list,
        idField: 'organizationId',
        parentIdField: 'parentId'
      });
      if (list.length) {
        if (typeof list[0].key === 'number') {
          selectedRowKeys.value = [list[0].key];
        }
        // current.value = list[0];
      } else {
        selectedRowKeys.value = [];
        // current.value = null;
      }
    })
    .catch((e) => {
      loading.value = false;
      message.error(e.message);
    });

  // 表格实例
  const tableRef = ref<InstanceType<typeof EleProTable> | null>(null);
  // 表格列配置
  const columns = ref<ColumnItem[]>([
    {
      title: 'ID',
      dataIndex: 'userId',
      width: 90,
      showSorterTooltip: false
    },
    {
      title: '头像',
      key: 'avatar',
      dataIndex: 'avatar',
      align: 'center',
      width: 90
    },
    {
      title: '手机号码',
      dataIndex: 'mobile',
      align: 'center',
      key: 'mobile',
      showSorterTooltip: false
    },
    {
      title: '真实姓名',
      dataIndex: 'realName',
      align: 'center',
      showSorterTooltip: false
    },
    {
      title: '昵称',
      key: 'nickname',
      align: 'center',
      dataIndex: 'nickname'
    },
    {
      title: '性别',
      dataIndex: 'sex',
      align: 'center',
      showSorterTooltip: false,
      customRender: ({ text }) => ['', '男', '女'][text]
    },
    {
      title: '邮箱',
      dataIndex: 'email',
      align: 'center',
      hideInTable: true,
      showSorterTooltip: false
    },
    {
      title: '可用余额',
      dataIndex: 'balance',
      align: 'center'
    },
    {
      title: '可用积分',
      dataIndex: 'points',
      align: 'center',
      sorter: true
    },
    {
      title: '实际消费金额',
      dataIndex: 'expendMoney',
      key: 'expendMoney',
      sorter: true,
      align: 'center',
      hideInTable: true,
      showSorterTooltip: false
    },
    {
      title: '注册来源',
      key: 'platform',
      align: 'center',
      dataIndex: 'platform',
      width: 120
    },
    {
      title: '证件号码',
      dataIndex: 'idCard',
      align: 'center',
      hideInTable: true
    },
    {
      title: '出生日期',
      dataIndex: 'birthday',
      key: 'birthday',
      hideInTable: true
    },
    {
      title: '省份',
      dataIndex: 'province',
      key: 'province',
      hideInTable: true
    },
    {
      title: '城市',
      dataIndex: 'city',
      key: 'city',
      hideInTable: true,
      showSorterTooltip: false
    },
    {
      title: '地区',
      dataIndex: 'region',
      key: 'region',
      hideInTable: true,
      showSorterTooltip: false
    },
    {
      title: '邮箱认证',
      dataIndex: 'emailVerified',
      hideInTable: true,
      showSorterTooltip: false,
      customRender: ({ text }) => ['未认证', '已认证'][text]
    },
    {
      title: '实名认证',
      dataIndex: 'certification',
      sorter: true,
      hideInTable: true,
      customRender: ({ text }) => ['未认证', '已认证'][text]
    },
    {
      title: '角色',
      dataIndex: 'roles',
      key: 'roles',
      align: 'center'
    },
    {
      title: '管理员',
      key: 'isAdmin',
      dataIndex: 'isAdmin',
      sorter: true,
      hideInTable: true,
      showSorterTooltip: false,
      align: 'center'
    },
    {
      title: '注册时间',
      dataIndex: 'createTime',
      sorter: true,
      showSorterTooltip: false,
      ellipsis: true,
      customRender: ({ text }) => toDateString(text, 'yyyy-MM-dd HH:mm:ss')
    },
    {
      title: '操作',
      key: 'action',
      width: 120,
      fixed: 'right',
      align: 'center'
    }
  ]);

  // 默认搜索条件
  const defaultWhere = reactive({
    username: '',
    nickname: ''
  });

  // 表格数据源
  const datasource: DatasourceFunction = ({
    page,
    limit,
    where,
    orders,
    filters
  }) => {
    where = {};
    where.roleId = filters.roles;
    where.keywords = searchText.value;
    return pageUsers({ page, limit, ...where, ...orders });
  };

  /* 搜索 */
  const reload = (where?: UserParam) => {
    selection.value = [];
    tableRef?.value?.reload({ where });
  };

  /* 打开编辑弹窗 */
  const openEdit = (row?: User) => {
    current.value = row ?? null;
    showEdit.value = true;
  };

  /* 打开用户详情弹窗 */
  const openInfo = (row?: User) => {
    current.value = row ?? null;
    showInfo.value = true;
  };

  /* 打开编辑弹窗 */
  const openImport = () => {
    showImport.value = true;
  };

  const handleTabs = (e) => {
    userType.value = Number(e.target.value);
    reload();
  };

  /* 删除单个 */
  const remove = (row: User) => {
    const hide = messageLoading('请求中..', 0);
    removeUser(row.userId)
      .then((msg) => {
        hide();
        message.success(msg);
        reload();
      })
      .catch((e) => {
        hide();
        message.error(e.message);
      });
  };

  /* 批量删除 */
  const removeBatch = () => {
    if (!selection.value.length) {
      message.error('请至少选择一条数据');
      return;
    }
    Modal.confirm({
      title: '提示',
      content: '确定要删除选中的用户吗?',
      icon: createVNode(ExclamationCircleOutlined),
      maskClosable: true,
      onOk: () => {
        const hide = messageLoading('请求中..', 0);
        removeUsers(selection.value.map((d) => d.userId))
          .then((msg) => {
            hide();
            message.success(msg);
            reload();
          })
          .catch((e) => {
            hide();
            message.error(e.message);
          });
      }
    });
  };

  /* 重置用户密码 */
  const resetPsw = (row: User) => {
    Modal.confirm({
      title: '提示',
      content: '确定要重置此用户的密码吗?',
      icon: createVNode(ExclamationCircleOutlined),
      maskClosable: true,
      onOk: () => {
        const hide = message.loading('请求中..', 0);
        const password = uuid(8);
        updateUserPassword(row.userId, password)
          .then((msg) => {
            hide();
            message.success(msg + '，新密码：' + password);
          })
          .catch((e) => {
            hide();
            message.error(e.message);
          });
      }
    });
  };

  /* 修改用户状态 */
  const updateIsAdmin = (row: User) => {
    row.isAdmin = !row.isAdmin;
    updateUser(row)
      .then((msg) => {
        message.success(msg);
      })
      .catch((e) => {
        message.error(e.message);
      });
  };

  /* 搜索处理 */
  const handleSearch = () => {
    reload();
  };

  /* 导出功能 */
  const handleExport = async () => {
    if (loading.value) {
      return;
    }

    loading.value = true;
    message.loading('正在准备导出数据...', 0);

    try {
      const array: (string | number)[][] = [
        [
          '用户ID',
          '头像',
          '手机号码',
          '真实姓名',
          '昵称',
          '性别',
          '邮箱',
          '可用余额',
          '可用积分',
          '注册来源',
          '注册时间'
        ]
      ];

      // 构建查询参数
      const queryParams = {
        keywords: searchText.value
      };

      // 按搜索结果导出
      const list = await listUsers(queryParams);

      if (!list || list.length === 0) {
        message.warning('没有数据可以导出');
        loading.value = false;
        return;
      }

      list.forEach((user: User) => {
        array.push([
          `${user.userId || ''}`,
          `${user.avatar ? '有头像' : '无头像'}`,
          `${user.phone || ''}`,
          `${user.realName || ''}`,
          `${user.nickname || ''}`,
          `${user.sex === '1' ? '男' : user.sex === '2' ? '女' : ''}`,
          `${user.email || ''}`,
          `${user.balance || '0'}`,
          `${user.points || '0'}`,
          `${user.platform || ''}`,
          `${user.createTime || ''}`
        ]);
      });

      const sheetName = `导出用户列表${dayjs(new Date()).format('YYYYMMDD')}`;
      const workbook = {
        SheetNames: [sheetName],
        Sheets: {}
      };
      const sheet = utils.aoa_to_sheet(array);
      workbook.Sheets[sheetName] = sheet;

      // 设置列宽
      sheet['!cols'] = [
        { wch: 10 }, // 用户ID
        { wch: 10 }, // 头像
        { wch: 15 }, // 手机号码
        { wch: 15 }, // 真实姓名
        { wch: 15 }, // 昵称
        { wch: 8 }, // 性别
        { wch: 20 }, // 邮箱
        { wch: 12 }, // 可用余额
        { wch: 12 }, // 可用积分
        { wch: 12 }, // 注册来源
        { wch: 20 } // 注册时间
      ];

      message.destroy();
      message.loading('正在生成Excel文件...', 0);

      setTimeout(() => {
        writeFile(workbook, `${sheetName}.xlsx`);
        loading.value = false;
        message.destroy();
        message.success(`成功导出 ${list.length} 条记录`);
      }, 1000);
    } catch (error: any) {
      loading.value = false;
      message.destroy();
      message.error(error.message || '导出失败，请重试');
    }
  };

  /* 自定义行属性 */
  const customRow = (record: User) => {
    return {
      // 行点击事件
      onClick: () => {
        // console.log(record);
      },
      // 行双击事件
      onDblclick: () => {
        openEdit(record);
      }
    };
  };
</script>

<script lang="ts">
  export default {
    name: 'SystemUser'
  };
</script>

<style lang="less" scoped>
  .user-box {
    display: flex;
    align-items: center;

    .user-info {
      display: flex;
      flex-direction: column;
      align-items: start;
    }
  }
</style>
