import os.path
import yaml
from glob import glob
from project.pages.utils import get_title_from_markdown, parse_content


def generate_config(base_dir):
    default = yaml.load(open(os.path.join(base_dir, 'data/context.yml'))) or {}
    page = yaml.load(open(os.path.join(base_dir, 'data/page_context.yml'))) or {}
    switcher = yaml.load(open(os.path.join(base_dir, 'data/path_switch.yml'))) or {}

    default['projects'] = generate_projects(base_dir)
    for i in range(len(default['projects'])):
        project = default['projects'][i]
        if project['path'] in page:  # If there's a custom config
            default['projects'][i] = dict(project, **page[project['path']])
            default['projects'][i]['url'] = '/' + project['path']
    return default, page, switcher


def generate_projects(base_dir):
    projects_path = os.path.join(base_dir, 'templates/projects')
    files = []
    for path in glob(projects_path + '/*.*'):
        filename = path.replace(projects_path, '')
        if filename == '/all.html':
            continue
        with open(path) as f:
            if filename.split('.')[1] == 'md':
                parsed_content = parse_content(f.read(), filename.split('.')[1])
                filename = get_title_from_markdown(parsed_content)
            else:
                filename = filename.split('.')[0]
            files.append({
                "name": filename,
                "path": 'projects' + path.replace(projects_path, '').split('.')[0]
            })
    return files