#!/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.
"""QueryHandler services all graph and package queries.

The QueryHandler is the business logic layer between the PackageManager and
the ComponentGraphGenerator. It is responsible for processing the data provided
by the PackageManager and forwarding it through for graph generation.
"""

import sys
import json
from server.util.logging import get_logger
from server.graph import ComponentGraphGenerator

class ComponentQueryError(Exception):
  """Raised when an unrecoverable query exception occurs"""
  pass

class QueryHandler:
    """ Core handler to respond to different queries """

    def __init__(self, package_manager):
        """ Verifies the package manager is online. """
        self.logger = get_logger(__name__)
        self.package_manager = package_manager
        self.graph_generator = ComponentGraphGenerator()

        if not self.package_manager.ping():
            self.logger.error(
                "Failed to connect to package manager please run fx serve.")
            sys.exit(1)

    def services(self, packages):
        """ Returns the list of service to component url mappings """
        return self.package_manager.get_services(packages)

    def packages(self):
        """ Returns a list of available packages """
        return self.package_manager.get_packages()

    def component_graph(self):
        """ Returns the component graph that shows all component connections """
        packages = self.packages()
        return self.graph_generator.generate(packages, self.services(packages)).export()