/*
 * Copyright (C) 2019 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <hidl-util/FQName.h>
#include <hidl-util/Formatter.h>
#include <json/json.h>

#include <algorithm>
#include <iostream>
#include <vector>

#include "AST.h"
#include "Coordinator.h"
#include "Lint.h"
#include "LintRegistry.h"
#include "Location.h"

using namespace android;

static void usage(const char* me) {
    Formatter out(stderr);

    out << "Usage: " << me << " [-j] ";
    Coordinator::emitOptionsUsageString(out);
    out << " FQNAME...\n\n";

    out << "Process FQNAME, PACKAGE(.SUBPACKAGE)*@[0-9]+.[0-9]+(::TYPE)?, and provide lints.\n\n";

    out.indent();
    out.indent();

    out << "-h: Prints this menu.\n";
    out << "-e: The script only errors if FQNAME does not compile (don't error on lints).\n";
    out << "-j: Prints output in JSON.\n";
    out.indent([&] {
        out << "{\n";
        out.indent([&] {
            out << "\"level\": \"warning\" | \"error\",\n";
            out << "\"message\": string,\n";
            out << "\"filename\": string,\n";

            out << "\"begin\": { \"line\" : number, \"column\" : number }\n";
            out << "\"end\": { \"line\" : number, \"column\" : number }\n";
        });
        out << "}\n\n";
    });
    Coordinator::emitOptionsDetailString(out);

    out.unindent();
    out.unindent();
}

int main(int argc, char** argv) {
    const char* me = argv[0];
    if (argc == 1) {
        usage(me);
        std::cerr << "ERROR: no fqname specified." << std::endl;
        exit(1);
    }

    bool machineReadable = false;
    bool errorOnLints = true;

    Coordinator coordinator;
    coordinator.parseOptions(argc, argv, "hje", [&](int res, char* /* arg */) {
        switch (res) {
            case 'j':
                machineReadable = true;
                break;
            case 'e':
                errorOnLints = false;
                break;
            case 'h':
            case '?':
            default: {
                usage(me);
                exit(1);
                break;
            }
        }
    });

    argc -= optind;
    argv += optind;

    if (argc == 0) {
        usage(me);
        std::cerr << "ERROR: no fqname specified." << std::endl;
        exit(1);
    }

    bool haveLints = false;
    Json::Value lintJsonArray(Json::arrayValue);
    for (int i = 0; i < argc; ++i) {
        const char* arg = argv[i];

        FQName fqName;
        if (!FQName::parse(arg, &fqName)) {
            std::cerr << "ERROR: Invalid fully-qualified name as argument: " << arg << "."
                      << std::endl;
            exit(1);
        }

        std::vector<FQName> targets;
        if (fqName.isFullyQualified()) {
            targets.push_back(fqName);
        } else {
            status_t err = coordinator.appendPackageInterfacesToVector(fqName, &targets);
            if (err != OK) {
                std::cerr << "ERROR: Could not get sources for: " << arg << "." << std::endl;
                exit(1);
            }
        }

        std::vector<Lint> lints;
        for (const FQName& target : targets) {
            AST* ast = coordinator.parse(target);
            if (ast == nullptr) {
                std::cerr << "ERROR: Could not parse " << target.name() << ". Aborting."
                          << std::endl;
                exit(1);
            }

            LintRegistry::get()->runAllLintFunctions(*ast, &lints);
        }

        haveLints = haveLints || !lints.empty();

        std::sort(lints.begin(), lints.end());
        if (machineReadable) {
            for (const Lint& lint : lints) {
                lintJsonArray.append(lint.asJson());
            }
        } else {
            if (!lints.empty()) {
                std::cout << "Lints for: " << fqName.string() << std::endl << std::endl;
            }

            for (const Lint& lint : lints) {
                std::cout << lint;
            }
        }
    }

    if (machineReadable) {
        Json::StreamWriterBuilder factory;
        std::unique_ptr<Json::StreamWriter> const writer(factory.newStreamWriter());
        writer->write(lintJsonArray, &std::cout);
    }

    return errorOnLints && haveLints;
}