auv-cli-invoke 0.0.22

Application Use Via... script-able OS automation library
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

use std::collections::HashSet;

use crate::{CommandGroup, CommandNode, InvokeCommand, commands};

pub struct InvokeRegistry {
  groups: Vec<CommandGroup>,
  commands: Vec<InvokeCommand>,
}

impl InvokeRegistry {
  pub fn from_groups(groups: Vec<CommandGroup>) -> Self {
    let mut commands = Vec::new();
    for group in &groups {
      flatten_group(group, &mut commands);
    }
    assert_unique_command_ids(&commands);
    Self { groups, commands }
  }

  pub fn resolve(&self, command_id: &str) -> Option<&InvokeCommand> {
    self.commands.iter().find(|command| command.id == command_id)
  }

  pub fn all(&self) -> &[InvokeCommand] {
    &self.commands
  }

  pub fn groups(&self) -> &[CommandGroup] {
    &self.groups
  }
}

fn assert_unique_command_ids(commands: &[InvokeCommand]) {
  let mut ids = HashSet::new();
  for command in commands {
    assert!(ids.insert(command.id), "duplicate invoke command id registered: {}", command.id);
  }
}

pub fn default_registry() -> InvokeRegistry {
  InvokeRegistry::from_groups(vec![
    commands::display::group(),
    commands::screen::group(),
    commands::window::group(),
    commands::input::group(),
    commands::app::group(),
    commands::overlay::group(),
    commands::media_control::group(),
    commands::scan::group(),
  ])
}

fn flatten_group(group: &CommandGroup, commands: &mut Vec<InvokeCommand>) {
  for child in &group.children {
    match child {
      CommandNode::Command(command) => commands.push(command.clone()),
      CommandNode::Group(group) => flatten_group(group, commands),
    }
  }
}