#!/usr/bin/env python3
# Copyright 2019 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Node implementation for the ComponentGraph.

The ComponentNode constructed with a package stores rich data about all the
services this component offers, uses and exposes. This is all stored directly
on the node as aggregation data during generation to make the structure as
simple as possible to use without any further aggregation.
"""

from server.util.url import package_resource_url


class ComponentNode:
    """ Node implementation for the component graph. """

    def __init__(self, resource_path, resource_data, ty):
        self.id = resource_path
        self.name = resource_path.split("/")[-1]
        self.manifest = resource_data
        self.exposes = []
        self.offers = []
        self.version = 0
        self.consumers = 0
        self.source = "package"
        self.ty = ty
        # Filled in when links are generated
        self.component_use_deps = []
        try:
            self.uses = self.manifest["sandbox"]["services"]
        except KeyError:
            self.uses = []
        try:
            self.features = self.manifest["sandbox"]["features"]
        except KeyError:
            self.features = []

    @classmethod
    def create_inferred(cls, pkg_resource_url):
        """ Named constructor to construct an inferred ComponentNode. """
        node = cls(pkg_resource_url, {}, "inferred")
        return node

    def append_offer(self, offer):
        """ Adds an additional service this component offers. """
        self.offers.append(offer)

    def __str__(self):
        return self.id

    def export(self):
        """ Returns an export of the node that can be transformed into JSON """
        export_data = {}
        export_data["id"] = self.id
        export_data["name"] = self.name
        export_data["manifest"] = self.manifest
        export_data["consumers"] = self.consumers
        export_data["routes"] = {}
        export_data["version"] = self.version
        export_data["source"] = self.source
        export_data["routes"]["exposes"] = self.exposes
        export_data["routes"]["offers"] = self.offers
        export_data["routes"]["uses"] = self.uses
        return export_data