#!/usr/bin/env python3

"""Write dependencies for all website pages in makefile syntax.

We want to compute:

- a list of leaf pages,
- a list of auto-generated indices,

- dependencies for leaf pages:
  OUTPUT/foo/bar.html: foo/bar.txt | OUTPUT/foo

    - special case for READMEs:
      OUTPUT/foo/index.html: foo/README.txt foo | OUTPUT/foo

- dependencies for auto-generated indices:
  OUTPUT/foo/index.html: foo | OUTPUT/foo
"""

from collections import defaultdict
from dataclasses import dataclass, field
from os import path
from sys import argv, exit
from typing import List, Set

from git import Repo


@dataclass
class Directory:
    files: List[str] = field(default_factory=list)
    subfolders: Set[str] = field(default_factory=set)


def parse_arguments(args):
    if len(args) != 3:
        exit(f'Usage: {argv[0]} EXTENSIONS OUTPUT-DIR')

    return argv[1].split(), argv[2]


def find_sources(extensions, repository):
    patterns = (f'**.{ext}' for ext in extensions)
    zero = '\x00'
    return repository.git.ls_files('-z', *patterns).strip(zero).split(zero)


def compute_directories(files, top_dir):
    directories = defaultdict(Directory)

    for f in files:
        fdir, fname = path.split(f)

        directories[fdir].files.append(fname)

        if fdir:
            parent, child = path.split(fdir)
            directories[parent].subfolders.add(child)

    return directories


def write_dependencies(deps, directories, top_dir, out_dir):
    pages = list()

    for dpath, d in directories.items():

        for f in d.files:
            name, _ = path.splitext(f)
            if name == 'README':
                name = 'index'

            html_dir = (
                path.join(out_dir, dpath)
                if dpath
                else out_dir
            )

            html_path = path.join(html_dir, name+'.html')
            src_path = path.join(top_dir, dpath, f)

            print(f'{html_path}: {src_path} | {html_dir}', file=deps)
            pages.append(html_path)

    print(file=deps)
    print(f'pages = {" ".join(pages)}', file=deps)


def main(arguments):
    extensions, out_dir = parse_arguments(arguments)

    repository = Repo(search_parent_directories=True)
    top_dir = repository.working_dir

    source_files = find_sources(extensions, repository)

    directories = compute_directories(source_files, top_dir)

    with open('deps.mk', 'w') as deps:
        write_dependencies(deps, directories, top_dir, out_dir)


if __name__ == '__main__':
    main(argv)