mirror of
https://github.com/kennethreitz-archive/gitx.git
synced 2026-06-05 23:40:18 +00:00
Pieter de Bie
530377a121
This seems to be the only sensible thing to do, as we can't guess what the encoding would otherwise be.
119 lines
3.3 KiB
Objective-C
119 lines
3.3 KiB
Objective-C
//
|
|
// PBEasyPipe.m
|
|
// GitX
|
|
//
|
|
// Created by Pieter de Bie on 16-06-08.
|
|
// Copyright 2008 __MyCompanyName__. All rights reserved.
|
|
//
|
|
|
|
#import "PBEasyPipe.h"
|
|
|
|
|
|
@implementation PBEasyPipe
|
|
|
|
+ (NSFileHandle*) handleForCommand: (NSString*) cmd withArgs: (NSArray*) args
|
|
{
|
|
return [self handleForCommand:cmd withArgs:args inDir:nil];
|
|
}
|
|
|
|
+ (NSTask *) taskForCommand:(NSString *)cmd withArgs:(NSArray *)args inDir:(NSString *)dir
|
|
{
|
|
NSTask* task = [[NSTask alloc] init];
|
|
task.launchPath = cmd;
|
|
task.arguments = args;
|
|
if (dir)
|
|
task.currentDirectoryPath = dir;
|
|
|
|
if ([[NSUserDefaults standardUserDefaults] boolForKey:@"Show Debug Messages"])
|
|
NSLog(@"Starting command `%@ %@` in dir %@", cmd, [args componentsJoinedByString:@" "], dir);
|
|
#ifdef CLI
|
|
NSLog(@"Starting command `%@ %@` in dir %@", cmd, [args componentsJoinedByString:@" "], dir);
|
|
#endif
|
|
|
|
NSPipe* pipe = [NSPipe pipe];
|
|
task.standardOutput = pipe;
|
|
return task;
|
|
}
|
|
|
|
+ (NSFileHandle*) handleForCommand: (NSString*) cmd withArgs: (NSArray*) args inDir: (NSString*) dir
|
|
{
|
|
NSTask *task = [self taskForCommand:cmd withArgs:args inDir:dir];
|
|
NSFileHandle* handle = [task.standardOutput fileHandleForReading];
|
|
|
|
[task launch];
|
|
return handle;
|
|
}
|
|
|
|
|
|
|
|
+ (NSString*) outputForCommand:(NSString *) cmd
|
|
withArgs:(NSArray *) args
|
|
inDir:(NSString *) dir
|
|
retValue:(int *) ret
|
|
{
|
|
return [self outputForCommand:cmd withArgs:args inDir:dir inputString:NULL retValue:ret];
|
|
}
|
|
|
|
// TODO: Refactor this to use the function above
|
|
+ (NSString*) outputForCommand:(NSString *) cmd
|
|
withArgs:(NSArray *) args
|
|
inDir:(NSString *) dir
|
|
inputString:(NSString *)input
|
|
retValue:(int *) ret
|
|
{
|
|
NSTask *task = [self taskForCommand:cmd withArgs:args inDir:dir];
|
|
NSFileHandle* handle = [task.standardOutput fileHandleForReading];
|
|
|
|
if (input) {
|
|
task.standardInput = [NSPipe pipe];
|
|
NSFileHandle *inHandle = [task.standardInput fileHandleForWriting];
|
|
[inHandle writeData:[input dataUsingEncoding:NSUTF8StringEncoding]];
|
|
[inHandle closeFile];
|
|
}
|
|
|
|
[task launch];
|
|
|
|
NSData* data = [handle readDataToEndOfFile];
|
|
NSString *string = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
|
|
if (!string)
|
|
string = [[NSString alloc] initWithData:data encoding:NSASCIIStringEncoding];
|
|
|
|
// Strip trailing newline
|
|
if ([string hasSuffix:@"\n"])
|
|
string = [string substringToIndex:[string length]-1];
|
|
|
|
[task waitUntilExit];
|
|
if (ret)
|
|
*ret = [task terminationStatus];
|
|
return string;
|
|
}
|
|
|
|
// We don't use the above function because then we'd have to wait until the program was finished
|
|
// with running
|
|
|
|
+ (NSString*) outputForCommand: (NSString*) cmd withArgs: (NSArray*) args inDir: (NSString*) dir
|
|
{
|
|
NSTask *task = [self taskForCommand:cmd withArgs:args inDir:dir];
|
|
NSFileHandle* handle = [task.standardOutput fileHandleForReading];
|
|
|
|
[task launch];
|
|
#warning This can cause a "Bad file descriptor"... when?
|
|
NSData* data = [handle readDataToEndOfFile];
|
|
NSString *string = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
|
|
if (!string)
|
|
string = [[NSString alloc] initWithData:data encoding:NSASCIIStringEncoding];
|
|
|
|
// Strip trailing newline
|
|
if ([string hasSuffix:@"\n"])
|
|
string = [string substringToIndex:[string length]-1];
|
|
return string;
|
|
}
|
|
|
|
|
|
+ (NSString*) outputForCommand: (NSString*) cmd withArgs: (NSArray*) args
|
|
{
|
|
return [self outputForCommand:cmd withArgs:args inDir:nil];
|
|
}
|
|
|
|
@end
|