firegex-traffic-viewer/frontend/src/components/PortInput.tsx

import { Input, NumberInput, NumberInputProps, TextInput, TextInputProps } from "@mantine/core"
import React, { useState } from "react"
import { regex_port, regex_range_port } from "../js/utils"

interface PortInputProps extends NumberInputProps {
    fullWidth?: boolean
}


const PortInput =  React.forwardRef<HTMLInputElement, PortInputProps>( (props, ref) => {
    const [oldValue, setOldValue] = useState<string>(props.defaultValue?props.defaultValue.toString():"")
    const {fullWidth, ...propeties} = props
    return <NumberInput
        variant={props.variant?props.variant:"filled"}
        hideControls
        placeholder="80"
        min={props.min?props.min:1}
        max={props.max?props.min:65535}
        style={fullWidth?props.style:{ width: "75px", ...props.style }}
        onInput={(e) => {
            const target = e.target as HTMLInputElement
            target.value.match(regex_port)?setOldValue(target.value):target.value = oldValue
            props.onInput?.(e)
        }}
        ref={ref}
        {...propeties}
    />

})


interface PortRangeInputProps extends TextInputProps {
    fullWidth?: boolean,
    defaultValues?: number[]
}

export const PortRangeInput =  React.forwardRef<HTMLInputElement, PortRangeInputProps>( (props, ref) => {
    const [oldValue, setOldValue] = useState<string>(props.defaultValue?props.defaultValue.toString():"")
    const {fullWidth, defaultValues, ...propeties} = props
    let defaultValuesInt = defaultValues
    if (defaultValuesInt?.length == 2 && defaultValuesInt[0] == defaultValuesInt[1]){
        defaultValuesInt = [defaultValuesInt[0]]
    }
    
    return <TextInput
        variant={props.variant?props.variant:"filled"}
        placeholder="1000-1337"
        style={fullWidth?props.style:{ width: "150px", ...props.style }}
        onInput={(e) => {
            const target = e.target as HTMLInputElement
            target.value.match(regex_range_port)?setOldValue(target.value):target.value = oldValue
            props.onInput?.(e)
        }}
        ref={ref}
        defaultValue={defaultValuesInt?defaultValuesInt.map(v => v.toString()).join("-"):props.defaultValue}
        {...propeties}
    />

})

export default PortInput