import { useState } from 'react'
import { Button, Card, Form, Input, InputNumber, Modal, Popconfirm, Select, Space, Switch, Table, Tag, Typography, message } from 'antd'
import type { ColumnsType } from 'antd/es/table'
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query'
import { useTranslation } from 'react-i18next'

import apiClient, { isEnvelope } from '../../api/client'

interface IPAddress {
  id: number
  interface_id: number
  address: string
  prefix: number
  is_vip: boolean
  vip_priority?: number | null
  description?: string | null
  active: boolean
  created_at: string
  updated_at: string
}

interface SystemInterface {
  ifname: string
  flags?: string[]
  link_type?: string
  addr_info?: Array<{ family: 'inet' | 'inet6'; local: string; prefixlen: number }>
}

interface SystemAddress {
  ifname: string
  family: 'inet' | 'inet6'
  address: string
  prefix: number
}

interface IPFormValues {
  interface_id: number
  address: string
  prefix: number
  is_vip: boolean
  vip_priority?: number
  description?: string
  active: boolean
}

interface NetworkInterface { id: number; name: string; role: string }

async function listIPs(): Promise<IPAddress[]> {
  const r = await apiClient.get('/ip-addresses')
  if (!isEnvelope(r.data)) return []
  return (r.data.data as { ip_addresses?: IPAddress[] }).ip_addresses ?? []
}
async function listIfaces(): Promise<NetworkInterface[]> {
  const r = await apiClient.get('/network-interfaces')
  if (!isEnvelope(r.data)) return []
  return (r.data.data as { interfaces?: NetworkInterface[] }).interfaces ?? []
}

// Flatten /system/interfaces into one row per (ifname, address) so
// the operator sees every kernel-side IP at a glance — including
// addresses that EdgeGuard hasn't taken under management yet.
async function listSystemAddresses(): Promise<SystemAddress[]> {
  const r = await apiClient.get('/system/interfaces')
  if (!isEnvelope(r.data)) return []
  const ifs = (r.data.data as { interfaces?: SystemInterface[] }).interfaces ?? []
  const out: SystemAddress[] = []
  for (const i of ifs) {
    for (const a of i.addr_info ?? []) {
      out.push({
        ifname: i.ifname,
        family: a.family,
        address: a.local,
        prefix: a.prefixlen,
      })
    }
  }
  return out
}

export default function IPAddressesPage() {
  const { t } = useTranslation()
  const qc = useQueryClient()

  const { data: ips, isLoading } = useQuery({ queryKey: ['ip-addresses'], queryFn: listIPs })
  const { data: ifs } = useQuery({ queryKey: ['network-interfaces'], queryFn: listIfaces })
  const { data: sysAddrs } = useQuery({
    queryKey: ['system', 'addresses'],
    queryFn: listSystemAddresses,
    refetchInterval: 60_000,
  })

  const ifaceLabel = (id: number) => {
    const i = ifs?.find((x) => x.id === id)
    return i ? `${i.name} (${i.role})` : `#${id}`
  }

  const [editing, setEditing] = useState<IPAddress | null>(null)
  const [creating, setCreating] = useState(false)
  const [form] = Form.useForm<IPFormValues>()

  const create = useMutation({
    mutationFn: async (v: IPFormValues) => { await apiClient.post('/ip-addresses', v) },
    onSuccess: () => {
      message.success(t('common.save'))
      setCreating(false); form.resetFields()
      void qc.invalidateQueries({ queryKey: ['ip-addresses'] })
    },
  })
  const update = useMutation({
    mutationFn: async ({ id, v }: { id: number; v: IPFormValues }) => { await apiClient.put(`/ip-addresses/${id}`, v) },
    onSuccess: () => {
      message.success(t('common.save'))
      setEditing(null); form.resetFields()
      void qc.invalidateQueries({ queryKey: ['ip-addresses'] })
    },
  })
  const del = useMutation({
    mutationFn: async (id: number) => { await apiClient.delete(`/ip-addresses/${id}`) },
    onSuccess: () => { void qc.invalidateQueries({ queryKey: ['ip-addresses'] }) },
  })

  const columns: ColumnsType<IPAddress> = [
    { title: t('ips.interface'), dataIndex: 'interface_id', key: 'iface', render: (id: number) => ifaceLabel(id) },
    {
      title: t('ips.address'), key: 'addr',
      render: (_, row) => <code>{row.address}/{row.prefix}</code>,
    },
    {
      title: t('ips.vip'), key: 'vip',
      render: (_, row) => row.is_vip
        ? <Tag color="gold">VIP{row.vip_priority != null ? ` · prio ${row.vip_priority}` : ''}</Tag>
        : '—',
    },
    { title: t('ips.active'), dataIndex: 'active', key: 'active', render: (v: boolean) => v ? t('common.yes') : t('common.no') },
    { title: t('ips.description'), dataIndex: 'description', key: 'desc', render: (v?: string) => v ?? '—' },
    {
      title: t('ips.actions'), key: 'actions',
      render: (_, row) => (
        <Space>
          <Button size="small" onClick={() => {
            setEditing(row)
            form.setFieldsValue({
              interface_id: row.interface_id,
              address: row.address, prefix: row.prefix,
              is_vip: row.is_vip, vip_priority: row.vip_priority ?? undefined,
              description: row.description ?? undefined,
              active: row.active,
            })
          }}>{t('common.edit')}</Button>
          <Popconfirm
            title={t('ips.deleteConfirm', { addr: row.address })}
            onConfirm={() => del.mutate(row.id)}
          >
            <Button size="small" danger>{t('common.delete')}</Button>
          </Popconfirm>
        </Space>
      ),
    },
  ]

  return (
    <div>
      <Typography.Title level={3}>{t('ips.title')}</Typography.Title>
      <Typography.Paragraph type="secondary">{t('ips.intro')}</Typography.Paragraph>

      <Card title={t('ips.systemDiscovered')} size="small" style={{ marginBottom: 16 }}>
        {(sysAddrs ?? []).length === 0
          ? <Typography.Text type="secondary">—</Typography.Text>
          : (
            <Table
              size="small"
              rowKey={(r) => `${r.ifname}-${r.address}`}
              dataSource={sysAddrs ?? []}
              pagination={false}
              columns={[
                { title: t('ips.interface'), dataIndex: 'ifname', key: 'ifname', render: (s: string) => <code>{s}</code> },
                { title: t('ips.address'), key: 'addr', render: (_, row: SystemAddress) => <code>{row.address}/{row.prefix}</code> },
                { title: t('ips.family'), dataIndex: 'family', key: 'family', render: (f: string) => <Tag>{f === 'inet' ? 'IPv4' : 'IPv6'}</Tag> },
              ]}
            />
          )
        }
      </Card>

      <Typography.Title level={5} style={{ marginTop: 8 }}>{t('ips.managedTitle')}</Typography.Title>
      <Button type="primary" style={{ marginBottom: 16 }} onClick={() => {
        setCreating(true); form.resetFields()
        form.setFieldsValue({ prefix: 24, is_vip: false, active: true })
      }}>
        {t('ips.addAddress')}
      </Button>
      <Table rowKey="id" loading={isLoading} dataSource={ips ?? []} columns={columns} pagination={false} />
      <Modal
        title={editing ? t('ips.editAddress') : t('ips.addAddress')}
        open={editing !== null || creating}
        onCancel={() => { setEditing(null); setCreating(false) }}
        onOk={() => { void form.submit() }}
        confirmLoading={create.isPending || update.isPending}
      >
        <Form
          form={form}
          layout="vertical"
          onFinish={(v) => {
            if (editing) update.mutate({ id: editing.id, v })
            else create.mutate(v)
          }}
        >
          <Form.Item label={t('ips.interface')} name="interface_id" rules={[{ required: true }]}>
            <Select
              options={(ifs ?? []).map((i) => ({ value: i.id, label: ifaceLabel(i.id) }))}
              placeholder={t('ips.selectInterface')}
            />
          </Form.Item>
          <Form.Item label={t('ips.address')} name="address" rules={[{ required: true }]}>
            <Input placeholder="192.0.2.10 oder fd00::1" />
          </Form.Item>
          <Form.Item label={t('ips.prefix')} name="prefix" rules={[{ required: true }]}>
            <InputNumber min={0} max={128} style={{ width: '100%' }} />
          </Form.Item>
          <Form.Item label={t('ips.vipFlag')} name="is_vip" valuePropName="checked">
            <Switch />
          </Form.Item>
          <Form.Item noStyle shouldUpdate={(p, c) => p.is_vip !== c.is_vip}>
            {({ getFieldValue }) => getFieldValue('is_vip') ? (
              <Form.Item label={t('ips.vipPriority')} name="vip_priority">
                <InputNumber min={0} max={255} style={{ width: '100%' }} placeholder="100" />
              </Form.Item>
            ) : null}
          </Form.Item>
          <Form.Item label={t('ips.description')} name="description">
            <Input />
          </Form.Item>
          <Form.Item label={t('ips.active')} name="active" valuePropName="checked">
            <Switch />
          </Form.Item>
        </Form>
      </Modal>
    </div>
  )
}