import argparse
from src.universal_refactor.pipelines import InferencePipeline
from src.universal_refactor.utils import read_text

parser=argparse.ArgumentParser()
parser.add_argument('--mode',choices=['refactor','convert'],default='refactor')
parser.add_argument('--file');parser.add_argument('--lang',default='python')
parser.add_argument('--src');parser.add_argument('--tgt')

def main():
    a=parser.parse_args();pipe=InferencePipeline()
    code=read_text(a.file)
    if a.mode=='refactor':
        out=pipe.analyze_and_refactor(code,a.lang);print(out)
    else:
        print(pipe.convert(code,a.src,a.tgt))

if __name__=='__main__': main()