#!/usr/bin/env python
# Copyright 2017 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.

import argparse
import sys

from package_imports_resolver import PackageImportsResolver

class PackageLangageObserver:
    def __init__(self):
        self.languages = set()
        self.labels = []

    def import_resolved(self, config, config_path):
        for label in config.get("labels", []):
            self.labels.append(label)
        if config.get("languages"):
            self.languages.update(config.get("languages"))


def get_dep_from_package_name(package_name):
    if package_name[0] == '/':
        return '"%s"' % package_name
    return '"//%s"' % package_name

def main():
    parser = argparse.ArgumentParser(description="Determine languages used by a"
                                     + " given set of packages")
    parser.add_argument("--packages", help="list of packages", required=True)
    args = parser.parse_args()


    language_observer = PackageLangageObserver()
    imports_resolver = PackageImportsResolver(language_observer)
    imported = imports_resolver.resolve_imports(args.packages.split(","))
    languages = language_observer.languages
    labels = language_observer.labels

    if languages is None:
        return -1

    for language in ["cpp", "dart", "go", "rust"]:
        sys.stdout.write("have_%s = %s\n" %
                         (language, str(language in languages).lower()))
    sys.stdout.write("imported = [%s]\n" %
                     ",".join(map(get_dep_from_package_name, imported)))
    sys.stdout.write("labels = [%s]\n" %
                     ",".join(['"%s"' % label for label in labels]))
    return 0

if __name__ == "__main__":
    sys.exit(main())