/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/*
 * Copyright (C) 2001-2003, Ximian, Inc.
 */

#include <ctype.h>
#include <fcntl.h>
#include <errno.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <unistd.h>

#include <glib.h>
#include <libsoup/soup-address.h>
#include <libsoup/soup-message.h>
#include <libsoup/soup-server.h>
#include <libsoup/soup-server-message.h>

static void
print_header (gpointer name, gpointer value, gpointer data)
{
	printf ("%s: %s\n", (char *)name, (char *)value);
}

static void
server_callback (SoupServerContext *context, SoupMessage *msg, gpointer data)
{
	char *path, *path_to_open, *slash;
	SoupMethodId method;
	struct stat st;
	int fd;

	path = soup_uri_to_string (soup_message_get_uri (msg), TRUE);
	printf ("%s %s HTTP/1.%d\n", msg->method, path,
		soup_message_get_http_version (msg));
	soup_message_foreach_header (msg->request_headers, print_header, NULL);
	if (msg->request.length)
		printf ("%.*s\n", msg->request.length, msg->request.body);

	method = soup_method_get_id (msg->method);
	if (method != SOUP_METHOD_ID_GET && method != SOUP_METHOD_ID_HEAD) {
		soup_message_set_status (msg, SOUP_STATUS_NOT_IMPLEMENTED);
		goto DONE;
	}

	if (path) {
		if (*path != '/') {
			soup_message_set_status (msg, SOUP_STATUS_BAD_REQUEST);
			goto DONE;
		}
	} else
		path = g_strdup ("");

	path_to_open = g_strdup_printf (".%s", path);

 AGAIN:
	if (stat (path_to_open, &st) == -1) {
		g_free (path_to_open);
		if (errno == EPERM)
			soup_message_set_status (msg, SOUP_STATUS_FORBIDDEN);
		else if (errno == ENOENT)
			soup_message_set_status (msg, SOUP_STATUS_NOT_FOUND);
		else
			soup_message_set_status (msg, SOUP_STATUS_INTERNAL_SERVER_ERROR);
		goto DONE;
	}

	if (S_ISDIR (st.st_mode)) {
		slash = strrchr (path_to_open, '/');
		if (!slash || slash[1]) {
			char *uri, *redir_uri;

			uri = soup_uri_to_string (soup_message_get_uri (msg), FALSE);
			redir_uri = g_strdup_printf ("%s/", uri);
			soup_message_add_header (msg->response_headers,
						 "Location", redir_uri);
			soup_message_set_status (msg, SOUP_STATUS_MOVED_PERMANENTLY);
			g_free (redir_uri);
			g_free (uri);
			g_free (path_to_open);
			goto DONE;
		}

		g_free (path_to_open);
		path_to_open = g_strdup_printf (".%s/index.html", path);
		goto AGAIN;
	}

	fd = open (path_to_open, O_RDONLY);
	g_free (path_to_open);
	if (fd == -1) {
		soup_message_set_status (msg, SOUP_STATUS_INTERNAL_SERVER_ERROR);
		goto DONE;
	}

	msg->response.owner = SOUP_BUFFER_SYSTEM_OWNED;
	msg->response.length = st.st_size;

	if (method == SOUP_METHOD_ID_GET) {
		msg->response.body = g_malloc (msg->response.length);
		read (fd, msg->response.body, msg->response.length);
	} else /* method == SOUP_METHOD_ID_HEAD */ {
		/* SoupServer will ignore response.body and only use
		 * response.length when responding to HEAD, so we
		 * could just use the same code for both GET and HEAD.
		 * But we'll optimize and avoid the extra malloc.
		 */
		msg->response.body = NULL;
	}

	close (fd);

	soup_message_set_status (msg, SOUP_STATUS_OK);

 DONE:
	g_free (path);
	printf ("  -> %d %s\n\n", msg->status_code, msg->reason_phrase);
}

static void
quit (int sig)
{
	/* Exit cleanly on ^C in case we're valgrinding. */
	exit (0);
}

int
main (int argc, char **argv)
{
	GMainLoop *loop;
	SoupServer *server, *ssl_server;
	int opt;
	int port = SOUP_ADDRESS_ANY_PORT;
	int ssl_port = SOUP_ADDRESS_ANY_PORT;
	const char *ssl_cert_file = NULL, *ssl_key_file = NULL;

	g_type_init ();
	g_thread_init (NULL);
	signal (SIGINT, quit);

	while ((opt = getopt (argc, argv, "p:k:c:s:")) != -1) {
		switch (opt) {
		case 'p':
			port = atoi (optarg);
			break;
		case 'k':
			ssl_key_file = optarg;
			break;
		case 'c':
			ssl_cert_file = optarg;
			break;
		case 's':
			ssl_port = atoi (optarg);
			break;
		default:
			fprintf (stderr, "Usage: %s [-p port] [-c ssl-cert-file -k ssl-key-file [-s ssl-port]]\n",
				 argv[0]);
			exit (1);
		}
	}

	server = soup_server_new (SOUP_SERVER_PORT, port,
				  NULL);
	if (!server) {
		fprintf (stderr, "Unable to bind to server port %d\n", port);
		exit (1);
	}
	soup_server_add_handler (server, NULL, NULL,
				 server_callback, NULL, NULL);
	printf ("\nStarting Server on port %d\n",
		soup_server_get_port (server));
	soup_server_run_async (server);

	if (ssl_cert_file && ssl_key_file) {
		ssl_server = soup_server_new (
			SOUP_SERVER_PORT, ssl_port,
			SOUP_SERVER_SSL_CERT_FILE, ssl_cert_file,
			SOUP_SERVER_SSL_KEY_FILE, ssl_key_file,
			NULL);

		if (!ssl_server) {
			fprintf (stderr, "Unable to bind to SSL server port %d\n", ssl_port);
			exit (1);
		}
		soup_server_add_handler (ssl_server, NULL, NULL,
					 server_callback, NULL, NULL);
		printf ("Starting SSL Server on port %d\n", 
			soup_server_get_port (ssl_server));
		soup_server_run_async (ssl_server);
	}

	printf ("\nWaiting for requests...\n");

	loop = g_main_loop_new (NULL, TRUE);
	g_main_loop_run (loop);

	return 0;
}