from charmhelpers.core.templating import render
from charmhelpers.core import hookenv
from charmhelpers.core import host

import toml
import os


def load_site():
    if not os.path.isfile('site.toml'):
        return {}

    with open('site.toml') as fp:
        conf = toml.loads(fp.read())

    return conf


def get_app_path():
    site = load_site()
    if 'app_path' in site:
        return site['app_path']
    return '/srv/app'


def remove_default_site():
    """
    Remove the default enabled 
    site.

    :return: Boolean
    """
    site_path = '/etc/nginx/sites-enabled/default'
    if os.path.isfile(site_path):
        os.remove(site_path)
        host.service_reload('nginx', restart_on_failure=True)
        return True

    return False


def configure_site(site, template, **kwargs):
    """ configures vhost

    Arguments:
    site: Site name
    template: template to process in templates/<template.conf>
    **kwargs: additional dict items to append to template variables exposed
              through the site.toml
    """
    hookenv.status_set('maintenance', 'Configuring site {}'.format(site))

    config = hookenv.config()
    context = load_site()
    context['host'] = config['host']
    context['port'] = config['port']
    context.update(**kwargs)
    conf_path = '/etc/nginx/sites-available/{}'.format(site)
    if os.path.exists(conf_path):
        os.remove(conf_path)
    render(source=template,
           target=conf_path,
           context=context)

    symlink_path = '/etc/nginx/sites-enabled/{}'.format(site)
    if os.path.exists(symlink_path):
        os.unlink(symlink_path)
    os.symlink(conf_path, symlink_path)
    hookenv.log('Wrote vhost config {} to {}'.format(context, template),
                'info')

    if not remove_default_site():
        host.service_reload('nginx', restart_on_failure=True)
    hookenv.status_set('active', '')