mirror of
https://gitlab.com/hyask/swaysome.git
synced 2025-12-10 07:44:43 +01:00
Rework everything to have a global SwaySome object with methods in it
This allows easily keeping state across the various function calls.
This commit is contained in:
parent
99a99648b1
commit
90ec4a2566
1 changed files with 551 additions and 521 deletions
356
src/main.rs
356
src/main.rs
|
|
@ -5,6 +5,7 @@ extern crate serde_json;
|
|||
use serde::{Deserialize, Serialize};
|
||||
|
||||
use clap::{Args, Parser, Subcommand};
|
||||
use std::cell::Cell;
|
||||
use std::env;
|
||||
use std::io::Cursor;
|
||||
use std::io::{Read, Write};
|
||||
|
|
@ -78,6 +79,42 @@ struct Index {
|
|||
index: usize,
|
||||
}
|
||||
|
||||
struct SwaySome {
|
||||
socket: Cell<Option<UnixStream>>,
|
||||
outputs: Vec<Output>,
|
||||
// current_output: Output,
|
||||
workspaces: Vec<Workspace>,
|
||||
// current_workspace: Workspace,
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize, Debug)]
|
||||
struct Output {
|
||||
name: String,
|
||||
#[serde(default)]
|
||||
focused: bool,
|
||||
#[serde(default)]
|
||||
active: bool,
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize, Debug)]
|
||||
struct Workspace {
|
||||
num: usize,
|
||||
output: String,
|
||||
visible: bool,
|
||||
}
|
||||
|
||||
impl SwaySome {
|
||||
fn new() -> SwaySome {
|
||||
let mut swaysome = SwaySome {
|
||||
socket: Cell::new(Some(SwaySome::get_stream())),
|
||||
outputs: vec![],
|
||||
workspaces: vec![],
|
||||
};
|
||||
swaysome.outputs = swaysome.get_outputs();
|
||||
swaysome.workspaces = swaysome.get_workspaces();
|
||||
swaysome
|
||||
}
|
||||
|
||||
fn get_stream() -> UnixStream {
|
||||
for socket_var in ["SWAYSOCK", "I3SOCK"] {
|
||||
let socket_path = match env::var(socket_var) {
|
||||
|
|
@ -110,7 +147,7 @@ fn get_stream() -> UnixStream {
|
|||
panic!("couldn't find any i3/sway socket")
|
||||
}
|
||||
|
||||
fn send_msg(mut stream: &UnixStream, msg_type: u32, payload: &str) {
|
||||
fn send_msg(&self, msg_type: u32, payload: &str) {
|
||||
let payload_length = payload.len() as u32;
|
||||
|
||||
let mut msg_prefix: [u8; 6 * mem::size_of::<u8>() + 2 * mem::size_of::<u32>()] =
|
||||
|
|
@ -129,20 +166,30 @@ fn send_msg(mut stream: &UnixStream, msg_type: u32, payload: &str) {
|
|||
let mut msg: Vec<u8> = msg_prefix[..].to_vec();
|
||||
msg.extend(payload.as_bytes());
|
||||
|
||||
if stream.write_all(&msg[..]).is_err() {
|
||||
let mut socket = self
|
||||
.socket
|
||||
.take()
|
||||
.expect("Unexisting socket, there probably is a logic error");
|
||||
|
||||
if socket.write_all(&msg[..]).is_err() {
|
||||
panic!("couldn't send message");
|
||||
}
|
||||
self.socket.set(Some(socket));
|
||||
}
|
||||
|
||||
fn send_command(stream: &UnixStream, command: &str) {
|
||||
fn send_command(&self, command: &str) {
|
||||
eprint!("Sending command: '{}' - ", &command);
|
||||
send_msg(stream, RUN_COMMAND, command);
|
||||
check_success(stream);
|
||||
self.send_msg(RUN_COMMAND, command);
|
||||
self.check_success();
|
||||
}
|
||||
|
||||
fn read_msg(mut stream: &UnixStream) -> Result<String, &str> {
|
||||
fn read_msg(&self) -> Result<String, &str> {
|
||||
let mut response_header: [u8; 14] = *b"uninitialized.";
|
||||
stream.read_exact(&mut response_header).unwrap();
|
||||
let mut socket = self
|
||||
.socket
|
||||
.take()
|
||||
.expect("Unexisting socket, there probably is a logic error");
|
||||
socket.read_exact(&mut response_header).unwrap();
|
||||
|
||||
if &response_header[0..6] == b"i3-ipc" {
|
||||
let mut v = Cursor::new(vec![
|
||||
|
|
@ -154,20 +201,22 @@ fn read_msg(mut stream: &UnixStream) -> Result<String, &str> {
|
|||
let payload_length = v.read_u32::<LittleEndian>().unwrap();
|
||||
|
||||
let mut payload = vec![0; payload_length as usize];
|
||||
stream.read_exact(&mut payload[..]).unwrap();
|
||||
socket.read_exact(&mut payload[..]).unwrap();
|
||||
let payload_str = String::from_utf8(payload).unwrap();
|
||||
self.socket.set(Some(socket));
|
||||
Ok(payload_str)
|
||||
} else {
|
||||
eprint!("Not an i3-icp packet, emptying the buffer: ");
|
||||
let mut v = vec![];
|
||||
stream.read_to_end(&mut v).unwrap();
|
||||
socket.read_to_end(&mut v).unwrap();
|
||||
eprintln!("{:?}", v);
|
||||
self.socket.set(Some(socket));
|
||||
Err("Unable to read i3-ipc packet")
|
||||
}
|
||||
}
|
||||
|
||||
fn check_success(stream: &UnixStream) {
|
||||
match read_msg(stream) {
|
||||
fn check_success(&self) {
|
||||
match self.read_msg() {
|
||||
Ok(msg) => {
|
||||
let r: Vec<serde_json::Value> = serde_json::from_str(&msg).unwrap();
|
||||
match r[0]["success"] {
|
||||
|
|
@ -179,18 +228,9 @@ fn check_success(stream: &UnixStream) {
|
|||
};
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize, Debug)]
|
||||
struct Output {
|
||||
name: String,
|
||||
#[serde(default)]
|
||||
focused: bool,
|
||||
#[serde(default)]
|
||||
active: bool,
|
||||
}
|
||||
|
||||
fn get_outputs(stream: &UnixStream) -> Vec<Output> {
|
||||
send_msg(stream, GET_OUTPUTS, "");
|
||||
let o = match read_msg(stream) {
|
||||
fn get_outputs(&self) -> Vec<Output> {
|
||||
self.send_msg(GET_OUTPUTS, "");
|
||||
let o = match self.read_msg() {
|
||||
Ok(msg) => msg,
|
||||
Err(_) => panic!("Unable to get outputs"),
|
||||
};
|
||||
|
|
@ -203,16 +243,9 @@ fn get_outputs(stream: &UnixStream) -> Vec<Output> {
|
|||
outputs
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize, Debug)]
|
||||
struct Workspace {
|
||||
num: usize,
|
||||
output: String,
|
||||
visible: bool,
|
||||
}
|
||||
|
||||
fn get_workspaces(stream: &UnixStream) -> Vec<Workspace> {
|
||||
send_msg(stream, GET_WORKSPACES, "");
|
||||
let ws = match read_msg(stream) {
|
||||
fn get_workspaces(&self) -> Vec<Workspace> {
|
||||
self.send_msg(GET_WORKSPACES, "");
|
||||
let ws = match self.read_msg() {
|
||||
Ok(msg) => msg,
|
||||
Err(_) => panic!("Unable to get current workspace"),
|
||||
};
|
||||
|
|
@ -221,19 +254,15 @@ fn get_workspaces(stream: &UnixStream) -> Vec<Workspace> {
|
|||
workspaces
|
||||
}
|
||||
|
||||
fn get_current_output_index(stream: &UnixStream) -> usize {
|
||||
let outputs = get_outputs(stream);
|
||||
|
||||
match outputs.iter().position(|x| x.focused) {
|
||||
fn get_current_output_index(&self) -> usize {
|
||||
match self.outputs.iter().position(|x| x.focused) {
|
||||
Some(i) => i,
|
||||
None => panic!("WTF! No focused output???"),
|
||||
}
|
||||
}
|
||||
|
||||
fn get_current_output_name(stream: &UnixStream) -> String {
|
||||
let outputs = get_outputs(stream);
|
||||
|
||||
let focused_output_index = match outputs.iter().find(|x| x.focused) {
|
||||
fn get_current_output_name(&self) -> String {
|
||||
let focused_output_index = match self.outputs.iter().find(|x| x.focused) {
|
||||
Some(i) => i.name.as_str(),
|
||||
None => panic!("WTF! No focused output???"),
|
||||
};
|
||||
|
|
@ -241,57 +270,62 @@ 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)
|
||||
fn get_current_workspace_index(&self) -> usize {
|
||||
self.workspaces
|
||||
.iter()
|
||||
.find(|w| {
|
||||
w.visible
|
||||
&& self
|
||||
.outputs
|
||||
.iter()
|
||||
.find(|o| o.name == w.output)
|
||||
.unwrap()
|
||||
.focused
|
||||
})
|
||||
.unwrap()
|
||||
.num
|
||||
}
|
||||
|
||||
fn move_container_to_workspace(stream: &UnixStream, workspace_index: usize) {
|
||||
fn move_container_to_workspace(&self, workspace_index: usize) {
|
||||
if workspace_index < MAX_GROUP_WS {
|
||||
move_container_to_workspace_relative(stream, workspace_index);
|
||||
self.move_container_to_workspace_relative(workspace_index);
|
||||
} else {
|
||||
move_container_to_workspace_absolute(stream, workspace_index);
|
||||
self.move_container_to_workspace_absolute(workspace_index);
|
||||
}
|
||||
}
|
||||
|
||||
fn move_container_to_workspace_group(stream: &UnixStream, target_group: usize) {
|
||||
let current_workspace_index = get_current_workspace(stream).num;
|
||||
fn move_container_to_workspace_group(&self, target_group: usize) {
|
||||
let current_workspace_index = self.get_current_workspace_index();
|
||||
let current_workspace_index_relative = (current_workspace_index % MAX_GROUP_WS) as usize;
|
||||
move_container_to_workspace_absolute(
|
||||
stream,
|
||||
self.move_container_to_workspace_absolute(
|
||||
current_workspace_index_relative + target_group * MAX_GROUP_WS,
|
||||
);
|
||||
}
|
||||
|
||||
fn move_container_to_workspace_absolute(stream: &UnixStream, workspace_index: usize) {
|
||||
fn move_container_to_workspace_absolute(&self, workspace_index: usize) {
|
||||
let group_index = (workspace_index / MAX_GROUP_WS) as usize;
|
||||
let outputs = get_outputs(stream);
|
||||
let workspaces = get_workspaces(stream);
|
||||
let full_ws_name = format!(
|
||||
"{}",
|
||||
group_index * MAX_GROUP_WS + workspace_index % MAX_GROUP_WS
|
||||
);
|
||||
|
||||
// If the workspace already exists
|
||||
match workspaces.iter().find(|w| w.num == workspace_index) {
|
||||
match self.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(&full_ws_name);
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
}
|
||||
None => {
|
||||
let target_group = workspace_index / MAX_GROUP_WS;
|
||||
let target_screen_index = match workspaces
|
||||
let target_screen_index = match self
|
||||
.workspaces
|
||||
.iter()
|
||||
.find(|w| w.num / MAX_GROUP_WS == target_group)
|
||||
{
|
||||
// If other workspaces on the same group exists
|
||||
Some(other_workspace) => Some(
|
||||
outputs
|
||||
self.outputs
|
||||
.iter()
|
||||
.enumerate()
|
||||
.find(|i| i.1.name == other_workspace.output)
|
||||
|
|
@ -300,7 +334,7 @@ fn move_container_to_workspace_absolute(stream: &UnixStream, workspace_index: us
|
|||
),
|
||||
None => {
|
||||
// Or if the targeted output is currently connected
|
||||
if group_index < outputs.len() {
|
||||
if group_index < self.outputs.len() {
|
||||
Some(group_index)
|
||||
} else {
|
||||
None
|
||||
|
|
@ -310,97 +344,96 @@ fn move_container_to_workspace_absolute(stream: &UnixStream, workspace_index: us
|
|||
|
||||
match target_screen_index {
|
||||
Some(target_screen_index) => {
|
||||
let target_output = &outputs[target_screen_index];
|
||||
let target_output = &self.outputs[target_screen_index];
|
||||
|
||||
let current_output_name = get_current_output_name(stream);
|
||||
let current_output_name = self.get_current_output_name();
|
||||
|
||||
if target_output.name == current_output_name {
|
||||
let mut focus_cmd: String = "move container to workspace ".to_string();
|
||||
focus_cmd.push_str(&full_ws_name);
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
} else {
|
||||
// If we have to send it to another screen
|
||||
let mut focus_cmd: String = "focus output ".to_string();
|
||||
focus_cmd.push_str(&target_output.name);
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
|
||||
let focused_workspace_index = get_current_workspace(stream).num;
|
||||
let focused_workspace_index = self.get_current_workspace_index();
|
||||
|
||||
let mut focus_cmd: String = "workspace ".to_string();
|
||||
focus_cmd.push_str(&full_ws_name);
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
|
||||
let mut focus_cmd: String = "focus output ".to_string();
|
||||
focus_cmd.push_str(¤t_output_name);
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
|
||||
let mut focus_cmd: String = "move container to workspace ".to_string();
|
||||
focus_cmd.push_str(&full_ws_name);
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
|
||||
let mut focus_cmd: String = "focus output ".to_string();
|
||||
focus_cmd.push_str(&target_output.name);
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
|
||||
let mut focus_cmd: String = "workspace ".to_string();
|
||||
focus_cmd.push_str(&focused_workspace_index.to_string());
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
|
||||
let mut focus_cmd: String = "focus output ".to_string();
|
||||
focus_cmd.push_str(¤t_output_name);
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&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(&full_ws_name);
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
}
|
||||
};
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn move_container_to_workspace_relative(stream: &UnixStream, workspace_index: usize) {
|
||||
let current_workspace_index: usize = get_current_workspace(stream).num;
|
||||
fn move_container_to_workspace_relative(&self, workspace_index: usize) {
|
||||
let current_workspace_index: usize = self.get_current_workspace_index();
|
||||
let focused_output_index = current_workspace_index / MAX_GROUP_WS;
|
||||
|
||||
let mut cmd: String = "move container to workspace number ".to_string();
|
||||
let full_ws_name = format!("{}", focused_output_index * MAX_GROUP_WS + workspace_index);
|
||||
cmd.push_str(&full_ws_name);
|
||||
send_command(stream, &cmd);
|
||||
self.send_command(&cmd);
|
||||
}
|
||||
|
||||
fn focus_to_workspace(stream: &UnixStream, workspace_index: usize) {
|
||||
fn focus_to_workspace(&self, workspace_index: usize) {
|
||||
if workspace_index < MAX_GROUP_WS {
|
||||
focus_to_workspace_relative(stream, workspace_index);
|
||||
self.focus_to_workspace_relative(workspace_index);
|
||||
} else {
|
||||
focus_to_workspace_absolute(stream, workspace_index);
|
||||
self.focus_to_workspace_absolute(workspace_index);
|
||||
}
|
||||
}
|
||||
|
||||
fn focus_to_workspace_absolute(stream: &UnixStream, workspace_index: usize) {
|
||||
fn focus_to_workspace_absolute(&self, workspace_index: usize) {
|
||||
let output_index = (workspace_index / MAX_GROUP_WS) 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) {
|
||||
match self.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);
|
||||
self.send_command(&focus_cmd);
|
||||
}
|
||||
None => {
|
||||
let target_group = workspace_index / MAX_GROUP_WS;
|
||||
let target_screen_index = match workspaces
|
||||
let target_screen_index = match self
|
||||
.workspaces
|
||||
.iter()
|
||||
.find(|w| w.num / MAX_GROUP_WS == target_group)
|
||||
{
|
||||
// If other workspaces on the same group exists
|
||||
Some(other_workspace) => Some(
|
||||
outputs
|
||||
self.outputs
|
||||
.iter()
|
||||
.enumerate()
|
||||
.find(|i| i.1.name == other_workspace.output)
|
||||
|
|
@ -409,7 +442,7 @@ fn focus_to_workspace_absolute(stream: &UnixStream, workspace_index: usize) {
|
|||
),
|
||||
None => {
|
||||
// Or if the targeted output is currently connected
|
||||
if output_index < outputs.len() {
|
||||
if output_index < self.outputs.len() {
|
||||
Some(output_index)
|
||||
} else {
|
||||
None
|
||||
|
|
@ -420,37 +453,34 @@ fn focus_to_workspace_absolute(stream: &UnixStream, workspace_index: usize) {
|
|||
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 target_output = &self.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);
|
||||
self.send_command(&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);
|
||||
self.send_command(&focus_cmd);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn focus_to_workspace_relative(stream: &UnixStream, workspace_index: usize) {
|
||||
let current_workspace_index: usize = get_current_workspace(stream).num;
|
||||
fn focus_to_workspace_relative(&self, workspace_index: usize) {
|
||||
let current_workspace_index: usize = self.get_current_workspace_index();
|
||||
let focused_output_index = current_workspace_index / MAX_GROUP_WS;
|
||||
|
||||
let mut cmd: String = "workspace number ".to_string();
|
||||
let full_ws_name = format!("{}", focused_output_index * MAX_GROUP_WS + workspace_index);
|
||||
cmd.push_str(&full_ws_name);
|
||||
send_command(stream, &cmd);
|
||||
self.send_command(&cmd);
|
||||
}
|
||||
|
||||
fn focus_to_group(stream: &UnixStream, group_index: usize) {
|
||||
let outputs = get_outputs(stream);
|
||||
let workspaces = get_workspaces(stream);
|
||||
|
||||
let current_workspace_index: usize = get_current_workspace(stream).num;
|
||||
fn focus_to_group(&self, group_index: usize) {
|
||||
let current_workspace_index: usize = self.get_current_workspace_index();
|
||||
let target_workspace_relative_index = current_workspace_index % MAX_GROUP_WS;
|
||||
|
||||
let target_workspace_index = group_index * MAX_GROUP_WS + target_workspace_relative_index;
|
||||
|
|
@ -460,20 +490,25 @@ fn focus_to_group(stream: &UnixStream, group_index: usize) {
|
|||
);
|
||||
|
||||
// If the workspace already exists
|
||||
match workspaces.iter().find(|w| w.num == target_workspace_index) {
|
||||
match self
|
||||
.workspaces
|
||||
.iter()
|
||||
.find(|w| w.num == target_workspace_index)
|
||||
{
|
||||
Some(_) => {
|
||||
let mut focus_cmd: String = "workspace number ".to_string();
|
||||
focus_cmd.push_str(&full_ws_name);
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
}
|
||||
None => {
|
||||
let target_screen_index = match workspaces
|
||||
let target_screen_index = match self
|
||||
.workspaces
|
||||
.iter()
|
||||
.find(|w| w.num / MAX_GROUP_WS == group_index)
|
||||
{
|
||||
// If other workspaces on the same group exists
|
||||
Some(other_workspace) => Some(
|
||||
outputs
|
||||
self.outputs
|
||||
.iter()
|
||||
.enumerate()
|
||||
.find(|i| i.1.name == other_workspace.output)
|
||||
|
|
@ -483,7 +518,7 @@ fn focus_to_group(stream: &UnixStream, group_index: usize) {
|
|||
),
|
||||
None => {
|
||||
// Or if the targeted output is currently connected
|
||||
if group_index > 0 && group_index <= outputs.len() {
|
||||
if group_index > 0 && group_index <= self.outputs.len() {
|
||||
Some(group_index)
|
||||
} else {
|
||||
None
|
||||
|
|
@ -494,60 +529,58 @@ fn focus_to_group(stream: &UnixStream, group_index: usize) {
|
|||
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 target_output = &self.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);
|
||||
self.send_command(&focus_cmd);
|
||||
}
|
||||
None => {}
|
||||
};
|
||||
// Then we focus the workspace
|
||||
let mut focus_cmd: String = "workspace number ".to_string();
|
||||
focus_cmd.push_str(&target_workspace_index.to_string());
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn focus_all_outputs_to_workspace(stream: &UnixStream, workspace_index: usize) {
|
||||
let current_output = get_current_output_name(stream);
|
||||
fn focus_all_outputs_to_workspace(&self, workspace_index: usize) {
|
||||
let current_output = self.get_current_output_name();
|
||||
|
||||
// Iterate on all outputs to focus on the given workspace
|
||||
let outputs = get_outputs(stream);
|
||||
for output in outputs.iter() {
|
||||
for output in self.outputs.iter() {
|
||||
let mut cmd: String = "focus output ".to_string();
|
||||
cmd.push_str(output.name.as_str());
|
||||
send_command(stream, &cmd);
|
||||
self.send_command(&cmd);
|
||||
|
||||
focus_to_workspace(stream, workspace_index);
|
||||
self.focus_to_workspace(workspace_index);
|
||||
}
|
||||
|
||||
// Get back to currently focused output
|
||||
let mut cmd: String = "focus output ".to_string();
|
||||
cmd.push_str(¤t_output);
|
||||
send_command(stream, &cmd);
|
||||
self.send_command(&cmd);
|
||||
}
|
||||
|
||||
fn move_container_to_next_output(stream: &UnixStream) {
|
||||
move_container_to_next_or_prev_output(stream, false);
|
||||
fn move_container_to_next_output(&self) {
|
||||
self.move_container_to_next_or_prev_output(false);
|
||||
}
|
||||
|
||||
fn move_container_to_prev_output(stream: &UnixStream) {
|
||||
move_container_to_next_or_prev_output(stream, true);
|
||||
fn move_container_to_prev_output(&self) {
|
||||
self.move_container_to_next_or_prev_output(true);
|
||||
}
|
||||
|
||||
fn move_container_to_next_or_prev_output(stream: &UnixStream, go_to_prev: bool) {
|
||||
let outputs = get_outputs(stream);
|
||||
let focused_output_index = get_current_output_index(stream);
|
||||
fn move_container_to_next_or_prev_output(&self, go_to_prev: bool) {
|
||||
let focused_output_index = self.get_current_output_index();
|
||||
|
||||
let target_output = if go_to_prev {
|
||||
&outputs[(focused_output_index + outputs.len() - 1) % outputs.len()]
|
||||
&self.outputs[(focused_output_index + self.outputs.len() - 1) % self.outputs.len()]
|
||||
} else {
|
||||
&outputs[(focused_output_index + 1) % outputs.len()]
|
||||
&self.outputs[(focused_output_index + 1) % self.outputs.len()]
|
||||
};
|
||||
|
||||
let workspaces = get_workspaces(stream);
|
||||
let workspaces = self.get_workspaces();
|
||||
let target_workspace = workspaces
|
||||
.iter()
|
||||
.find(|x| x.output == target_output.name && x.visible)
|
||||
|
|
@ -561,109 +594,106 @@ fn move_container_to_next_or_prev_output(stream: &UnixStream, go_to_prev: bool)
|
|||
// Move container to target workspace
|
||||
let mut cmd: String = "move container to workspace number ".to_string();
|
||||
cmd.push_str(&full_ws_name);
|
||||
send_command(stream, &cmd);
|
||||
self.send_command(&cmd);
|
||||
|
||||
// Focus that workspace to follow the container
|
||||
let mut cmd: String = "workspace number ".to_string();
|
||||
cmd.push_str(&target_workspace.num.to_string());
|
||||
send_command(stream, &cmd);
|
||||
cmd.push_str(&full_ws_name);
|
||||
self.send_command(&cmd);
|
||||
}
|
||||
|
||||
fn focus_container_to_next_group(stream: &UnixStream) {
|
||||
focus_container_to_next_or_prev_group(stream, false);
|
||||
fn focus_container_to_next_group(&self) {
|
||||
self.focus_container_to_next_or_prev_group(false);
|
||||
}
|
||||
|
||||
fn focus_container_to_prev_group(stream: &UnixStream) {
|
||||
focus_container_to_next_or_prev_group(stream, true);
|
||||
fn focus_container_to_prev_group(&self) {
|
||||
self.focus_container_to_next_or_prev_group(true);
|
||||
}
|
||||
|
||||
fn focus_container_to_next_or_prev_group(stream: &UnixStream, go_to_prev: bool) {
|
||||
let current_workspace_index: usize = get_current_workspace(stream).num;
|
||||
fn focus_container_to_next_or_prev_group(&self, go_to_prev: bool) {
|
||||
let current_workspace_index: usize = self.get_current_workspace_index();
|
||||
let focused_group_index = current_workspace_index / MAX_GROUP_WS;
|
||||
|
||||
if go_to_prev {
|
||||
focus_to_group(stream, focused_group_index - 1);
|
||||
self.focus_to_group(focused_group_index - 1);
|
||||
} else {
|
||||
focus_to_group(stream, focused_group_index + 1);
|
||||
self.focus_to_group(focused_group_index + 1);
|
||||
};
|
||||
}
|
||||
|
||||
fn init_workspaces(stream: &UnixStream, workspace_index: usize) {
|
||||
let outputs = get_outputs(stream);
|
||||
|
||||
fn init_workspaces(&self, workspace_index: usize) {
|
||||
let cmd_prefix: String = "focus output ".to_string();
|
||||
for output in outputs.iter().rev() {
|
||||
for output in self.outputs.iter().rev() {
|
||||
let mut cmd = cmd_prefix.clone();
|
||||
cmd.push_str(output.name.as_str());
|
||||
send_command(stream, &cmd);
|
||||
self.send_command(&cmd);
|
||||
|
||||
let mut cmd: String = "workspace number ".to_string();
|
||||
let full_ws_name = format!(
|
||||
"{}",
|
||||
(get_current_output_index(stream) + 1) * MAX_GROUP_WS + workspace_index
|
||||
(self.get_current_output_index() + 1) * MAX_GROUP_WS + workspace_index
|
||||
);
|
||||
cmd.push_str(&full_ws_name);
|
||||
send_command(stream, &cmd);
|
||||
self.send_command(&cmd);
|
||||
}
|
||||
}
|
||||
|
||||
fn rearrange_workspaces(stream: &UnixStream) {
|
||||
let outputs = get_outputs(stream);
|
||||
let workspaces = get_workspaces(stream);
|
||||
|
||||
fn rearrange_workspaces(&self) {
|
||||
let focus_cmd_prefix: String = "workspace number ".to_string();
|
||||
let move_cmd_prefix: String = "move workspace to ".to_string();
|
||||
for workspace in workspaces.iter() {
|
||||
for workspace in self.workspaces.iter() {
|
||||
let mut focus_cmd = focus_cmd_prefix.clone();
|
||||
focus_cmd.push_str(&workspace.num.to_string());
|
||||
send_command(stream, &focus_cmd);
|
||||
self.send_command(&focus_cmd);
|
||||
|
||||
let group_index = workspace.num / MAX_GROUP_WS;
|
||||
if group_index <= outputs.len() - 1 {
|
||||
if group_index <= self.outputs.len() - 1 {
|
||||
let mut move_cmd = move_cmd_prefix.clone();
|
||||
move_cmd.push_str(&outputs[group_index.max(1) - 1].name);
|
||||
send_command(stream, &move_cmd);
|
||||
move_cmd.push_str(&self.outputs[group_index.max(1) - 1].name);
|
||||
self.send_command(&move_cmd);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn main() {
|
||||
let cli = Cli::parse();
|
||||
let stream = get_stream();
|
||||
|
||||
let swaysome = SwaySome::new();
|
||||
|
||||
match &cli.command {
|
||||
Command::Init(action) => {
|
||||
init_workspaces(&stream, action.index);
|
||||
swaysome.init_workspaces(action.index);
|
||||
}
|
||||
Command::Move(action) => {
|
||||
move_container_to_workspace(&stream, action.index);
|
||||
swaysome.move_container_to_workspace(action.index);
|
||||
}
|
||||
Command::MoveToGroup(action) => {
|
||||
move_container_to_workspace_group(&stream, action.index);
|
||||
swaysome.move_container_to_workspace_group(action.index);
|
||||
}
|
||||
Command::Focus(action) => {
|
||||
focus_to_workspace(&stream, action.index);
|
||||
swaysome.focus_to_workspace(action.index);
|
||||
}
|
||||
Command::FocusGroup(action) => {
|
||||
focus_to_group(&stream, action.index);
|
||||
swaysome.focus_to_group(action.index);
|
||||
}
|
||||
Command::FocusAllOutputs(action) => {
|
||||
focus_all_outputs_to_workspace(&stream, action.index);
|
||||
swaysome.focus_all_outputs_to_workspace(action.index);
|
||||
}
|
||||
Command::NextOutput => {
|
||||
move_container_to_next_output(&stream);
|
||||
swaysome.move_container_to_next_output();
|
||||
}
|
||||
Command::PrevOutput => {
|
||||
move_container_to_prev_output(&stream);
|
||||
swaysome.move_container_to_prev_output();
|
||||
}
|
||||
Command::NextGroup => {
|
||||
focus_container_to_next_group(&stream);
|
||||
swaysome.focus_container_to_next_group();
|
||||
}
|
||||
Command::PrevGroup => {
|
||||
focus_container_to_prev_group(&stream);
|
||||
swaysome.focus_container_to_prev_group();
|
||||
}
|
||||
Command::RearrangeWorkspaces => {
|
||||
rearrange_workspaces(&stream);
|
||||
swaysome.rearrange_workspaces();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue