mirror of https://github.com/langgenius/dify.git
115 lines
3.1 KiB
TypeScript
115 lines
3.1 KiB
TypeScript
import type { FC } from 'react'
|
|
import { RiArrowDownSLine, RiArrowUpSLine } from '@remixicon/react'
|
|
import Input, { type InputProps } from '../input'
|
|
import classNames from '@/utils/classnames'
|
|
|
|
export type InputNumberProps = {
|
|
unit?: string
|
|
value?: number
|
|
onChange: (value?: number) => void
|
|
amount?: number
|
|
size?: 'regular' | 'large'
|
|
max?: number
|
|
min?: number
|
|
defaultValue?: number
|
|
disabled?: boolean
|
|
wrapClassName?: string
|
|
controlWrapClassName?: string
|
|
controlClassName?: string
|
|
} & Omit<InputProps, 'value' | 'onChange' | 'size' | 'min' | 'max' | 'defaultValue'>
|
|
|
|
export const InputNumber: FC<InputNumberProps> = (props) => {
|
|
const { unit, className, onChange, amount = 1, value, size = 'regular', max, min, defaultValue, wrapClassName, controlWrapClassName, controlClassName, disabled, ...rest } = props
|
|
|
|
const isValidValue = (v: number) => {
|
|
if (typeof max === 'number' && v > max)
|
|
return false
|
|
return !(typeof min === 'number' && v < min)
|
|
}
|
|
|
|
const inc = () => {
|
|
if (disabled) return
|
|
|
|
if (value === undefined) {
|
|
onChange(defaultValue)
|
|
return
|
|
}
|
|
const newValue = value + amount
|
|
if (!isValidValue(newValue))
|
|
return
|
|
onChange(newValue)
|
|
}
|
|
const dec = () => {
|
|
if (disabled) return
|
|
|
|
if (value === undefined) {
|
|
onChange(defaultValue)
|
|
return
|
|
}
|
|
const newValue = value - amount
|
|
if (!isValidValue(newValue))
|
|
return
|
|
onChange(newValue)
|
|
}
|
|
|
|
return <div className={classNames('flex', wrapClassName)}>
|
|
<Input {...rest}
|
|
// disable default controller
|
|
type='text'
|
|
className={classNames('rounded-r-none', className)}
|
|
value={value}
|
|
max={max}
|
|
min={min}
|
|
disabled={disabled}
|
|
onChange={(e) => {
|
|
if (e.target.value === '')
|
|
onChange(undefined)
|
|
|
|
const parsed = Number(e.target.value)
|
|
if (Number.isNaN(parsed))
|
|
return
|
|
|
|
if (!isValidValue(parsed))
|
|
return
|
|
onChange(parsed)
|
|
}}
|
|
unit={unit}
|
|
size={size}
|
|
/>
|
|
<div className={classNames(
|
|
'flex flex-col bg-components-input-bg-normal rounded-r-md border-l border-divider-subtle text-text-tertiary focus:shadow-xs',
|
|
disabled && 'opacity-50 cursor-not-allowed',
|
|
controlWrapClassName)}
|
|
>
|
|
<button
|
|
type='button'
|
|
onClick={inc}
|
|
disabled={disabled}
|
|
aria-label='increment'
|
|
className={classNames(
|
|
size === 'regular' ? 'pt-1' : 'pt-1.5',
|
|
'px-1.5 hover:bg-components-input-bg-hover',
|
|
disabled && 'cursor-not-allowed hover:bg-transparent',
|
|
controlClassName,
|
|
)}
|
|
>
|
|
<RiArrowUpSLine className='size-3' />
|
|
</button>
|
|
<button
|
|
type='button'
|
|
onClick={dec}
|
|
disabled={disabled}
|
|
aria-label='decrement'
|
|
className={classNames(
|
|
size === 'regular' ? 'pb-1' : 'pb-1.5',
|
|
'px-1.5 hover:bg-components-input-bg-hover',
|
|
disabled && 'cursor-not-allowed hover:bg-transparent',
|
|
controlClassName,
|
|
)}
|
|
>
|
|
<RiArrowDownSLine className='size-3' />
|
|
</button>
|
|
</div>
|
|
</div>
|
|
}
|