import { useEffect, useRef, useState } from 'react';
import { Drawer, Form, Input, Select, Radio, TreeSelect, Button, Space, Card } from 'antd';
import type { DefaultOptionType } from 'antd/es/select';
import { getUser, createUser, updateUser } from '@/api/system/user';
import { getDeptList } from '@/api/system/dept';
import type { SystemDept } from '@/api/system/dept';
import { getSimplePostList } from '@/api/system/post';
import type { SystemPost } from '@/api/system/post';
import type { SystemUser } from '@/api/system/user';
import { useTranslation } from 'react-i18next';
import { App } from 'antd';

const DEFAULT_FORM = { status: 0 };

/** Build tree-select data from flat dept list */
function buildDeptTreeSelect(list: SystemDept[]): DefaultOptionType[] {
  const map = new Map<number, DefaultOptionType>();
  const roots: DefaultOptionType[] = [];

  for (const item of list) {
    map.set(item.id!, {
      value: item.id,
      title: item.name,
      children: [],
    });
  }

  for (const item of list) {
    const node = map.get(item.id!)!;
    if (item.parentId === 0 || !map.has(item.parentId)) {
      roots.push(node);
    } else {
      map.get(item.parentId)!.children.push(node);
    }
  }

  return roots;
}

interface UserDrawerProps {
  open: boolean;
  id: number | null;
  onSuccess: () => void;
  onClose: () => void;
}

const UserDrawer: React.FC<UserDrawerProps> = ({ open, id, onSuccess, onClose }) => {
  const { t } = useTranslation('system/user');
  const { t: tc } = useTranslation('common');
  const { message } = App.useApp();
  const [form] = Form.useForm();
  const [confirmLoading, setConfirmLoading] = useState(false);
  const [deptTree, setDeptTree] = useState<DefaultOptionType[]>([]);
  const [postList, setPostList] = useState<SystemPost[]>([]);
  // 保留服务端原始字段，避免编辑时丢掉未在表单里出现的字段
  const originData = useRef<Partial<SystemUser>>({});
  const isEdit = id != null;

  useEffect(() => {
    if (open) {
      form.resetFields();
      originData.current = {};
      // Load dept tree and post list
      getDeptList().then(({ data }) => setDeptTree(buildDeptTreeSelect(data))).catch(() => { });
      getSimplePostList().then(({ data }) => setPostList(data)).catch(() => { });
      // Load user data in edit mode
      if (id) {
        getUser(id).then(({ data }) => {
          originData.current = { ...data };
          form.setFieldsValue(data);
        }).catch(() => { });
      } else {
        form.setFieldsValue(DEFAULT_FORM);
      }
    }
  }, [open, id, form]);

  const handleOk = async () => {
    let values: SystemUser;
    try { values = await form.validateFields(); } catch { return; }
    setConfirmLoading(true);
    try {
      const payload = { ...originData.current, ...values, ...(id != null ? { id } : {}) } as SystemUser;
      if (isEdit) {
        await updateUser(payload);
        message.success(tc('success'));
      } else {
        await createUser(payload);
        message.success(tc('success'));
      }
      onSuccess();
    } finally {
      setConfirmLoading(false);
    }
  };

  return (
    <Drawer
      title={isEdit ? t('editTitle') : t('addTitle')}
      open={open}
      onClose={onClose}
      size={'15%'}
      closable={{ placement: 'end' }}
      destroyOnHidden={true}
      mask={{ closable: false }}
      styles={{ body: { background: '#f5f5f5' } }}
      footer={
        <Space style={{ float: 'right' }}>
          <Button onClick={onClose}>{tc('cancel')}</Button>
          <Button type="primary" loading={confirmLoading} onClick={handleOk}>
            {tc('confirm')}
          </Button>
        </Space>
      }
    >
      <Card style={{ border: 'none' }}>
        <Form form={form} layout="vertical" initialValues={DEFAULT_FORM} autoComplete="off">
          <Form.Item
            name="nickname"
            label={t('nickname')}
            rules={[{ required: true, message: t('nicknamePlaceholder') }]}
          >
            <Input placeholder={t('nicknamePlaceholder')} />
          </Form.Item>

          {!id && (
            <Form.Item
              name="username"
              label={t('username')}
              rules={[{ required: true, message: t('usernamePlaceholder') }]}
            >
              <Input placeholder={t('usernamePlaceholder')} />
            </Form.Item>
          )}

          {!id && (
            <Form.Item
              name="password"
              label={t('password')}
              rules={[{ required: true, message: t('passwordPlaceholder') }]}
            >
              <Input.Password placeholder={t('passwordPlaceholder')} />
            </Form.Item>
          )}

          <Form.Item name="mobile" label={t('mobile')}>
            <Input placeholder={t('mobilePlaceholder')} />
          </Form.Item>

          <Form.Item name="email" label={t('email')}>
            <Input placeholder={t('emailPlaceholder')} type="email" />
          </Form.Item>

          <Form.Item name="sex" label={t('sex')}>
            <Select
              placeholder={t('sexPlaceholder')}
              options={[
                { value: 1, label: t('male') },
                { value: 2, label: t('female') },
              ]}
            />
          </Form.Item>

          <Form.Item name="deptId" label={t('dept')}>
            <TreeSelect
              placeholder={t('deptPlaceholder')}
              treeData={deptTree}
              treeDefaultExpandAll
              allowClear
            />
          </Form.Item>

          <Form.Item name="postIds" label={t('post')}>
            <Select
              mode="multiple"
              placeholder={t('postPlaceholder')}
              options={postList.map((p) => ({ value: p.id, label: p.name }))}
            />
          </Form.Item>

          <Form.Item name="status" label={t('status')}>
            <Radio.Group>
              <Radio value={0}>{tc('enable')}</Radio>
              <Radio value={1}>{tc('disable')}</Radio>
            </Radio.Group>
          </Form.Item>

          <Form.Item name="remark" label={t('remark')}>
            <Input.TextArea rows={3} placeholder={t('remarkPlaceholder')} />
          </Form.Item>
        </Form>
      </Card>
    </Drawer >
  );
};

export default UserDrawer;