#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

"""Add election timer data
"""

import os
import yaml
import calendar
import time

from jinja2 import FileSystemLoader
from jinja2.environment import Environment

PATH_PREFIX = 'candidates'


def build_timer(app):
    app.add_javascript("event_timer.js")
    data = yaml.load(open("events.yaml").read())
    # add better text for tabulatted planning
    now = time.time()
    outdated = '*'
    for ev in data:
        d, h = ev['date'].split('T')
        epoch_time = calendar.timegm(
            time.strptime(ev['date'], "%Y-%m-%dT%H:%M"))
        ev['status'] = 'past'
        if epoch_time > now:
            if outdated == '*':
                # Mark upcoming event as bold
                ev['status'] = 'current'
                outdated = '**'
            else:
                ev['status'] = 'future'
                outdated = ''
        ev['date_str'] = "%s%s, %s UTC%s" % (outdated, d, h, outdated)
        ev['name_str'] = "%s%s%s" % (outdated, ev['name'], outdated)
    output_file = os.path.join(PATH_PREFIX, "events.rst")
    with open(output_file, "w") as out:
        template_dir = os.path.join(".", "doc", "source", "_exts")
        loader = FileSystemLoader(template_dir)
        env = Environment(trim_blocks=True, loader=loader)
        template = env.get_template("events.jinja")
        out.write(template.render({'events': data}))


def setup(app):
    if not os.path.isfile("events.yaml"):
        app.info('No events.yaml found, not loading events_timer extension')
        return
    app.info('loading events_timer extension')
    app.connect('builder-inited', build_timer)