/*
Copyright © 2009 Jim Porter

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

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

#include "ui.h"

#include <cstdio>
#include <commctrl.h>

#include "proto.h"
#include "twitter.h"

static const TCHAR *sites[] = {
	_T("https://api.twitter.com/"),
	_T("https://identi.ca/api/")
};

INT_PTR CALLBACK first_run_dialog(HWND hwndDlg,UINT msg,WPARAM wParam,LPARAM lParam)
{
	TwitterProto *proto;

	switch(msg) 
	{
	case WM_INITDIALOG:
		TranslateDialogDefault(hwndDlg);

		proto = reinterpret_cast<TwitterProto*>(lParam);
		SetWindowLongPtr(hwndDlg,GWLP_USERDATA,lParam);

		DBVARIANT dbv;

		if(!db_get_ts(0,proto->ModuleName(),TWITTER_KEY_GROUP,&dbv))
		{
			SetDlgItemText(hwndDlg,IDC_GROUP,dbv.ptszVal);
			db_free(&dbv);
		}
		else
		{
			SetDlgItemText(hwndDlg,IDC_GROUP,L"Twitter");
		}

		if(!db_get_s(0,proto->ModuleName(),TWITTER_KEY_UN,&dbv))
		{
			SetDlgItemTextA(hwndDlg,IDC_USERNAME,dbv.pszVal);
			db_free(&dbv);
		}

		/*if ( !db_get_s(0,proto->ModuleName(),TWITTER_KEY_PASS,&dbv))
		{
			CallService(MS_DB_CRYPT_DECODESTRING,strlen(dbv.pszVal)+1,
				reinterpret_cast<LPARAM>(dbv.pszVal));
			SetDlgItemTextA(hwndDlg,IDC_PW,dbv.pszVal);
			db_free(&dbv);
		}*/

		for(size_t i=0; i<SIZEOF(sites); i++)
		{
			SendDlgItemMessage(hwndDlg,IDC_SERVER,CB_ADDSTRING,0,reinterpret_cast<LPARAM>(sites[i]));
		}
		if(!db_get_s(0,proto->ModuleName(),TWITTER_KEY_BASEURL,&dbv))
		{
			SetDlgItemTextA(hwndDlg,IDC_SERVER,dbv.pszVal);
			db_free(&dbv);
		}
		else
		{
			SendDlgItemMessage(hwndDlg,IDC_SERVER,CB_SETCURSEL,0,0);
		}

		return true;
	case WM_COMMAND:
		if(LOWORD(wParam) == IDC_NEWACCOUNTLINK)
		{
			CallService(MS_UTILS_OPENURL,1,reinterpret_cast<LPARAM>("https://twitter.com/signup"));
			return true;
		}

		if(GetWindowLongPtr(hwndDlg,GWLP_USERDATA)) // Window is done initializing
		{
			switch(HIWORD(wParam))
			{
			case EN_CHANGE:
			case CBN_EDITCHANGE:
			case CBN_SELCHANGE:
				SendMessage(GetParent(hwndDlg),PSM_CHANGED,0,0);
			}
		}
		break;

	case WM_NOTIFY: // might be able to get rid of this bit?
		if(reinterpret_cast<NMHDR*>(lParam)->code == PSN_APPLY) 
		{
			proto = reinterpret_cast<TwitterProto*>(GetWindowLongPtr(hwndDlg,GWLP_USERDATA));
			char str[128];
			TCHAR tstr[128];
			
			/*
			GetDlgItemTextA(hwndDlg,IDC_UN,str,sizeof(str));
			db_set_s(0,proto->ModuleName(),TWITTER_KEY_UN,str);

			GetDlgItemTextA(hwndDlg,IDC_PW,str,sizeof(str));
			CallService(MS_DB_CRYPT_ENCODESTRING,sizeof(str),reinterpret_cast<LPARAM>(str));
			db_set_s(0,proto->ModuleName(),TWITTER_KEY_PASS,str);
			*/

			GetDlgItemTextA(hwndDlg,IDC_SERVER,str,sizeof(str)-1);
			if(str[strlen(str)-1] != '/')
				strncat(str,"/",sizeof(str));
			db_set_s(0,proto->ModuleName(),TWITTER_KEY_BASEURL,str);

			GetDlgItemText(hwndDlg,IDC_GROUP,tstr,SIZEOF(tstr));
			db_set_ts(0,proto->ModuleName(),TWITTER_KEY_GROUP,tstr);

			return true;
		}
		break;
	}

	return false;
}

INT_PTR CALLBACK tweet_proc(HWND hwndDlg,UINT msg,WPARAM wParam,LPARAM lParam)
{
	TwitterProto *proto;

	switch(msg)
	{
	case WM_INITDIALOG:
		TranslateDialogDefault(hwndDlg);

		proto = reinterpret_cast<TwitterProto*>(lParam);
		SetWindowLongPtr(hwndDlg,GWLP_USERDATA,lParam);
		SendDlgItemMessage(hwndDlg,IDC_TWEETMSG,EM_LIMITTEXT,140,0);
		SetDlgItemText(hwndDlg,IDC_CHARACTERS,_T("140"));

		// Set window title
		TCHAR title[512];
		mir_sntprintf(title,SIZEOF(title),_T("Send Tweet for %s"),proto->m_tszUserName);
		SendMessage(hwndDlg,WM_SETTEXT,0,(LPARAM)title);

		return true;
	case WM_COMMAND:
		if(LOWORD(wParam) == IDOK)
		{
			TCHAR msg[141];
			proto = reinterpret_cast<TwitterProto*>(GetWindowLongPtr(hwndDlg,GWLP_USERDATA));

			GetDlgItemText(hwndDlg,IDC_TWEETMSG,msg,SIZEOF(msg));
			ShowWindow(hwndDlg,SW_HIDE);

			char *narrow = mir_t2a_cp(msg,CP_UTF8);
			proto->ForkThread(&TwitterProto::SendTweetWorker, narrow);

			EndDialog(hwndDlg, wParam); 
			return true;
		}
		else if(LOWORD(wParam) == IDCANCEL)
		{
			EndDialog(hwndDlg, wParam);
			return true;
		}
		else if(LOWORD(wParam) == IDC_TWEETMSG && HIWORD(wParam) == EN_CHANGE)
		{
			size_t len = SendDlgItemMessage(hwndDlg,IDC_TWEETMSG,WM_GETTEXTLENGTH,0,0);
			char str[4];
			mir_snprintf(str,sizeof(str),"%d",140-len);
			SetDlgItemTextA(hwndDlg,IDC_CHARACTERS,str);

			return true;
		}

		break;
	case WM_SETREPLY:
		{
			char foo[512];
			mir_snprintf(foo,sizeof(foo),"@%s ",(char*)wParam);
			size_t len = strlen(foo);

			SetDlgItemTextA(hwndDlg,IDC_TWEETMSG,foo);
			SendDlgItemMessage(hwndDlg,IDC_TWEETMSG,EM_SETSEL,len,len);

			char str[4];
			mir_snprintf(str,sizeof(str),"%d",140-len);
			SetDlgItemTextA(hwndDlg,IDC_CHARACTERS,str);

			return true;
		}
		break;
	}

	return false;
}

INT_PTR CALLBACK options_proc(HWND hwndDlg,UINT msg,WPARAM wParam,LPARAM lParam)
{
	TwitterProto *proto;

	switch(msg) 
	{
	case WM_INITDIALOG:
		TranslateDialogDefault(hwndDlg);

		proto = reinterpret_cast<TwitterProto*>(lParam);

		DBVARIANT dbv;
		if(!db_get_s(0,proto->ModuleName(),TWITTER_KEY_UN,&dbv))
		{
			SetDlgItemTextA(hwndDlg,IDC_UN,dbv.pszVal);
			db_free(&dbv);
		}

		/*if( !db_get_s(0,proto->ModuleName(),TWITTER_KEY_PASS,&dbv))
		{
			CallService(MS_DB_CRYPT_DECODESTRING,strlen(dbv.pszVal)+1,
				reinterpret_cast<LPARAM>(dbv.pszVal));
			SetDlgItemTextA(hwndDlg,IDC_PW,dbv.pszVal);
			db_free(&dbv);
		}*/

		CheckDlgButton(hwndDlg,IDC_CHATFEED,db_get_b(0,proto->ModuleName(),TWITTER_KEY_CHATFEED,0));

		for(size_t i=0; i<SIZEOF(sites); i++)
		{
			SendDlgItemMessage(hwndDlg,IDC_BASEURL,CB_ADDSTRING,0,reinterpret_cast<LPARAM>(sites[i]));
		}

		if(!db_get_s(0,proto->ModuleName(),TWITTER_KEY_BASEURL,&dbv))
		{
			SetDlgItemTextA(hwndDlg,IDC_BASEURL,dbv.pszVal);
			db_free(&dbv);
		}
		else
		{
			SendDlgItemMessage(hwndDlg,IDC_BASEURL,CB_SETCURSEL,0,0);
		}
		
		char pollrate_str[32];
		mir_snprintf(pollrate_str,sizeof(pollrate_str),"%d",db_get_dw(0,proto->ModuleName(),TWITTER_KEY_POLLRATE,80));
		SetDlgItemTextA(hwndDlg,IDC_POLLRATE,pollrate_str);

		CheckDlgButton(hwndDlg,IDC_TWEET_MSG,db_get_b(0,proto->ModuleName(),TWITTER_KEY_TWEET_TO_MSG,0));


		// Do this last so that any events propagated by pre-filling the form don't
		// instigate a PSM_CHANGED message
		SetWindowLongPtr(hwndDlg,GWLP_USERDATA,lParam);

		break;
	case WM_COMMAND:
		if(GetWindowLongPtr(hwndDlg,GWLP_USERDATA)) // Window is done initializing
		{
			switch(HIWORD(wParam))
			{
			case EN_CHANGE:
			case BN_CLICKED:
			case CBN_EDITCHANGE:
			case CBN_SELCHANGE:
				switch(LOWORD(wParam))
				{
				case IDC_UN:
				case IDC_PW:
				case IDC_BASEURL:
					ShowWindow(GetDlgItem(hwndDlg,IDC_RECONNECT),SW_SHOW);
				}
				SendMessage(GetParent(hwndDlg),PSM_CHANGED,0,0);
			}
		}
			
		break;
	case WM_NOTIFY:
		if(reinterpret_cast<NMHDR*>(lParam)->code == PSN_APPLY) 
		{
			proto = reinterpret_cast<TwitterProto*>(GetWindowLongPtr(hwndDlg,GWLP_USERDATA));
			char str[128];

			GetDlgItemTextA(hwndDlg,IDC_UN,str,sizeof(str));
			db_set_s(0,proto->ModuleName(),TWITTER_KEY_UN,str);

			/*GetDlgItemTextA(hwndDlg,IDC_PW,str,sizeof(str));
			CallService(MS_DB_CRYPT_ENCODESTRING,sizeof(str),reinterpret_cast<LPARAM>(str));
			db_set_s(0,proto->ModuleName(),TWITTER_KEY_PASS,str);*/

			GetDlgItemTextA(hwndDlg,IDC_BASEURL,str,sizeof(str)-1);
			if(str[strlen(str)-1] != '/')
				strncat(str,"/",sizeof(str));
			db_set_s(0,proto->ModuleName(),TWITTER_KEY_BASEURL,str);

			db_set_b(0,proto->ModuleName(),TWITTER_KEY_CHATFEED,IsDlgButtonChecked(hwndDlg,IDC_CHATFEED));

			GetDlgItemTextA(hwndDlg,IDC_POLLRATE,str,sizeof(str));
			int rate = atoi(str);
			if(rate == 0)
				rate = 80;
			db_set_dw(0,proto->ModuleName(),TWITTER_KEY_POLLRATE,rate);

			db_set_b(0,proto->ModuleName(),TWITTER_KEY_TWEET_TO_MSG,IsDlgButtonChecked(hwndDlg,IDC_TWEET_MSG));

			proto->UpdateSettings();
			return true;
		}
	}

	return false;
}

namespace popup_options
{
	static int get_timeout(HWND hwndDlg)
	{
		if(IsDlgButtonChecked(hwndDlg,IDC_TIMEOUT_PERMANENT))
			return -1;
		else if(IsDlgButtonChecked(hwndDlg,IDC_TIMEOUT_CUSTOM))
		{
			char str[32];
			GetDlgItemTextA(hwndDlg,IDC_TIMEOUT,str,sizeof(str));
			return atoi(str);
		}
		else // Default checked (probably)
			return 0;
	}

	static COLORREF get_text_color(HWND hwndDlg,bool for_db)
	{
		if(IsDlgButtonChecked(hwndDlg,IDC_COL_WINDOWS))
		{
			if(for_db)
				return -1;
			else
				return GetSysColor(COLOR_WINDOWTEXT);
		}
		else if(IsDlgButtonChecked(hwndDlg,IDC_COL_CUSTOM))
			return (COLORREF)SendDlgItemMessage(hwndDlg,IDC_COLTEXT,CPM_GETCOLOUR,0,0);
		else // Default checked (probably)
			return 0;
	}

	static COLORREF get_back_color(HWND hwndDlg,bool for_db)
	{
		if(IsDlgButtonChecked(hwndDlg,IDC_COL_WINDOWS))
		{
			if(for_db)
				return -1;
			else
				return GetSysColor(COLOR_WINDOW);
		}
		else if(IsDlgButtonChecked(hwndDlg,IDC_COL_CUSTOM))
			return (COLORREF)SendDlgItemMessage(hwndDlg,IDC_COLBACK,CPM_GETCOLOUR,0,0);
		else // Default checked (probably)
			return 0;
	}

	struct
	{
		TCHAR *name;
		TCHAR *text;
	} const quotes[] = {
		{ _T("Dorothy Parker"), _T("If, with the literate, I am\n")
		                        _T("Impelled to try an epigram,\n")
		                        _T("I never seek to take the credit;\n")
		                        _T("We all assume that Oscar said it.") },
		{ _T("Steve Ballmer"),  _T("I have never, honestly, thrown a chair in my life.") },
		{ _T("James Joyce"),    _T("I think I would know Nora's fart anywhere. I think ")
		                        _T("I could pick hers out in a roomful of farting women.") },
		{ _T("Brooke Shields"), _T("Smoking kills. If you're killed, you've lost a very ")
		                        _T("important part of your life.") },
		{ _T("Yogi Berra"),     _T("Always go to other peoples' funerals, otherwise ")
		                        _T("they won't go to yours.") },
	};

	static void preview(HWND hwndDlg)
	{
		POPUPDATAT popup = {};

		// Pick a random contact
		MCONTACT hContact = 0;
		int n_contacts = (int)CallService(MS_DB_CONTACT_GETCOUNT,0,0);

		if(n_contacts != 0)
		{
			int contact = rand() % n_contacts;
			hContact = db_find_first();
			for(int i=0; i<contact; i++)
				hContact = db_find_next(hContact);
		}

		// Pick a random quote
		int q = rand() % SIZEOF(quotes);
		_tcsncpy(popup.lptzContactName,quotes[q].name,MAX_CONTACTNAME);
		_tcsncpy(popup.lptzText,       quotes[q].text,MAX_SECONDLINE);

		popup.lchContact = hContact;
		popup.iSeconds   = get_timeout(hwndDlg);
		popup.colorText  = get_text_color(hwndDlg,false);
		popup.colorBack  = get_back_color(hwndDlg,false);

		PUAddPopupT(&popup);
	}
}

void CheckAndUpdateDlgButton(HWND hWnd,int button,BOOL check)
{
	CheckDlgButton(hWnd,button,check);
	SendMessage(hWnd,WM_COMMAND,MAKELONG(button,BN_CLICKED),
		(LPARAM)GetDlgItem(hWnd,button));
}

INT_PTR CALLBACK popup_options_proc(HWND hwndDlg,UINT msg,WPARAM wParam,LPARAM lParam)
{
	using namespace popup_options;
	TwitterProto *proto;

	int text_color,back_color,timeout;

	switch(msg) 
	{
	case WM_INITDIALOG:
		TranslateDialogDefault(hwndDlg);

		proto = reinterpret_cast<TwitterProto*>(lParam);

		CheckAndUpdateDlgButton(hwndDlg,IDC_SHOWPOPUPS,db_get_b(0,proto->ModuleName(),TWITTER_KEY_POPUP_SHOW,0));
		CheckDlgButton(hwndDlg,IDC_NOSIGNONPOPUPS,!db_get_b(0,proto->ModuleName(),TWITTER_KEY_POPUP_SIGNON,0));


		// ***** Get color information
		back_color = db_get_dw(0,proto->ModuleName(),TWITTER_KEY_POPUP_COLBACK,0);
		text_color = db_get_dw(0,proto->ModuleName(),TWITTER_KEY_POPUP_COLTEXT,0);

		SendDlgItemMessage(hwndDlg,IDC_COLBACK,CPM_SETCOLOUR,0,RGB(255,255,255));
		SendDlgItemMessage(hwndDlg,IDC_COLTEXT,CPM_SETCOLOUR,0,RGB(  0,  0,  0));

		if(back_color == -1 && text_color == -1)            // Windows defaults
			CheckAndUpdateDlgButton(hwndDlg,IDC_COL_WINDOWS,true);
		else if(back_color == 0 && text_color == 0)         // Popup defaults
			CheckAndUpdateDlgButton(hwndDlg,IDC_COL_POPUP,true);
		else                                                // Custom colors
		{
			CheckAndUpdateDlgButton(hwndDlg,IDC_COL_CUSTOM,true);
			SendDlgItemMessage(hwndDlg,IDC_COLBACK,CPM_SETCOLOUR,0,back_color);
			SendDlgItemMessage(hwndDlg,IDC_COLTEXT,CPM_SETCOLOUR,0,text_color);
		}

		// ***** Get timeout information
		timeout = db_get_dw(0,proto->ModuleName(),TWITTER_KEY_POPUP_TIMEOUT,0);
		SetDlgItemTextA(hwndDlg,IDC_TIMEOUT,"5");
		
		if(timeout == 0)
			CheckAndUpdateDlgButton(hwndDlg,IDC_TIMEOUT_DEFAULT,true);
		else if(timeout < 0)
			CheckAndUpdateDlgButton(hwndDlg,IDC_TIMEOUT_PERMANENT,true);
		else
		{
			char str[32];
			mir_snprintf(str,sizeof(str),"%d",timeout);
			SetDlgItemTextA(hwndDlg,IDC_TIMEOUT,str);
			CheckAndUpdateDlgButton(hwndDlg,IDC_TIMEOUT_CUSTOM,true);
		}

		SendDlgItemMessage(hwndDlg,IDC_TIMEOUT_SPIN,UDM_SETRANGE32,1,INT_MAX);
		SetWindowLongPtr(hwndDlg,GWLP_USERDATA,lParam);

		return true;
	case WM_COMMAND:
		switch(HIWORD(wParam))
		{
		case BN_CLICKED:
			switch(LOWORD(wParam))
			{
			case IDC_SHOWPOPUPS:
				EnableWindow(GetDlgItem(hwndDlg,IDC_NOSIGNONPOPUPS),IsDlgButtonChecked(hwndDlg,IDC_SHOWPOPUPS));
				break;

			case IDC_COL_CUSTOM:
				EnableWindow(GetDlgItem(hwndDlg,IDC_COLBACK),true);
				EnableWindow(GetDlgItem(hwndDlg,IDC_COLTEXT),true);
				break;
			case IDC_COL_WINDOWS:
			case IDC_COL_POPUP:
				EnableWindow(GetDlgItem(hwndDlg,IDC_COLBACK),false);
				EnableWindow(GetDlgItem(hwndDlg,IDC_COLTEXT),false);
				break;

			case IDC_TIMEOUT_CUSTOM:
				EnableWindow(GetDlgItem(hwndDlg,IDC_TIMEOUT),true);
				EnableWindow(GetDlgItem(hwndDlg,IDC_TIMEOUT_SPIN),true);
				break;
			case IDC_TIMEOUT_DEFAULT:
			case IDC_TIMEOUT_PERMANENT:
				EnableWindow(GetDlgItem(hwndDlg,IDC_TIMEOUT),false);
				EnableWindow(GetDlgItem(hwndDlg,IDC_TIMEOUT_SPIN),false);
				break;

			case IDC_PREVIEW:
				preview(hwndDlg);
				break;
			}

		case EN_CHANGE:
			if(GetWindowLongPtr(hwndDlg,GWLP_USERDATA)) // Window is done initializing
				SendMessage(GetParent(hwndDlg),PSM_CHANGED,0,0);
		}
		break;
	case WM_NOTIFY:
		if(reinterpret_cast<NMHDR*>(lParam)->code == PSN_APPLY) 
		{
			proto = reinterpret_cast<TwitterProto*>(GetWindowLongPtr(hwndDlg,GWLP_USERDATA));

			db_set_b(0,proto->ModuleName(),TWITTER_KEY_POPUP_SHOW,IsDlgButtonChecked(hwndDlg,IDC_SHOWPOPUPS));
			db_set_b(0,proto->ModuleName(),TWITTER_KEY_POPUP_SIGNON,!IsDlgButtonChecked(hwndDlg,IDC_NOSIGNONPOPUPS));

			// ***** Write color settings
			db_set_dw(0,proto->ModuleName(),TWITTER_KEY_POPUP_COLBACK,get_back_color(hwndDlg,true));
			db_set_dw(0,proto->ModuleName(),TWITTER_KEY_POPUP_COLTEXT,get_text_color(hwndDlg,true));

			// ***** Write timeout setting
			db_set_dw(0,proto->ModuleName(),TWITTER_KEY_POPUP_TIMEOUT,get_timeout(hwndDlg));

			return true;
		}
		break;
	}

	return false;
}


INT_PTR CALLBACK pin_proc(HWND hwndDlg,UINT msg,WPARAM wParam,LPARAM lParam)
{
	TwitterProto *proto;

	switch(msg)
	{
		case WM_INITDIALOG:
			TranslateDialogDefault(hwndDlg);

			SetWindowLongPtr(hwndDlg,GWLP_USERDATA,lParam);

			return true;
		case WM_COMMAND:
			if(LOWORD(wParam) == IDOK)
			{
				proto = reinterpret_cast<TwitterProto*>(GetWindowLongPtr(hwndDlg,GWLP_USERDATA));
				char str[128];

				GetDlgItemTextA(hwndDlg,IDC_PIN,str,sizeof(str));

				db_set_s(0,proto->ModuleName(),TWITTER_KEY_OAUTH_PIN,str);
				EndDialog(hwndDlg, wParam); 
				return true;
			}
			else if(LOWORD(wParam) == IDCANCEL)
			{
				EndDialog(hwndDlg, wParam);
				return true;
			}
			break;
	}

	return false;
}