icinga2/lib/remote/consolehandler.hpp

/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */

#ifndef CONSOLEHANDLER_H
#define CONSOLEHANDLER_H

#include "remote/httphandler.hpp"
#include "base/scriptframe.hpp"

namespace icinga
{

struct ApiScriptFrame
{
	double Seen{0};
	int NextLine{1};
	std::map<String, String> Lines;
	Dictionary::Ptr Locals;
};

class ConsoleHandler final : public HttpHandler
{
public:
	DECLARE_PTR_TYPEDEFS(ConsoleHandler);

	bool HandleRequest(
		AsioTlsStream& stream,
		const ApiUser::Ptr& user,
		boost::beast::http::request<boost::beast::http::string_body>& request,
		const Url::Ptr& url,
		boost::beast::http::response<boost::beast::http::string_body>& response,
		const Dictionary::Ptr& params,
		boost::asio::yield_context& yc,
		HttpServerConnection& server
	) override;

	static std::vector<String> GetAutocompletionSuggestions(const String& word, ScriptFrame& frame);

private:
	static bool ExecuteScriptHelper(boost::beast::http::request<boost::beast::http::string_body>& request,
		boost::beast::http::response<boost::beast::http::string_body>& response,
		const Dictionary::Ptr& params, const String& command, const String& session, bool sandboxed);
	static bool AutocompleteScriptHelper(boost::beast::http::request<boost::beast::http::string_body>& request,
		boost::beast::http::response<boost::beast::http::string_body>& response,
		const Dictionary::Ptr& params, const String& command, const String& session, bool sandboxed);

};

}

#endif /* CONSOLEHANDLER_H */