/*  XMMS - Cross-platform multimedia player
 *  Copyright (C) 1998-2000  Peter Alm, Mikael Alm, Olle Hallnas, Thomas Nilsson and 4Front Technologies
 *  Modified by Jens Burkal <jzb@rapanden.dk> for use with the Speex Xmms plugin
 *
 *  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, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <glib.h>
#include <gtk/gtk.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>

#include <pthread.h>

#include "http.h"
#include "xmms/util.h"
#include "xmms/plugin.h"

#include <speex.h>
#include "libspeex.h"


#define min(x,y) ((x)<(y)?(x):(y))
#define min3(x,y,z) (min(x,y)<(z)?min(x,y):(z))
#define min4(x,y,z,w) (min3(x,y,z)<(w)?min3(x,y,z):(w))

static gchar *ice_name = NULL;

static gboolean prebuffering;
static gint sock, rd_index, wr_index, buffer_length, prebuffer_length;
static guint64 buffer_read = 0;
static gchar *buffer;
static pthread_t thread;
static GtkWidget *error_dialog = NULL;


static FILE *output_file = NULL;

#define BASE64_LENGTH(len) (4 * (((len) + 2) / 3))

/* Encode the string S of length LENGTH to base64 format and place it
   to STORE.  STORE will be 0-terminated, and must point to a writable
   buffer of at least 1+BASE64_LENGTH(length) bytes.  */
static void base64_encode (const gchar *s, gchar *store, gint length)
{
	/* Conversion table.  */
	static gchar tbl[64] = {
		'A','B','C','D','E','F','G','H',
		'I','J','K','L','M','N','O','P',
		'Q','R','S','T','U','V','W','X',
		'Y','Z','a','b','c','d','e','f',
		'g','h','i','j','k','l','m','n',
		'o','p','q','r','s','t','u','v',
		'w','x','y','z','0','1','2','3',
		'4','5','6','7','8','9','+','/'
	};
	gint i;
	guchar *p = (guchar *)store;

	/* Transform the 3x8 bits to 4x6 bits, as required by base64.  */
	for (i = 0; i < length; i += 3)
	{
		*p++ = tbl[s[0] >> 2];
		*p++ = tbl[((s[0] & 3) << 4) + (s[1] >> 4)];
		*p++ = tbl[((s[1] & 0xf) << 2) + (s[2] >> 6)];
		*p++ = tbl[s[2] & 0x3f];
		s += 3;
	}
	/* Pad the result if necessary...  */
	if (i == length + 1)
		*(p - 1) = '=';
	else if (i == length + 2)
		*(p - 1) = *(p - 2) = '=';
	/* ...and zero-terminate it.  */
	*p = '\0';
}

/* Create the authentication header contents for the `Basic' scheme.
   This is done by encoding the string `USER:PASS' in base64 and
   prepending `HEADER: Basic ' to it.  */
static gchar *basic_authentication_encode (const gchar *user, const gchar *passwd, const gchar *header)
{
	gchar *t1, *t2, *res;
	gint len1 = strlen (user) + 1 + strlen (passwd);
	gint len2 = BASE64_LENGTH (len1);

	t1 = g_strdup_printf("%s:%s", user, passwd);
	t2 = g_malloc0(len2 + 1);
	base64_encode (t1, t2, len1);
	res = g_strdup_printf("%s: Basic %s\r\n", header, t2);
	g_free(t2);
	g_free(t1);
	
	return res;
}

static void parse_url(const gchar * url, gchar ** user, gchar ** pass, gchar ** host, int *port, gchar ** filename)
{
	gchar *h, *p, *pt, *f, *temp, *ptr;

	temp = g_strdup(url);
	ptr = temp;

	if (!strncasecmp("http://", ptr, 7))
		ptr += 7;
	h = strchr(ptr, '@');
	f = strchr(ptr, '/');
	if (h != NULL && (!f || h < f))
	{
		*h = '\0';
		p = strchr(ptr, ':');
		if (p != NULL && p < h)
		{
			*p = '\0';
			p++;
			*pass = g_strdup(p);
		}
		else
			*pass = NULL;
		*user = g_strdup(ptr);
		h++;
		ptr = h;
	}
	else
	{
		*user = NULL;
		*pass = NULL;
		h = ptr;
	}
	pt = strchr(ptr, ':');
	if (pt != NULL && (f == NULL || pt < f))
	{
		*pt = '\0';
		*port = atoi(pt + 1);
	}
	else
	{
		if (f)
			*f = '\0';
		*port = 80;
	}
	*host = g_strdup(h);
	
	if (f)
		*filename = g_strdup(f + 1);
	else
		*filename = NULL;
	g_free(temp);
}

void speex_http_close(void)
{
	speex_fs->playing = FALSE;

	pthread_join(thread, NULL);
	g_free(ice_name);
	ice_name = NULL;
}


static gint http_used(void)
{
	if (wr_index >= rd_index)
		return wr_index - rd_index;
	return buffer_length - (rd_index - wr_index);
}

static gint http_free(void)
{
	if (rd_index > wr_index)
		return (rd_index - wr_index) - 1;
	return (buffer_length - (wr_index - rd_index)) - 1;
}

static void http_wait_for_data(gint bytes)
{
	while ((prebuffering || http_used() < bytes) && !speex_fs->eof  && speex_fs->playing) 
		xmms_usleep(10000);
}

static void show_error_message(gchar *error)
{
	if(!error_dialog)
	{
		GDK_THREADS_ENTER();
		error_dialog = xmms_show_message(g_strdup("Error"), error, g_strdup("Ok"), FALSE,
						 NULL, NULL);
		gtk_signal_connect(GTK_OBJECT(error_dialog),
				   "destroy",
				   GTK_SIGNAL_FUNC(gtk_widget_destroyed),
				   &error_dialog);
		GDK_THREADS_LEAVE();
	}
}

int speex_http_read(gpointer data, gint length)
{
	gint len, cnt, off = 0;

	http_wait_for_data(length);

	if (!speex_fs->playing) 
		return 0;
	len = min(http_used(), length);

	while (len && http_used())
	{
		cnt = min3(len, buffer_length - rd_index, http_used());
		if (output_file)
			fwrite(buffer + rd_index, 1, cnt, output_file);

		memcpy((gchar *)data + off, buffer + rd_index, cnt);
		rd_index = (rd_index + cnt) % buffer_length;
		buffer_read += cnt;
		len -= cnt;
		off += cnt;
	}
	return off;
}

static gboolean http_check_for_data(void)
{

	fd_set set;
	struct timeval tv;
	gint ret;

	tv.tv_sec = 0;
	tv.tv_usec = 20000;
	FD_ZERO(&set);
	FD_SET(sock, &set);
	ret = select(sock + 1, &set, NULL, NULL, &tv);
	if (ret > 0)
		return TRUE;
	return FALSE;
}

gint speex_http_read_line(gchar * buf, gint size)
{
	gint i = 0;

	while (speex_fs->playing && i < size - 1)
	{
		if (http_check_for_data())
		{
			if (read(sock, buf + i, 1) <= 0)
				return -1;
			if (buf[i] == '\n')
				break;
			if (buf[i] != '\r')
				i++;
		}
	}
	if (!speex_fs->playing)
		return -1;
	buf[i] = '\0';
	return i;
}

static void *http_buffer_loop(void *arg)
{
	gchar line[1024], *user, *pass, *host, *filename,
	     *status, *url, *temp, *file;
	gchar *chost;
	gint cnt, written, error, err_len, port, cport;
	gboolean redirect;
	fd_set set;
	struct hostent *hp;
	struct sockaddr_in address;
	struct timeval tv;

	url = (gchar *) arg;
	do
	{
		redirect=FALSE;
	
		g_strstrip(url);

		parse_url(url, &user, &pass, &host, &port, &filename);

		if ((!filename || !*filename) && url[strlen(url) - 1] != '/')
			temp = g_strconcat(url, "/", NULL);
		else
			temp = g_strdup(url);
		g_free(url);
		url = temp;

		chost = speex_cfg->use_proxy ? speex_cfg->proxy_host : host;
		cport = speex_cfg->use_proxy ? speex_cfg->proxy_port : port;

		sock = socket(AF_INET, SOCK_STREAM, 0);
		fcntl(sock, F_SETFL, O_NONBLOCK);
		address.sin_family = AF_INET;

		status = g_strdup_printf("LOOKING UP %s", chost);
		speex_ip.set_info_text(status);
		g_free(status);

		if (!(hp = gethostbyname(chost)))
		{
			status = g_strdup_printf("Couldn't look up host %s", chost);
			show_error_message(status);
			g_free(status);

			speex_ip.set_info_text(NULL);
			speex_fs->eof = TRUE;
		}

		if (!speex_fs->eof)
		{
			memcpy(&address.sin_addr.s_addr, *(hp->h_addr_list), sizeof (address.sin_addr.s_addr));
			address.sin_port = g_htons(cport);

			status = g_strdup_printf("CONNECTING TO %s:%d", chost, cport);
			speex_ip.set_info_text(status);
			g_free(status);
			if (connect(sock, (struct sockaddr *) &address, sizeof (struct sockaddr_in)) == -1)
			{
				if (errno != EINPROGRESS)
				{
					status = g_strdup_printf("Couldn't connect to host %s", chost);
					show_error_message(status);
					g_free(status);

					speex_ip.set_info_text(NULL);
					speex_fs->eof = TRUE;
				}
			}
			while (speex_fs->playing)
			{
				tv.tv_sec = 0;
				tv.tv_usec = 10000;
				FD_ZERO(&set);
				FD_SET(sock, &set);
				if (select(sock + 1, NULL, &set, NULL, &tv) > 0)
				{
					err_len = sizeof (error);
					getsockopt(sock, SOL_SOCKET, SO_ERROR, &error, &err_len);
					if (error)
					{
						status = g_strdup_printf(("Couldn't connect to host %s"),
									 chost);
						show_error_message(status);
						g_free(status);

						speex_ip.set_info_text(NULL);
						speex_fs->eof = TRUE;
						
					}
					break;
				}
			}
			if (!speex_fs->eof)
			{
				gchar *auth = NULL, *proxy_auth = NULL;

				if(user && pass)
					auth = basic_authentication_encode(user, pass, "Authorization");

				if (speex_cfg->use_proxy)
				{
					file = g_strdup(url);
					if(speex_cfg->use_auth && speex_cfg->proxy_user && speex_cfg->proxy_passwd)
					{
						proxy_auth = basic_authentication_encode(speex_cfg->proxy_user,
											 speex_cfg->proxy_passwd,
											 "Proxy-Authorization");
					}
				}
				else
					file = g_strconcat("/", filename, NULL);
				temp = g_strdup_printf("GET %s HTTP/1.0\r\n"
						       "Host: %s\r\n"
						       "User-Agent: %s/%s\r\n"
						       "%s%s\r\n",
						       file, host, PACKAGE, VERSION, 
						       proxy_auth ? proxy_auth : "", auth ? auth : "");
				g_free(file);
				if(proxy_auth)
					g_free(proxy_auth);
				if(auth)
					g_free(auth);
				write(sock, temp, strlen(temp));
				g_free(temp);
				speex_ip.set_info_text("CONNECTED: WAITING FOR REPLY");
				while (speex_fs->playing && !speex_fs->eof)
				  {
					if (http_check_for_data())
					{
						if (speex_http_read_line(line, 1024))
						{
							status = strchr(line, ' ');
							if (status)
							{
								if (status[1] == '2')
									break;
								else if(status[1] == '3' && status[2] == '0' && status[3] == '2')
								{
									while(speex_fs->playing)
									{
										if(http_check_for_data())
										{
											if((cnt = speex_http_read_line(line, 1024)) != -1)
											{
												if(!cnt)
													break;
												if(!strncmp(line, "Location:", 9))
												{
													g_free(url);
													url = g_strdup(line+10);
												}
											}
											else
											{
												speex_fs->eof=TRUE;
												speex_ip.set_info_text(NULL);
												break;
											}
										}
									}			
									redirect=TRUE;
									break;
								}
								else
								{
									status = g_strdup_printf(("Couldn't connect to host %s\nServer reported: %s"), chost, status);
									show_error_message(status);
									g_free(status);
									break;
								}
							}
						}
						else
						{
							speex_fs->eof = TRUE;
							speex_ip.set_info_text(NULL);
						}
					}
				}

				while (speex_fs->playing && !redirect)
				{
					if (http_check_for_data())
					{
						if ((cnt = speex_http_read_line(line, 1024)) != -1)
						{
							if (!cnt)
								break;
							if (!strncmp(line, "ice-name:", 9))
								ice_name = g_strdup(line + 9);
							
						}
						else
						{
							speex_fs->eof = TRUE;
							speex_ip.set_info_text(NULL);
							break;
						}
					}
				}
			}
		}
	
		if(redirect)
		{
			if (output_file)
			{
				fclose(output_file);
				output_file = NULL;
			}
			close(sock);
			g_free(user);
			g_free(pass);
			g_free(host);
			g_free(filename);
		}		
	} while(redirect);
	
	if (speex_cfg->save_stream)
	{
		gchar *output_name;
		file = g_strdup(generate_title(url, NULL));
		output_name = file;
		if (!strncasecmp(output_name, "http://", 7))
			output_name += 7;
		temp = strrchr(output_name, '.');
		if (temp && !strcasecmp(temp, SPEEX_EXT))
			*temp = '\0';

		while ((temp = strchr(output_name, '/')))
			*temp = '_';
		output_name = g_strdup_printf("%s/%s" SPEEX_EXT, speex_cfg->save_path, output_name);

		g_free(file);

		output_file = fopen(output_name, "wb");
		g_free(output_name);
	}

	while (speex_fs->playing)
	{

		if (!http_used() && !speex_ip.output->buffer_playing())
			prebuffering = TRUE;
		if (http_free() > 0 && !speex_fs->eof)
		{
			if (http_check_for_data())
			{
				cnt = min(http_free(), buffer_length - wr_index);
				if (cnt > 1024)
					cnt = 1024;
				written = read(sock, buffer + wr_index, cnt);
				if (written <= 0)
				{
					speex_fs->eof = TRUE;
					if (prebuffering)
					{
						prebuffering = FALSE;

						speex_ip.set_info_text(NULL);
					}

				}
				else
					wr_index = (wr_index + written) % buffer_length;
			}

			if (prebuffering)
			{
				if (http_used() > prebuffer_length)
				{
					prebuffering = FALSE;
					speex_ip.set_info_text(NULL);
				}
				else
				{
					status = g_strdup_printf("PRE-BUFFERING: %dKB/%dKB", http_used() / 1024, prebuffer_length / 1024);
					speex_ip.set_info_text(status);
					g_free(status);
				}

			}
		}
		else
			xmms_usleep(10000);

	}
	if (output_file)
	{
		fclose(output_file);
		output_file = NULL;
	}
	close(sock);


	g_free(user);
	g_free(pass);
	g_free(host);
	g_free(filename);
	g_free(buffer);
	g_free(url);
	
	pthread_exit(NULL);
}

int speex_http_open(gchar * _url)
{
	gchar *url;


	
	url = g_strdup(_url);

	rd_index = 0;
	wr_index = 0;
	buffer_length = speex_cfg->buffersize * 1024;

	// Minimum buffer length
	if (buffer_length == 0) buffer_length = 1024;
	
	prebuffer_length = (buffer_length * speex_cfg->prebuffer) / 100;
	buffer_read = 0;
	prebuffering = TRUE;
	speex_fs->playing = TRUE;
	speex_fs->eof = FALSE;
	buffer = g_malloc(buffer_length);

	pthread_create(&thread, NULL, http_buffer_loop, url);

	return 0;
}

/* Unused

char* speex_http_get_title(char *url)
{
	if (ice_name)
		return g_strdup(ice_name);
	if (g_basename(url) && strlen(g_basename(url)) > 0)
		return g_strdup(g_basename(url));
	return g_strdup(url);
}

*/