工具库 zx - 使用 JavaScript 更便捷地编写 shell 脚本-工具盒子

在编写一些复杂脚本做处理的时候，比起传统 shell 脚本，由于对 JS 更加熟悉，我更倾向于使用 JS 来编写。但一旦涉及到文件相关操作，几条 shell 指令肯定是更加方便的，但 Node 本身的 child_process 又差了那么点意思，手感并不是很好。

今年初发布的 zx 包于近日更新了 2.0 版本，它本质上是一个对原生 child_process 的包裹，提供了对传参等操作和部分默认行为的优化。同时它还是基于 Promise 的，因此可以随心所欲的在异步函数中使用。

作为一个常规的 npm 包，安装部分不再多说了，Node 肯定是需要的，直接进入使用部分。

基本使用 {#基本使用}

首先创建一个可运行的 zx 脚本：

                    
                      echo '#!/usr/bin/env zx \n console.log("Hello, zx")' >> script.mjs
chmod +x script.mjs
                &lt;/code&gt;
              &lt;/pre&gt;

指定运行环境为 zx 后，我们就可以像普通 shell 脚本一样直接运行 zx 脚本了。
<br />

<br />
执行 shell 指令 {#执行-shell-指令}
当需要执行 shell 指令时，使用 $ 符号和模板字符串对其进行包裹即可，其内部使用了 child_process 的 spawn ，返回值类型为 ProcessPromise&lt;ProcessOutput&gt; 。
例如，尝试创建删除文件并显示目录内文件个数：
                    
                      await $`echo '' &gt;&gt; ~/abc`;
let count = parseInt(await $`ls -1 ~ | wc -l`);
console.log(`Files: ${count}`);
await $`rm ~/abc`;
count = parseInt(await $`ls -1 ~ | wc -l`);
console.log(`Files: ${count}`);
                &amp;lt;/code&amp;gt;
              &amp;lt;/pre&amp;gt;

&lt;br /&gt;

&lt;br /&gt;
又或者是在一些录播脚本中并发上传文件：
                    
                      const HOST = '1.1.1.1:3000';
let files = ['a.mp4', 'b.mp4'];
let target = '/home/ubuntu/rsync';
await Promise.all(files.map((filename) =&amp;gt; $`rsync -azP ./${filename} ${HOST}:${target}`));

                    &amp;lt;/code&amp;gt;
                  &amp;lt;/pre&amp;gt;



 
同时，zx 也支持错误处理：

 
`````js
                    
                      try {
  await $`exit 1`;
} catch (p) {
  console.log(`Exit code: ${p.exitCode}`);
  console.log(`Error: ${p.stderr}`);
}

                    &amp;lt;/code&amp;gt;
                  &amp;lt;/pre&amp;gt;



 
内置函数和工具库 {#内置函数和工具库}
--------------------


 
除此之外 zx 还对一些常用指令和工具库进行了包裹以方便使用：

 
````js
                    
                      // shell cd
cd('/tmp');

// node-fetch 包
const res = await fetch('https://wttr.in');
if (res.ok) {
console.log(await res.text());
}


// readline 包
const env = await question('Choose a env: ', {
choices: Object.keys(process.env),
});


// setTimeout 实现的 sleep
await sleep(1000);


// 彩色输出 chalk
console.log(chalk.blue('This is blue'));


// fs-extra
const content = await fs.readFile('package.json');


                    &amp;lt;/code&amp;gt;
                  &amp;lt;/pre&amp;gt;



 
额外的 Polyfill {#额外的-polyfill}
----------------------------


 
zx 为 esm 脚本添加了一些额外的 polyfill 以方便使用：

 
```js
                    
                      // __filename 和 __dirname
console.log(__filename, __dirname);
// require
const { version } = require('package.json');

                    &amp;lt;/code&amp;gt;
                  &amp;lt;/pre&amp;gt;



 
参考 {#参考}
--------


 

  
* https://github.com/google/zx

 

```


````


`````