#include "common.h"
#include "formatting.h"

void entitize_msg(char *buff, int size) {
	char *tmp = new char[size];

	int in = 0, out = 0;
	while(buff[in] && out < size) {
		switch(buff[in]) {
			case '\"': in++; strncpy(tmp + out, "&quot;", size - out); out += 6; break;
			case '&': in++; strncpy(tmp + out, "&amp;", size - out); out += 5; break;
			case '\'': in++; strncpy(tmp + out, "&apos;", size - out); out += 6; break;
			case '<': in++; strncpy(tmp + out, "&lt;", size - out); out += 4; break;
			case '>': in++; strncpy(tmp + out, "&gt;", size - out); out += 4; break;
			default:
				tmp[out++] = buff[in++];
		}
	}
	tmp[out] = 0;

	strncpy(buff, tmp, size);
	delete tmp;
	buff[size - 1] = 0;
}

void unentitize_msg(char *buff) {
	int in = 0, out = 0;
	while(buff[in]) {
		if(buff[in] == '&') {
			if(strncmp(buff + in, "&quot;", 6) == 0) {
				buff[out++] = '\"'; in += 6;
			} else if(strncmp(buff + in, "&amp;", 5) == 0) {
				buff[out++] = '&'; in += 5;
			} else if(strncmp(buff + in, "&apos;", 6) == 0) {
				buff[out++] = '\''; in += 6;
			} else if(strncmp(buff + in, "&lt;", 4) == 0) {
				buff[out++] = '<'; in += 4;
			} else if(strncmp(buff + in, "&gt;", 4) == 0) {
				buff[out++] = '>'; in += 4;
			} else
				buff[out++] = buff[in++];
		} else
			buff[out++] = buff[in++];
	}
	buff[out] = 0;
}

void entitize_nick(char *buff, int size) {
	char *tmp = new char[size];

	int in = 0, out = 0;
	while(buff[in] && out < size) {
		switch(buff[in]) {
			case '\'': in++; strncpy(tmp + out, "&#39;", size - out); out += 5; break;
			default:
				tmp[out++] = buff[in++];
		}
	}
	tmp[out] = 0;

	strncpy(buff, tmp, size);
	delete tmp;
	buff[size - 1] = 0;
}

void unentitize_nick(char *buff) {
	int in = 0, out = 0;
	while(buff[in]) {
		if(buff[in] == '&') {
			if(strncmp(buff + in, "&#39;", 5) == 0) {
				buff[out++] = '\''; in += 5;
			} else
				buff[out++] = buff[in++];
		} else
			buff[out++] = buff[in++];
	}
	buff[out] = 0;
}


void strip_tags(char *buff) {
	int in = 0, out = 0;
	while(buff[in]) {
		if(buff[in] == '<' && buff[in + 1] != 'i' && buff[in + 1] != 'a') {
			while(buff[in] && buff[in] != '>') in++;
			in++;
		} else if(buff[in] == '<' && buff[in + 1] == 'a') {
			// link
			while(buff[in] && buff[in] != '\'' && buff[in] != '\"') in++;
			in++;
			while(buff[in] && buff[in] != '\'' && buff[in] != '\"')
				buff[out++] = buff[in++];
			while(buff[in] && buff[in] != '>') in++;
			in++;
			// add a space after a link in case there's something inside the <a></a> tags
			if(buff[in] && buff[in] != ' ' && buff[in] != '\r' && buff[in] != '\n' && buff[in] != '\t')
				buff[out++] = ' ';
		} else
			buff[out++] = buff[in++];
	}
	buff[out] = 0;
}

char *words[25] = {"bigsmile", "growl", "mad", "scared", "tongue", "devil", "happy", "messed", "sidefrown", "upset", "frazzled", 
	"heart", "nerd", "sinister", "wink", "geek", "laugh", "oops", "smirk", "worried", "googles", "mohawk", "pirate", "straight", "kiss"};
char *symbols[25] = {":D", ":E", "X(", ":O", ":p", "}:)", ":)", "X)", ":{", "B|", ":Z", 
	":X", "Q)", ":B", ";-)", "B)", ":))", ":G", ":,", ":[", "%)", "-:", "P)", ":|", ":X"};

void encode_smileys(char *buff, int size) {
	char *tmp = new char[size];

	int in = 0, out = 0;
	char sb[128];
	bool done;
	int i, len, len_s;
	while(buff[in] && out < size) {
		done = false;
		for(i = 0; i < 25; i++) {
			len_s = strlen(symbols[i]);
			if(strncmp(buff + in, symbols[i], len_s) == 0) {
				mir_snprintf(sb, 128, "<i n=\"%s\"/>", words[i]);
				len = strlen(sb);
				strncpy(tmp + out, sb, size - out);
				out += len;
				in += len_s;
				done = true;
				break;
			}
		}
		if(!done) tmp[out++] = buff[in++];
	}
	tmp[out] = 0;

	strncpy(buff, tmp, size);
	delete tmp;
	buff[size - 1] = 0;
}

void decode_smileys(char *buff) {
	int in = 0, out = 0;
	char sb[128];
	bool done;
	int i, j, len, len_s;
	while(buff[in]) {
		if(buff[in] == '<') {
			done = false;
			for(i = 0; i < 25; i++) {
				mir_snprintf(sb, 128, "<i n=\"%s\"/>", words[i]);
				len = strlen(sb);
				if(strncmp(buff + in, sb, len) == 0) {
					len_s = strlen(symbols[i]);
					strncpy(buff + out, symbols[i], len_s);
					out += len_s;
					in += len;
					done = true;
					break;
				}
				mir_snprintf(sb, 128, "<i n='%s\'/>", words[i]);
				if(strncmp(buff + in, sb, len) == 0) {
					len_s = strlen(symbols[i]);
					strncpy(buff + out, symbols[i], len_s);
					out += len_s;
					in += len;
					done = true;
					break;
				}
			}
			if(!done) buff[out++] = buff[in++];
		} else
			buff[out++] = buff[in++];
	}
	buff[out] = 0;
}