如何在Go语言中获取终端尺寸：cgo与ioctl的实践

#include <sys/ioctl.h> // 包含 ioctl 函数和 TIOCGWINSZ 定义
#include <stdio.h>

struct winsize {
    unsigned short ws_row;    /* rows, in characters */
    unsigned short ws_col;    /* columns, in characters */
    unsigned short ws_xpixel; /* horizontal size, in pixels */
    unsigned short ws_ypixel; /* vertical size, in pixels */
};

int main() {
    struct winsize ws;
    // ioctl(文件描述符, 请求, 参数)
    // 0 通常代表标准输入文件描述符
    if (ioctl(0, TIOCGWINSZ, &ws) == 0) {
        printf("Terminal size: %d rows, %d columns\n", ws.ws_row, ws.ws_col);
    } else {
        perror("ioctl failed");
    }
    return 0;
}

// #include <sys/ioctl.h>
// #include <termios.h> // 包含 winsize 结构体定义，或在C代码中自定义
import "C"

// 在Linux系统上，TIOCGWINSZ的值通常为0x5413。
// 请注意，这个值可能因操作系统而异（例如，macOS的值不同）。
const TIOCGWINSZ C.ulong = 0x5413

// #include <sys/ioctl.h>
// #include <termios.h> // 包含 winsize 结构体定义

// 定义一个C语言函数，用于包装ioctl调用
// 这里的 winsize 结构体通常在 <termios.h> 或 <sys/ioctl.h> 中定义
// 如果你的系统使用 ttysize，则需要相应修改
// 例如：typedef struct ttysize { unsigned short ts_lines; unsigned short ts_cols; } ttysize;
// 在大多数现代Unix-like系统中，通常是 struct winsize
void myioctl(int fd, unsigned long request, struct winsize *ws) {
    ioctl(fd, request, ws);
}
import "C"

package main

import (
    "fmt"
    "os"
    "syscall"
    "unsafe" // 用于类型转换
)

/*
#include <sys/ioctl.h>
#include <termios.h> // 通常包含 struct winsize 的定义

// 定义一个C语言函数，用于包装ioctl调用
// 这里的 struct winsize 是终端尺寸结构体，包含行和列信息
// 注意：在某些旧系统或特定配置下，可能是 struct ttysize
// 如果编译报错，请检查你的系统上 struct winsize 的定义
// 并确保其字段名和类型与 Go 中的 C.winsize 对应
void myioctl(int fd, unsigned long request, struct winsize *ws) {
    ioctl(fd, request, ws);
}
*/
import "C"

// TIOCGWINSZ 的值在不同操作系统上可能不同。
// 0x5413 是 Linux 上的值。
// 如果在 macOS 上，它可能是 0x40087468。
const TIOCGWINSZ C.ulong = 0x5413

// GetTerminalSize 获取当前终端的行数和列数
func GetTerminalSize() (rows, cols int, err error) {
    // 创建一个 C 语言的 winsize 结构体实例
    var ws C.struct_winsize

    // 调用包装后的 C 函数 myioctl
    // os.Stdin.Fd() 获取标准输入的文件描述符
    // unsafe.Pointer(&ws) 将 Go 变量的地址转换为 C 指针
    // 注意：myioctl 不返回错误码，需要自行处理 ioctl 的返回值
    // 但由于我们是在 C 包装函数中调用，Go 无法直接获取 ioctl 的返回值
    // 通常，如果 ioctl 失败，ws 的内容可能不会被修改，或者返回全0
    // 更好的做法是让 C 包装函数返回 int，表示 ioctl 的结果
    // 这里简化处理，假设成功
    C.myioctl(C.int(os.Stdin.Fd()), TIOCGWINSZ, &ws)

    // 如果 myioctl 失败，ws 可能包含无效数据。
    // 在生产环境中，应该通过 C 函数返回 ioctl 的结果码来判断是否成功。
    // 例如：
    // int ret = ioctl(fd, request, ws); return ret;
    // 然后在 Go 中检查 C.myioctl 的返回值。
    // 这里为了与原始答案保持一致，简化处理。

    // 检查是否获取到有效尺寸
    if ws.ws_row == 0 || ws.ws_col == 0 {
        // 尝试使用 syscall.Syscall6 来直接调用 ioctl，可以获取返回值
        // 这种方式更底层，但可以处理错误
        // TIOCGWINSZ 是一个平台相关的常量，需要确保其值正确
        // 这里的 TIOCGWINSZ 值为 0x5413 是 Linux 的，对于其他系统需要调整
        _, _, errno := syscall.Syscall6(
            syscall.SYS_IOCTL,
            os.Stdin.Fd(),
            uintptr(TIOCGWINSZ),
            uintptr(unsafe.Pointer(&ws)),
            0, 0, 0,
        )
        if errno != 0 {
            return 0, 0, errno
        }
    }

    return int(ws.ws_row), int(ws.ws_col), nil
}

func main() {
    rows, cols, err := GetTerminalSize()
    if err != nil {
        fmt.Printf("获取终端尺寸失败: %v\n", err)
        // 如果在非TTY环境下运行，例如管道或重定向，ioctl可能会失败
        // 此时可以提供默认值或退出
        return
    }
    fmt.Printf("当前终端尺寸: %d 行, %d 列\n", rows, cols)
}

   package main

   import (
       "fmt"
       "os"
       "golang.org/x/term" // 导入 golang.org/x/term
   )

   func main() {
       // GetSize 函数会自动处理底层系统调用和平台差异
       width, height, err := term.GetSize(int(os.Stdout.Fd()))
       if err != nil {
           fmt.Printf("获取终端尺寸失败: %v\n", err)
           return
       }
       fmt.Printf("当前终端尺寸: %d 行, %d 列 (通过 golang.org/x/term)\n", height, width)
   }