MemoryBear/web/src/views/ApplicationManagement/components/ApplicationModal.tsx

/*
 * @Author: ZhaoYing 
 * @Date: 2026-02-03 16:34:09 
 * @Last Modified by: ZhaoYing
 * @Last Modified time: 2026-02-03 16:35:30
 */
/**
 * Application Modal
 * Modal for creating and editing applications
 * Supports three application types: agent, multi_agent, and workflow
 */

import { forwardRef, useImperativeHandle, useState } from 'react';
import { Form, Input } from 'antd';
import { useTranslation } from 'react-i18next';

import RadioGroupCard from '@/components/RadioGroupCard'
import AgentIcon from '@/assets/images/application/agent.svg'
import ClusterIcon from '@/assets/images/application/cluster.svg'
import WorkflowIcon from '@/assets/images/application/workflow.svg'
import type { ApplicationModalData, ApplicationModalRef, Application } from '../types'
import RbModal from '@/components/RbModal'
import { addApplication, updateApplication } from '@/api/application'

const FormItem = Form.Item;

/**
 * Component props
 */
interface ApplicationModalProps {
  /** Callback to refresh application list */
  refresh: () => void;
}

/**
 * Supported application types
 */
const types = [
  'agent',
  'multi_agent',
  'workflow'
]
/**
 * Application type icon mapping
 */
const typeIcons: Record<string, string> = {
  agent: AgentIcon,
  multi_agent: ClusterIcon,
  workflow: WorkflowIcon
}

/**
 * Modal for creating and editing applications
 */
const ApplicationModal = forwardRef<ApplicationModalRef, ApplicationModalProps>(({
  refresh
}, ref) => {
  const { t } = useTranslation();
  const [visible, setVisible] = useState(false);
  const [form] = Form.useForm<ApplicationModalData>();
  const [loading, setLoading] = useState(false)
  const [editVo, setEditVo] = useState<Application | null>(null)

  const values = Form.useWatch([], form);

  /** Close modal and reset form */
  const handleClose = () => {
    setVisible(false);
    form.resetFields();
    setLoading(false)
    setEditVo(null)
  };

  /** Open modal with optional application data for editing */
  const handleOpen = (application?: Application) => {
    if (application) {
      setEditVo(application || null)
      form.setFieldsValue({
        name: application.name,
        type: application.type,
        description: application.description,
      })
    } else {
      form.resetFields();
    }
    setVisible(true);
  };
  /** Save application (create or update) */
  const handleSave = () => {
    form
      .validateFields()
      .then(() => {
        setLoading(true)

        const response = editVo?.id ? updateApplication(editVo.id, {
          ...editVo,
          ...values,
        }) : addApplication(values)
        response.then(() => {
          refresh()
          handleClose()
        })
        .finally(() => {
          setLoading(false)
        });
      })
      .catch((err) => {
        console.log('err', err)
      });
  }

  /** Expose methods to parent component */
  useImperativeHandle(ref, () => ({
    handleOpen,
    handleClose
  }));

  return (
    <RbModal
      title={t(`application.${editVo?.id ? 'editApplication' : 'createApplication'}`)}
      open={visible}
      onCancel={handleClose}
      okText={t('common.save')}
      onOk={handleSave}
      confirmLoading={loading}
    >
      <Form
        form={form}
        layout="vertical"
      >
        <FormItem
          name="name"
          label={t('application.applicationName')}
          rules={[{ required: true, message: t('common.pleaseEnter') }]}
        >
          <Input placeholder={t('common.enter')} />
        </FormItem>
        <FormItem
          name="description"
          label={t('application.description')}
        >
          <Input.TextArea placeholder={t('common.enter')} />
        </FormItem>
        
        <FormItem
          name="type"
          label={t('application.applicationType')}
          rules={[{ required: true, message: t('common.pleaseSelect') }]}
        >
          <RadioGroupCard
            options={types.map((type) => ({
              value: type,
              label: t(`application.${type}`),
              labelDesc: t(`application.${type}Desc`),
              icon: typeIcons[type],
            }))}
          />
        </FormItem>
      </Form>
    </RbModal>
  );
});

export default ApplicationModal;