#import <CoreFoundation/CoreFoundation.h>

#import <CoreServices/CoreServices.h>

#import <Foundation/Foundation.h>

#import <limits.h> // PATH_MAX

#include "Common.h"

NSData *runTask(NSString *script, NSDictionary *env, int *exitCode) {

NSTask *task = [[NSTask alloc] init];

[task setCurrentDirectoryPath:@"/tmp"]; /* XXX: Fix this */

[task setEnvironment:env];

[task setLaunchPath:@"/bin/sh"];

[task setArguments:[NSArray arrayWithObjects:@"-c", script, nil]];

NSPipe *pipe;

pipe = [NSPipe pipe];

[task setStandardOutput: pipe];

// Let stderr go to the usual place

//[task setStandardError: pipe];

NSFileHandle *file;

file = [pipe fileHandleForReading];

[task launch];

NSData *data;

data = [file readDataToEndOfFile];

[task waitUntilExit];

*exitCode = [task terminationStatus];

[task release];

/* The docs claim this isn't needed, but we leak descriptors otherwise */

[file closeFile];

/*[pipe release];*/

return data;

}

NSString *pathOfURL(CFURLRef url)

{

NSString *targetCFS = [[(NSURL *)url absoluteURL] path];

n8log(@"targetCFS = %@", targetCFS);

//return [targetCFS stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];

return targetCFS;

}

NSData *colorizeURL(CFBundleRef bundle, CFURLRef url, int *status, int thumbnail)

{

NSData *output = NULL;

CFURLRef rsrcDirURL = CFBundleCopyResourcesDirectoryURL(bundle);

//n8log(@"rsrcDirURL = %@", CFURLGetString(rsrcDirURL));

NSString *rsrcEsc = pathOfURL(rsrcDirURL);

CFRelease(rsrcDirURL);

n8log(@"url = %@", url);

NSString *targetEsc = pathOfURL(url);

n8log(@"targetEsc = %@", targetEsc);

// Set up preferences

NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];

NSMutableDictionary *env = [NSMutableDictionary dictionaryWithDictionary:

[[NSProcessInfo processInfo] environment]];

[env addEntriesFromDictionary:[NSDictionary dictionaryWithObjectsAndKeys:

#ifdef DEBUG

@"1", @"qlcc_debug",

#endif

@"9", @"fontSizePoints",

@"Monaco", @"font",

@"ide-xcode", @"hlTheme",

@"", @"extraHLFlags",

@"", @"maxFileSize",

@"UTF-8", @"textEncoding",

@"UTF-8", @"webkitTextEncoding", nil]];

[env addEntriesFromDictionary:[defaults persistentDomainForName:myDomain]];

NSString *cmd = [NSString stringWithFormat:

@"'%@/colorize.sh' '%@' '%@' %s",

rsrcEsc, rsrcEsc, targetEsc, thumbnail ? "1" : "0"];

n8log(@"cmd = %@", cmd);

output = runTask(cmd, env, status);

if (*status != 0) {

NSLog(@"QLColorCode: colorize.sh failed with exit code %d. Command was (%@).",

*status, cmd);

}

return output;

}