import pathlib

import click
import tqdm


@click.command('Check if all TextGrids are generated')
@click.option('--wavs', required=True, help='Path to the segments directory')
@click.option('--tg', required=True, help='Path to the TextGrids directory')
def check_tg(wavs, tg):
    wavs = pathlib.Path(wavs)
    tg = pathlib.Path(tg)
    missing = []
    filelist = list(wavs.glob('*.wav'))
    for wavfile in tqdm.tqdm(filelist):
        tgfile = tg / wavfile.with_suffix('.TextGrid').name
        if not tgfile.exists():
            missing.append(tgfile)
    if len(missing) > 0:
        print(
            'These TextGrids are missing! There are possible severe errors in labels of those corresponding segments. '
            'If you do believe there are no errors, consider increase the \'--beam\' argument for MFA.')
        for fn in missing:
            print(f' - {fn}')
    else:
        print('All alignments have been successfully generated.')


if __name__ == '__main__':
    check_tg()