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

from cloudinit import sources

from cloudinit.sources import DataSourceEc2 as EC2

from cloudinit import util

ALIYUN_PRODUCT = "Alibaba Cloud ECS"

class DataSourceAliYun(EC2.DataSourceEc2):

    dsname = 'AliYun'

    metadata_urls = ['http://100.100.100.200']

    # The minimum supported metadata_version from the ec2 metadata apis

    min_metadata_version = '2016-01-01'

    extended_metadata_versions = []

    def get_hostname(self, fqdn=False, resolve_ip=False, metadata_only=False):

        return self.metadata.get('hostname', 'localhost.localdomain')

    def get_public_ssh_keys(self):

        return parse_public_keys(self.metadata.get('public-keys', {}))

    def _get_cloud_name(self):

        if _is_aliyun():

            return EC2.CloudNames.ALIYUN

        else:

            return EC2.CloudNames.NO_EC2_METADATA

def _is_aliyun():

    return util.read_dmi_data('system-product-name') == ALIYUN_PRODUCT

def parse_public_keys(public_keys):

    keys = []

    for _key_id, key_body in public_keys.items():

        if isinstance(key_body, str):

            keys.append(key_body.strip())

        elif isinstance(key_body, list):

            keys.extend(key_body)

        elif isinstance(key_body, dict):

            key = key_body.get('openssh-key', [])

            if isinstance(key, str):

                keys.append(key.strip())

            elif isinstance(key, list):

                keys.extend(key)

    return keys

# Used to match classes to dependencies

datasources = [

    (DataSourceAliYun, (sources.DEP_FILESYSTEM, sources.DEP_NETWORK)),

]

# Return a list of data sources that match this set of dependencies

def get_datasource_list(depends):

    return sources.list_from_depends(depends, datasources)

# vi: ts=4 expandtab