orotool/src/timer_base.hpp

/* Copyright 2020, Michele Santullo
 * This file is part of orotool.
 *
 * Orotool is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Orotool is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Orotool.  If not, see <http://www.gnu.org/licenses/>.
 */

#pragma once

#include "eventia/private/context.hpp"
#include "eventia/timer.hpp"
#include "oro/dboperation.hpp"
#include "oro/source.hpp"

namespace roar11 {
	class ThreadPool;
} //namespace roar11

namespace oro {
	class Api;
	class OriginsDB;
	struct Header;
	struct Shops;
	struct Icons;
	struct Items;
	struct Creators;
} //namespace oro

namespace duck {

struct TimerSettings {
	TimerSettings (double min_wait, double extra_delay, bool with_raw_json) :
		min_wait(min_wait),
		extra_delay(extra_delay),
		store_raw_json(with_raw_json)
	{ }

	double min_wait;
	double extra_delay;
	bool store_raw_json;
};

class TimerBase : public eve::Timer {
public:
	TimerBase (
		const eve::Context& ctx,
		oro::DBOperation type,
		const TimerSettings& settings,
		roar11::ThreadPool* pool,
		oro::Api* oro_api,
		oro::OriginsDB* db
	);
	virtual ~TimerBase() noexcept = default;

	virtual void on_timer() override;

protected:
	void set_next_timer (const oro::Header& header);
	void update_db (const oro::Shops& shops, const oro::Header& header, const oro::Source& source);
	void update_db (const oro::Items& items, const oro::Header& header, const oro::Source& source);
	void update_db (const oro::Icons& icons, const oro::Header& header, const oro::Source& source);
	void update_db (const oro::Creators& creators, const oro::Header& header, const oro::Source& source);
	roar11::ThreadPool& pool();
	oro::Api& oro_api();
	oro::OriginsDB& db();

private:
	virtual void fetch_data(bool with_raw) = 0;

	double m_extra_delay;
	roar11::ThreadPool* m_pool;
	oro::Api* m_oro_api;
	oro::OriginsDB* m_db;
	bool m_store_raw_response;
};

} //namespace duck