# Copyright (C) 2015-2016 Bigstep Cloud Ltd.

#

# Author: Alexandru Sirbu <alexandru.sirbu@bigstep.com>

#

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

import errno

import json

from cloudinit import log as logging

from cloudinit import sources

from cloudinit import url_helper

from cloudinit import util

LOG = logging.getLogger(__name__)

class DataSourceBigstep(sources.DataSource):

    dsname = 'Bigstep'

    def __init__(self, sys_cfg, distro, paths):

        sources.DataSource.__init__(self, sys_cfg, distro, paths)

        self.metadata = {}

        self.vendordata_raw = ""

        self.userdata_raw = ""

    def _get_data(self, apply_filter=False):

        url = get_url_from_file()

        if url is None:

            return False

        response = url_helper.readurl(url)

        decoded = json.loads(response.contents.decode())

        self.metadata = decoded["metadata"]

        self.vendordata_raw = decoded["vendordata_raw"]

        self.userdata_raw = decoded["userdata_raw"]

        return True

    def _get_subplatform(self):

        """Return the subplatform metadata source details."""

        return 'metadata (%s)' % get_url_from_file()

def get_url_from_file():

    try:

        content = util.load_file("/var/lib/cloud/data/seed/bigstep/url")

    except IOError as e:

        # If the file doesn't exist, then the server probably isn't a Bigstep

        # instance; otherwise, another problem exists which needs investigation

        if e.errno == errno.ENOENT:

            return None

        else:

            raise

    return content

# Used to match classes to dependencies

datasources = [

    (DataSourceBigstep, (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