Reworked util.spawn to mergable version

Author: elektronikworkshop

src/arduino/arduino.ts

@@ -244,9 +244,9 @@ Please make sure the folder is not occupied by other procedures .`);
         }
         try {
             await util.spawn(this._settings.commandPath,
-                showOutput ? arduinoChannel.channel : null,
-                ["--install-boards", `${packageName}${arch && ":" + arch}${version && ":" + version}`]);
-
+                             ["--install-boards", `${packageName}${arch && ":" + arch}${version && ":" + version}`],
+                             undefined,
+                             { channel: showOutput ? arduinoChannel.channel : null });
             if (updatingIndex) {
                 arduinoChannel.end("Updated package index files.");
             } else {
@@ -282,9 +282,9 @@ Please make sure the folder is not occupied by other procedures .`);
         }
         try {
             await util.spawn(this._settings.commandPath,
-                showOutput ? arduinoChannel.channel : null,
-                ["--install-library", `${libName}${version && ":" + version}`]);
-
+                             ["--install-library", `${libName}${version && ":" + version}`],
+                             undefined,
+                             { channel: showOutput ? arduinoChannel.channel : undefined });
             if (updatingIndex) {
                 arduinoChannel.end("Updated library index files.");
             } else {
@@ -418,9 +418,9 @@ Please make sure the folder is not occupied by other procedures .`);
             const cmd = args.shift();
             try {
                 await util.spawn(cmd,
-                                 arduinoChannel.channel,
                                  args,
-                                 { shell: true, cwd: ArduinoWorkspace.rootPath });
+                                 { shell: true, cwd: ArduinoWorkspace.rootPath },
+                                 { channel: arduinoChannel.channel });
             } catch (ex) {
                 arduinoChannel.error(`Running pre-build command failed: ${os.EOL}${ex.error}`);
                 return false;
@@ -548,8 +548,6 @@ Please make sure the folder is not occupied by other procedures .`);
         // Push sketch as last argument
         args.push(path.join(ArduinoWorkspace.rootPath, dc.sketch));
 
-        let success = false;
-
         const cleanup = async () => {
             if (cocopa) {
                 await cocopa.conclude();
@@ -562,25 +560,14 @@ Please make sure the folder is not occupied by other procedures .`);
             }
         }
 
-        // TODO: Get rid of spawn's channel parameter and just support
-        // stdout and stderr callbacks
-        const stdoutCallback = (line: string) => {
-            if (cocopa) {
-                cocopa.callback(line);
-                if (verbose) {
-                    arduinoChannel.channel.append(line);
-                }
-            } else {
-                arduinoChannel.channel.append(line);
-            }
-        }
-
-        await util.spawn(
+        return await util.spawn(
             this._settings.commandPath,
-            arduinoChannel.channel,
             args,
             undefined,
-            stdoutCallback,
+            {
+                channel: !cocopa || cocopa && verbose ? arduinoChannel.channel : undefined,
+                stdout: cocopa ? cocopa.callback : undefined,
+            }
         ).then(async () => {
             await cleanup();
             if (mode !== BuildMode.Analyze) {
@@ -590,17 +577,14 @@ Please make sure the folder is not occupied by other procedures .`);
                 arduinoChannel.info(`To rebuild your IntelliSense configuration run "${cmd}"`);
             }
             arduinoChannel.end(`${mode} sketch '${dc.sketch}${os.EOL}`);
-            success = true;
+            return true;
         }, async (reason) => {
             await cleanup();
             const msg = reason.code
                 ? `Exit with code=${reason.code}`
-                : reason.message
-                    ? reason.message
-                    : JSON.stringify(reason);
+                : JSON.stringify(reason);
             arduinoChannel.error(`${mode} sketch '${dc.sketch}': ${msg}${os.EOL}`);
+            return false;
         });
-
-        return success;
     }
 }

src/common/util.ts

@@ -185,16 +185,13 @@ export function isArduinoFile(filePath): boolean {
     return fileExistsSync(filePath) && (path.extname(filePath) === ".ino" || path.extname(filePath) === ".pde");
 }
 
-// TODO: remove output channel and just provide callback hooks for stdout and
-// stderr
 export function spawn(command: string,
-                      outputChannel: vscode.OutputChannel,
                       args: string[] = [],
                       options: any = {},
-                      stdoutCallback?: (string) => void): Thenable<object> {
+                      output?: {channel?: vscode.OutputChannel,
+                                stdout?: (string) => void,
+                                stderr?: (string) => void}): Thenable<object> {
     return new Promise((resolve, reject) => {
-        const stdout = "";
-        const stderr = "";
         options.cwd = options.cwd || path.resolve(path.join(__dirname, ".."));
         const child = childProcess.spawn(command, args, options);
 
@@ -210,27 +207,37 @@ export function spawn(command: string,
             }
         }
 
-        if (outputChannel) {
-            child.stdout.on("data", (data: Buffer) => {
-                const decoded = decodeData(data, codepage);
-                if (stdoutCallback) {
-                    stdoutCallback(decoded);
-                } else {
-                    outputChannel.append(decoded);
-                }
-            });
-            child.stderr.on("data", (data: Buffer) => {
-                outputChannel.append(decodeData(data, codepage));
-            });
+        if (output) {
+            if (output.channel || output.stdout) {
+                child.stdout.on("data", (data: Buffer) => {
+                    const decoded = decodeData(data, codepage);
+                    if (output.stdout) {
+                        output.stdout(decoded);
+                    }
+                    if (output.channel) {
+                        output.channel.append(decoded);
+                    }
+                });
+            }
+            if (output.channel || output.stderr) {
+                child.stderr.on("data", (data: Buffer) => {
+                    const decoded = decodeData(data, codepage);
+                    if (output.stderr) {
+                        output.stderr(decoded);
+                    }
+                    if (output.channel) {
+                        output.channel.append(decoded);
+                    }
+                });
+            }
         }
 
-        child.on("error", (error) => reject({ error, stderr, stdout }));
-
+        child.on("error", (error) => reject({ error }));
         child.on("exit", (code) => {
             if (code === 0) {
-                resolve({ code, stdout, stderr });
+                resolve({ code });
             } else {
-                reject({ code, stdout, stderr });
+                reject({ code });
             }
         });
     });