<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>
