获取Go语言中终端尺寸的Cgo实现详解

#include <sys/ioctl.h> // For ioctl and TIOCGWINSZ
#include <termios.h>   // For struct winsize (often included via sys/ioctl or unistd.h)

struct winsize ws;
ioctl(0, TIOCGWINSZ, &ws); // 0 is stdin file descriptor
// ws.ws_row and ws.ws_col now hold the terminal dimensions

// #include <sys/ioctl.h>
// #include <termios.h> // Usually where winsize is defined

// 定义 Go 语言中使用的 winsize 结构体，与 C 语言的 struct winsize 对应
typedef struct winsize {
    unsigned short ws_row;    // 终端行数
    unsigned short ws_col;    // 终端列数
    unsigned short ws_xpixel; // 终端像素宽度 (不常用)
    unsigned short ws_ypixel; // 终端像素高度 (不常用)
} winsize;

// TIOCGWINSZ 是获取终端尺寸的 ioctl 命令，其值在不同系统上可能不同。
// 0x5413 是 Linux 系统上的常用值。
const TIOCGWINSZ C.ulong = 0x5413

// void myioctl(int fd, unsigned long request, winsize *ws){ioctl(fd, request, ws);}

package main

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

// 定义 Go 语言中使用的 winsize 结构体，与 C 语言的 struct winsize 对应
typedef struct winsize {
    unsigned short ws_row;    // 终端行数
    unsigned short ws_col;    // 终端列数
    unsigned short ws_xpixel; // 终端像素宽度 (不常用)
    unsigned short ws_ypixel; // 终端像素高度 (不常用)
} winsize;

// 封装 ioctl 函数，使其成为一个固定参数的 C 函数，以便 cgo 调用
// ioctl 的第三个参数类型取决于第二个参数 (request)。
// 对于 TIOCGWINSZ，第三个参数是指向 winsize 结构体的指针。
void myioctl(int fd, unsigned long request, winsize *ws) {
    ioctl(fd, request, ws);
}
*/
import "C" // 导入 Cgo

import (
    "fmt"
    "os"
)

// TIOCGWINSZ 是获取终端尺寸的 ioctl 命令，其值在不同系统上可能不同。
// 0x5413 是 Linux 系统上的常用值。
const TIOCGWINSZ C.ulong = 0x5413

// GetTerminalSize 获取当前终端的行数和列数
func GetTerminalSize() (rows, cols int, err error) {
    var ws C.winsize // 声明一个 C.winsize 类型的变量
    // 获取标准输入的文件描述符
    fd := int(os.Stdin.Fd())

    // 调用封装的 C 函数 myioctl 来获取终端尺寸
    // C.myioctl(C.int(fd), TIOCGWINSZ, &ws)
    // ioctl 函数返回 -1 表示失败，0 表示成功
    ret := C.myioctl(C.int(fd), TIOCGWINSZ, &ws)
    if ret == -1 {
        return 0, 0, fmt.Errorf("failed to get terminal size via ioctl: %w", os.NewSyscallError("ioctl", nil))
    }

    // 将 C.winsize 的值转换为 Go 的 int 类型
    rows = int(ws.ws_row)
    cols = int(ws.ws_col)
    return rows, cols, nil
}

func main() {
    rows, cols, err := GetTerminalSize()
    if err != nil {
        fmt.Printf("Error getting terminal size: %v\n", err)
        return
    }
    fmt.Printf("Terminal size: %d rows, %d columns\n", rows, cols)
}