hellpipe/hellpipe/convert.py

13 lines
374 B
Python

from typing import Optional
from . import mappers
MAPPERS = {"curl": mappers.CurlMapper,
"xargs": mappers.XargsMapper,
"jq": mappers.JqMapper}
def convert_command(command: list[str], input_name: Optional[str]):
mapper_cls = MAPPERS.get(command[0], mappers.ShellMapper)
mapper = mapper_cls(command)
return mapper.generate(input_name, ["bytes"])