blob: b72f5a3cce563bc2a38546e73dbb32d6c78e8f5f [file] [log] [blame]
// Copyright 2016 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 GARNET_BIN_TRACE_COMMAND_H_
#define GARNET_BIN_TRACE_COMMAND_H_
#include <functional>
#include <iosfwd>
#include <map>
#include <memory>
#include <string>
#include <tracing/cpp/fidl.h>
#include "lib/app/cpp/application_context.h"
#include "lib/fxl/command_line.h"
#include "lib/fxl/functional/closure.h"
#include "lib/fxl/macros.h"
namespace tracing {
class Command {
public:
// OnDoneCallback is the callback type invoked when a command finished
// running. It takes as argument the return code to exit the process with.
using OnDoneCallback = std::function<void(int32_t)>;
struct Info {
using CommandFactory =
std::function<std::unique_ptr<Command>(component::ApplicationContext*)>;
CommandFactory factory;
std::string name;
std::string usage;
std::map<std::string, std::string> options;
};
virtual ~Command();
void Run(const fxl::CommandLine& command_line, OnDoneCallback on_done);
protected:
static std::ostream& out();
explicit Command(component::ApplicationContext* context);
component::ApplicationContext* context();
component::ApplicationContext* context() const;
// Starts running the command.
// The command must invoke Done() when finished.
virtual void Start(const fxl::CommandLine& command_line) = 0;
void Done(int32_t return_code);
private:
component::ApplicationContext* context_;
OnDoneCallback on_done_;
int32_t return_code_ = -1;
FXL_DISALLOW_COPY_AND_ASSIGN(Command);
};
class CommandWithTraceController : public Command {
protected:
explicit CommandWithTraceController(component::ApplicationContext* context);
TraceControllerPtr& trace_controller();
const TraceControllerPtr& trace_controller() const;
private:
std::unique_ptr<component::ApplicationContext> context_;
TraceControllerPtr trace_controller_;
FXL_DISALLOW_COPY_AND_ASSIGN(CommandWithTraceController);
};
} // namespace tracing
#endif // GARNET_BIN_TRACE_COMMAND_H_