Rustでterminalのwindow sizeを取得する

はじめに

RustでCLIツールを作る際にターミナル等のwindow size変更に応じて表示を変更したい場合がある。

その際に画面サイズの取得について調査したログ。

はじめに
画面サイズの取得
調査ログ

画面サイズの取得

画面サイズの取得には、libcのioctlを利用し、TIOCGWINSZをシステムコールする。
Cargo.tomlにはlibcを追記する。

[dependencies]
libc = "0.2"

　
ioctlにはファイルディスクリプタを渡す必要があり、libc::STDOUT_FILENOが使えるが、アプリケーション側にSIGWINCHが上手く送られない場合（CLIツール等を作成している場合）に、Terminalを制御するウィンドウのサイズ変更に動的に対応できない。そのため"/dev/tty"を読む方法も用意する。

use libc::{ioctl, winsize, TIOCGWINSZ, STDOUT_FILENO};
use std::{mem, fs::File, os::unix::io::IntoRawFd};
use std::{thread, time};


pub fn terminal_size() -> Option<winsize> {
    // STDOUT_FILENOか/dev/ttyを利用する
    let fd = if let Ok(file) = File::open("/dev/tty"){
        file.into_raw_fd()
    }else {
        STDOUT_FILENO
    };

    // ファイルディスクリプタに対してTIOCGWINSZをシステムコール
    let mut ws: winsize = unsafe { mem::zeroed() };
    if unsafe { ioctl(fd, TIOCGWINSZ, &mut ws) } == -1 {
        None
    } else {
        Some(ws)
    }
}

// 1秒毎に画面サイズを表示
fn main(){
    loop{
        match terminal_size(){
            Some(ws) => println!("h:{},w:{}", ws.ws_row, ws.ws_col),
            None => println!("failure")
        }
        thread::sleep(time::Duration::from_millis(1000));
    }
}