AbstractOutputStream.hpp

/*
 * AbstractOutputStream.hpp
 *
 * Copyright (C) 2020 by RStudio, PBC
 *
 * Unless you have received this program directly from RStudio pursuant to the terms of a commercial license agreement
 * with RStudio, then this program is licensed to you under the following terms:
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
 * documentation files (the "Software"), to deal in the Software without restriction, including without limitation the
 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all copies or substantial portions of the
 * Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
 * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
 * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 *
 */
 
#ifndef LAUNCHER_PLUGINS_ABSTRACT_OUTPUT_STREAM_HPP
#define LAUNCHER_PLUGINS_ABSTRACT_OUTPUT_STREAM_HPP
 
#include <memory>
 
#include <functional>
 
#include <PImpl.hpp>
#include <api/Job.hpp>
 
namespace rstudio {
namespace launcher_plugins {
namespace api {
 
enum class OutputType
{
   FIRST    = 0,
 
   STDOUT   = 0,
 
   STDERR   = 1,
 
   BOTH     = 2,
 
   LAST
};
 
class AbstractOutputStream
{
public:
   typedef std::function<void(uint64_t)> OnComplete;
   typedef std::function<void(const Error&)> OnError;
   typedef std::function<void(const std::string&, OutputType, uint64_t)> OnOutput;
 
   virtual ~AbstractOutputStream() = default;
 
   virtual Error start() = 0;
 
   virtual void stop() = 0;
 
protected:
   AbstractOutputStream(
      OutputType in_outputType,
      JobPtr in_job,
      OnOutput in_onOutput,
      OnComplete in_onComplete,
      OnError in_onError);
 
   void reportData(const std::string& in_data, OutputType in_outputType);
 
   void reportError(const Error& in_error);
 
   void setStreamComplete();
 
   OutputType m_outputType;
 
   JobPtr m_job;
 
private:
   // The private implementation of AbstractOutputStream.
   PRIVATE_IMPL(m_baseImpl);
};
 
typedef std::shared_ptr<AbstractOutputStream> OutputStreamPtr;
 
} // namespace api
} // namespace launcher_plugins
} // namespace rstudio
 
#endif