import os
from html5tagger import Document, E, HTML
from shutil import rmtree
import markdown
import html

#if homepage is at Home.page, set homepage to "Home"
homepage = "Home"

try:
  os.mkdir("./website-output")
except FileExistsError:
  print("Output directory already exists, let's clean it!")

  #deleting contents of folder without deleting the folder, to increase compatibility with various systems
  for root, dirs, files in os.walk('./website-output'):
    for f in files:
      os.unlink(os.path.join(root, f))
    for d in dirs:
      rmtree(os.path.join(root, d))

def generateNavigationBar(lines):
  global navbar
  navbar = E
  for line in lines:

    title, link = line.strip().split(";", 1)
    navbar = navbar.li(HTML("<a href='" + link + "'>" + title + "</a>"))

def parseRawHTML(doc, line, htmlstring, rawhtml):
  #raw html start
  if line.strip() == ">":
    rawhtml = True
    htmlstring = ""

  #parse indented raw html
  elif rawhtml:
    if line.startswith("  "):
      htmlstring = htmlstring + line.strip()
    else:
      rawhtml = False
      doc.div(HTML(html))
      htmlstring = ""
  return rawhtml, doc, htmlstring

def parseMarkdown(doc, line):
  line = html.escape(line.strip())
  return doc(HTML(markdown.markdown(line)))

def generateLines(title, lines):
  title = title.replace(".page", "")
  doc = Document(title, lang="en")
  rawhtml = False
  htmlstring = ""
  for line in lines:

    #parse raw HTML
    rawhtml, doc, htmlstring = parseRawHTML(doc, line, htmlstring, rawhtml)

    #parse markdown
    if not rawhtml:
      doc = parseMarkdown(doc, line)

  #if indented html was the last line, this is needed for it to not be ignored
  if rawhtml:
     doc.div(HTML(htmlstring))

  generatePage(title, doc)

def generatePage(title, doc):
  #creates ./website-output/pagetitle/index.html file if it is not homepage
  foldername = ""

  if title != homepage:
    foldername = "/" + title.replace(" ", "-").lower()
    os.mkdir("./website-output" + foldername)

  with open("./website-output" + foldername + "/index.html", 'w') as newpage:
    if 'navbar' in globals():
      newpage.write(str(E.ul(navbar)) + str(doc))
    else:
      newpage.write(str(doc))

if os.path.exists("./navbar"):
  with open("./navbar", 'r') as navbarfile:
    generateNavigationBar(navbarfile.readlines())
else:
  print("No 'navbar' file found, there will be no navigation bar.")

pagescount = 0
for file in os.listdir("./"):
  if file.endswith(".page"):
    with open(file, 'r') as page:
      generateLines(os.path.basename(file), page.readlines())
    pagescount=pagescount+1
print("Generated " + str(pagescount) + " pages")