"""
A function to create citations for Books or Ebooks. Different forms of the citation will be created depending on which parameters have arguments passed to them. Each citation is compliant with APA7. 
"""

import numpy as np
from author_names import *


def citing_books(book_title=False, publisher=False, publishing_year=False,
                 DOI=False, alternate_description=False, **names):
    # Check for source
    if publisher is False:
        return "Not enough information to generate a citation. Please provide publisher information."
    # Check for date or n.d.
    if publishing_year is False:
        publishing_year = "n.d."
    # Author name should be LastName, first initial, middle initial 
    authors_list = author_names(**names)

    # Begin generating final citation
    if authors_list:
        if book_title:  # author and title and source are present
            reference = authors_list + " (" + publishing_year + "). " + book_title + ". " + publisher + "."
        else:  # Title missing
            if alternate_description:
                reference = authors_list + " (" + publishing_year + "). [" + alternate_description + "]. " + publisher + "."
            else:
                return "Not enough information to generate a citation. Please provide a description in liu of a title."
    else:  # authors missing
        if book_title:  # author missing, title present
            reference = book_title + ". (" + publishing_year + "). " + publisher + "."
        else:  # author and title missing
            if alternate_description:
                reference = "[" + alternate_description + "]. (" + publishing_year + "). " + publisher + "."
            else:
                return "Not enough information to generate a citation. Please provide a description in liu of a title."
    # Add optional DOI
    if DOI:
        reference = reference + " " + DOI
    return reference


#### Function for citing a webpage ####

def citing_webpage(publishing_year=False, publishing_month=False, publishing_day=False, title_of_webpage=False,
                   webpage_name=False, url=False, alternate_description=False, **names):
    # Author name should be LastName, first initial, middle initial
    authors_list = author_names(**names)

    # check for required values
    if url is False:
        return "Not enough information to generate a citation. Electronic sources require URL."
    if webpage_name is False:  # Check for source before evaluating if source and author are the same.
        return "Not enough information to generate a citation. Please provide a website name."
    # If author and site name are the same, omit the site name from the citation
    if str(authors_list).lower() == str(webpage_name).lower():
        webpage_name = False
        authors_list += "."

    # Create date
    if publishing_year:
        if publishing_month:
            if publishing_day:
                pubdate = "(" + publishing_year + ", " + publishing_month + " " + publishing_day + ")"
            else:
                pubdate = "(" + publishing_year + ", " + publishing_month + ")"
        else:
            pubdate = "(" + publishing_year + ")"
    else:
        pubdate = "(n.d.)"

    # Begin generating final citation
    if authors_list:
        if title_of_webpage:
            if webpage_name:
                reference = authors_list + " " + pubdate + ". " + title_of_webpage + ". " + webpage_name + ". " + url
            else:  # Webpage is the same as author
                reference = authors_list + " " + pubdate + ". " + title_of_webpage + ". " + url
        else:  # title is missing, author present
            if alternate_description:
                reference = authors_list + " " + pubdate + ". [" + alternate_description + "]. " + url
            else:
                return "Not enough information to generate a citation. Please provide a description in liu of a title."
    else:  # authors missing
        if title_of_webpage:  # missing author
            if webpage_name:
                reference = title_of_webpage + ". " + pubdate + ". " + webpage_name + ". " + url
            else:  # webpage is same as author, omit
                reference = title_of_webpage + ". " + pubdate + ". " + url
        else:  # author and title missing
            if alternate_description:
                reference = "[" + alternate_description + "]. " + pubdate + ". " + webpage_name + ". " + url
            else:
                return "Not enough information to generate a citation. Please provide a description in liu of a title."
    return reference


def citing_journal(publishing_year=False, article_title=False, journal_name=False, page_start=False, page_end=False,
                   volume=False, issue=False, alt_description=False, DOI=False, **names):
    # Author name should be LastName, first initial, middle initial
    authors_list = author_names(**names)

    # Check for required values
    if journal_name is False:
        return "Not enough information to generate a citation. Please provide a website name."

    if publishing_year is False:
        publishing_year = "n.d."

    # Create page-range
    page_range = False
    if page_start:
        if page_end:
            page_range = page_start + "-" + page_end

    # Create journal name, volume(issue)
    if journal_name:
        if volume:
            if issue:
                if page_range:
                    journal = journal_name + ", " + volume + "(" + issue + "), " + page_range
                else:  # no page range
                    journal = journal_name + ", " + volume + "(" + issue + ")"
            else:  # no issue
                journal = journal_name + ", " + volume
        else:  # no volume
            journal = journal_name
    else:
        return "Not enough information to generate a citation. Please provide publisher information."
        # Check for date or n.d.
    if publishing_year is False:
        publishing_year = "n.d."

    # Begin generating final citation
    if authors_list:
        if article_title:  # author and title and source are present
            reference = authors_list + " (" + publishing_year + "). " + article_title + ". " + journal + "."
        else:  # Title missing
            if alt_description:
                reference = authors_list + " (" + publishing_year + "). [" + alt_description + "]. " + journal + "."
            else:
                return "Not enough information to generate a citation. Please provide a description in liu of a title."
    else:  # authors missing
        if article_title:  # author missing, title present
            reference = article_title + ". (" + publishing_year + "). " + journal + "."
        else:  # author and title missing
            if alt_description:
                reference = "[" + alt_description + "]. (" + publishing_year + "). " + journal + "."
            else:
                return "Not enough information to generate a citation. Please provide a description in liu of a title."
    # Add optional DOI
    if DOI:
        reference = reference + " " + DOI
    return reference