/*
 * Copyright 2018 Google Inc.
 *
 * 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 "aemu/base/testing/ProtobufMatchers.h"

#include <algorithm>
#include <regex>
#include <string>
#include <string_view>

#include "absl/log/check.h"
#include "gmock/gmock-matchers.h"
#include "gmock/gmock-more-matchers.h"
#include "google/protobuf/descriptor.h"
#include "google/protobuf/io/tokenizer.h"
#include "google/protobuf/message.h"
#include "google/protobuf/text_format.h"
#include "google/protobuf/util/message_differencer.h"

namespace android {
namespace internal {

// Utilities.
using google::protobuf::io::ColumnNumber;

class StringErrorCollector : public google::protobuf::io::ErrorCollector {
   public:
    explicit StringErrorCollector(std::string* error_text) : error_text_(error_text) {}

    void RecordError(int line, ColumnNumber column, absl::string_view message) override {
        std::ostringstream ss;
        ss << "ERROR: " << line << "(" << column << ")" << message << "\n";
        *error_text_ += ss.str();
    }

    void RecordWarning(int line, ColumnNumber column, absl::string_view message) override {
        std::ostringstream ss;
        ss << "WARNING: " << line << "(" << column << ")" << message << "\n";
        *error_text_ += ss.str();
    }

   private:
    std::string* error_text_;
    StringErrorCollector(const StringErrorCollector&) = delete;
    StringErrorCollector& operator=(const StringErrorCollector&) = delete;
};

bool ParsePartialFromAscii(const std::string& pb_ascii, google::protobuf::Message* proto,
                           std::string* error_text) {
    google::protobuf::TextFormat::Parser parser;
    StringErrorCollector collector(error_text);
    parser.RecordErrorsTo(&collector);
    parser.AllowPartialMessage(true);
    return parser.ParseFromString(pb_ascii, proto);
}

// Returns true iff p and q can be compared (i.e. have the same descriptor).
bool ProtoComparable(const google::protobuf::Message& p, const google::protobuf::Message& q) {
    return p.GetDescriptor() == q.GetDescriptor();
}

template <typename Container>
std::string JoinStringPieces(const Container& strings, std::string_view separator) {
    std::stringstream stream;
    std::string_view sep = "";
    for (const std::string_view str : strings) {
        stream << sep << str;
        sep = separator;
    }
    return stream.str();
}

// Find all the descriptors for the ignore_fields.
std::vector<const google::protobuf::FieldDescriptor*> GetFieldDescriptors(
    const google::protobuf::Descriptor* proto_descriptor,
    const std::vector<std::string>& ignore_fields) {
    std::vector<const google::protobuf::FieldDescriptor*> ignore_descriptors;
    std::vector<std::string_view> remaining_descriptors;

    const google::protobuf::DescriptorPool* pool = proto_descriptor->file()->pool();
    for (const std::string& name : ignore_fields) {
        if (const google::protobuf::FieldDescriptor* field = pool->FindFieldByName(name)) {
            ignore_descriptors.push_back(field);
        } else {
            remaining_descriptors.push_back(name);
        }
    }

    DCHECK(remaining_descriptors.empty())
        << "Could not find fields for proto " << proto_descriptor->full_name()
        << " with fully qualified names: " << JoinStringPieces(remaining_descriptors, ",");
    return ignore_descriptors;
}

// Sets the ignored fields corresponding to ignore_fields in differencer. Dies
// if any is invalid.
void SetIgnoredFieldsOrDie(const google::protobuf::Descriptor& root_descriptor,
                           const std::vector<std::string>& ignore_fields,
                           google::protobuf::util::MessageDifferencer* differencer) {
    if (!ignore_fields.empty()) {
        std::vector<const google::protobuf::FieldDescriptor*> ignore_descriptors =
            GetFieldDescriptors(&root_descriptor, ignore_fields);
        for (std::vector<const google::protobuf::FieldDescriptor*>::iterator it =
                 ignore_descriptors.begin();
             it != ignore_descriptors.end(); ++it) {
            differencer->IgnoreField(*it);
        }
    }
}

// Configures a MessageDifferencer and DefaultFieldComparator to use the logic
// described in comp. The configured differencer is the output of this function,
// but a FieldComparator must be provided to keep ownership clear.
void ConfigureDifferencer(const internal::ProtoComparison& comp,
                          google::protobuf::util::DefaultFieldComparator* comparator,
                          google::protobuf::util::MessageDifferencer* differencer,
                          const google::protobuf::Descriptor* descriptor) {
    differencer->set_message_field_comparison(comp.field_comp);
    differencer->set_scope(comp.scope);
    comparator->set_float_comparison(comp.float_comp);
    comparator->set_treat_nan_as_equal(comp.treating_nan_as_equal);
    differencer->set_repeated_field_comparison(comp.repeated_field_comp);
    SetIgnoredFieldsOrDie(*descriptor, comp.ignore_fields, differencer);
    if (comp.float_comp == internal::kProtoApproximate &&
        (comp.has_custom_margin || comp.has_custom_fraction)) {
        // Two fields will be considered equal if they're within the fraction
        // _or_ within the margin. So setting the fraction to 0.0 makes this
        // effectively a "SetMargin". Similarly, setting the margin to 0.0 makes
        // this effectively a "SetFraction".
        comparator->SetDefaultFractionAndMargin(comp.float_fraction, comp.float_margin);
    }
    differencer->set_field_comparator(comparator);
}

// Returns true iff actual and expected are comparable and match.  The
// comp argument specifies how two are compared.
bool ProtoCompare(const internal::ProtoComparison& comp, const google::protobuf::Message& actual,
                  const google::protobuf::Message& expected) {
    if (!ProtoComparable(actual, expected)) return false;

    google::protobuf::util::MessageDifferencer differencer;
    google::protobuf::util::DefaultFieldComparator field_comparator;
    ConfigureDifferencer(comp, &field_comparator, &differencer, actual.GetDescriptor());

    // It's important for 'expected' to be the first argument here, as
    // Compare() is not symmetric.  When we do a partial comparison,
    // only fields present in the first argument of Compare() are
    // considered.
    return differencer.Compare(expected, actual);
}

// Describes the types of the expected and the actual protocol buffer.
std::string DescribeTypes(const google::protobuf::Message& expected,
                          const google::protobuf::Message& actual) {
    return "whose type should be " + expected.GetDescriptor()->full_name() + " but actually is " +
           actual.GetDescriptor()->full_name();
}

// Prints the protocol buffer pointed to by proto.
std::string PrintProtoPointee(const google::protobuf::Message* proto) {
    if (proto == NULL) return "";

    return "which points to " + ::testing::PrintToString(*proto);
}

// Describes the differences between the two protocol buffers.
std::string DescribeDiff(const internal::ProtoComparison& comp,
                         const google::protobuf::Message& actual,
                         const google::protobuf::Message& expected) {
    google::protobuf::util::MessageDifferencer differencer;
    google::protobuf::util::DefaultFieldComparator field_comparator;
    ConfigureDifferencer(comp, &field_comparator, &differencer, actual.GetDescriptor());

    std::string diff;
    differencer.ReportDifferencesToString(&diff);

    // We must put 'expected' as the first argument here, as Compare()
    // reports the diff in terms of how the protobuf changes from the
    // first argument to the second argument.
    differencer.Compare(expected, actual);

    // Removes the trailing '\n' in the diff to make the output look nicer.
    if (diff.length() > 0 && *(diff.end() - 1) == '\n') {
        diff.erase(diff.end() - 1);
    }

    return "with the difference:\n" + diff;
}

bool ProtoMatcherBase::MatchAndExplain(
    const google::protobuf::Message& arg,
    bool is_matcher_for_pointer,  // true iff this matcher is used to match
                                  // a protobuf pointer.
    ::testing::MatchResultListener* listener) const {
    if (must_be_initialized_ && !arg.IsInitialized()) {
        *listener << "which isn't fully initialized";
        return false;
    }

    const google::protobuf::Message* const expected = CreateExpectedProto(arg, listener);
    if (expected == NULL) return false;

    // Protobufs of different types cannot be compared.
    const bool comparable = ProtoComparable(arg, *expected);
    const bool match = comparable && ProtoCompare(comp(), arg, *expected);

    // Explaining the match result is expensive.  We don't want to waste
    // time calculating an explanation if the listener isn't interested.
    if (listener->IsInterested()) {
        const char* sep = "";
        if (is_matcher_for_pointer) {
            *listener << PrintProtoPointee(&arg);
            sep = ",\n";
        }

        if (!comparable) {
            *listener << sep << DescribeTypes(*expected, arg);
        } else if (!match) {
            *listener << sep << DescribeDiff(comp(), arg, *expected);
        }
    }

    DeleteExpectedProto(expected);
    return match;
}

}  // namespace internal
}  // namespace android