#include <string.h>
#include <3ds.h>

Handle __httpc_servhandle = 0;

Result httpcInit()
{
	Result ret=0;

	if(__httpc_servhandle)return 0;
	if((ret=srvGetServiceHandle(&__httpc_servhandle, "http:C")))return ret;

	//*((u32*)0x600) = __httpc_servhandle;

	ret = HTTPC_Initialize(__httpc_servhandle);
	if(ret!=0)return ret;

	return 0;
}

void httpcExit()
{
	if(__httpc_servhandle==0)return;

	svcCloseHandle(__httpc_servhandle);
}

Result httpcOpenContext(httpcContext *context, char* url, u32 use_defaultproxy)
{
	Result ret=0;

	ret = HTTPC_CreateContext(__httpc_servhandle, url, &context->httphandle);
	if(ret!=0)return ret;

	ret = srvGetServiceHandle(&context->servhandle, "http:C");
	if(ret!=0)return ret;

	ret = HTTPC_InitializeConnectionSession(context->servhandle, context->httphandle);
	if(ret!=0)return ret;

	if(use_defaultproxy==0)return 0;

	ret = HTTPC_SetProxyDefault(context->servhandle, context->httphandle);
	if(ret!=0)return ret;

	return 0;
}

Result httpcCloseContext(httpcContext *context)
{
	Result ret=0;

	ret = HTTPC_CloseContext(context->servhandle, context->httphandle);

	svcCloseHandle(context->servhandle);

	return ret;
}

Result httpcAddRequestHeaderField(httpcContext *context, char* name, char* value)
{
	return HTTPC_AddRequestHeaderField(context->servhandle, context->httphandle, name, value);
}

Result httpcBeginRequest(httpcContext *context)
{
	return HTTPC_BeginRequest(context->servhandle, context->httphandle);
}

Result httpcReceiveData(httpcContext *context, u8* buffer, u32 size)
{
	return HTTPC_ReceiveData(context->servhandle, context->httphandle, buffer, size);
}

Result httpcGetRequestState(httpcContext *context, httpcReqStatus* out)
{
	return HTTPC_GetRequestState(context->servhandle, context->httphandle, out);
}

Result httpcGetDownloadSizeState(httpcContext *context, u32* downloadedsize, u32* contentsize)
{
	return HTTPC_GetDownloadSizeState(context->servhandle, context->httphandle, downloadedsize, contentsize);
}

Result httpcGetResponseStatusCode(httpcContext *context, u32* out, u64 delay)
{
	return HTTPC_GetResponseStatusCode(context->servhandle, context->httphandle, out);
}

Result httpcDownloadData(httpcContext *context, u8* buffer, u32 size, u32 *downloadedsize)
{
	Result ret=0;
	u32 contentsize=0;
	u32 pos=0, sz=0;

	if(downloadedsize)*downloadedsize = 0;

	ret=httpcGetDownloadSizeState(context, NULL, &contentsize);
	if(ret!=0)return ret;

	while(pos < size)
	{
		sz = size - pos;

		ret=httpcReceiveData(context, &buffer[pos], sz);

		if(ret==HTTPC_RESULTCODE_DOWNLOADPENDING)
		{
			ret=httpcGetDownloadSizeState(context, &pos, NULL);
			if(ret!=0)return ret;
		}
		else if(ret!=0)
		{
			return ret;
		}
		else
		{
			pos+= sz;
		}

		if(downloadedsize)*downloadedsize = pos;
	}

	return 0;
}

Result HTTPC_Initialize(Handle handle)
{
	u32* cmdbuf=getThreadCommandBuffer();

	cmdbuf[0]=0x10044; //request header code
	cmdbuf[1]=0x1000; //unk
	cmdbuf[2]=0x20;//processID header, following word is set to processID by the arm11kernel.
	cmdbuf[4]=0;
	cmdbuf[5]=0;//Some sort of handle.
	
	Result ret=0;
	if((ret=svcSendSyncRequest(handle)))return ret;

	return cmdbuf[1];
}

Result HTTPC_CreateContext(Handle handle, char* url, Handle* contextHandle)
{
	u32* cmdbuf=getThreadCommandBuffer();
	u32 l=strlen(url)+1;

	cmdbuf[0]=0x20082; //request header code
	cmdbuf[1]=l;
	cmdbuf[2]=0x01; //unk
	cmdbuf[3]=(l<<4)|0xA;
	cmdbuf[4]=(u32)url;
	
	Result ret=0;
	if((ret=svcSendSyncRequest(handle)))return ret;
	
	if(contextHandle)*contextHandle=cmdbuf[2];

	return cmdbuf[1];
}

Result HTTPC_InitializeConnectionSession(Handle handle, Handle contextHandle)
{
	u32* cmdbuf=getThreadCommandBuffer();

	cmdbuf[0]=0x80042; //request header code
	cmdbuf[1]=contextHandle;
	cmdbuf[2]=0x20; //unk, constant afaict
	
	Result ret=0;
	if((ret=svcSendSyncRequest(handle)))return ret;

	return cmdbuf[1];
}

Result HTTPC_SetProxyDefault(Handle handle, Handle contextHandle)
{
	u32* cmdbuf=getThreadCommandBuffer();

	cmdbuf[0]=0xe0040; //request header code
	cmdbuf[1]=contextHandle;
	
	Result ret=0;
	if((ret=svcSendSyncRequest(handle)))return ret;

	return cmdbuf[1];
}

Result HTTPC_CloseContext(Handle handle, Handle contextHandle)
{
	u32* cmdbuf=getThreadCommandBuffer();

	cmdbuf[0]=0x30040; //request header code
	cmdbuf[1]=contextHandle;
	
	Result ret=0;
	if((ret=svcSendSyncRequest(handle)))return ret;

	return cmdbuf[1];
}

Result HTTPC_AddRequestHeaderField(Handle handle, Handle contextHandle, char* name, char* value)
{
	u32* cmdbuf=getThreadCommandBuffer();

	int l1=strlen(name)+1;
	int l2=strlen(value)+1;

	cmdbuf[0]=0x1100c4; //request header code
	cmdbuf[1]=contextHandle;
	cmdbuf[2]=l1;
	cmdbuf[3]=l2;
	cmdbuf[4]=(l1<<14)|0xC02;
	cmdbuf[5]=(u32)name;
	cmdbuf[6]=(l1<<4)|0xA;
	cmdbuf[7]=(u32)value;
	
	Result ret=0;
	if((ret=svcSendSyncRequest(handle)))return ret;

	return cmdbuf[1];
}

Result HTTPC_BeginRequest(Handle handle, Handle contextHandle)
{
	u32* cmdbuf=getThreadCommandBuffer();

	cmdbuf[0]=0x90040; //request header code
	cmdbuf[1]=contextHandle;
	
	Result ret=0;
	if((ret=svcSendSyncRequest(handle)))return ret;

	return cmdbuf[1];
}

Result HTTPC_ReceiveData(Handle handle, Handle contextHandle, u8* buffer, u32 size)
{
	u32* cmdbuf=getThreadCommandBuffer();

	cmdbuf[0]=0xB0082; //request header code
	cmdbuf[1]=contextHandle;
	cmdbuf[2]=size;
	cmdbuf[3]=(size<<4)|12;
	cmdbuf[4]=(u32)buffer;
	
	Result ret=0;
	if((ret=svcSendSyncRequest(handle)))return ret;

	return cmdbuf[1];
}

Result HTTPC_GetRequestState(Handle handle, Handle contextHandle, httpcReqStatus* out)
{
	u32* cmdbuf=getThreadCommandBuffer();

	cmdbuf[0]=0x50040; //request header code
	cmdbuf[1]=contextHandle;
	
	Result ret=0;
	if((ret=svcSendSyncRequest(handle)))return ret;

	*out = cmdbuf[2];

	return cmdbuf[1];
}

Result HTTPC_GetDownloadSizeState(Handle handle, Handle contextHandle, u32* downloadedsize, u32* contentsize)
{
	u32* cmdbuf=getThreadCommandBuffer();

	cmdbuf[0]=0x60040; //request header code
	cmdbuf[1]=contextHandle;
	
	Result ret=0;
	if((ret=svcSendSyncRequest(handle)))return ret;

	if(downloadedsize)*downloadedsize = cmdbuf[2];
	if(contentsize)*contentsize = cmdbuf[3];

	return cmdbuf[1];
}

Result HTTPC_GetResponseStatusCode(Handle handle, Handle contextHandle, u32* out)
{
	u32* cmdbuf=getThreadCommandBuffer();

	cmdbuf[0]=0x220040; //request header code
	cmdbuf[1]=contextHandle;
	
	Result ret=0;
	if((ret=svcSendSyncRequest(handle)))return ret;

	*out = cmdbuf[2];

	return cmdbuf[1];
}