jan/web/app/_components/ModelActionButton/index.tsx

import React from 'react'
import { Button } from '@uikit'
import ModelActionMenu from '../ModelActionMenu'

export enum ModelActionType {
  Start = 'Start',
  Stop = 'Stop',
}

type ModelActionStyle = {
  title: string
}

const modelActionMapper: Record<ModelActionType, ModelActionStyle> = {
  [ModelActionType.Start]: {
    title: 'Start',
  },
  [ModelActionType.Stop]: {
    title: 'Stop',
  },
}

type Props = {
  type: ModelActionType
  onActionClick: (type: ModelActionType) => void
  onDeleteClick: () => void
}

const ModelActionButton: React.FC<Props> = ({
  type,
  onActionClick,
  onDeleteClick,
}) => {
  const styles = modelActionMapper[type]
  const onClick = () => {
    onActionClick(type)
  }

  return (
    <td className="whitespace-nowrap px-3 py-2 text-right">
      <div className="flex items-center justify-end gap-x-4">
        <ModelActionMenu onDeleteClick={onDeleteClick} />
        <Button
          size="sm"
          themes={styles.title === 'Start' ? 'accent' : 'default'}
          onClick={onClick}
        >
          {styles.title} Model
        </Button>
      </div>
    </td>
  )
}

export default ModelActionButton