import argparse
import yaml
import locale

from pathlib import Path
from datetime import datetime

from invoice_generator import html_generator


class DataObject:
    def __init__(self, **kwargs):
        for key, value in kwargs.items():
            setattr(self, key, value)


def parse_yaml_file(file_path):
    with open(file_path, 'r') as file:
        return yaml.safe_load(file)


def check_file(file_path):
    try_out_extensions = ['.yaml', '.yml']

    for ext in try_out_extensions:
        if file_path.exists():
            return file_path
        file_path = file_path.with_suffix(ext)
    return None


def parse_date(date_string):
    return datetime.strptime(date_string, '%Y-%m-%d').date()


def merge_envelope_data_into_invoice_data(invoice_data, source_data):
    if (source_data is None) or (invoice_data is None):
        return
    for key, value in source_data.items():
        if not hasattr(invoice_data, key):
            setattr(invoice_data, key, value)


# Utility function


def main():
    locale.setlocale(locale.LC_ALL, 'de_DE.utf8')

    print('Simple invoice generator for freelancers and small businesses by Torsten Ueberschar')
    print()
    parser = argparse.ArgumentParser(description='Read invoice and envelope data from yaml file')
    parser.add_argument('-b', '--base', type=str, required=True, help='base directory for invoice and envelope files')
    parser.add_argument('-i', '--invoice', type=str, required=True, help='Invoice file name')
    parser.add_argument('-e', '--envelope', type=str, required=False, default='envelope.yaml',
                        help='Envelope file name')
    parser.add_argument('-t', '--template', type=str, required=False, default='template',
                        help='directory for template files')

    args = parser.parse_args()

    invoice_file = args.invoice
    envelope_file = args.envelope

    print(f'Searching for invoice files in: {args.base}')

    invoice_file = Path(args.base) / invoice_file
    envelope_file = Path(args.base) / envelope_file

    invoice_file = check_file(invoice_file)
    envelope_file = check_file(envelope_file)

    print(f'Found invoice file: {invoice_file}')
    print(f'Found envelope file: {envelope_file}')

    if invoice_file is None:
        print('Error: No invoice file found')
        return
    if envelope_file is None:
        print('Error: No envelope file found')
        return

    try:
        invoice = parse_yaml_file(invoice_file)
        envelope = parse_yaml_file(envelope_file)

        print('Envelope data:')
        envelope_data = DataObject(**envelope)
        print(envelope_data.__dict__)

        print('Invoice data:')
        invoice_data = DataObject(**invoice)
        merge_envelope_data_into_invoice_data(invoice_data, envelope_data.Invoice)

        selected_customer = next((x for x in envelope_data.Customers if x['CustomerId'] == invoice_data.CustomerId), None)
        merge_envelope_data_into_invoice_data(invoice_data, selected_customer)

        if not hasattr(invoice_data, 'InvoiceDate'):
            invoice_data.InvoiceDate = datetime.now().date()
        else:
            invoice_data.InvoiceDate = parse_date(invoice_data.InvoiceDate)

        if not hasattr(invoice_data, 'Id'):
            invoice_data.Id = None

        invoice_data.Id = invoice_data.Id or invoice_file.stem
        print(invoice_data.__dict__)

    except FileNotFoundError as e:
        print(f'Error: {e}')
        return
    except yaml.YAMLError as e:
        print(f'Error: {e}')
        return
    except Exception as e:
        print(f'Error: {e}')
        return

    generator = html_generator.HtmlTemplate(args.template)
    template = generator.prepare_template(invoice_data, envelope_data)

    try:
        print('Generating invoice...')
        invoice_pdf = Path(invoice_data.Id).with_suffix('.pdf')
        print(f'Invoice PDF: {invoice_pdf}')
        html_generator.HtmlTemplate.convert_html_to_pdf(template, invoice_pdf)
    except Exception as e:
        print(f'Error: {e}')
        return


if __name__ == "__main__":
    main()