diff --git a/src/main.rs b/src/main.rs index ce7718b..08319f5 100644 --- a/src/main.rs +++ b/src/main.rs @@ -38,6 +38,9 @@ enum Command { #[clap(about = "Focus to another workspace on the same output")] Focus(FocusAction), + #[clap(about = "Focus to output")] + FocusOutput(FocusAction), + #[clap(about = "Focus to another workspace on all the outputs")] FocusAllOutputs(FocusAction), @@ -47,6 +50,12 @@ enum Command { #[clap(about = "Move the focused container to the previous output")] PrevOutput, + #[clap(about = "Move the focused container to the next group")] + NextGroup, + + #[clap(about = "Move the focused container to the previous group")] + PrevGroup, + #[clap(about = "Rearrange already opened workspaces")] RearrangeWorkspaces, } @@ -195,7 +204,7 @@ fn get_current_output_index(stream: &UnixStream) -> usize { let outputs = get_outputs(stream); match outputs.iter().position(|x| x.focused) { - Some(i) => i, + Some(i) => i + 1, None => panic!("WTF! No focused output???"), } } @@ -211,22 +220,200 @@ fn get_current_output_name(stream: &UnixStream) -> String { focused_output_index.to_string() } +fn get_current_workspace(stream: &UnixStream) -> Workspace { + let outputs = get_outputs(stream); + let workspaces = get_workspaces(stream); + workspaces + .into_iter() + .find(|w| w.visible && outputs.iter().find(|o| o.name == w.output).unwrap().focused) + .unwrap() +} + fn move_container_to_workspace(stream: &UnixStream, workspace_index: usize) { - let mut cmd: String = "move container to workspace number ".to_string(); - let full_ws_name = format!("{}{}", get_current_output_index(stream), workspace_index) + if workspace_index < 10 { + let mut cmd: String = "move container to workspace number ".to_string(); + let full_ws_name = format!("{}{}", get_current_output_index(stream), workspace_index) + .parse::() + .unwrap(); + cmd.push_str(&full_ws_name.to_string()); + send_command(stream, &cmd); + } else { + move_container_to_workspace_absolute(stream, workspace_index); + } +} + +fn move_container_to_workspace_absolute(stream: &UnixStream, workspace_index: usize) { + let output_index = (workspace_index / 10) as usize; + let outputs = get_outputs(stream); + let workspaces = get_workspaces(stream); + + // If the workspace already exists + match workspaces.iter().find(|w| w.num == workspace_index) { + Some(_) => { + let mut focus_cmd: String = "move container to workspace number ".to_string(); + focus_cmd.push_str(&workspace_index.to_string()); + send_command(stream, &focus_cmd); + } + None => { + let target_group = workspace_index / 10; + let target_screen_index = match workspaces.iter().find(|w| w.num / 10 == target_group) { + // If other workspaces on the same group exists + Some(other_workspace) => Some( + outputs + .iter() + .enumerate() + .find(|i| i.1.name == other_workspace.output) + .unwrap() + .0, + ), + None => { + // Or if the targeted output is currently connected + if output_index < outputs.len() { + Some(output_index) + } else { + None + } + } + }; + + match target_screen_index { + // If we have to send it to another screen + Some(target_screen_index) => { + let target_output = &outputs[target_screen_index - 1]; + + let current_output_name = get_current_output_name(stream); + + let mut focus_cmd: String = "focus output ".to_string(); + focus_cmd.push_str(&target_output.name); + send_command(stream, &focus_cmd); + + let focused_workspace_index = get_current_workspace(stream).num; + + let mut focus_cmd: String = "workspace ".to_string(); + focus_cmd.push_str(&workspace_index.to_string()); + send_command(stream, &focus_cmd); + + let mut focus_cmd: String = "focus output ".to_string(); + focus_cmd.push_str(¤t_output_name); + send_command(stream, &focus_cmd); + + let mut focus_cmd: String = "move container to workspace ".to_string(); + focus_cmd.push_str(&workspace_index.to_string()); + send_command(stream, &focus_cmd); + + let mut focus_cmd: String = "focus output ".to_string(); + focus_cmd.push_str(&target_output.name); + send_command(stream, &focus_cmd); + + let mut focus_cmd: String = "workspace ".to_string(); + focus_cmd.push_str(&focused_workspace_index.to_string()); + send_command(stream, &focus_cmd); + + let mut focus_cmd: String = "focus output ".to_string(); + focus_cmd.push_str(¤t_output_name); + send_command(stream, &focus_cmd); + } + None => { + // Else, we send the container on the current output + let mut focus_cmd: String = "move container to workspace ".to_string(); + focus_cmd.push_str(&workspace_index.to_string()); + send_command(stream, &focus_cmd); + } + }; + } + } +} + +fn focus_to_workspace(stream: &UnixStream, workspace_index: usize) { + if workspace_index < 10 { + let mut cmd: String = "workspace number ".to_string(); + let full_ws_name = format!("{}{}", get_current_output_index(stream), &workspace_index) + .parse::() + .unwrap(); + cmd.push_str(&full_ws_name.to_string()); + send_command(stream, &cmd); + } else { + focus_to_workspace_absolute(stream, workspace_index); + } +} + +fn focus_to_workspace_absolute(stream: &UnixStream, workspace_index: usize) { + let output_index = (workspace_index / 10) as usize; + let outputs = get_outputs(stream); + let workspaces = get_workspaces(stream); + + // If the workspace already exists + match workspaces.iter().find(|w| w.num == workspace_index) { + Some(_) => { + let mut focus_cmd: String = "workspace number ".to_string(); + focus_cmd.push_str(&workspace_index.to_string()); + send_command(stream, &focus_cmd); + } + None => { + let target_group = workspace_index / 10; + let target_screen_index = match workspaces.iter().find(|w| w.num / 10 == target_group) { + // If other workspaces on the same group exists + Some(other_workspace) => Some( + outputs + .iter() + .enumerate() + .find(|i| i.1.name == other_workspace.output) + .unwrap() + .0, + ), + None => { + // Or if the targeted output is currently connected + if output_index < outputs.len() { + Some(output_index) + } else { + None + } + } + }; + + match target_screen_index { + // If we have to send it to another screen + Some(target_screen_index) => { + let target_output = &outputs[target_screen_index - 1]; + + let mut focus_cmd: String = "focus output ".to_string(); + focus_cmd.push_str(&target_output.name); + send_command(stream, &focus_cmd); + } + None => {} + }; + // Then we focus the workspace + let mut focus_cmd: String = "workspace number ".to_string(); + focus_cmd.push_str(&workspace_index.to_string()); + send_command(stream, &focus_cmd); + } + } +} + +fn focus_to_workspace_virtual_output(stream: &UnixStream, workspace_index: usize) { + let current_workspace_index: usize = get_current_workspace(stream).num; + let focused_output_index = current_workspace_index / 10; + + let mut cmd: String = "workspace number ".to_string(); + let full_ws_name = format!("{}{}", focused_output_index, &workspace_index) .parse::() .unwrap(); cmd.push_str(&full_ws_name.to_string()); send_command(stream, &cmd); } -fn focus_to_workspace(stream: &UnixStream, workspace_index: usize) { - let mut cmd: String = "workspace number ".to_string(); - let full_ws_name = format!("{}{}", get_current_output_index(stream), &workspace_index) - .parse::() - .unwrap(); - cmd.push_str(&full_ws_name.to_string()); - send_command(stream, &cmd); +fn focus_to_output(stream: &UnixStream, target_output_index: usize) { + let outputs = get_outputs(stream); + if target_output_index < outputs.len() { + let mut cmd: String = "focus output ".to_string(); + cmd.push_str(&outputs[target_output_index - 1].name); + send_command(stream, &cmd); + } else { + let current_workspace_index = get_current_workspace(stream).num; + let contextual_workspace_index = + current_workspace_index - (current_workspace_index / 10) * 10; + focus_to_workspace_virtual_output(stream, contextual_workspace_index); + } } fn focus_all_outputs_to_workspace(stream: &UnixStream, workspace_index: usize) { @@ -258,15 +445,12 @@ fn move_container_to_prev_output(stream: &UnixStream) { fn move_container_to_next_or_prev_output(stream: &UnixStream, go_to_prev: bool) { let outputs = get_outputs(stream); - let focused_output_index = match outputs.iter().position(|x| x.focused) { - Some(i) => i, - None => panic!("WTF! No focused output???"), - }; + let focused_output_index = get_current_output_index(stream); let target_output = if go_to_prev { - &outputs[(focused_output_index + outputs.len() - 1) % outputs.len()] + &outputs[(focused_output_index + outputs.len() - 1) % outputs.len() - 1] } else { - &outputs[(focused_output_index + 1) % outputs.len()] + &outputs[(focused_output_index + 1) % outputs.len() - 1] }; let workspaces = get_workspaces(stream); @@ -286,6 +470,42 @@ fn move_container_to_next_or_prev_output(stream: &UnixStream, go_to_prev: bool) send_command(stream, &cmd); } +fn move_container_to_next_group(stream: &UnixStream) { + move_container_to_next_or_prev_group(stream, false); +} + +fn move_container_to_prev_group(stream: &UnixStream) { + move_container_to_next_or_prev_group(stream, true); +} + +fn move_container_to_next_or_prev_group(stream: &UnixStream, go_to_prev: bool) { + let current_workspace_index: usize = get_current_workspace(stream).num; + let focused_group_index = current_workspace_index / 10; + + let outputs = get_outputs(stream); + if focused_group_index < outputs.len() { + let target_output = if go_to_prev { + &outputs[(focused_group_index - 1) - 1] + } else { + &outputs[(focused_group_index + 1) - 1] + }; + + let mut cmd: String = "focus output ".to_string(); + cmd.push_str(&target_output.name); + send_command(stream, &cmd); + } else { + let target_workspace = if go_to_prev { + current_workspace_index - 10 + } else { + current_workspace_index + 10 + }; + + let mut cmd: String = "workspace number ".to_string(); + cmd.push_str(&target_workspace.to_string()); + send_command(stream, &cmd); + }; +} + fn init_workspaces(stream: &UnixStream, workspace_index: usize) { let outputs = get_outputs(stream); @@ -313,7 +533,7 @@ fn rearrange_workspaces(stream: &UnixStream) { let output_index = workspace.num / 10; if output_index < outputs.len() { let mut move_cmd = move_cmd_prefix.clone(); - move_cmd.push_str(&outputs[output_index].name); + move_cmd.push_str(&outputs[output_index - 1].name); send_command(stream, &move_cmd); } } @@ -333,6 +553,9 @@ fn main() { Command::Focus(action) => { focus_to_workspace(&stream, action.index); } + Command::FocusOutput(action) => { + focus_to_output(&stream, action.index); + } Command::FocusAllOutputs(action) => { focus_all_outputs_to_workspace(&stream, action.index); } @@ -342,6 +565,12 @@ fn main() { Command::PrevOutput => { move_container_to_prev_output(&stream); } + Command::NextGroup => { + move_container_to_next_group(&stream); + } + Command::PrevGroup => { + move_container_to_prev_group(&stream); + } Command::RearrangeWorkspaces => { rearrange_workspaces(&stream); }