// $Id: SOPEXWebConnection.m 1 2004-08-20 11:17:52Z znek $

#import "SOPEXWebConnection.h"
#import "SOPEXWebMetaParser.h"
#import "NSString+Ext.h"
#import "NSBundle+Ext.h"


@implementation SOPEXWebConnection

static BOOL debugOn = NO;

+ (void)initialize
{
    static BOOL isInitialized = NO;

    if(isInitialized)
        return;
    
    debugOn = [[NSUserDefaults standardUserDefaults] boolForKey:@"SOPEXDebugEnabled"];
    if(debugOn)
        debugOn = [[NSUserDefaults standardUserDefaults] boolForKey:@"SOPEXDebugWebConnection"];
    isInitialized = YES;
}

- (id)initWithURL:(id)_url localResourceBundle:(NSBundle *)_resourceBundle
{
    if ((self = [super init])) {
        if ([_url isKindOfClass:[NSURL class]])
            self->url = [_url copy];
        else
            self->url = [[NSURL alloc] initWithString:_url];
        
        if(_resourceBundle == nil)
            _resourceBundle = [NSBundle mainBundle];
        self->localResourceBundle = [_resourceBundle retain];
        self->resourceCache = [[NSMutableDictionary alloc] init];
        self->appPrefix = [[NSString stringWithFormat:@"/%@", [[self->url path] lastPathComponent]] retain];
    }
    return self;
}

- (id)init {
    return [self initWithURL:nil localResourceBundle:nil];
}

- (void)dealloc {
    [self->resourceCache release];
    [self->localResourceBundle release];
    [self->url release];
    [self->sessionID release];
    [self->appPrefix release];
    [super dealloc];
}

/* accessors */

- (NSURL *)url {
    return self->url;
}
- (NSString *)sessionID {
    return self->sessionID;
}

/* session tracking */

- (void)_useSessionID:(NSString *)_sid {
    [self->sessionID autorelease];
    self->sessionID = [_sid copy];
}

- (void)handleNoSessionInResponse:(NSURLResponse *)_r {
    if(debugOn)
        NSLog(@"%s: NO session-id", __PRETTY_FUNCTION__);
}
- (void)handleInitialSessionID:(NSString *)_s inResponse:(NSURLResponse *)_r {
    if(debugOn)
        NSLog(@"%s: initial sid: %@", __PRETTY_FUNCTION__, _s);
    [self _useSessionID:_s];
}
- (void)handleChangedSessionID:(NSString *)_s inResponse:(NSURLResponse *)_r {
    if(debugOn)
        NSLog(@"%s: changed sid: %@", __PRETTY_FUNCTION__, _s);
    [self _useSessionID:_s];
}

- (void)processSessionID:(NSString *)_sid ofResponse:(NSURLResponse *)_r {
    if (_sid) {
        if (self->sessionID && ![self->sessionID isEqualToString:_sid])
            [self handleChangedSessionID:_sid inResponse:_r];
        else if (self->sessionID == nil)
            [self handleInitialSessionID:_sid inResponse:_r];
    }
    else {
        [self handleNoSessionInResponse:_r];
    }
}

/* operations */

- (void)processHTML:(NSString *)_html ofResponse:(NSURLResponse *)_r {
    NSArray      *links;
    NSDictionary *meta;
    
    [[SOPEXWebMetaParser sharedWebMetaParser]
      processHTML:_html meta:&meta links:&links];
#if 0
    if(debugOn)
        NSLog(@"%s: meta: %@\n  links: %@", __PRETTY_FUNCTION__,
          [meta descriptionInStringsFileFormat], links);
#endif
    [self processSessionID:[meta objectForKey:@"OGoSessionID"] ofResponse:_r];
}

- (void)processResponse:(NSURLResponse *)_r data:(NSData *)_data {
    NSString *s;
    
    if (![[_r MIMEType] hasPrefix:@"text/html"])
        return;
    
    s = [[NSString alloc] initWithData:_data encoding:NSISOLatin1StringEncoding];
    if (s == nil)
        return;
    
    [self processHTML:s ofResponse:_r];
    [s release];
}

- (BOOL)shouldRewriteRequestURL:(NSURL *)_url {
    NSString *path;
    BOOL shouldRewrite = NO;

    if(debugOn)
        NSLog(@"%s testing if I should rewrite:%@", __PRETTY_FUNCTION__, _url);

    if ([_url isFileURL])
        return shouldRewrite;
    
    if ((path = [_url path]) == nil) {
        if(debugOn)
            NSLog(@"%s could not get path for URL: %@", __PRETTY_FUNCTION__, path);
        return shouldRewrite;
    }
    
    shouldRewrite = ([path rangeOfString:@"WebServerResources"].location != NSNotFound || (![path hasPrefix:self->appPrefix]));
    if(debugOn)
        NSLog(@"%s shouldRewrite:%@ ->%@", __PRETTY_FUNCTION__, _url, shouldRewrite ? @"YES" : @"NO");
    return shouldRewrite;
}

- (NSURL *)rewriteRequestURL:(NSURL *)_url
{
    NSString *resourcePath, *urlPath;
    NSURL *cacheURL;

    if ([_url isFileURL])
        return _url;

    urlPath = [_url path];
#if 0
    if([urlPath hasPrefix:self->appPrefix])
        return _url;
#endif
    if(debugOn)
        NSLog(@"%s [_url path] will be rewritten:%@", __PRETTY_FUNCTION__, _url);

    cacheURL = [self->resourceCache objectForKey:urlPath];
    if(debugOn && cacheURL)
        NSLog(@"%s found cached URL for resource:%@", __PRETTY_FUNCTION__, urlPath);
    if(cacheURL)
        return cacheURL;

    if(debugOn)
        NSLog(@"%s trying to find resource:%@", __PRETTY_FUNCTION__, urlPath);

    resourcePath= [self->localResourceBundle pathForResourceWithURLPath:urlPath];
    if(resourcePath == nil) {
        if(debugOn)
            NSLog(@"%s didn't find resource:%@ in bundle:%@", __PRETTY_FUNCTION__, urlPath, self->localResourceBundle);
        /* not cached locally */
        return _url;
    }
    if(debugOn)
        NSLog(@"%s found resource:%@", __PRETTY_FUNCTION__, urlPath);

    cacheURL = [NSURL fileURLWithPath:resourcePath];
    [self->resourceCache setObject:cacheURL forKey:urlPath];

    return cacheURL;
}

/* description */

- (NSString *)description {
    NSMutableString *ms = [NSMutableString stringWithCapacity:32];
    
    [ms appendFormat:@"<0x%08X[%@]:", self, NSStringFromClass([self class])];
    
    [ms appendFormat:@" url=%@", [self->url absoluteString]];
    
    if (self->sessionID)
        [ms appendFormat:@" sid=%@", self->sessionID];
    else
        [ms appendString:@" no-sid"];
    
    [ms appendString:@">"];
    return ms;
}

@end /* SOPEXWebConnection */