最近在學習bash腳本語法,但是如果對bash語法不是熟手的話,感覺非常容易出錯,比如說:顯示未定義的變量shell中變量沒有定義,仍然是可以使用的,但是它的結果可能不是你所預期的。舉個例子:
#!/bin/bash
# 這裡是判斷變量var是否等於字符串abc,但是var這個變量並沒有聲明
if [ "$var" = "abc" ]
then
# 如果if判斷裡是true就在控制臺列印 「 not abc」
echo " not abc"
else
# 如果if判斷裡是false就在控制臺列印 「 abc」
echo " abc "
fi
複製代碼結果是列印了abc,但問題是,這個腳本應該報錯啊,變量並沒有賦值算是錯誤吧。
為了彌補這些錯誤,我們學會在腳本開頭加入:set \-u 這句命令的意思是腳本在頭部加上它,遇到不存在的變量就會報錯,並停止執行。
再次運行就會提示:test.sh: 3: test.sh: num: parameter not set
再想像一下,你本來想刪除:rm \-rf $dir/*然後dir是空的時候,變成了什麼?rm \-rf是刪除命令,$dir是空的話,相當於執行 rm \-rf /*,這是刪除所有文件和文件夾。。。然後,你的系統就沒了,這就是傳說中的刪庫跑路嗎~~~~
如果是node或者瀏覽器環境,我們直接var === 'abc' 肯定是會報錯的,也就是說很多javascript編程經驗無法復用到bash來,如果能復用的話,該多好啊。
後來就開始探索,如果用node腳本代替bash該多好啊,經過一天折騰逐漸發現一個神器,Google旗下的zx庫,先別著急,我先不介紹這個庫,我們先看看目前主流用node如何編寫bash腳本,就知道為啥它是神器了。
node執行bash腳本: 勉強解決方案:child_process API例如 child_process的API裡面exec命令
const { exec } = require("child_process");
exec("ls -la", (error, stdout, stderr) => {
if (error) {
console.log(`error: ${error.message}`);
return;
}
if (stderr) {
console.log(`stderr: ${stderr}`);
return;
}
console.log(`stdout: ${stdout}`);
});
複製代碼這裡需要注意的是,首先exec是異步的,但是我們bash腳本命令很多都是同步的。
而且注意:error對象不同於stderr. error當child_process模塊無法執行命令時,該對象不為空。例如,查找一個文件找不到該文件,則error對象不為空。但是,如果命令成功運行並將消息寫入標準錯誤流,則該stderr對象不會為空。
當然我們可以使用同步的exec命令,execSync
// 引入 exec 命令 from child_process 模塊
const { execSync } = require("child_process");
// 同步創建了一個hello的文件夾
execSync("mkdir hello");
複製代碼再簡單介紹一下child_process的其它能夠執行bash命令的api
exec:啟動一個子進程來執行命令,與spawn不同的是,它有一個回調函數能知道子進程的情況fork:與spawn類似,不同點是它需要指定子進程需要需執行的javascript文件exec跟ececFile不同的是,exec適合執行命令,eexecFile適合執行文件。
node執行bash腳本: 進階方案 shelljsconst shell = require('shelljs');
# 刪除文件命令
shell.rm('-rf', 'out/Release');
// 拷貝文件命令
shell.cp('-R', 'stuff/', 'out/Release');
# 切換到lib目錄,並且列出目錄下到.js結尾到文件,並替換文件內容(sed -i 是替換文字命令)
shell.cd('lib');
shell.ls('*.js').forEach(function (file) {
shell.sed('-i', 'BUILD_VERSION', 'v0.1.2', file);
shell.sed('-i', /^.*REMOVE_THIS_LINE.*$/, '', file);
shell.sed('-i', /.*REPLACE_LINE_WITH_MACRO.*\n/, shell.cat('macro.js'), file);
});
shell.cd('..');
# 除非另有說明,否則同步執行給定的命令。 在同步模式下,這將返回一個 ShellString
#(與 ShellJS v0.6.x 兼容,它返回一個形式為 { code:..., stdout:..., stderr:... } 的對象)。
# 否則,這將返回子進程對象,並且回調接收參數(代碼、標準輸出、標準錯誤)。
if (shell.exec('git commit -am "Auto-commit"').code !== 0) {
shell.echo('Error: Git commit failed');
shell.exit(1);
}
複製代碼從上面代碼上看來,shelljs真的已經算是非常棒的nodejs寫bash腳本的方案了,如果你們那邊的node環境不能隨便升級,我覺得shelljs確實夠用了。
接著我們看看今天的主角zx,start已經17.4k了。
zx庫官方網址:www.npmjs.com/package/zx[1]
我們先看看怎麼用
#!/usr/bin/env zx
await $`cat package.json | grep name`
let branch = await $`git branch --show-current`
await $`dep deploy --branch=${branch}`
await Promise.all([
$`sleep 1; echo 1`,
$`sleep 2; echo 2`,
$`sleep 3; echo 3`,
])
let name = 'foo bar'
await $`mkdir /tmp/${name}
複製代碼各位看官覺得咋樣,是不是就是在寫linux命令而已,bash語法可以忽略很多,直接上js就行,而且它的優點還不止這些,有一些特點挺有意思的:
1、支持ts,自動編譯.ts為.mjs文件,.mjs文件是node高版本自帶的支持es6 module的文件結尾,也就是這個文件直接import模塊就行,不用其它工具轉義
2、自帶支持管道操作pipe方法
3、自帶fetch庫,可以進行網絡請求,自帶chalk庫,可以列印有顏色的字體,自帶錯誤處理nothrow方法,如果bash命令出錯,可以包裹在這個方法裡忽略錯誤
完整中文文檔(在下翻譯水平一般,請見諒)#!/usr/bin/env zx
await $`cat package.json | grep name`
let branch = await $`git branch --show-current`
await $`dep deploy --branch=${branch}`
await Promise.all([
$`sleep 1; echo 1`,
$`sleep 2; echo 2`,
$`sleep 3; echo 3`,
])
let name = 'foo bar'
await $`mkdir /tmp/${name}
複製代碼Bash 很棒,但是在編寫腳本時,人們通常會選擇更方便的程式語言。JavaScript 是一個完美的選擇,但標準的 Node.js 庫在使用之前需要額外的做一些事情。zx 基於 child_process ,轉義參數並提供合理的默認值。
安裝
npm i -g zx
複製代碼需要的環境
Node.js >= 14.8.0
複製代碼將腳本寫入擴展名為 .mjs 的文件中,以便能夠在頂層使用await。
將以下 shebang添加到 zx 腳本的開頭:
#!/usr/bin/env zx
現在您將能夠像這樣運行您的腳本:
chmod +x ./script.mjs
./script.mjs
複製代碼或者通過 zx可執行文件:
zx ./script.mjs
複製代碼所有函數($、cd、fetch 等)都可以直接使用,無需任何導入。
$`command`使用 child_process 包中的 spawn 函數執行給定的字符串, 並返回 ProcessPromise.
let count = parseInt(await $`ls -1 | wc -l`)
console.log(`Files count: ${count}`)
複製代碼例如,要並行上傳文件:
如果執行的程序返回非零退出代碼,ProcessOutput 將被拋出
try {
await $`exit 1`
} catch (p) {
console.log(`Exit code: ${p.exitCode}`)
console.log(`Error: ${p.stderr}`)
}
複製代碼ProcessPromise,以下是promise typescript的接口定義
class ProcessPromise<T> extends Promise<T> {
readonly stdin: Writable
readonly stdout: Readable
readonly stderr: Readable
readonly exitCode: Promise<number>
pipe(dest): ProcessPromise<T>
}
複製代碼pipe() 方法可用於重定向標準輸出:
await $`cat file.txt`.pipe(process.stdout)
複製代碼閱讀更多的關於管道的信息 github.com/google/zx/b…[2]
ProcessOutput的typescript接口定義
class ProcessOutput {
readonly stdout: string
readonly stderr: string
readonly exitCode: number
toString(): string
}
複製代碼函數:
cd()更改當前工作目錄
cd('/tmp')
await $`pwd` // outputs /tmp
複製代碼
fetch()node-fetch 包。
let resp = await fetch('http://wttr.in')
if (resp.ok) {
console.log(await resp.text())
}
複製代碼
question()readline包
let bear = await question('What kind of bear is best? ')
let token = await question('Choose env variable: ', {
choices: Object.keys(process.env)
})
複製代碼在第二個參數中,可以指定選項卡自動完成的選項數組
以下是接口定義
function question(query?: string, options?: QuestionOptions): Promise<string>
type QuestionOptions = { choices: string[] }
複製代碼
sleep()基於setTimeout 函數
await sleep(1000)
複製代碼
nothrow()將 $ 的行為更改, 如果退出碼不是0,不跑出異常.
ts接口定義
function nothrow<P>(p: P): P
複製代碼await nothrow($`grep something from-file`)
// 在管道內:
await $`find ./examples -type f -print0`
.pipe(nothrow($`xargs -0 grep something`))
.pipe($`wc -l`)
複製代碼以下的包,無需導入,直接使用
chalkconsole.log(chalk.blue('Hello world!'))
複製代碼
fs類似於如下的使用方式
import {promises as fs} from 'fs'
let content = await fs.readFile('./package.json')
複製代碼
osawait $`cd ${os.homedir()} && mkdir example`
複製代碼配置:
$.shell指定要用的bash.
$.shell = '/usr/bin/bash'
複製代碼
$.quote指定用於在命令替換期間轉義特殊字符的函數
默認用的是 shq 包.
注意:
__filename & __dirname這兩個變量是在commonjs中的。我們用的是.mjs結尾的es6 模塊。
在ESM模塊中,Node.js 不提供__filename和 __dirname 全局變量。由於此類全局變量在腳本中非常方便,因此 zx 提供了這些以在 .mjs 文件中使用(當使用 zx 可執行文件時)
require也是commonjs中的導入模塊方法, 在 ESM 模塊中,沒有定義 require() 函數。zx提供了 require() 函數,因此它可以與 .mjs 文件中的導入一起使用(當使用 zx 可執行文件時)
傳遞環境變量process.env.FOO = 'bar'
await $`echo $FOO`
複製代碼
傳遞數組如果值數組作為參數傳遞給 $,數組的項目將被單獨轉義並通過空格連接 Example:
let files = [1,2,3]
await $`tar cz ${files}`
複製代碼可以通過顯式導入來使用 $ 和其他函數
#!/usr/bin/env node
import {$} from 'zx'
await $`date`
複製代碼zx 可以將 .ts 腳本編譯為 .mjs 並執行它們
zx examples/typescript.ts
複製代碼