PBGitXProtocol.m

   1 //
   2 //  PBGitXProtocol.m
   3 //  GitX
   4 //
   5 //  Created by Pieter de Bie on 01-11-08.
   6 //  Copyright 2008 Pieter de Bie. All rights reserved.
   7 //
   8
   9 #import "PBGitXProtocol.h"
  10
  11
  12 @implementation PBGitXProtocol
  13
  14 + (BOOL) canInitWithRequest:(NSURLRequest *)request
  15 {
  16         return [[[request URL] scheme] isEqualToString:@"GitX"];
  17 }
  18
  19 + (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request
  20 {
  21     return request;
  22 }
  23
  24 -(void)startLoading
  25 {
  26     NSURL *url = [[self request] URL];
  27         PBGitRepository *repo = [[self request] repository];
  28
  29         if(!repo) {
  30                 [[self client] URLProtocol:self didFailWithError:[NSError errorWithDomain:NSURLErrorDomain code:0 userInfo:nil]];
  31                 return;
  32     }
  33
  34         NSString *specifier = [NSString stringWithFormat:@"%@:%@", [url host], [[url path] substringFromIndex:1]];
  35         handle = [repo handleInWorkDirForArguments:[NSArray arrayWithObjects:@"cat-file", @"blob", specifier, nil]];
  36         [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(didFinishFileLoad:) name:NSFileHandleReadToEndOfFileCompletionNotification object:handle];
  37         [handle readToEndOfFileInBackgroundAndNotify];
  38
  39     NSURLResponse *response = [[NSURLResponse alloc] initWithURL:[[self request] URL]
  40                                                                                                                 MIMEType:nil
  41                                                                                    expectedContentLength:-1
  42                                                                                                 textEncodingName:nil];
  43
  44     [[self client] URLProtocol:self
  45                         didReceiveResponse:response
  46                         cacheStoragePolicy:NSURLCacheStorageNotAllowed];
  47 }
  48
  49 - (void) didFinishFileLoad:(NSNotification *)notification
  50 {
  51         NSData *data = [[notification userInfo] valueForKey:NSFileHandleNotificationDataItem];
  52     [[self client] URLProtocol:self didLoadData:data];
  53     [[self client] URLProtocolDidFinishLoading:self];
  54 }
  55
  56 - (void) stopLoading
  57 {
  58         [[NSNotificationCenter defaultCenter] removeObserver:self];
  59 }
  60
  61 @end
  62
  63 @implementation NSURLRequest (PBGitXProtocol)
  64
  65 - (PBGitRepository *) repository
  66 {
  67         return [NSURLProtocol propertyForKey:@"PBGitRepository" inRequest:self];
  68 }
  69 @end
  70
  71 @implementation NSMutableURLRequest (PBGitXProtocol)
  72 @dynamic repository;
  73
  74 - (void) setRepository:(PBGitRepository *)repository
  75 {
  76         [NSURLProtocol setProperty:repository forKey:@"PBGitRepository" inRequest:self];
  77 }
  78
  79 @end