#    Copyright 2016 Mirantis, Inc.
#
#    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 exec_helpers

from devops.helpers.helpers import format_data
from devops import logger


def generate_cloud_image_settings(cloud_image_settings_path, meta_data_path,
                                  user_data_path, admin_network,
                                  interface_name, admin_ip, admin_netmask,
                                  gateway,
                                  hostname,
                                  meta_data_content=None,
                                  user_data_content=None):

    # create dir for meta_data, user_data and cloud_ISO
    dir_path = os.path.dirname(cloud_image_settings_path)

    if not os.path.exists(dir_path):
        os.makedirs(dir_path)

    # create meta_data and user_data

    # Variables in the '{ }' that will be substituted with real values
    data_context = {
        "interface_name": interface_name,
        "address": admin_ip,
        "network": admin_network,
        "netmask": admin_netmask,
        "gateway": gateway,
        "hostname": hostname
    }

    if meta_data_content is None:
        meta_data_content = ("instance-id: iid-local1\n"
                             "network-interfaces: |\n"
                             " auto {interface_name}\n"
                             " iface {interface_name} inet static\n"
                             " address {address}\n"
                             " network {network}\n"
                             " netmask {netmask}\n"
                             " gateway {gateway}\n"
                             " dns-nameservers 8.8.8.8\n"
                             "local-hostname: {hostname}")

    logger.debug("meta_data contains next data: \n{}".format(
        meta_data_content.format(**data_context)))

    with open(meta_data_path, 'w') as f:
        f.write(meta_data_content.format(**data_context))

    if user_data_content is None:
        user_data_content = ("\n#cloud-config\n"
                             "ssh_pwauth: True\n"
                             "chpasswd:\n"
                             " list: |\n"
                             "  root:r00tme\n"
                             " expire: False\n\n"
                             "runcmd:\n"
                             " - sudo ifup {interface_name}\n"
                             " - sudo sed -i -e '/^PermitRootLogin/s/^"
                             ".*$/PermitRootLogin yes/' /etc/ssh/sshd_config\n"
                             " - sudo service ssh restart\n"
                             " - sudo route add default gw "
                             "{gateway} {interface_name}")

    fmt_user_data = format_data(user_data_content, data_context)
    logger.debug("user_data contains next data: \n{}".format(fmt_user_data))

    with open(user_data_path, 'w') as f:
        f.write(fmt_user_data)

    # Generate cloud_ISO
    cmd = "genisoimage -output {} " \
          "-volid cidata -joliet " \
          "-rock {} {}".format(cloud_image_settings_path,
                               user_data_path,
                               meta_data_path)

    exec_helpers.Subprocess().check_call(cmd)