Go 语言中获取终端尺寸的实践指南

#include <sys/ioctl.h>
#include <stdio.h>

struct ttysize {
    unsigned short ts_lines;
    unsigned short ts_cols;
    unsigned short ts_xpixel;
    unsigned short ts_ypixel;
};

int main() {
    struct ttysize ts;
    if (ioctl(0, TIOCGWINSZ, &ts) == 0) {
        printf("Terminal size: %d rows, %d columns\n", ts.ts_lines, ts.ts_cols);
    } else {
        perror("ioctl");
    }
    return 0;
}

// #include <sys/ioctl.h>
// typedef struct ttysize ttysize; // 假设系统定义了ttysize，或需要自己定义
import "C"

const TIOCGWINSZ C.ulong = 0x5413 // TIOCGWINSZ 的常量值

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 */
};

// #include <sys/ioctl.h>
// typedef struct ttysize ttysize;
// void myioctl(int i, unsigned long l, ttysize * t){ioctl(i,l,t);} // 封装 ioctl
import "C"

// ... (Go code)

package main

import (
    "fmt"
    "os"
    "syscall"
)

/*
#include <sys/ioctl.h>

// 定义 ttysize 结构体，与系统保持一致
// 注意：在现代 Linux 系统中，通常是 struct winsize
// 为了兼容原始问题，这里使用 ttysize
typedef struct ttysize {
    unsigned short ts_lines;
    unsigned short ts_cols;
    unsigned short ts_xpixel;
    unsigned short ts_ypixel;
} ttysize;

// 封装 ioctl 函数，避免 cgo 处理变参的限制
void myioctl(int fd, unsigned long request, ttysize *ts) {
    ioctl(fd, request, ts);
}
*/
import "C"

// TIOCGWINSZ 的常量值，通常为 0x5413
// 这个值可能因操作系统或架构而异，但 0x5413 在许多 Unix-like 系统上是通用的。
const TIOCGWINSZ C.ulong = 0x5413

// GetTerminalSize 获取终端的行数和列数
func GetTerminalSize() (rows, cols int, err error) {
    var ts C.ttysize
    // 使用 os.Stdin.Fd() 获取标准输入的描述符
    fd := int(os.Stdin.Fd())

    // 调用封装后的 C 函数
    C.myioctl(C.int(fd), TIOCGWINSZ, &ts)

    // 检查 ioctl 调用是否成功（虽然 myioctl 封装后没有返回错误，
    // 但通常 ioctl 会返回 -1 表示失败，并设置 errno）
    // 我们可以通过检查 Go 的 syscall.Errno 来判断
    if errno := syscall.Errno(C.myioctl(C.int(fd), TIOCGWINSZ, &ts)); errno != 0 {
        return 0, 0, fmt.Errorf("ioctl failed: %w", errno)
    }

    return int(ts.ts_lines), int(ts.ts_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)
}