如何在Go中实现终端底部固定提示符的聊天客户端-Golang-PHP中文网

如何在Go中实现终端底部固定提示符的聊天客户端

霞舞

发布： 2025-09-22 16:02:01

原创

437人浏览过

如何在go中实现终端底部固定提示符的聊天客户端

本文介绍了如何使用Go语言创建一个终端聊天客户端，该客户端能够保持提示符固定在屏幕底部，即使在用户输入时收到新消息也能正确显示。我们将探讨如何利用termbox-go库来实现这一功能，该库提供了对终端的底层控制，可以方便地实现复杂的终端交互效果。

使用 termbox-go 构建终端聊天客户端

要实现一个在终端底部固定提示符的聊天客户端，我们需要一个能够控制终端输出和输入的库。termbox-go 是一个不错的选择，它提供了一组函数，允许我们直接操作终端的屏幕缓冲区，从而实现复杂的布局和交互。

1. 安装 termbox-go

首先，你需要安装 termbox-go 库。可以使用以下命令：

go get github.com/nsf/termbox-go

登录后复制

2. 初始化 termbox

在使用 termbox-go 之前，需要初始化它：

package main

import (
    "fmt"
    "github.com/nsf/termbox-go"
    "log"
)

func main() {
    err := termbox.Init()
    if err != nil {
        log.Fatal(err)
    }
    defer termbox.Close()

    // ... 你的代码 ...
}

登录后复制

termbox.Init() 函数会初始化终端，并将其设置为原始模式，这意味着你可以直接控制终端的输入和输出。defer termbox.Close() 确保在程序退出时正确关闭终端。

3. 绘制屏幕

接下来，我们需要编写代码来绘制屏幕。我们需要一个区域来显示聊天消息，一个区域来显示提示符和用户输入。

func draw(messages []string, prompt string) {
    termbox.Clear(termbox.ColorDefault, termbox.ColorDefault) // 清空屏幕
    width, height := termbox.Size()

    // 绘制消息
    for i, message := range messages {
        y := height - 2 - len(messages) + i // 从倒数第二行开始向上绘制
        if y >= 0 {
            for x, r := range message {
                termbox.SetCell(x, y, r, termbox.ColorDefault, termbox.ColorDefault)
            }
        }
    }

    // 绘制提示符和用户输入
    promptText := "> " + prompt
    for x, r := range promptText {
        termbox.SetCell(x, height-1, r, termbox.ColorDefault, termbox.ColorDefault)
    }

    termbox.Flush() // 刷新屏幕
}

登录后复制

这个 draw 函数接受一个消息列表和一个提示符字符串，并在终端上绘制它们。它首先清空屏幕，然后从倒数第二行开始向上绘制消息。最后，它在最后一行绘制提示符和用户输入。

4. 处理用户输入

知我AI·PC客户端

离线运行 AI 大模型，构建你的私有个人知识库，对话式提取文件知识，保证个人文件数据安全

查看详情

我们需要一个循环来监听用户的输入，并将输入添加到提示符字符串中。

func handleInput(prompt *string, messages *[]string) {
    for {
        event := termbox.PollEvent()
        switch event.Type {
        case termbox.EventKey:
            if event.Key == termbox.KeyEsc {
                return // 退出程序
            } else if event.Key == termbox.KeyEnter {
                *messages = append(*messages, *prompt) // 将输入的消息添加到消息列表中
                *prompt = ""                             // 清空提示符
            } else if event.Key == termbox.KeyBackspace2 || event.Key == termbox.KeyBackspace {
                if len(*prompt) > 0 {
                    *prompt = (*prompt)[:len(*prompt)-1] // 删除最后一个字符
                }
            } else if event.Ch != 0 {
                *prompt += string(event.Ch) // 添加字符到提示符
            }
        case termbox.EventError:
            panic(event.Err)
        }

        draw(*messages, *prompt) // 重新绘制屏幕
    }
}

登录后复制

这个 handleInput 函数监听终端的事件。如果用户按下 Esc 键，它会退出程序。如果用户按下 Enter 键，它会将当前的提示符字符串添加到消息列表中，并清空提示符。如果用户按下 Backspace 键，它会删除提示符字符串的最后一个字符。如果用户输入了其他字符，它会将字符添加到提示符字符串中。每次事件发生后，它都会调用 draw 函数来重新绘制屏幕。

5. 整合代码

最后，我们需要将所有的代码整合在一起：

package main

import (
    "fmt"
    "github.com/nsf/termbox-go"
    "log"
)

func draw(messages []string, prompt string) {
    termbox.Clear(termbox.ColorDefault, termbox.ColorDefault)
    width, height := termbox.Size()

    for i, message := range messages {
        y := height - 2 - len(messages) + i
        if y >= 0 {
            for x, r := range message {
                termbox.SetCell(x, y, r, termbox.ColorDefault, termbox.ColorDefault)
            }
        }
    }

    promptText := "> " + prompt
    for x, r := range promptText {
        termbox.SetCell(x, height-1, r, termbox.ColorDefault, termbox.ColorDefault)
    }

    termbox.Flush()
}

func handleInput(prompt *string, messages *[]string) {
    for {
        event := termbox.PollEvent()
        switch event.Type {
        case termbox.EventKey:
            if event.Key == termbox.KeyEsc {
                return
            } else if event.Key == termbox.KeyEnter {
                *messages = append(*messages, *prompt)
                *prompt = ""
            } else if event.Key == termbox.KeyBackspace2 || event.Key == termbox.KeyBackspace {
                if len(*prompt) > 0 {
                    *prompt = (*prompt)[:len(*prompt)-1]
                }
            } else if event.Ch != 0 {
                *prompt += string(event.Ch)
            }
        case termbox.EventError:
            panic(event.Err)
        }

        draw(*messages, *prompt)
    }
}

func main() {
    err := termbox.Init()
    if err != nil {
        log.Fatal(err)
    }
    defer termbox.Close()

    messages := []string{}
    prompt := ""

    draw(messages, prompt)
    handleInput(&prompt, &messages)
}

登录后复制

6. 运行代码

保存代码为 main.go，然后运行它：