cog/Plugins/HTTPSource/HTTPSource.m

101 lines
1.7 KiB
Objective-C

//
// HTTPSource.m
// HTTPSource
//
// Created by Vincent Spader on 3/1/07.
// Copyright 2007 __MyCompanyName__. All rights reserved.
//
#import "HTTPSource.h"
#import "HTTPConnection.h"
@implementation HTTPSource
- (BOOL)open:(NSURL *)url
{
_connection = [[HTTPConnection alloc] initWithURL:url];
NSString *userAgent = [NSString stringWithFormat:@"Cog %@", [[NSBundle mainBundle] objectForInfoDictionaryKey:(NSString *)kCFBundleVersionKey]];
[_connection setValue:userAgent forRequestHeader:@"User-Agent"];
[_connection setValue:@"close" forRequestHeader:@"Connection"];
[_connection setValue:@"*/*" forRequestHeader:@"Accept"];
BOOL success = [_connection connect];
if (NO == success) {
return NO;
}
_mimeType = [[_connection valueForResponseHeader:@"Content-type"] copy];
return YES;
}
- (NSString *)mimeType
{
NSLog(@"Returning mimetype! %@", _mimeType);
return _mimeType;
}
- (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 totalRead = 0;
while (totalRead < amount) {
NSInteger amountReceived = [_connection receiveData:((uint8_t *)buffer) + totalRead amount:amount - totalRead];
if (amountReceived <= 0) {
break;
}
totalRead += amountReceived;
}
_byteCount += totalRead;
return totalRead;
}
- (void)close
{
[_connection close];
[_connection release];
_connection = nil;
[_mimeType release];
_mimeType = nil;
}
- (void)dealloc
{
[self close];
[super dealloc];
}
- (NSURL *)url
{
return [_connection URL];
}
+ (NSArray *)schemes
{
return [NSArray arrayWithObject:@"http"];
}
@end