import yaml
import os

from .utils import get_files


__all__ = ['load_fonts', 'DSFont']


class DSFont:
    def __init__(self, path, language):
        self.path = path
        self.language = language


def load_fonts(config_path='configs/dataset.yml') -> list[DSFont]:
    with open(config_path, 'r', encoding='utf-8') as f:
        config = yaml.safe_load(f)

    ds_config = config['dataset']
    ds_path = ds_config['path']

    font_list = []

    for spec in ds_config['specs']:
        for spec_path in spec['path']:
            spec_path = os.path.join(ds_path, spec_path)
            spec_files = get_files(spec_path)

            if spec.keys().__contains__('rule'):
                rule = eval(spec['rule'])
            else:
                rule = None

            for file in spec_files:
                if rule is not None and not rule(file):
                    print('skip: ' + file)
                    continue
                font_list.append(DSFont(file, spec['language']))

    return font_list