#!/usr/bin/env python3
from datetime import date
import os
import subprocess
import sys


def main():
    base_dir = get_base_dir()
    today = date.today()

    if len(sys.argv) == 1:
        edit_note(base_dir, today)
    elif sys.argv[1] == "backup":
        backup(base_dir, today)
    elif sys.argv[1] == "merge":
        merge(base_dir)
    else:
        print('Command not found')


def get_base_dir() -> str:
    """
    Creates the directories necessary for the program
    to work, if they are not present.

    Returns the path to the directory where data can be stored.
    """
    base_dir = os.path.expanduser('~/.local/share/histd')
    os.makedirs(base_dir, exist_ok=True)
    return base_dir


def edit_note(base_dir: str, note_date):
    """
    Creates the required directories and opens a text editor
    so that the user can describe the day.
    """
    # Create dirs (base_dir/year/month)
    year = str(note_date.year)
    month = f'{note_date.month:02}'
    workdir = os.path.join(base_dir, year, month)
    os.makedirs(workdir, exist_ok=True)

    # Open file (base_dir/year/month/day.md) with default editor
    filename = f'{note_date.day:02}.md'
    path_to_file = os.path.join(workdir, filename)
    editor = os.environ.get('EDITOR', 'nano')
    try:
        subprocess.run([editor, path_to_file], check=True, cwd=base_dir)
    except FileNotFoundError:
        print("Error: I can't find your text editor")
        print("Make sure the 'EDITOR' environment variable is set correctly")
    except subprocess.CalledProcessError:
        print("Your editor returned non-zero exit code")


def backup(base_dir: str, current_date):
    """
    Creates an archive with all notes
    """
    date_str = f'{current_date.year}-{current_date.month:02}-{current_date.day:02}'
    archive_path = os.path.expanduser(f"~/histd-{date_str}.tar.xz")
    cmd = ["tar", "cfJ", archive_path, "."]
    try:
        subprocess.run(cmd, check=True, cwd=base_dir)
        print(f'Saved to {archive_path}')
    except FileNotFoundError:
        print("Error: I can't find tar program")
    except subprocess.CalledProcessError:
        print("Archiver returned non-zero exit code")


def merge(base_dir: str):
    """
    This function concatenates all files and prefixes each with the filename.
    The result will be printed to stdout.
    """
    
    def read_files(path: str) -> str:
        """
        Recursive function to read all files in a directory
        """
        strings = []
        contents = os.listdir(path)
        
        for entry in contents:
            entry_path = os.path.join(path, entry)
            
            # It's a directory
            if os.path.isdir(entry_path):
                # Read all files in this directory
                res = read_files(entry_path)
                strings.append(res)
            # It's a file
            else:
                with open(entry_path, 'r') as note:
                    strings.append(f'## {entry_path}')
                    strings.append(note.read())
        
        return '\n\n'.join(strings)

    res = read_files(base_dir)
    print(res)


if __name__ == '__main__':
    main()