Merge pull request #13 from NathanFlurry/master

Add `backgroundExecuteData` for getting raw `Data` response
vapor · Sep 8, 2016 · 820642e · 820642e
2 parents ef60f09 + 7074061
commit 820642e
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/Sources/Console/Console/Console.swift b/Sources/Console/Console/Console.swift
@@ -93,11 +93,11 @@ extension ConsoleProtocol {
         )
     }
 
-    public func backgroundExecute(program: String, arguments: [String]) throws -> String {
+    public func backgroundExecuteData(program: String, arguments: [String]) throws -> Data {
         let input = Pipe()
         let output = Pipe()
         let error = Pipe()
-
+        
         do {
             try execute(
                 program: program,
@@ -111,9 +111,13 @@ extension ConsoleProtocol {
             let error = String(data: error.fileHandleForReading.readDataToEndOfFile(), encoding: .utf8) ?? "Unknown"
             throw ConsoleError.backgroundExecute(result, error)
         }
-
+        
         close(output.fileHandleForWriting.fileDescriptor)
-        let data = output.fileHandleForReading.readDataToEndOfFile()
+        return output.fileHandleForReading.readDataToEndOfFile()
+    }
+
+    public func backgroundExecute(program: String, arguments: [String]) throws -> String {
+        let data = try backgroundExecuteData(program: program, arguments: arguments)
         return String(data: data, encoding: .utf8) ?? ""
     }
 }