from mako.template import Template
from mako.lookup import TemplateLookup
import distutils

import argparse
import urllib.parse
import glob
import os
import os.path
import lxml.html


parser = argparse.ArgumentParser()
parser.add_argument('indir')
parser.add_argument('--base', '-b', default='base.html', nargs='?')
parser.add_argument('--odir', '-o')
args = parser.parse_args()
if args.odir is None:
    args.odir = args.indir

mylookup = TemplateLookup(directories='templates')
template = mylookup.get_template('from-markdown.html')

items = []

counter = 0
for file in glob.glob1(args.indir, '*.md'):
    oname = os.path.join(args.odir, '%s.html' % file.rpartition('.')[0])

    with open(os.path.join(args.indir, file)) as f:
      out = template.render(content=f.read(), base=args.base)
      with open(oname, 'wb') as of:
          of.write(out.encode('utf-8'))
        
    counter += 1
    items.append({})
    htmldoc = lxml.html.fromstring(out)

    desc_el = htmldoc.xpath('//div[contains(@class,"container") and contains(@class, "has_description")]/p[1]')
    print(desc_el)
    desc_el = '' if not desc_el else desc_el[0]

    title_el = htmldoc.xpath('//h1 | //h2 | //h3 | //h4 | //h4 | //h6')
    title_el = '' if not title_el else title_el[0]

    items[-1].update(
            link=urllib.parse.quote(os.path.relpath(oname, args.odir)),
            counter = counter,
            title = oname if title_el is '' else title_el.text_content().strip(),
            description = desc_el if desc_el is '' else lxml.etree.tostring(desc_el).decode('utf-8')
    )

index_template = mylookup.get_template('index.html')
index = index_template.render(items=items, base=args.base)
with open(os.path.join(args.odir, 'index.html'), 'wb') as of:
    of.write(index.encode('utf-8'))