src/developer/forensics/feedback_data/attachments/system_log_ptr.h

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

#ifndef SRC_DEVELOPER_FORENSICS_FEEDBACK_DATA_ATTACHMENTS_SYSTEM_LOG_PTR_H_
#define SRC_DEVELOPER_FORENSICS_FEEDBACK_DATA_ATTACHMENTS_SYSTEM_LOG_PTR_H_

#include <fuchsia/logger/cpp/fidl.h>
#include <lib/async/dispatcher.h>
#include <lib/fidl/cpp/binding.h>
#include <lib/fit/promise.h>
#include <lib/sys/cpp/service_directory.h>

#include <cinttypes>
#include <vector>

#include "src/developer/forensics/feedback_data/attachments/types.h"
#include "src/developer/forensics/utils/fidl/oneshot_ptr.h"
#include "src/developer/forensics/utils/fit/timeout.h"

namespace forensics {
namespace feedback_data {

// Collects and orders the system log.
//
// fuchsia.logger.Log is expected to be in |services|.
::fit::promise<AttachmentValue> CollectSystemLog(async_dispatcher_t* dispatcher,
                                                 std::shared_ptr<sys::ServiceDirectory> services,
                                                 fit::Timeout timeout);

// Wraps around fuchsia::logger::LogListenerSafePtr to handle establishing the connection, losing
// the connection, waiting for the callback, enforcing a timeout, etc.
//
// CollectLogs() is expected to be called only once.
class LogListener : public fuchsia::logger::LogListenerSafe {
 public:
  explicit LogListener(async_dispatcher_t* dispatcher,
                       std::shared_ptr<sys::ServiceDirectory> services);

  // Collects the logs and returns a promise to when the collection is done or the timeout over.
  ::fit::promise<AttachmentValue> CollectLogs(fit::Timeout timeout);

 private:
  // |fuchsia::logger::LogListenerSafe|
  void LogMany(::std::vector<fuchsia::logger::LogMessage> log, LogManyCallback done) override;
  void Log(fuchsia::logger::LogMessage log, LogCallback done) override;
  void Done() override;

  ::fidl::Binding<fuchsia::logger::LogListenerSafe> binding_;

  fidl::OneShotPtr<fuchsia::logger::Log> logger_;

  std::vector<fuchsia::logger::LogMessage> log_messages_;
};

}  // namespace feedback_data
}  // namespace forensics

#endif  // SRC_DEVELOPER_FORENSICS_FEEDBACK_DATA_ATTACHMENTS_SYSTEM_LOG_PTR_H_