/**
	This file is a part of rexy's matrix bot
	Copyright (C) 2019 rexy712

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU Affero General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	This program 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 Affero General Public License for more details.

	You should have received a copy of the GNU Affero General Public License
	along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef MATRIX_HPP
#define MATRIX_HPP

#include "raii/curler.hpp"
#include "raii/string.hpp"
#include "raii/rjp_string.hpp"
#include "raii/filerd.hpp"
#include "raii/rjp_ptr.hpp"

#include <vector>
#include <functional>

#define HAS_FREEIMAGE
#define HAS_FFMPEG

namespace matrix{
	struct auth_data{
		raii::rjp_string name;
		raii::rjp_string pass;
		raii::rjp_string homeserver;
		raii::rjp_string access_token;

		operator bool(void)const{
			return (name && pass && homeserver);
		}
	};

	//upload info structs
	struct file_info{
		raii::rjp_string fileurl;
		raii::string filename;
		raii::string mimetype;
		size_t filesize;
	};
	struct image_info : file_info{
		size_t width;
		size_t height;

		raii::rjp_string thumburl;
		size_t thumb_width;
		size_t thumb_height;
		size_t thumbsize;
	};
	struct video_info : public image_info{};
	struct audio_info : public file_info{};

	//message handling structs
	//enumerate message type but also give a string representation
	class msgtype{
	private:
		const char* m_str;
		const int m_num;
	public:
		constexpr msgtype(const char* s, int n):
			m_str(s), m_num(n){}
		constexpr msgtype(const msgtype&) = default;
		~msgtype(void) = default;
		constexpr msgtype& operator=(const msgtype&) = default;

		constexpr bool operator==(const msgtype& m){
			return m_num == m.m_num;
		}
		constexpr bool operator!=(const msgtype& m){
			return m_num != m.m_num;
		}
		constexpr const char* str(void)const{
			return m_str;
		}
		constexpr operator int(void)const{
			return m_num;
		}
	};
	//class enumeration
	struct msg{
	private:
		enum _types{
			_text, _audio, _video, _file, _image, _other
		};
	public:
		constexpr static msgtype text = msgtype("text", _text);
		constexpr static msgtype audio = msgtype("audio", _audio);
		constexpr static msgtype video = msgtype("video", _video);
		constexpr static msgtype file = msgtype("file", _file);
		constexpr static msgtype image = msgtype("image", _image);
		constexpr static msgtype other = msgtype("other", _other);

		constexpr static const msgtype& from_str(const char* str){
			if(!strcmp(str, "m.text")){
				return text;
			}else if(!strcmp(str, "m.audio")){
				return audio;
			}else if(!strcmp(str, "m.video")){
				return video;
			}else if(!strcmp(str, "m.file")){
				return file;
			}else if(!strcmp(str, "m.image")){
				return image;
			}else{
				return other;
			}
		}
	};

	struct event_info
	{
		const raii::rjp_string roomid;
		const raii::rjp_string sender;
		const raii::rjp_string eventid;
		const raii::rjp_string eventtype;
		const int serverts;
		const int age;
	};
	struct msg_info : public event_info
	{
		const msgtype type = msg::other;
		const raii::rjp_string body;
	};
	struct membership_info : public event_info
	{
		const raii::rjp_string recipient;
	};

	//main class
	class client
	{
	private:
		class mat_url_list
		{
		private:
			raii::string m_create_room;
			raii::string m_file_upload;
			raii::string m_room_list;
			raii::string m_login;
			raii::string m_alias_lookup;
			raii::string m_whoami;
			raii::string m_displayname;
			raii::string m_profile_picture;

		public:
			mat_url_list(void) = default;
			mat_url_list(const raii::string_base& homeserver);
			mat_url_list(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& userid);
			mat_url_list(const mat_url_list&) = default;
			mat_url_list(mat_url_list&&) = default;
			mat_url_list& operator=(const mat_url_list&) = default;
			mat_url_list& operator=(mat_url_list&&) = default;

			const raii::string& create_room(void)const;
			const raii::string& file_upload(void)const;
			const raii::string& room_list(void)const;
			const raii::string& login(void)const;
			const raii::string& alias_lookup(void)const;
			const raii::string& whoami(void)const;
			const raii::string& displayname(void)const;
			const raii::string& profile_picture(void)const;
			raii::string logout(const raii::string_base& homeserver, const raii::string_base& access_token)const;
			raii::string join_room(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid)const;
			raii::string leave_room(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid)const;
			raii::string sync(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& next_batch, const raii::string_base& timeout)const;
			raii::string read_receipt(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid, const raii::string_base& eventid)const;
			raii::string send(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid)const;
			raii::string redact(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid, const raii::string_base& eventid)const;
			raii::string power_level(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid)const;
			raii::string presence(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& userid)const;
			raii::string typing(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid, const raii::string_base& userid)const;
			raii::string kick(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid)const;
			raii::string ban(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid)const;
			raii::string unban(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid)const;
			raii::string invite(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid)const;
			raii::string room_members(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& roomid)const;
			raii::string password(const raii::string_base& homeserver, const raii::string_base& access_token)const;


			void repopulate_accesstoken(const raii::string_base& homeserver, const raii::string_base& access_token);
			void repopulate_userid(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& userid);
			void repopulate(const raii::string_base& homeserver, const raii::string_base& access_token, const raii::string_base& userid);
			void invalidate_accesstoken(void);

		private:
			void _initial_populate(const raii::string_base& homeserver);
			static constexpr const char* s_proto = "https://";
			static constexpr const char* s_revision = "r0";
		};

	private:
		mutable raii::curler m_curl;     //https access
		raii::string m_useragent;        //useragent to identify our application
		raii::string m_homeserver;       //name of our homeserver
		raii::rjp_string m_access_token; //authentication
		raii::rjp_string m_userid;       //userid including homeserver
		mat_url_list m_urls;

		raii::rjp_string m_next_batch;   //string which tracks where we are in the server history

		std::function<void(const client&,const msg_info&)> m_message_callback;
		std::function<void(const client&, const membership_info&)> m_membership_callback;
		std::function<void(const client&, const raii::rjp_ptr&)> m_raw_callback;

	public:
		client(const auth_data& a, const raii::string_base& useragent);
		client(const auth_data& a, raii::string&& useragent);
		client(const client& b) = default;
		client(client&& b) = default;
		~client(void) = default;

		client& operator=(const client&) = default;
		client& operator=(client&&) = default;

		//local getter
		const raii::rjp_string& access_token(void)const;
		const raii::rjp_string& userid(void)const;
		const raii::string& useragent(void)const;

		//local setter
		void set_useragent(const raii::string_base&);
		void set_useragent(raii::string&&);

		//networked setter
		void set_display_name(const raii::string_base&);
		void set_profile_picture(const raii::string_base&);


		//networked getter
		raii::rjp_string get_display_name(void)const;
		raii::rjp_string get_profile_picture(void)const;
		raii::rjp_string room_alias_to_id(const raii::string_base& alias)const;
		std::vector<raii::rjp_string> list_rooms(void)const;

		//other networked operations
		raii::string create_room(const raii::string_base& name, const raii::string_base& alias)const;
		bool join_room(const raii::string_base& roomid)const;
		bool leave_room(const raii::string_base& roomid)const;

		bool accept_invite(const membership_info& invite)const;
		bool reject_invite(const membership_info& invite)const;

		//upload media
		file_info upload_file(const raii::string_base& filename)const;
		file_info upload_file(const raii::string_base& filename, const raii::string_base& alias)const;
		image_info upload_image(const raii::string_base& filename)const;
		image_info upload_image(const raii::string_base& filename, const raii::string_base& alias)const;
		video_info upload_video(const raii::string_base& filename)const;
		video_info upload_video(const raii::string_base& filename, const raii::string_base& alias)const;
		audio_info upload_audio(const raii::string_base& filename)const;
		audio_info upload_audio(const raii::string_base& filename, const raii::string_base& alias)const;

		//send messages
		raii::rjp_string send_image(const raii::string_base& room, const image_info& image)const;
		raii::rjp_string send_video(const raii::string_base& room, const video_info& video)const;
		raii::rjp_string send_message(const raii::string_base& room, const raii::string_base& text)const;
		raii::rjp_string send_audio(const raii::string_base& room, const audio_info& audio)const;
		raii::rjp_string send_file(const raii::string_base& room, const file_info& file)const;
		void send_typing(const raii::string_base& room, bool active, int timeout = 5000)const;

		raii::rjp_string redact_event(const raii::string_base& roomid, const raii::string_base& eventid, const raii::string_base& reason)const;
		raii::rjp_string redact_event(const raii::string_base& roomid, const raii::string_base& eventid)const;

		template<class Func>
		void set_message_callback(Func&& f){
			m_message_callback = std::forward<Func>(f);
		}
		template<class Func>
		void set_membership_callback(Func&& f){
			m_membership_callback = std::forward<Func>(f);
		}
		template<class Func>
		void set_raw_callback(Func&& f){
			m_raw_callback = std::forward<Func>(f);
		}
		raii::string sync(size_t timeout);
		void logout(void);

	protected:
		void _handle_membership_events(RJP_value* rooms);
		void _handle_other_membership(RJP_value* join);
		void _handle_invites(RJP_value* invites);
		void _handle_messages(RJP_value* messages);
		void _send_read_receipt(const raii::string_base& roomid, const raii::string_base& eventid)const;
		raii::rjp_string _upload_file(raii::filerd& fp, const raii::curl_llist& header)const;
		raii::rjp_string _send_message(const raii::string_base& room, const raii::string_base& msg)const;
		static size_t _post_reply_curl_callback(char* ptr, size_t size, size_t nmemb, void* userdata);
		raii::string _get_curl(const raii::string_base& url)const;
		raii::string _post_curl(const raii::string_base& postdata, const raii::string_base& url, const raii::curl_llist& header)const;
		raii::string _put_curl(const raii::string_base& postdata, const raii::string_base& url, const raii::curl_llist& header)const;
		raii::rjp_string _post_and_find(const raii::string_base& data, const raii::string_base& url, const raii::curl_llist& header, const raii::string_base& target)const;
		raii::rjp_string _get_and_find(const raii::string_base& url, const raii::string_base& search)const;
		raii::rjp_string _curl_reply_search(const raii::string_base& reply, const raii::string_base& search)const;
		void _set_curl_defaults(void)const;
		raii::string _request_access_token(const auth_data& a)const;
		void _get_new_access_token(const auth_data& a);
		void _acquire_access_token(const auth_data& a);
	};


	auth_data parse_auth_data(RJP_value* root);
}

#endif