#include <algorithm>
#include <memory>
#include <cstdio>
#include <cstring>
#include <time.h>

#include <json-c/json.h>
#include <curl/curl.h>

#include <ocnews_api.h>
#include <utils.h>

#define OCNEWS_API "/index.php/apps/news/api/v1-2/"

namespace newsbeuter {

typedef std::unique_ptr<json_object, decltype(*json_object_put)> json_uptr;
typedef std::unique_ptr<CURL, decltype(*curl_easy_cleanup)> curl_uptr;

ocnews_api::ocnews_api(configcontainer* c) : remote_api(c) {
	server = cfg->get_configvalue("ocnews-url");

	if (server.empty())
		LOG(level::CRITICAL, "ocnews_api::ocnews_api: No owncloud server set");
}

ocnews_api::~ocnews_api() {
}

bool ocnews_api::authenticate() {
	auth = retrieve_auth();
	if (auth.empty()) {
		return false;
	}
	return query("status");
}

std::string ocnews_api::retrieve_auth() {
	credentials cred = get_credentials("ocnews", "ocNews");
	if (cred.user.empty() || cred.pass.empty()) {
		LOG(level::CRITICAL, "ocnews_api::retrieve_auth: No user and/or password set");
		return "";
	}
	return cred.user+":"+cred.pass;
}

std::vector<tagged_feedurl> ocnews_api::get_subscribed_urls() {
	std::vector<tagged_feedurl> result;
	std::map<long, std::string> folders_map;

	json_object* feeds_query;
	if (!query("feeds", &feeds_query))
		return result;
	json_uptr feeds_uptr(feeds_query, json_object_put);

	json_object* folders_query;
	if (!query("folders", &folders_query))
		return result;
	json_uptr folders_uptr(folders_query, json_object_put);

	json_object* folders;
	json_object_object_get_ex(folders_query, "folders", &folders);

	array_list* folders_list = json_object_get_array(folders);
	int folders_length = folders_list->length;

	for (int i=0; i<folders_length; i++) {
		json_object* folder = static_cast<json_object*>(folders_list->array[i]);
		json_object* node;

		json_object_object_get_ex(folder, "id", &node);
		long folder_id = json_object_get_int(node);

		json_object_object_get_ex(folder, "name", &node);
		folders_map[folder_id] = json_object_get_string(node);
	}

	rsspp::feed starred;
	starred.title = "Starred";
	starred.link = server;
	starred.rss_version = rsspp::OCNEWS_JSON;

	known_feeds["Starred"] = std::make_pair(starred, 0);
	result.push_back(tagged_feedurl("Starred", std::vector<std::string>()));

	json_object* feeds;
	json_object_object_get_ex(feeds_query, "feeds", &feeds);

	array_list* feeds_list = json_object_get_array(feeds);
	int feeds_length = feeds_list->length;

	for (int i=0; i<feeds_length; i++) {
		json_object* feed = static_cast<json_object*>(feeds_list->array[i]);
		json_object* node;
		rsspp::feed current_feed;

		current_feed.rss_version = rsspp::OCNEWS_JSON;

		json_object_object_get_ex(feed, "id", &node);
		long feed_id = json_object_get_int(node);

		json_object_object_get_ex(feed, "title", &node);
		current_feed.title = json_object_get_string(node);

		json_object_object_get_ex(feed, "url", &node);
		current_feed.link = json_object_get_string(node);

		while (known_feeds.find(current_feed.title) != known_feeds.end())
			current_feed.title += "*";
		known_feeds[current_feed.title] = std::make_pair(current_feed, feed_id);

		json_object_object_get_ex(feed, "folderId", &node);
		long folder_id = json_object_get_int(node);

		std::vector<std::string> tags;
		if (folder_id != 0)
			tags.push_back(folders_map[folder_id]);

		result.push_back(tagged_feedurl(current_feed.title, tags));
	}

	std::sort(++begin(result), end(result),
	[](const tagged_feedurl& a, const tagged_feedurl& b) {
		return a.first < b.first;
	});

	return result;
}

bool ocnews_api::mark_all_read(const std::string& feedurl) {
	long id = known_feeds[feedurl].second;

	std::string max = std::to_string(std::numeric_limits<long>::max());
	std::string query = "feeds/" + std::to_string(id)
	                    + "/read?newestItemId=" + max;

	return this->query(query, nullptr, "{}");
}

bool ocnews_api::mark_article_read(const std::string& guid, bool read) {
	std::string query = "items/";
	query += guid.substr(0, guid.find_first_of(":"));

	if (read)
		query += "/read";
	else
		query += "/unread";

	return this->query(query, nullptr, "{}");
}

bool ocnews_api::update_article_flags(const std::string& oldflags, const std::string& newflags, const std::string& guid) {
	std::string star_flag = cfg->get_configvalue("ocnews-flag-star");
	std::string query = "items/";
	query += guid.substr(guid.find_first_of(":")+1);

	if (star_flag.length() > 0) {
		if (strchr(oldflags.c_str(), star_flag[0])==nullptr && strchr(newflags.c_str(), star_flag[0])!=nullptr) {
			query += "/star";
		} else if (strchr(oldflags.c_str(), star_flag[0])!=nullptr && strchr(newflags.c_str(), star_flag[0])==nullptr) {
			query += "/unstar";
		}
	}

	return this->query(query, nullptr, "{}");;
}

rsspp::feed ocnews_api::fetch_feed(const std::string& feed_id) {
	rsspp::feed feed = known_feeds[feed_id].first;

	std::string query = "items?";
	query += "type="+std::to_string(known_feeds[feed_id].second != 0 ? 0 : 2);
	query += "&id="+std::to_string(known_feeds[feed_id].second);

	json_object* response;
	if (!this->query(query, &response))
		return feed;
	json_uptr response_uptr(response, json_object_put);

	json_object* items;
	json_object_object_get_ex(response, "items", &items);
	if (json_object_get_type(items) != json_type_array) {
		LOG(level::ERROR, "ocnews_api::fetch_feed: items is not an array");
		return feed;
	}

	array_list* list = json_object_get_array(items);
	int array_length = list->length;

	feed.items.clear();

	for (int i=0; i<array_length; i++) {
		json_object* item_j = static_cast<json_object*>(list->array[i]);
		json_object* node;
		rsspp::item item;

		json_object_object_get_ex(item_j, "title", &node);
		item.title = json_object_get_string(node);

		json_object_object_get_ex(item_j, "url", &node);
		item.link = json_object_get_string(node);

		json_object_object_get_ex(item_j, "author", &node);
		item.author = json_object_get_string(node);

		json_object_object_get_ex(item_j, "body", &node);
		item.content_encoded = json_object_get_string(node);

		json_object_object_get_ex(item_j, "id", &node);
		long id = json_object_get_int(node);

		json_object_object_get_ex(item_j, "feedId", &node);
		long f_id = json_object_get_int(node);

		json_object_object_get_ex(item_j, "guid", &node);
		item.guid = std::to_string(id) + ":" + std::to_string(f_id)
		            + "/" + json_object_get_string(node);

		json_object_object_get_ex(item_j, "unread", &node);
		bool unread = json_object_get_boolean(node);
		if (unread) {
			item.labels.push_back("ocnews:unread");
		} else {
			item.labels.push_back("ocnews:read");
		}

		json_object_object_get_ex(item_j, "pubDate", &node);
		time_t updated = (time_t)json_object_get_int(node);
		char rfc822_date[128];
		strftime(rfc822_date, sizeof(rfc822_date), "%a, %d %b %Y %H:%M:%S %z", gmtime(&updated));
		item.pubDate = rfc822_date;

		feed.items.push_back(item);
	}

	return feed;
}

void ocnews_api::add_custom_headers(curl_slist** /* custom_headers */) {
	// nothing required
}

bool ocnews_api::query(const std::string& query, json_object** result, const std::string& post) {
	curl_uptr curlhandle(curl_easy_init(), curl_easy_cleanup);
	CURL * handle = curlhandle.get();

	std::string url = server + OCNEWS_API + query;
	curl_easy_setopt(handle, CURLOPT_URL, url.c_str());

	utils::set_common_curl_options(handle, cfg);

	static auto write_fn = [] (void *buffer, size_t size, size_t nmemb, void *userp) {
		std::string * pbuf = static_cast<std::string *>(userp);
		pbuf->append(static_cast<const char *>(buffer), size * nmemb);
		return size * nmemb;
	};
	std::string buff;
	curl_easy_setopt(handle, CURLOPT_WRITEFUNCTION, *write_fn);
	curl_easy_setopt(handle, CURLOPT_WRITEDATA, &buff);

	if (!post.empty()) {
		curl_easy_setopt(handle, CURLOPT_POST, 1);
		curl_easy_setopt(handle, CURLOPT_POSTFIELDS, post.c_str());
		curl_easy_setopt(handle, CURLOPT_CUSTOMREQUEST, "PUT");
	}

	curl_easy_setopt(handle, CURLOPT_HTTPAUTH, CURLAUTH_BASIC);
	curl_easy_setopt(handle, CURLOPT_USERPWD, auth.c_str());

	CURLcode res = curl_easy_perform(handle);

	if (res != CURLE_OK && res != CURLE_HTTP_RETURNED_ERROR) {
		std::string msg = "ocnews_api::query: connection error code ";
		msg += std::to_string(res);
		LOG(level::CRITICAL, msg);
		return false;
	}

	long response_code;
	curl_easy_getinfo(handle, CURLINFO_RESPONSE_CODE, &response_code);
	if (response_code != 200) {
		if (response_code == 401)
			LOG(level::CRITICAL, "ocnews_api::query: authentication error");
		else {
			std::string msg = "ocnews_api::query: error ";
			msg += response_code;
			LOG(level::CRITICAL, msg);
		}
		return false;
	}

	if (result)
		*result = json_tokener_parse(buff.c_str());
	return true;
}

}