//
//  HTTPSource.m
//  HTTPSource
//
//  Created by Vincent Spader on 3/1/07.
//  Copyright 2007 __MyCompanyName__. All rights reserved.
//

#import "HTTPSource.h"


@implementation HTTPSource

- (BOOL)buffered
{
	return NO;
}

- (BOOL)open:(NSURL *)url
{

	unsigned int port = [[url port] unsignedIntValue];
	if (!port)
		port = 80;
		
	_socket = [[Socket alloc] initWithHost:[url host] port:port];

	if (_socket) {
		NSData *request = [[NSString stringWithFormat:@"GET %@ HTTP/1.0\nHOST: %@\n\n",[url path],[url host]] dataUsingEncoding:NSUTF8StringEncoding];
		[_socket send:(void *)[request bytes] amount:[request length]];
		pastHeader = NO;
	}
	
	return (_socket != nil);
}

- (NSDictionary *)properties
{
	return nil;
}

- (BOOL)seekable
{
	return NO;
}

- (BOOL)seek:(long)position whence:(int)whence
{
	return NO;
}

- (long)tell
{
	return byteCount;
}

- (int)read:(void *)buffer amount:(int)amount
{
	int l = [_socket receive:buffer amount:amount];

	if (!pastHeader) {
		uint8_t *f = (uint8_t *)strnstr((const char *)buffer, "\r\n\r\n", l);
		if (f)
		{
			pastHeader = YES;
			
			uint8_t *bufferOffset = f + 4; //\r\n\r\n
			uint8_t *bufferEnd = (uint8_t *)buffer + l;
			int amountRemaining = bufferEnd - bufferOffset;
			
			/*
			//For testing only
			FILE *testFout = fopen("header.raw", "w");
			fwrite(buffer, 1, bufferOffset - (uint8_t *)buffer, testFout);
			fclose(testFout);

			testFout = fopen("test.raw", "w");
			fwrite(bufferOffset, 1, amountRemaining, testFout);
			fclose(testFout);
			*/
			
			memmove(buffer,bufferOffset, amountRemaining);
			
			return amountRemaining + [self read:((uint8_t *)buffer + amountRemaining) amount:(amount - amountRemaining)];
		}
		else {
			//Keep searching for header. Note: NEED TO WATCH BOUNDARY CASES
			return [self read:buffer amount:amount];
		}
	}

	/*
	//FOR TESTING ONLY
	FILE *testFout = fopen("test.raw", "a");
	fwrite(buffer, 1, l, testFout);
	fclose(testFout);
	*/
	if (l > 0)
		byteCount += l;

	return l;
}

- (void)close
{
	[_socket close];
}

+ (NSArray *)schemes
{
	return [NSArray arrayWithObject:@"http"];
}

@end