squeekboard/src/tests.rs

/*! Testing functionality */

use ::data::Layout;
use xkbcommon::xkb;

use ::util::WarningHandler;


pub struct CountAndPrint(u32);

impl WarningHandler for CountAndPrint {
    fn handle(&mut self, warning: &str) {
        self.0 = self.0 + 1;
        println!("{}", warning);
    }
}

impl CountAndPrint {
    fn new() -> CountAndPrint {
        CountAndPrint(0)
    }
}

pub fn check_builtin_layout(name: &str) {
    check_layout(Layout::from_resource(name).expect("Invalid layout data"))
}

pub fn check_layout_file(path: &str) {
    check_layout(Layout::from_file(path.into()).expect("Invalid layout file"))
}

fn check_layout(layout: Layout) {
    let handler = CountAndPrint::new();
    let (layout, handler) = layout.build(handler);

    if handler.0 > 0 {
        println!("{} mistakes in layout", handler.0)
    }

    let layout = layout.expect("layout broken");

    let context = xkb::Context::new(xkb::CONTEXT_NO_FLAGS);
    
    let keymap_str = layout.keymap_str
        .clone()
        .into_string().expect("Failed to decode keymap string");
    
    let keymap = xkb::Keymap::new_from_string(
        &context,
        keymap_str.clone(),
        xkb::KEYMAP_FORMAT_TEXT_V1,
        xkb::KEYMAP_COMPILE_NO_FLAGS,
    ).expect("Failed to create keymap");

    let state = xkb::State::new(&keymap);
    
    // "Press" each button with keysyms
    for view in layout.views.values() {
        for (_y, row) in &view.get_rows() {
            for (_x, button) in &row.buttons {
                let keystate = button.state.borrow();
                for keycode in &keystate.keycodes {
                    match state.key_get_one_sym(*keycode) {
                        xkb::KEY_NoSymbol => {
                            eprintln!("{}", keymap_str);
                            panic!("Keysym {} on key {:?} can't be resolved", keycode, button.name);
                        },
                        _ => {},
                    }
                }
            }
        }
    }

    if handler.0 > 0 {
        panic!("Layout contains mistakes");
    }
}