Como passar mais de um argumento no react-imask com MUI?

[dsslucas]

``Olá a todos. Tô com um problema dentro do Material UI correspondente a mascara no input.

Estou criando um input com máscaras dinâmicas (telefone, CPF, CNPJ, CEP...) e até que funciona bem com Material UI, entretanto, não consigo passar a string referente a máscara dentro deste TextMaskCustom. Não quero criar um código para cada máscara, portanto, essa é a necessidade.

Tentei salvar por variável externa mas ela atualiza a cada inserção, o que torna inviável. Tentei também passar os valores pelo InputProps > inputComponent mas também sem resultados.

O que posso estar fazendo?

import React, { useState } from "react";
import { IMaskInput } from "react-imask";
import TextField from "@mui/material/TextField";

interface CustomProps {
  onChange: (event: { target: { name: string; value: string } }) => void;
  name: string;
  mask: string;
}

const TextMaskCustom = React.forwardRef<HTMLElement, CustomProps>(
  function TextMaskCustom(props, ref) {
    const { onChange, mask, ...other } = props;
    console.log(props)
    return (
      <IMaskInput
        {...other}
        mask="(00) 0 0000-0000"
        onAccept={(value: any) =>
          onChange({ target: { name: props.name, value } })
        }
        overwrite
      />
    );
  }
);

export default function TextMask(props: any) {
  const { id, label, name, style, disabled, value, type, placeholder, size, required, mask } = props

  const [valor, setValor] = useState(value)

  const handleChange = (event: any) => {
    setValor(event.target.value)
    props.returnInput(event.target.value)
  }

  return (
    <>
      <TextField
        {...props}
        type={type}
        id={id}
        label={label}
        name={name}
        variant="outlined"
        sx={style}
        disabled={disabled}
        value={valor}
        placeholder={placeholder}
        onChange={handleChange}
        size={size}
        required={required}
        InputProps={{
          inputComponent: TextMaskCustom,
        }}
      />
    </>
  );
}