#include <u.h>
#include <libc.h>
#include <ctype.h>
#include <fcall.h>
#include <thread.h>
#include <9p.h>

#include "dat.h"
#include "fns.h"

typedef struct Webfid Webfid;
typedef struct Client Client;

struct Client
{
	Ref;

	char	request[16];
	Url	*baseurl;
	Url	*url;
	Key	*hdr;

	int	obody;	/* body opend */
	int	cbody;	/* body closed */
	Buq	*qbody;
};

struct Webfid
{
	int	level;

	Client	*client;
	Key	*key;	/* copy for Qheader */
	Buq	*buq;	/* reference for Qbody, Qpost */
};

enum {
	Qroot,
		Qrctl,
		Qclone,
		Qclient,
			Qctl,
			Qbody,
			Qpost,
			Qparsed,
				Qurl,
				Qurlschm,
				Qurluser,
				Qurlpass,
				Qurlhost,
				Qurlport,
				Qurlpath,
				Qurlqwry,
				Qurlfrag,
			Qheader,
};

static char *nametab[] = {
	"/",
		"ctl",
		"clone",
		nil,
			"ctl",
			"body",
			"postbody",
			"parsed",
				"url",
				"scheme",
				"user",
				"passwd",
				"host",
				"port",
				"path",
				"query",
				"fragment",
			nil,
};

static long time0;
static char *user;
static char *agent;
static Client client[64];
static int nclient;

#define	CLIENTID(c)	(((Client*)(c)) - client)

Client*
newclient(void)
{
	Client *cl;
	int i;

	for(i = 0; i < nclient; i++)
		if(client[i].ref == 0)
			break;
	if(i >= nelem(client))
		return nil;
	if(i == nclient)
		nclient++;
	cl = &client[i];
	incref(cl);

	cl->request[0] = 0;
	cl->baseurl = nil;
	cl->url = nil;
	cl->hdr = nil;
	cl->qbody = nil;
	
	return cl;
}

void
freeclient(Client *cl)
{
	Key *k;

	if(cl == nil || decref(cl))
		return;

	buclose(cl->qbody, 0);
	bufree(cl->qbody);

	while(k = cl->hdr){
		cl->hdr = k->next;
		free(k);
	}

	freeurl(cl->url);
	freeurl(cl->baseurl);

	memset(cl, 0, sizeof(*cl));
}

static Url*
clienturl(Client *cl)
{
	static Url nullurl;

	if(cl->qbody && cl->qbody->url)
		return cl->qbody->url;
	if(cl->url)
		return cl->url;
	return &nullurl;
}

static void*
wfaux(Webfid *f)
{
	if(f->level < Qclient)
		return nil;
	else if(f->level < Qurl)
		return f->client;
	else if(f->level < Qheader)
		return clienturl(f->client);
	else
		return f->key;
}

static void
fsmkqid(Qid *q, int level, void *aux)
{
	q->type = 0;
	q->vers = 0;
	switch(level){
	case Qroot:
	case Qparsed:
	case Qclient:
		q->type = QTDIR;
	default:
		q->path = (level<<24) | (((ulong)aux ^ time0) & 0x00ffffff);
	}
}

static char*
fshdrname(char *s)
{
	char *k, *w;

	for(k=w=s; *k; k++)
		if(isalnum(*k))
			*w++ = tolower(*k);
	*w = 0;
	return s;
}

static int
urlstr(char *buf, int nbuf, Url *u, int level)
{
	char *s;

	if(level == Qurl)
		return snprint(buf, nbuf, "%U", u);
	if(level == Qurlpath)
		return snprint(buf, nbuf, "%s", Upath(u));
	if((s = (&u->scheme)[level - Qurlschm]) == nil){
		buf[0] = 0;
		return 0;
	}
	return snprint(buf, nbuf, "%s", s);
}


static void
fsmkdir(Dir *d, int level, void *aux)
{
	char buf[1024];

	memset(d, 0, sizeof(*d));
	fsmkqid(&d->qid, level, aux);
	d->mode = 0444;
	d->atime = d->mtime = time0;
	d->uid = estrdup(user);
	d->gid = estrdup(user);
	d->muid = estrdup(user);
	if(d->qid.type & QTDIR)
		d->mode |= DMDIR | 0111;
	switch(level){
	case Qheader:
		d->name = fshdrname(estrdup(((Key*)aux)->key));
		d->length = strlen(((Key*)aux)->val);
		break;
	case Qclient:
		snprint(buf, sizeof(buf), "%ld", CLIENTID(aux));
		d->name = estrdup(buf);
		break;
	case Qctl:
	case Qrctl:
	case Qclone:
		d->mode = 0666;
		if(0){
	case Qpost:
		d->mode = 0222;
		}
	default:
		d->name = estrdup(nametab[level]);
		if(level >= Qurl && level <= Qurlfrag)
			d->length = urlstr(buf, sizeof(buf), (Url*)aux, level);
	}
}

static void
fsattach(Req *r)
{
	Webfid *f;

	if(r->ifcall.aname && r->ifcall.aname[0]){
		respond(r, "invalid attach specifier");
		return;
	}
	f = emalloc(sizeof(*f));
	f->level = Qroot;
	fsmkqid(&r->fid->qid, f->level, wfaux(f));
	r->ofcall.qid = r->fid->qid;
	r->fid->aux = f;
	respond(r, nil);
}

static void
fsstat(Req *r)
{
	Webfid *f;

	f = r->fid->aux;
	fsmkdir(&r->d, f->level, wfaux(f));
	respond(r, nil);
}

static char*
fswalk1(Fid *fid, char *name, Qid *qid)
{
	Webfid *f;
	int i, j;

	if(!(fid->qid.type&QTDIR))
		return "walk in non-directory";

	f = fid->aux;
	if(strcmp(name, "..") == 0){
		switch(f->level){
		case Qroot:
			break;
		case Qclient:
			freeclient(f->client);
			f->client = nil;
			break;
		default:
			if(f->level > Qparsed)
				f->level = Qparsed;
			else
				f->level = Qclient;
		}
	} else {
		for(i=f->level+1; i < nelem(nametab); i++){
			if(nametab[i]){
				if(strcmp(name, nametab[i]) == 0)
					break;
				if(i == Qbody && strncmp(name, "body.", 5) == 0)
					break;
			}
			if(i == Qclient){
				j = atoi(name);
				if(j >= 0 && j < nclient){
					f->client = &client[j];
					incref(f->client);
					break;
				}
			}
			if(i == Qheader && f->client && f->client->qbody){
				char buf[128];
				Key *k;

				for(k = f->client->qbody->hdr; k; k = k->next){
					nstrcpy(buf, k->key, sizeof(buf));
					if(!strcmp(name, fshdrname(buf)))
						break;
				}
				if(k != nil){
					/* need to copy as key is owned by qbody wich might go away */
					f->key = addkey(0, k->key, k->val);
					break;
				}
			}
		}
		if(i >= nelem(nametab))
			return "directory entry not found";
		f->level = i;
	}
	fsmkqid(qid, f->level, wfaux(f));
	fid->qid = *qid;
	return nil;
}

static char*
fsclone(Fid *oldfid, Fid *newfid)
{
	Webfid *f, *o;

	o = oldfid->aux;
	if(o == nil || o->key || o->buq)
		return "bad fid";
	f = emalloc(sizeof(*f));
	memmove(f, o, sizeof(*f));
	if(f->client)
		incref(f->client);
	newfid->aux = f;
	return nil;
}

static void
fsopen(Req *r)
{
	Webfid *f;
	Client *cl;

	f = r->fid->aux;
	cl = f->client;
	switch(f->level){
	case Qclone:
		if((cl = newclient()) == nil){
			respond(r, "no more clients");
			return;
		}
		f->level = Qctl;
		f->client = cl;
		fsmkqid(&r->fid->qid, f->level, wfaux(f));
		r->ofcall.qid = r->fid->qid;
		break;
	case Qpost:
		if(cl->qbody && !cl->cbody){
		Inuse:
			respond(r, "client in use");
			return;
		}
	case Qbody:
		if(cl->obody)
			goto Inuse;
		if(cl->cbody){
			bufree(cl->qbody);
			cl->qbody = nil;
			cl->cbody = 0;
		}
		if(cl->qbody == nil){
			char *m;

			if(cl->url == nil){
				respond(r, "no url set");
				return;
			}
			cl->qbody = bualloc(16*1024);
			if(strcmp(cl->url->scheme, "gopher") == 0) {
				int p[2];
				Url *u;
				char buf[16*1024];
				int n;

				switch(rfork(RFPROC|RFMEM|RFNOWAIT)) {
				case -1:
					buclose(cl->qbody, "can't fork");
					bufree(cl->qbody);
					break;
				case 0:
					cl->qbody->hdr = addkey(cl->qbody->hdr, "Content-Type", "text/html");
					u = cl->qbody->url = cl->url;
					pipe(p);
					switch(fork()) {
					case -1:
						buclose(cl->qbody, "can't fork");
						bufree(cl->qbody);
						break;
					case 0:
						dup(p[0], 1);
						close(p[0]);
						close(p[1]);
						execl("/bin/aux/gopher2html", "gopher2html",
							u->host ? u->host : "",
							u->port ? u->port : "gopher",
							u->path ? u->path+1 : "", nil);
						buclose(cl->qbody, "exec gopher2html: %r");
						bufree(cl->qbody);
						exits("exec");
					default:
						close(p[0]);
						while((n = read(p[1], buf, sizeof buf)) > 0)
							buwrite(cl->qbody, buf, n);
						if(n < 0) {
							rerrstr(buf, sizeof buf);
							buclose(cl->qbody, buf);
							bufree(cl->qbody);
						} else {
							buclose(cl->qbody, nil);
						}
						close(p[1]);
						exits(nil);
					}
				}
			} else {
				if(f->level != Qbody){
					f->buq = bualloc(64*1024);
					if(!lookkey(cl->hdr, "Content-Type"))
						cl->hdr = addkey(cl->hdr, "Content-Type", 
							"application/x-www-form-urlencoded");
					m = "POST";
				} else
					m = "GET";
				if(cl->request[0])
					m = cl->request;

				/*
				 * some sites give a 403 Forbidden if we dont include
				 * a meaningless Accept header in the request.
				 */
				if(!lookkey(cl->hdr, "Accept"))
					cl->hdr = addkey(cl->hdr, "Accept", "*/*");

				if(!lookkey(cl->hdr, "Referer")){
					char *r;
					Url *u;

					/*
					 * Referer header is often required on broken
					 * websites even if the spec makes them optional,
					 * so we make one up.
					 */
					if(u = url("/", cl->url)){
						if(r = smprint("%U", u)){
							cl->hdr = addkey(cl->hdr, "Referer", r);
							free(r);
						}
						freeurl(u);
					}
				}

				if(!lookkey(cl->hdr, "Connection"))
					cl->hdr = addkey(cl->hdr, "Connection", "keep-alive");

				if(agent && !lookkey(cl->hdr, "User-Agent"))
					cl->hdr = addkey(cl->hdr, "User-Agent", agent);

				http(m, cl->url, cl->hdr, cl->qbody, f->buq);
			}
			cl->request[0] = 0;
			cl->url = nil;
			cl->hdr = nil;
		}
		if(f->buq)
			break;
		cl->obody = 1;
		incref(cl->qbody);
		bureq(f->buq = cl->qbody, r);
		return;
	}
	respond(r, nil);
}

static int
rootgen(int i, Dir *d, void *)
{
	i += Qroot+1;
	if(i < Qclient){
		fsmkdir(d, i, 0);
		return 0;
	}
	i -= Qclient;
	if(i < nclient){
		fsmkdir(d, Qclient, &client[i]);
		return 0;
	}
	return -1;
}

static int
clientgen(int i, Dir *d, void *aux)
{
	i += Qclient+1;
	if(i > Qparsed){
		Client *cl = aux;
		Key *k;

		i -= Qparsed+1;
		if(cl == nil || cl->qbody == nil)
			return -1;
		for(k = cl->qbody->hdr; i > 0 && k; i--, k = k->next)
			;
		if(k == nil || i > 0)
			return -1;
		i = Qheader;
		aux = k;
	}
	fsmkdir(d, i, aux);
	return 0;
}

static int
parsedgen(int i, Dir *d, void *aux)
{
	i += Qparsed+1;
	if(i > Qurlfrag)
		return -1;
	fsmkdir(d, i, aux);
	return 0;
}

static void
fsread(Req *r)
{
	char buf[1024];
	Webfid *f;

	f = r->fid->aux;
	switch(f->level){
	case Qroot:
		dirread9p(r, rootgen, nil);
		respond(r, nil);
		return;
	case Qclient:
		dirread9p(r, clientgen, f->client);
		respond(r, nil);
		return;
	case Qparsed:
		dirread9p(r, parsedgen, clienturl(f->client));
		respond(r, nil);
		return;
	case Qrctl:
		snprint(buf, sizeof(buf), "useragent %s\ntimeout %d\n", agent, timeout);
	String:
		readstr(r, buf);
		respond(r, nil);
		return;
	case Qctl:
		snprint(buf, sizeof(buf), "%ld\n", CLIENTID(f->client));
		goto String;
	case Qheader:
		snprint(buf, sizeof(buf), "%s", f->key->val);
		goto String;
	case Qurl:
	case Qurlschm:
	case Qurluser:
	case Qurlpass:
	case Qurlhost:
	case Qurlport:
	case Qurlpath:
	case Qurlqwry:
	case Qurlfrag:
		urlstr(buf, sizeof(buf), clienturl(f->client), f->level);
		goto String;
	case Qbody:
		bureq(f->buq, r);
		return;
	}
	respond(r, "not implemented");
}

static char*
rootctl(char *ctl, char *arg)
{
	Url *u;

	if(debug)
		fprint(2, "rootctl: %q %q\n", ctl, arg);

	if(!strcmp(ctl, "useragent")){
		free(agent);
		if(arg && *arg)
			agent = estrdup(arg);
		else
			agent = nil;
		return nil;
	}

	if(!strcmp(ctl, "flushauth")){
		u = nil;
		if(arg && *arg)
			u = saneurl(url(arg, 0));
		flushauth(u, 0);
		freeurl(u);
		return nil;
	}

	if(!strcmp(ctl, "timeout")){
		if(arg && *arg)
			timeout = atoi(arg);
		else
			timeout = 0;
		if(timeout < 0)
			timeout = 0;
		return nil;
	}

	return "bad ctl message";
}

static char*
clientctl(Client *cl, char *ctl, char *arg)
{
	char *p;
	Url *u;
	Key *k;

	if(debug)
		fprint(2, "clientctl: %q %q\n", ctl, arg);

	if(!strcmp(ctl, "url")){
		if((u = saneurl(url(arg, cl->baseurl))) == nil)
			return "bad url";
		freeurl(cl->url);
		cl->url = u;
	}
	else if(!strcmp(ctl, "baseurl")){
		if((u = url(arg, 0)) == nil)
			return "bad baseurl";
		freeurl(cl->baseurl);
		cl->baseurl = u;
	}
	else if(!strcmp(ctl, "request")){
		p = cl->request;
		nstrcpy(p, arg, sizeof(cl->request));
		for(; *p && isalpha(*p); p++)
			*p = toupper(*p);
		*p = 0;
	}
	else if(!strcmp(ctl, "headers")){
		while(arg && *arg){
			ctl = arg;
			while(*ctl && strchr(whitespace, *ctl))
				ctl++;
			if(arg = strchr(ctl, '\n'))
				*arg++ = 0;
			if(k = parsehdr(ctl)){
				k->next = cl->hdr;
				cl->hdr = k;
			}
		}
	}
	else {
		char buf[128], **t;
		static char *tab[] = {
			"User-Agent",
			"Content-Type",
			nil,
		};
		for(t = tab; *t; t++){
			nstrcpy(buf, *t, sizeof(buf));
			if(!strcmp(ctl, fshdrname(buf))){
				cl->hdr = delkey(cl->hdr, *t);
				if(arg && *arg)
					cl->hdr = addkey(cl->hdr, *t, arg);
				break;
			}
		}
		if(*t == nil)
			return "bad ctl message";
	}
	return nil;
}

static void
fswrite(Req *r)
{
	int n;
	Webfid *f;
	char *s, *t;

	f = r->fid->aux;
	switch(f->level){
	case Qrctl:
	case Qctl:
		n = r->ofcall.count = r->ifcall.count;
		s = emalloc(n+1);
		memmove(s, r->ifcall.data, n);
		while(n > 0 && strchr("\r\n", s[n-1]))
			n--;
		s[n] = 0;
		t = s;
		while(*t && strchr(whitespace, *t)==0)
			t++;
		while(*t && strchr(whitespace, *t))
			*t++ = 0;
		if(f->level == Qctl)
			t = clientctl(f->client, s, t);
		else
			t = rootctl(s, t);
		free(s);
		respond(r, t);
		return;
	case Qpost:
		bureq(f->buq, r);
		return;
	}
	respond(r, "not implemented");
}

static void
fsflush(Req *r)
{
	Webfid *f;
	Req *o;

	if(o = r->oldreq)
	if(f = o->fid->aux)
		buflushreq(f->buq, o);
	respond(r, nil);
}

static void
fsdestroyfid(Fid *fid)
{
	Webfid *f;

	if(f = fid->aux){
		fid->aux = nil;
		if(f->buq){
			buclose(f->buq, 0);
			if(f->client->qbody == f->buq){
				f->client->obody = 0;
				f->client->cbody = 1;
			}
			bufree(f->buq);
		}
		if(f->key)
			free(f->key);
		freeclient(f->client);
		free(f);
	}
}

Srv fs = 
{
	.attach=fsattach,
	.stat=fsstat,
	.walk1=fswalk1,
	.clone=fsclone,
	.open=fsopen,
	.read=fsread,
	.write=fswrite,
	.flush=fsflush,
	.destroyfid=fsdestroyfid,
};

void
usage(void)
{
	fprint(2, "usage: %s [-D] [-A useragent] [-T timeout] [-m mtpt] [-s srv]\n", argv0);
	exits("usage");
}

void
main(int argc, char *argv[])
{
	char *srv, *mtpt, *s;

	quotefmtinstall();
	fmtinstall('U', Ufmt);
	fmtinstall('E', Efmt);

	srv = nil;
	mtpt = "/mnt/web";
	user = getuser();
	time0 = time(0);
	timeout = 10000;
	agent = nil;

	ARGBEGIN {
	case 'D':
		chatty9p++;
		break;
	case 'A':
		agent = EARGF(usage());
		break;
	case 'T':
		timeout = atoi(EARGF(usage()));
		if(timeout < 0)
			timeout = 0;
		break;
	case 'm':
		mtpt = EARGF(usage());
		break;
	case 's':
		srv = EARGF(usage());
		break;
	case 'd':
		debug++;
		break;
	default:
		usage();
	} ARGEND;

	rfork(RFNOTEG);

	if(agent == nil)
		agent = "hjdicks";
	agent = estrdup(agent);

	if(s = getenv("httpproxy")){
		proxy = saneurl(url(s, 0));
		free(s);
	}

	postmountsrv(&fs, srv, mtpt, MREPL);
}