// Copyright 2018 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.

#include "trace-test-utils/read_records.h"

#include <stdio.h>

#include <utility>

#include <trace-reader/reader_internal.h>

namespace trace_testing {

bool ReadRecordsAndErrors(const uint8_t* buffer, size_t buffer_size,
                          fbl::Vector<trace::Record>* out_records,
                          fbl::Vector<fbl::String>* out_errors) {
    trace::TraceReader reader(
        [out_records](trace::Record record) {
            out_records->push_back(std::move(record));
        },
        [out_errors](fbl::String error) {
            out_errors->push_back(std::move(error));
        });

    trace::internal::TraceBufferReader buffer_reader(
        [&reader](trace::Chunk chunk) {
            if (!reader.ReadRecords(chunk)) {
                // Nothing to do, error already recorded.
            }
        },
        [out_errors](fbl::String error) {
            out_errors->push_back(std::move(error));
        });

    return buffer_reader.ReadChunks(buffer, buffer_size);
}

bool ReadRecords(const uint8_t* buffer, size_t buffer_size,
                 fbl::Vector<trace::Record>* out_records) {
    fbl::Vector<fbl::String> errors;
    if (!ReadRecordsAndErrors(buffer, buffer_size, out_records, &errors)) {
        fprintf(stderr, "Error reading buffer\n");
        return false;
    }

    for (const auto& error : errors) {
        fprintf(stderr, "error: %s\n", error.c_str());
    }
    if (errors.size() != 0) {
        fprintf(stderr, "Errors encountered\n");
        return false;
    }

    return true;
}

}  // namespace trace_testing