"""
Copyright (C) 2022  Daniel Mowitz

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with this program.  If not, see <https://www.gnu.org/licenses/>.
"""

import sys, os, cook2tex

def create_book(argv):
    """
    Takes: [ string ] argv
    Returns: [ string ] tex
    
    Takes multiple recipe paths as arguments and returns a complete latex document.
    One element per line.
    """
    tex = []

    for path in argv:
        tex.append("\n")
        tex += cook2tex.parse_recipe_from_file(path)

    return latex_boilerplate(tex)

def create_book_recursive(source_path, dest_path):
    """
    Takes: string source_path
           string dest_path
    Returns: 
    
    Reads all .cook files found in source_path and it's subdirectories.
    Creates a new directory at dest_dir if it doesn't exist already.
    When a recipe path reads '/source_dir/.../last_dir/recipe.cook',
    creates /dest_dir/last_dir/recipe.tex and writes texified recipe to it.
    Creates /dest_dir/recipe-book.tex which includes all generated recipe files,
    has one chapter per directory and a table of contents.
    """
    
    try:
        os.mkdir(dest_path)
    except FileExistsError:
        pass

    chapters = {}
    
    for root, dirs, files in os.walk(source_path):
        cook_files = {}
        
        for name in files:
            split_name = os.path.splitext(name)
            if split_name[-1] == ".cook":
                cook_files[split_name[0]] = os.path.join(root, name)

        if cook_files:
            dir_name = os.path.basename(root)
            dir_path = os.path.join(dest_path, dir_name)

            chapters[dir_name] = []
            
            try:
                os.mkdir(dir_path)
            except FileExistsError:
                pass

            for name in cook_files.keys():
                recipe_path = os.path.join(dir_path, name + ".tex")
                chapters[dir_name].append(recipe_path)
                
                with open(recipe_path, "w") as file:
                    for line in cook2tex.parse_recipe_from_file(cook_files[name]):
                        file.write(line + "\n")

    tex = []
    
    for chap in chapters.keys():
        tex.append("\\section{" + chap + "}")
        for file_name in chapters[chap]:
            tex.append("\\input{" +
                       os.path.relpath(file_name, dest_path) + "}")

        tex.append("\n")

    with open(os.path.join(dest_path, "recipe-book.tex"), "w") as file:
        for line in latex_boilerplate(tex):
            file.write(line + "\n")

        
def latex_boilerplate(body):
    tex = []

    tex.append("\\documentclass{article}")
    tex.append("\\usepackage{cuisine}")
    tex.append("\n")
    tex.append("\\begin{document}")
    tex.append("\n")
    tex.append("\\tableofcontents")
    tex.append("\n")

    tex += body

    tex.append("\n")
    tex.append("\\end{document}")

    return tex
        
if __name__ == "__main__":
    if sys.argv[1] == "-r":
        tex = create_book_recursive(sys.argv[2], sys.argv[3])
    else:
        for line in create_book(sys.argv[1:]):
            print(line)