LocalJobSource.hpp

/*
 * LocalJobSource.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_LOCAL_JOB_SOURCE_HPP
#define LAUNCHER_PLUGINS_LOCAL_JOB_SOURCE_HPP
 
#include <api/IJobSource.hpp>
 
#include <vector>
 
#include <api/Job.hpp>
#include <jobs/AbstractJobRepository.hpp>
#include <jobs/JobStatusNotifier.hpp>
 
#include "LocalJobRepository.hpp"
#include "LocalSecureCookie.hpp"
#include "LocalJobRunner.hpp"
 
namespace rstudio {
namespace launcher_plugins {
namespace local {
 
class LocalJobSource : public api::IJobSource
{
public:
   LocalJobSource(
      std::string in_hostname,
      jobs::JobStatusNotifierPtr in_jobStatusNotifier,
      std::shared_ptr<LocalJobRepository> in_jobRepository);
 
   Error initialize() override;
 
   bool cancelJob(api::JobPtr in_job, bool& out_isComplete, std::string& out_statusMessage) override;
 
   Error getConfiguration(const system::User&, api::JobSourceConfiguration& out_configuration) const override;
 
   Error getNetworkInfo(api::JobPtr in_job, api::NetworkInfo& out_networkInfo) const override;
 
   bool killJob(api::JobPtr in_job, bool& out_isComplete, std::string& out_statusMessage) override;
 
   bool resumeJob(api::JobPtr in_job, bool& out_isComplete, std::string& out_statusMessage) override;
 
   bool stopJob(api::JobPtr in_job, bool& out_isComplete, std::string& out_statusMessage) override;
 
   bool suspendJob(api::JobPtr in_job, bool& out_isComplete, std::string& out_statusMessage) override;
 
   Error submitJob(api::JobPtr io_job, bool& out_wasInvalidRequest) const override;
 
   Error createOutputStream(
      api::OutputType in_outputType,
      api::JobPtr in_job,
      api::AbstractOutputStream::OnOutput in_onOutput,
      api::AbstractOutputStream::OnComplete in_onComplete,
      api::AbstractOutputStream::OnError in_onError,
      api::OutputStreamPtr& out_outputStream) override;
 
   Error createResourceStream(
      api::ConstJobPtr in_job,
      comms::AbstractLauncherCommunicatorPtr in_launcherCommunicator,
      api::AbstractResourceStreamPtr& out_resourceStream) override;
 
private:
   const std::string m_hostname;
 
   std::shared_ptr<LocalJobRunner> m_jobRunner;
};
 
} // namespace local
} // namespace launcher_plugins
} // namespace rstudio
 
#endif