stephen/swayout
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

code format

Browse Source
This commit is contained in:
Stephen Byrne
2024-12-31 02:51:58 +00:00
parent c5f35ebf06
commit 20ec42d126
4 changed files with 98 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/config.rs
View File

@@ -1,6 +1,6 @@
use kdl::KdlDocument;
use std::collections::HashMap;
use std::fs;
use kdl::KdlDocument;
/// Configuration for swayout.
pub struct Config {
@@ -59,25 +59,32 @@ pub fn get_config() -> Config {
fn parse_config(kdl: String) -> Config {
let doc: KdlDocument = kdl.parse().expect("failed to parse config KDL");
let monitors:Vec<Monitor> = doc.nodes().iter()
let monitors: Vec<Monitor> = doc
.nodes()
.iter()
.filter(|node| node.name().value() == "monitor")
.map(|monitor_node| {
Monitor {
.map(|monitor_node| Monitor {
name: String::from(monitor_node[0].as_string().unwrap()),
make: String::from(monitor_node["make"].as_string().unwrap()),
model: String::from(monitor_node["model"].as_string().unwrap()),
serial: String::from(monitor_node["serial"].as_string().unwrap()),
}
})
.collect();
let layouts:Vec<Layout> = doc.nodes().iter()
let layouts: Vec<Layout> = doc
.nodes()
.iter()
.filter(|node| node.name().value() == "layout")
.map(|layout_node| {
let layout_name = String::from(layout_node[0].as_string().unwrap());
let automatic = layout_node.get("automatic").is_some_and(|v| v.as_bool().is_some_and(|a| a));
let monitor_name_to_output_config:HashMap<String,OutputConfig> = layout_node
.children().unwrap().nodes().iter()
let name = String::from(layout_node[0].as_string().unwrap());
let automatic = layout_node
.get("automatic")
.is_some_and(|v| v.as_bool().is_some_and(|a| a));
let outputs: HashMap<String, OutputConfig> = layout_node
.children()
.unwrap()
.nodes()
.iter()
.map(|output_node| {
let output = String::from(output_node[0].as_string().unwrap());
let output_config = OutputConfig {
@@ -91,9 +98,9 @@ fn parse_config(kdl:String) -> Config {
})
.collect();
Layout {
name : layout_name,
automatic: automatic,
outputs: monitor_name_to_output_config
name,
automatic,
outputs,
}
})
.collect();

64
src/lib.rs
View File

@@ -27,8 +27,7 @@ fn get_layouts() -> HashSet<String> {
// add each individual output (by monitor name if the monitor is defined, else by output name)
available_outputs.iter().for_each(|output| {
// if there is a monitor for the output, use the monitor name, else use the output name
let monitor_opt = config.monitors.iter()
.find(|monitor| {
let monitor_opt = config.monitors.iter().find(|monitor| {
monitor.make == output.make
&& monitor.model == output.model
&& monitor.serial == output.serial
@@ -44,28 +43,39 @@ fn get_layouts() -> HashSet<String> {
}
/// Get the names of monitors that are available (that match an available output)
fn get_available_monitor_names<'a,'b>(config:&'a Config, available_outputs:&'b Vec<Output>) -> Vec<&'a String> {
config.monitors.iter()
.filter(|monitor|
available_outputs.iter().any(|output|
fn get_available_monitor_names<'a, 'b>(
config: &'a Config,
available_outputs: &'b Vec<Output>,
) -> Vec<&'a String> {
config
.monitors
.iter()
.filter(|monitor| {
available_outputs.iter().any(|output| {
monitor.make == output.make
&& monitor.model == output.model
&& output.serial == monitor.serial
)
)
})
})
.map(|monitor| &monitor.name)
.collect()
}
/// Get the layouts for which all monitors are available.
fn get_available_layouts<'a,'b>(config:&'a Config, available_outputs:&'b Vec<Output>) -> Vec<&'a Layout> {
let available_monitor_names:Vec<&String> = get_available_monitor_names(&config, available_outputs);
fn get_available_layouts<'a, 'b>(
config: &'a Config,
available_outputs: &'b Vec<Output>,
) -> Vec<&'a Layout> {
let available_monitor_names: Vec<&String> =
get_available_monitor_names(&config, available_outputs);
let mut layouts: Vec<&'a Layout> = Vec::new();
// Add each layout defined in the config file for which all outputs are available
config.layouts.iter().for_each(|layout| {
if layout.outputs.iter()
if layout
.outputs
.iter()
.all(|(output_name, _output)| available_monitor_names.contains(&output_name))
{
layouts.push(&layout);
@@ -92,10 +102,21 @@ pub fn apply_layout(layout_name: &String) {
let mut output_config_map: HashMap<&String, &OutputConfig> = HashMap::new();
// First check if the layout is defined in the rules
if let Some(layout) = rules.layouts.iter().find(|layout| &layout.name == layout_name) {
if let Some(layout) = rules
.layouts
.iter()
.find(|layout| &layout.name == layout_name)
{
// The layout is defined in the rules.
layout.outputs.iter().for_each(|(monitor_name, output_config)| {
let monitor = &rules.monitors.iter().find(|monitor| &monitor.name == monitor_name).unwrap();
layout
.outputs
.iter()
.for_each(|(monitor_name, output_config)| {
let monitor = &rules
.monitors
.iter()
.find(|monitor| &monitor.name == monitor_name)
.unwrap();
// find the output for the monitor to get the output name.
let output_opt = outputs.iter().find(|output| {
@@ -112,10 +133,15 @@ pub fn apply_layout(layout_name: &String) {
} else {
// The layout is not defined in the rules.
// See if it is a monitor name...
if let Some(monitor) = rules.monitors.iter().find(|monitor| &monitor.name == layout_name) {
if let Some(monitor) = rules
.monitors
.iter()
.find(|monitor| &monitor.name == layout_name)
{
// It is a monitor name. Find the matching output...
if let Some(output) = outputs.iter().find(|output| {
output.make == monitor.make && output.model == monitor.model
output.make == monitor.make
&& output.model == monitor.model
&& output.serial == monitor.serial
}) {
output_config_map.insert(&output.name, &output.output_config);
@@ -124,8 +150,7 @@ pub fn apply_layout(layout_name: &String) {
}
} else {
// See if it is an output name...
if let Some(output) = outputs.iter()
.find(|output| &output.name == layout_name) {
if let Some(output) = outputs.iter().find(|output| &output.name == layout_name) {
output_config_map.insert(&output.name, &output.output_config);
} else {
panic!("could not find layout, monitor, or output {layout_name}")
@@ -144,7 +169,8 @@ pub fn apply_automatic() -> Option<String> {
if let Some(layout) = get_available_layouts(&config, &outputs)
.iter()
.find(|layout| layout.automatic) {
.find(|layout| layout.automatic)
{
apply_layout(&layout.name);
Some(String::from(&layout.name))
} else {

5
src/sway.rs
View File

@@ -1,8 +1,8 @@
use crate::config::OutputConfig;
use serde_json::{from_str, Value};
use std::collections::HashMap;
use std::process::Command;
use std::str::from_utf8;
use serde_json::{from_str, Value};
use crate::config::OutputConfig;
/// An output, as returned by `swaymsg -t get_outputs`.
pub struct Output {
@@ -68,7 +68,6 @@ fn get_mode(output: &Value) -> String {
/// Apply the specified outputs. Enable all outputs in [outputs], disable others.
pub fn apply_outputs(all_outputs: &Vec<Output>, outputs: &HashMap<&String, &OutputConfig>) {
// set enabled outputs first, then set disabled outputs.
// That way if some work before an error, you have at least one output enabled.