linux shell go to语句,[译] 使用 Go 语言编写一个简单的 SHELL

介绍

在本文中，我们将使用 Go 语言，编写一个最小的 UNIX(-like)操作系统 SHELL，它只需要大概 60 行代码。你需要稍微了解一些 Go 语言(知道如何编译简单的项目)，以及简单使用 UNIX Shell。UNIX 非常简单，简单到一个天才都能理解它的简单性 - Dennis Ritchie

当然，我并非天才，我也不太确定 Dennis Ritchie 所说的，是否也包括运行于用户空间的工具。Shell 只是完整操作系统的一小部分(相较于内核，它真的是一个简单的部分)，但我希望在本文的结尾，你可以感到吃惊，吃惊于编写一个 SHELL，所用到的知识如此少。

什么是SHELL

给 SHELL 下定义有点困难。我认为 SHELL 可以理解为你所使用的操作系统，基本的用户界面。你可以在 SHELL 中输入命令，然后接收一些反馈输出。如果想了解更多信息，或者更明确的定义，请查阅 维基百科)。

一些 SHELL 的例子：

有像 Windows 和 GNOME 这种图形界面 SHELL，但大多数 IT 相关人员(至少我是)，当谈论起 SHELL，指的是基于文本的 SHELL(上面列表的头两项)。当然，也可以简化的定义为非图形界面 SHELL。

事实上，SHELL 的功能可以定义为输入命令，然后接收该命令的输出。想看个例子？运行 ls 命令，输出目录的内容。Input:

ls

Output:

Applications etc

Library home

...

就是这样，十分简单。让我们开始吧！

输入循环

要执行一个命令，我们必须接收输入。而输入来自我们人类，使用键盘进行的。

键盘是我们的标准输入设备(os.Stdin)，我们可以访问并读取它。当我们按下回车键的时候，会创建新的一行。这行新的文本以 \n 结尾。当敲击回车键的时候，所有存储在输入区的内容将被输入。reader := bufio.NewReader(os.Stdin)

input, err := reader.ReadString('\n')

让我们将这些代码输入进我们的 main.go 文件，ReadingString 方法被嵌套在 for 循环中，所以我们可以反复输入命令。当读取输入，发生错误时，我们可以将错误信息输出到标准错误处理设备(os.Stderr)。如果我们使用 fmt.Println，但并没有指定输出设备，这个错误信息还是会输出到标准输出设备中(os.Stdout)。这并不会改变 SHELL 的功能，但是输出到单独的设备，可以方便过滤输出，以进行下一步处理。func main() {

reader := bufio.NewReader(os.Stdin)

for {

// Read the keyboad input.

input, err := reader.ReadString('\n')

if err != nil {

fmt.Fprintln(os.Stderr, err)

}

}

}

执行命令

现在，我们打算执行输入的命令。增加一个名为 execInput 的新的函数，他接收输入的字符串作为参数。首先，我们移除输入结尾的换行符。接下来，通过 exec.Command(input) 来准备执行命令，设置参数，以及捕获输出的结果和错误。最后，通过 cmd.Run() 来执行。func execInput(input string) error {

// Remove the newline character.

input = strings.TrimSuffix(input, "\n")

// Prepare the command to execute.

cmd := exec.Command(input)

// Set the correct output device.

cmd.Stderr = os.Stderr

cmd.Stdout = os.Stdout

// Execute the command and return the error.

return cmd.Run()

}

原型

接着，在循环语句上面，添加一个美化作用的指示器(>)，在循环语句下面，添加新的 execInput 函数，此时，主要功能就完成了。func main() {

reader := bufio.NewReader(os.Stdin)

for {

fmt.Print("> ")

// Read the keyboad input.

input, err := reader.ReadString('\n')

if err != nil {

fmt.Fprintln(os.Stderr, err)

}

// Handle the execution of the input.

if err = execInput(input); err != nil {

fmt.Fprintln(os.Stderr, err)

}

}

}

是时候执行一次测试了。使用 go run main.go 构建并运行咱们的 SHELL。你将看到输入标识符 >，此时可以接受输入。举个例子，我们可以执行 ls 命令。> ls

LICENSE

main.go

main_test.go

不错，可以运行！咱们的程序此时可以执行 ls 命令，并输出当前目录的内容。你可以像退出其他程序一样，使用 ctrl+c，退出它。

参数

让我们命令后面加个参数，如 ls -l。> ls -l

执此时执行会报错：exec: "ls -l": executable file not found in $PATH。

这是因为我们的 SHELL 尝试执行 ls -l，但是并没有找到叫这个名字的程序。我们的意思是执行 ls，带上 -l 的参数。当前，我们的程序还不支持接受命令参数。要修复这个问题，需要修改 execLine 函数，将要执行的命令以空格拆分。func execInput(input string) error {

// Remove the newline character.

input = strings.TrimSuffix(input, "\n")

// Split the input to separate the command and the arguments.

args := strings.Split(input, " ")

// Pass the program and the arguments separately.

cmd := exec.Command(args[0], args[1:]...)

...

}

程序的名字现在存储在 args[0] 中，程序执行的参数存储在数组其他索引中。执行 ls -l 现在可以得到预期的结果。> ls -l

total 24

-rw-r--r-- 1 simon staff 1076 30 Jun 09:49 LICENSE

-rw-r--r-- 1 simon staff 1058 30 Jun 10:10 main.go

-rw-r--r-- 1 simon staff 897 30 Jun 09:49 main_test.go

更改目录(cd)

现在，我们已经可以带着参数执行命令了。现有的这些功能，距离达到一个最小的可用性，只差了一点点。你也许在使用我们的 Shell 时候，已经注意到了：你无法通过 cd 改变当前命令执行的目录。> cd /

> ls

LICENSE

main.go

main_test.go

不，这不是我们根目录的内容。那为什么 cd 命令不起作用呢？要理解这点很容易：没有真正的 cd 程序，该功能是 SHELL 的内置命令。

我们必须对 execInput 函数再次进行修改。在 Split 方法后面，我们添加 switch 结构语句，并将 args[0] 作为它的参数。当这个命令是 cd，我们检查它后面是否还有参数，如果没有指定参数，我们无法改变当前目录(在大多数 SHELL 中，不指定参数，将跳转到主目录)。当 args[1] 中有一个后续参数时(存储路径的参数)，我们使用 os.Chdir(args[1]) 更改目录。在 case 块的末尾，我们返回 execInput 函数以停止其他处理。

因为如此简单，我们在 cd 块后面，再添加一个 exit 命令，exit 可以用来退出当前SHELL(另一个退出方法是 CTRL+C)。// Split the input to separate the command and the arguments.

args := strings.Split(input, " ")

// Check for built-in commands.

switch args[0] {

case "cd":

// 'cd' to home dir with empty path not yet supported.

if len(args) < 2 {

return errors.New("path required")

}

// Change the directory and return the error.

return os.Chdir(args[1])

case "exit":

os.Exit(0)

}

...

可以看到，此时输出的内容，相较于之前的输出结果，更像是我们的根目录。> cd /

> ls

Applications

Library

Network

System

...

至此，我们已经完成了这个简单的 SHEEL 的编写。

考虑改善的地方

此时，如果你觉得有些无聊，你可以尝试改进这个 SHELL。下面是一些可以改善的点：修改光标所在行的显示：增加当前目录

增加机器名称

增加当前用户

通过输入 up/down 键，来翻阅输入的历史

结尾

至此，本文已接近尾声，我希望你读的愉快。如你所见，SHELL 背后的概念十分简单。

Go 同样是一门十分简单的编程语言，它帮助我们更快的得到想要的结果。我们无需关心内存管理。Rob Pike 和 Ken Thompson，以及 Robert Griesemer 共同创造了 Go，他们也创造了 Unix，所以，使用 Go 编写 SHELL 是个很好的选择。

我也一直在学习，如果你发现本文有哪些可改进的地方，请联系我。