PBWebController.m

   1 //
   2 //  PBWebController.m
   3 //  GitX
   4 //
   5 //  Created by Pieter de Bie on 08-10-08.
   6 //  Copyright 2008 __MyCompanyName__. All rights reserved.
   7 //
   8
   9 #import "PBWebController.h"
  10 #import "PBGitRepository.h"
  11 #import "PBGitXProtocol.h"
  12
  13 #include <SystemConfiguration/SCNetworkReachability.h>
  14
  15 @implementation PBWebController
  16
  17 @synthesize startFile, repository;
  18
  19 - (void) awakeFromNib
  20 {
  21         NSString *path = [NSString stringWithFormat:@"html/views/%@", startFile];
  22         NSString* file = [[NSBundle mainBundle] pathForResource:@"index" ofType:@"html" inDirectory:path];
  23         NSURLRequest * request = [NSURLRequest requestWithURL:[NSURL fileURLWithPath:file]];
  24         callbacks = [NSMapTable mapTableWithKeyOptions:(NSPointerFunctionsObjectPointerPersonality|NSPointerFunctionsStrongMemory) valueOptions:(NSPointerFunctionsObjectPointerPersonality|NSPointerFunctionsStrongMemory)];
  25
  26         finishedLoading = NO;
  27         [view setUIDelegate:self];
  28         [view setFrameLoadDelegate:self];
  29         [view setResourceLoadDelegate:self];
  30         [[view mainFrame] loadRequest:request];
  31 }
  32
  33 - (WebScriptObject *) script
  34 {
  35         return [view windowScriptObject];
  36 }
  37
  38 # pragma mark Delegate methods
  39
  40 - (void) webView:(id) v didFinishLoadForFrame:(id) frame
  41 {
  42         id script = [view windowScriptObject];
  43         [script setValue: self forKey:@"Controller"];
  44
  45         finishedLoading = YES;
  46         if ([self respondsToSelector:@selector(didLoad)])
  47                 [self performSelector:@selector(didLoad)];
  48 }
  49
  50 - (void)webView:(WebView *)webView addMessageToConsole:(NSDictionary *)dictionary
  51 {
  52         NSLog(@"Error from webkit: %@", dictionary);
  53 }
  54
  55 - (NSURLRequest *)webView:(WebView *)sender
  56                  resource:(id)identifier
  57           willSendRequest:(NSURLRequest *)request
  58          redirectResponse:(NSURLResponse *)redirectResponse
  59            fromDataSource:(WebDataSource *)dataSource
  60 {
  61         if (!self.repository)
  62                 return request;
  63
  64         // TODO: Change this to canInitWithRequest
  65         if ([[[request URL] scheme] isEqualToString:@"GitX"]) {
  66                 NSMutableURLRequest *newRequest = [request mutableCopy];
  67                 [newRequest setRepository:self.repository];
  68                 return newRequest;
  69         }
  70
  71         return request;
  72 }
  73
  74
  75 + (BOOL)isSelectorExcludedFromWebScript:(SEL)aSelector
  76 {
  77         return NO;
  78 }
  79
  80 + (BOOL)isKeyExcludedFromWebScript:(const char *)name {
  81         return NO;
  82 }
  83
  84 #pragma mark Functions to be used from JavaScript
  85
  86 - (void) log: (NSString*) logMessage
  87 {
  88         NSLog(@"%@", logMessage);
  89 }
  90
  91 - (BOOL) isReachable:(NSString *)hostname
  92 {
  93         SCNetworkConnectionFlags flags;
  94         if (!SCNetworkCheckReachabilityByName([hostname cStringUsingEncoding:NSASCIIStringEncoding], &flags))
  95                 return FALSE;
  96
  97         // If a connection is required, then it's not reachable
  98         if (flags & (kSCNetworkFlagsConnectionRequired | kSCNetworkFlagsConnectionAutomatic | kSCNetworkFlagsInterventionRequired))
  99                 return FALSE;
 100
 101         return flags > 0;
 102 }
 103
 104 #pragma mark Using async function from JS
 105
 106 - (void) runCommand:(WebScriptObject *)arguments inRepository:(PBGitRepository *)repo callBack:(WebScriptObject *)callBack
 107 {
 108         // The JS bridge does not handle JS Arrays, even though the docs say it does. So, we convert it ourselves.
 109         int length = [[arguments valueForKey:@"length"] intValue];
 110         NSMutableArray *realArguments = [NSMutableArray arrayWithCapacity:length];
 111         int i = 0;
 112         for (i = 0; i < length; i++)
 113                 [realArguments addObject:[arguments webScriptValueAtIndex:i]];
 114
 115         NSFileHandle *handle = [repo handleInWorkDirForArguments:realArguments];
 116         [callbacks setObject:callBack forKey:handle];
 117         [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(JSRunCommandDone:) name:NSFileHandleReadToEndOfFileCompletionNotification object:handle];
 118         [handle readToEndOfFileInBackgroundAndNotify];
 119 }
 120
 121 - (void) callSelector:(NSString *)selectorString onObject:(id)object callBack:(WebScriptObject *)callBack
 122 {
 123         NSArray *arguments = [NSArray arrayWithObjects:selectorString, object, nil];
 124         NSThread *thread = [[NSThread alloc] initWithTarget:self selector:@selector(runInThread:) object:arguments];
 125         [callbacks setObject:callBack forKey:thread];
 126         [thread start];
 127 }
 128
 129 - (void) runInThread:(NSArray *)arguments
 130 {
 131         SEL selector = NSSelectorFromString([arguments objectAtIndex:0]);
 132         id object = [arguments objectAtIndex:1];
 133         id ret = [object performSelector:selector];
 134         NSArray *returnArray = [NSArray arrayWithObjects:[NSThread currentThread], ret, nil];
 135         [self performSelectorOnMainThread:@selector(threadFinished:) withObject:returnArray waitUntilDone:NO];
 136 }
 137
 138
 139 - (void) returnCallBackForObject:(id)object withData:(id)data
 140 {
 141         WebScriptObject *a = [callbacks objectForKey: object];
 142         if (!a) {
 143                 NSLog(@"Could not find a callback for object: %@", object);
 144                 return;
 145         }
 146
 147         [callbacks removeObjectForKey:object];
 148         [a callWebScriptMethod:@"call" withArguments:[NSArray arrayWithObjects:@"", data, nil]];
 149 }
 150
 151 - (void) threadFinished:(NSArray *)arguments
 152 {
 153         [self returnCallBackForObject:[arguments objectAtIndex:0] withData:[arguments objectAtIndex:1]];
 154 }
 155
 156 - (void) JSRunCommandDone:(NSNotification *)notification
 157 {
 158         NSString *data = [[NSString alloc] initWithData:[[notification userInfo] valueForKey:NSFileHandleNotificationDataItem] encoding:NSUTF8StringEncoding];
 159         [self returnCallBackForObject:[notification object] withData:data];
 160 }
 161
 162 @end