<template>
  <a-page-header :title="getPageTitle()" @back="() => $router.go(-1)">
    <a-card :bordered="false">
      <!-- 表格 -->
      <ele-pro-table
        ref="tableRef"
        row-key="userId"
        :columns="columns"
        :datasource="datasource"
        class="sys-org-table"
        :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 class="ele-btn-icon" @click="openImport()">
              <template #icon>
                <cloud-upload-outlined />
              </template>
              <span>导入</span>
            </a-button>
            <a-button
              class="ele-btn-icon"
              @click="exportData()"
              :loading="exportLoading"
            >
              <template #icon>
                <download-outlined />
              </template>
              <span>导出</span>
            </a-button>
            <a-input-search
              allow-clear
              v-model:value="searchText"
              placeholder="请输入关键词"
              @search="reload"
              @pressEnter="reload"
            />
          </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'">
            <div>{{ record.nickname }}</div>
            <div class="text-gray-400">{{ record.realName }}</div>
          </template>
          <template v-if="column.key === 'phone'">
            <span v-if="hasRole('superAdmin')">{{ record.phone }}</span>
            <span v-else>{{ record.phone }}</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-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-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" />
  </a-page-header>
</template>

<script lang="ts" setup>
  import { createVNode, ref, reactive, watch } from 'vue';
  import { message, Modal } from 'ant-design-vue/es';
  import {
    PlusOutlined,
    UserOutlined,
    Html5Outlined,
    ChromeOutlined,
    WechatOutlined,
    CloudUploadOutlined,
    DownloadOutlined,
    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 { toDateString } from 'ele-admin-pro';
  import { utils, writeFile } from 'xlsx';
  import dayjs from 'dayjs';
  import {
    pageShopUser,
    removeShopUser,
    removeBatchShopUser,
    updateShopUser,
    listShopUser
  } from '@/api/shop/shopUser';
  import type { ShopUser, ShopUserParam } from '@/api/shop/shopUser/model';
  import { toTreeData, uuid } from 'ele-admin-pro';
  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 router from '@/router';
  import { getTenantId } from '@/utils/domain';

  // 加载状态
  const loading = ref(true);
  // 树形数据
  const data = ref<Organization[]>([]);
  // 树展开的key
  const expandedRowKeys = ref<number[]>([]);
  // 树选中的key
  const selectedRowKeys = ref<number[]>([]);
  // 表格选中数据
  const selection = ref<ShopUser[]>([]);
  // 当前编辑数据
  const current = ref<ShopUser | null>(null);
  // 是否显示编辑弹窗
  const showEdit = ref(false);
  // 是否显示用户详情
  const showInfo = ref(false);
  // 是否显示用户导入弹窗
  const showImport = ref(false);
  // 导出加载状态
  const exportLoading = ref(false);
  const userType = ref<number>();
  const searchText = ref('');

  // 加载角色
  const roles = ref<any[]>([]);
  // 加载机构
  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: '昵称/姓名',
      dataIndex: 'nickname',
      key: 'nickname',
      align: 'center',
      showSorterTooltip: false
    },
    {
      title: '手机号码',
      dataIndex: 'phone',
      align: 'center',
      showSorterTooltip: false
    },
    {
      title: '积分',
      dataIndex: 'points',
      align: 'center',
      width: 100
    },
    {
      title: '余额',
      dataIndex: 'balance',
      align: 'center',
      width: 100
    },
    // {
    //   title: '角色',
    //   dataIndex: 'roles',
    //   key: 'roles',
    //   align: 'center'
    // },
    {
      title: '备注',
      dataIndex: 'comments',
      align: 'center'
    },
    {
      title: '状态',
      dataIndex: 'status',
      align: 'center',
      sorter: true,
      customRender: ({ text }) => {
        return text === 1
          ? createVNode(
              'span',
              {
                class: 'text-red-400'
              },
              '封号'
            )
          : createVNode(
              'span',
              {
                class: 'text-gray-400'
              },
              '正常'
            );
      }
    },
    {
      title: '创建时间',
      dataIndex: 'createTime',
      sorter: true,
      align: 'center',
      showSorterTooltip: false,
      ellipsis: true,
      customRender: ({ text }) => toDateString(text, 'yyyy-MM-dd HH:mm:ss')
    },
    {
      title: '操作',
      key: 'action',
      width: 180,
      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 pageShopUser({ page, limit, ...where, ...orders });
  };

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

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

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

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

  /* 导出数据 */
  const exportData = async () => {
    exportLoading.value = true;

    try {
      // 定义表头
      const array: (string | number)[][] = [
        [
          '用户ID',
          '账号',
          '昵称',
          '真实姓名',
          '手机号',
          '邮箱',
          '性别',
          '状态',
          '注册时间'
        ]
      ];

      // 构建查询参数，使用当前搜索条件
      const params = {
        keywords: searchText.value,
        isAdmin: 0
      };

      // 获取用户列表数据
      const list = await listShopUser(params);

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

      // 将数据转换为Excel行
      list.forEach((user: ShopUser) => {
        array.push([
          `${user.userId || ''}`,
          `${user.username || ''}`,
          `${user.nickname || ''}`,
          `${user.realName || ''}`,
          `${user.phone || ''}`,
          `${user.email || ''}`,
          `${user.sex == 1 ? '男' : '女'}`,
          `${user.status === 0 ? '正常' : '冻结'}`,
          `${user.createTime || ''}`
        ]);
      });

      // 生成Excel文件
      const sheetName = `shop_user_${getTenantId()}_${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: 15 }, // 账号
        { wch: 12 }, // 昵称
        { wch: 12 }, // 真实姓名
        { wch: 15 }, // 手机号
        { wch: 20 }, // 邮箱
        { wch: 8 }, // 性别
        { wch: 15 }, // 所属部门
        { wch: 20 }, // 角色
        { wch: 8 }, // 状态
        { wch: 20 } // 注册时间
      ];

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

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

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

  /* 删除单个 */
  const remove = (row: ShopUser) => {
    const hide = messageLoading('请求中..', 0);
    removeShopUser(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);
        removeShopUser(selection.value.map((d) => d.userId))
          .then((msg) => {
            hide();
            message.success(msg);
            reload();
          })
          .catch((e) => {
            hide();
            message.error(e.message);
          });
      }
    });
  };

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

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

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

  const query = async () => {
    const info = await listRoles({});
    if (info) {
      roles.value = info;
    }
  };

  watch(
    () => router.currentRoute.value.query,
    () => {
      query();
    },
    { immediate: true }
  );
</script>

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

<style lang="less" scoped>
  .sys-org-table {
    :deep(.ant-table) {
      .ant-table-thead > tr > th {
        background: #fafafa;
        font-weight: 600;
        color: #262626;
        border-bottom: 2px solid #f0f0f0;
      }

      .ant-table-tbody > tr > td {
        padding: 12px 8px;
        border-bottom: 1px solid #f5f5f5;
      }

      .ant-table-tbody > tr:hover > td {
        background: #f8f9ff;
      }

      .ant-tag {
        margin: 0;
        border-radius: 4px;
        font-size: 12px;
        padding: 2px 8px;
      }
    }
  }

  .ele-text-primary {
    color: #1890ff;

    &:hover {
      color: #40a9ff;
    }
  }

  .ele-text-danger {
    color: #ff4d4f;

    &:hover {
      color: #ff7875;
    }
  }
</style>
