#pragma once

#include <memory>
#include <functional>
#include <sqlite3.h>

class DB;
class DBStatement;

class DBQueryRow {
public:
	DBQueryRow(DBStatement &);

public:
	template <typename T>
	T col(int index, const T & default_value) const;
	template <typename T>
	T col(int index) const;

private:
	DBStatement & parent;
};

class DBQueryRowRange {
public:
	DBQueryRowRange(DBStatement & stmt);

private:
	class DBQueryRowIterator {
	public:
		DBQueryRowIterator(DBQueryRowRange & parent);

	public:
		DBQueryRow & operator * () const;
		DBQueryRowIterator & operator ++ ();
		bool operator != (const DBQueryRowIterator &) const;

	private:
		bool end = false;
		DBQueryRowRange & parent;
	};

public:
	DBQueryRowIterator begin();
	DBQueryRowIterator end();

private:
	DBStatement & parent;
	DBQueryRow row;
};

class DBStatement {
	friend class DBQueryRow;
	friend class DBQueryRowRange;

public:
	DBStatement(DB &, const std::string & query);

public:
	DBStatement & reset();
	DBStatement & bind(const std::string & text);
	DBStatement & bind(const int & number);
public:
	void execute();
	DBQueryRow row();
	DBQueryRowRange rows();

private:
	std::unique_ptr<sqlite3_stmt, std::function<void(sqlite3_stmt*)>> stmt;
	int param_index = 1;
	DB & parent;
};

class DB {
	friend class DBStatement;

public:
	DB(const std::string & path);
	DBStatement prepare(const std::string & query);

private:
	std::unique_ptr<sqlite3, std::function<void(sqlite3*)>> db = NULL;
};