# vim: tabstop=4 shiftwidth=4 softtabstop=4

#    Copyright (C) 2012 Yahoo! Inc. All Rights Reserved.
#
#    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.

import os
import re
import subprocess

from devstack import env

#general extraction cfg keys
CFG_MAKE = {
    'ADMIN_PASSWORD': ('passwords', 'horizon_keystone_admin'),
    'MYSQL_PASSWORD': ('passwords', 'sql'),
    'RABBIT_PASSWORD': ('passwords', 'rabbit'),
    'SERVICE_TOKEN': ('passwords', 'service_token'),
    'FLAT_INTERFACE': ('nova', 'flat_interface'),
    'HOST_IP': ('host', 'ip'),
}

#various settings we will output
EC2_PORT = 8773
S3_PORT = 3333

#these are pretty useless
EC2_USER_ID = 42
EC2_ACCESS_KEY = 'demo'

#change if you adjust keystone
NOVA_PORT = 5000
NOVA_VERSION = '1.1'
NOVA_PROJECT = 'demo'
NOVA_REGION = 'RegionOne'

#change if you adjust keystone
OS_TENANT_NAME = 'demo'
OS_USERNAME = 'demo'
OS_AUTH_PORT = 5000

#how we know if a line is an export or if it isn't
EXP_PAT = re.compile("^export (.*?)=(.*?)$", re.IGNORECASE)


def _write_line(text, fh):
    fh.write(text)
    fh.write(os.linesep)


def _write_env(name, value, fh):
    str_value = str(value)
    escaped_val = subprocess.list2cmdline([str_value])
    if str_value != escaped_val:
        _write_line("export %s=\"%s\"" % (name, escaped_val), fh)
    else:
        _write_line("export %s=%s" % (name, str_value), fh)


def _generate_ec2_env(fh, cfg):
    _write_line('# EC2 and/or S3 stuff', fh)
    ip = cfg.get('host', 'ip')
    _write_env('EC2_URL', 'http://%s:%s/services/Cloud' % (ip, EC2_PORT), fh)
    _write_env('S3_URL', 'http://%s:%s/services/Cloud' % (ip, S3_PORT), fh)
    _write_env('EC2_ACCESS_KEY', EC2_ACCESS_KEY, fh)
    hkpw = cfg.get('passwords', 'horizon_keystone_admin', auto_pw=False)
    if hkpw:
        _write_env('EC2_SECRET_KEY', hkpw, fh)
    _write_env('EC2_USER_ID', EC2_USER_ID, fh)
    _write_env('EC2_CERT', '~/cert.pem', fh)
    _write_line("", fh)


def _generate_nova_env(fh, cfg):
    _write_line('# Nova stuff', fh)
    ip = cfg.get('host', 'ip')
    hkpw = cfg.get('passwords', 'horizon_keystone_admin', auto_pw=False)
    if hkpw:
        _write_env('NOVA_PASSWORD', hkpw, fh)
    _write_env('NOVA_URL', 'http://%s:%s/v2.0' % (ip, NOVA_PORT), fh)
    _write_env('NOVA_PROJECT_ID', NOVA_PROJECT, fh)
    _write_env('NOVA_REGION_NAME', NOVA_REGION, fh)
    _write_env('NOVA_VERSION', NOVA_VERSION, fh)
    _write_env('NOVA_CERT', '~/cacert.pem', fh)
    _write_line("", fh)


def _generate_os_env(fh, cfg):
    _write_line('# Openstack stuff', fh)
    ip = cfg.get('host', 'ip')
    hkpw = cfg.get('passwords', 'horizon_keystone_admin', auto_pw=False)
    if hkpw:
        _write_env('OS_PASSWORD', hkpw, fh)
    _write_env('OS_TENANT_NAME', OS_TENANT_NAME, fh)
    _write_env('OS_USERNAME', OS_USERNAME, fh)
    _write_env('OS_AUTH_URL', 'http://%s:%s/v2.0' % (ip, OS_AUTH_PORT), fh)
    _write_line("", fh)


def generate_local_rc(fn, cfg):
    with open(fn, "w") as fh:
        _write_line('# General stuff', fh)
        for (out_name, cfg_data) in CFG_MAKE.items():
            section = cfg_data[0]
            key = cfg_data[1]
            value = cfg.get(section, key, auto_pw=False)
            if value:
                _write_env(out_name, value, fh)
        _write_line("", fh)
        _generate_ec2_env(fh, cfg)
        _generate_nova_env(fh, cfg)
        _generate_os_env(fh, cfg)


def load_local_rc(fn):
    am_set = 0
    with open(fn, "r") as fh:
        for line in fh:
            m = EXP_PAT.search(line)
            if m:
                key = m.group(1).strip()
                value = m.group(2).strip()
                if len(key):
                    env.set(key, value)
                    am_set += 1
    return am_set