#!/usr/bin/env python
#
# Copyright (C) 2015 GNS3 Technologies Inc.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import os
import sys
import json
import shutil
import copy
import base64

from jinja2 import Environment, FileSystemLoader

import logging
log = logging.getLogger(__name__)
log.setLevel(logging.DEBUG)

handler = logging.StreamHandler(sys.stdout)
formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
handler.setFormatter(formatter)
log.addHandler(handler)

if os.path.exists('build'):
    for file in os.listdir('build'):
        if os.path.isdir(os.path.join('build', file)):
            shutil.rmtree(os.path.join('build', file))
        else:
            os.remove(os.path.join('build', file))
else:
    os.mkdir('build')
os.mkdir(os.path.join('build', 'devices'))
os.mkdir(os.path.join('build', 'images'))


def render(template_file, out, **kwargs):
    log.info('Build %s', out)
    env = Environment(loader=FileSystemLoader('templates'))
    env.filters['jsonify'] = json.dumps
    env.filters['b64encode'] = lambda s: base64.b64encode(s.encode()).decode("utf-8")
    template = env.get_template(template_file)
    template.stream(**kwargs).dump(os.path.join('build', out))


def keep_only_version_with_device(md5sum, device):
    """
    Filter device version in order to keep only the
    version where the image is present.

    :param md5sum: Md5sum of the image
    :param device: Device hash
    :returns: List of version
    """

    new_versions = []
    for version in device["versions"]:
        found = False
        for image in version["images"].values():
            if image["md5sum"] == md5sum:
                found = True
                break
        if found:
            new_versions.append(version)
    return new_versions


render('index.html', 'index.html')
render('chat.html', 'chat.html')
render('downloads.html', 'downloads.html')


devices = []
for device_file in os.listdir('devices'):
    log.info("Process " + device_file)
    out_filename = device_file[:-5]
    with open(os.path.join('devices', device_file)) as f:
        device = json.load(f)
    device['id'] = out_filename

    # Resolve version image to the corresponding file
    for version in device['versions']:
        for image_type, filename in version['images'].items():
            found = False
            for file in device['images']:
                if file['filename'] == filename:
                    version['images'][image_type] = copy.copy(file)
                    version['images'][image_type]["type"] = image_type
                    found = True
                    break
            if not found:
                log.critical('Image for {} {} with filename {} is missing'.format(device["name"], version["name"], file['filename']))
                sys.exit(1)


    render('device.html', os.path.join('devices', out_filename + '.html'), device=device)
    devices.append(device)

    # Build a page named with the md5sum of each file of the device
    # it's allow to get the device informations via HTTP with just an md5sum
    # it's what powered the import feature
    for image in device['images']:
        # We keep only version with this image in the page
        image_device = copy.copy(device)
        image_device['versions'] = keep_only_version_with_device(image['md5sum'], device)
        render('device.html', os.path.join('images', image['md5sum'] + '.html'), device=image_device)

render('devices.html', os.path.join('devices', 'index.html'), devices=devices)