# This file is part of cloud-init. See LICENSE file for license information.

"""Base NoCloud KVM snapshot."""
import os

from tests.cloud_tests.snapshots import base


class NoCloudKVMSnapshot(base.Snapshot):
    """NoCloud KVM image copy backed snapshot."""

    platform_name = "nocloud-kvm"

    def __init__(self, platform, properties, config, features,
                 instance):
        """Set up snapshot.

        @param platform: platform object
        @param properties: image properties
        @param config: image config
        @param features: supported feature flags
        """
        self.instance = instance

        super(NoCloudKVMSnapshot, self).__init__(
            platform, properties, config, features)

    def launch(self, user_data, meta_data=None, block=True, start=True,
               use_desc=None):
        """Launch instance.

        @param user_data: user-data for the instance
        @param instance_id: instance-id for the instance
        @param block: wait until instance is created
        @param start: start instance and wait until fully started
        @param use_desc: description of snapshot instance use
        @return_value: an Instance
        """
        key_file = os.path.join(self.platform.config['data_dir'],
                                self.platform.config['public_key'])
        user_data = self.inject_ssh_key(user_data, key_file)

        instance = self.platform.create_image(
            self.properties, self.config, self.features,
            self.instance.name, image_desc=str(self), use_desc=use_desc,
            user_data=user_data, meta_data=meta_data)

        if start:
            instance.start()

        return instance

    def inject_ssh_key(self, user_data, key_file):
        """Inject the authorized key into the user_data."""
        with open(key_file) as f:
            value = f.read()

        key = 'ssh_authorized_keys:'
        value = '  - %s' % value.strip()
        user_data = user_data.split('\n')
        if key in user_data:
            user_data.insert(user_data.index(key) + 1, '%s' % value)
        else:
            user_data.insert(-1, '%s' % key)
            user_data.insert(-1, '%s' % value)

        return '\n'.join(user_data)

    def destroy(self):
        """Clean up snapshot data."""
        self.instance.destroy()
        super(NoCloudKVMSnapshot, self).destroy()

# vi: ts=4 expandtab