Skip to content

Commit

Permalink
feat: new focused window module
Browse files Browse the repository at this point in the history
  • Loading branch information
JakeStanger committed Aug 14, 2022
1 parent e416e03 commit dc14cb0
Show file tree
Hide file tree
Showing 13 changed files with 222 additions and 88 deletions.
25 changes: 19 additions & 6 deletions src/bar.rs
Original file line number Diff line number Diff line change
Expand Up @@ -48,14 +48,14 @@ fn load_modules(
right: &gtk::Box,
app: &Application,
config: Config,
output_name: &str
output_name: &str,
) {
if let Some(modules) = config.left {
let info = ModuleInfo {
app,
location: ModuleLocation::Left,
bar_position: &config.position,
output_name
output_name,
};

add_modules(left, modules, info);
Expand All @@ -66,7 +66,7 @@ fn load_modules(
app,
location: ModuleLocation::Center,
bar_position: &config.position,
output_name
output_name,
};

add_modules(center, modules, info);
Expand All @@ -77,7 +77,7 @@ fn load_modules(
app,
location: ModuleLocation::Right,
bar_position: &config.position,
output_name
output_name,
};

add_modules(right, modules, info);
Expand Down Expand Up @@ -122,6 +122,11 @@ fn add_modules(content: &gtk::Box, modules: Vec<ModuleConfig>, info: ModuleInfo)
widget.set_widget_name("script");
content.add(&widget);
}
ModuleConfig::Focused(module) => {
let widget = module.into_widget(&info);
widget.set_widget_name("focused");
content.add(&widget);
}
}
}
}
Expand All @@ -137,8 +142,16 @@ fn setup_layer_shell(win: &ApplicationWindow, monitor: &Monitor, position: &BarP
gtk_layer_shell::set_margin(win, gtk_layer_shell::Edge::Left, 0);
gtk_layer_shell::set_margin(win, gtk_layer_shell::Edge::Right, 0);

gtk_layer_shell::set_anchor(win, gtk_layer_shell::Edge::Top, position == &BarPosition::Top);
gtk_layer_shell::set_anchor(win, gtk_layer_shell::Edge::Bottom, position == &BarPosition::Bottom);
gtk_layer_shell::set_anchor(
win,
gtk_layer_shell::Edge::Top,
position == &BarPosition::Top,
);
gtk_layer_shell::set_anchor(
win,
gtk_layer_shell::Edge::Bottom,
position == &BarPosition::Bottom,
);
gtk_layer_shell::set_anchor(win, gtk_layer_shell::Edge::Left, true);
gtk_layer_shell::set_anchor(win, gtk_layer_shell::Edge::Right, true);
}
13 changes: 11 additions & 2 deletions src/config.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
use crate::modules::clock::ClockModule;
use crate::modules::focused::FocusedModule;
use crate::modules::launcher::LauncherModule;
use crate::modules::mpd::MpdModule;
use crate::modules::script::ScriptModule;
Expand All @@ -19,13 +20,14 @@ pub enum ModuleConfig {
SysInfo(SysInfoModule),
Launcher(LauncherModule),
Script(ScriptModule),
Focused(FocusedModule),
}

#[derive(Debug, Deserialize, Clone, PartialEq)]
#[derive(Debug, Deserialize, Clone, PartialEq, Eq)]
#[serde(rename_all = "kebab-case")]
pub enum BarPosition {
Top,
Bottom
Bottom,
}

impl Default for BarPosition {
Expand Down Expand Up @@ -83,3 +85,10 @@ impl Config {
})
}
}

pub const fn default_false() -> bool {
false
}
pub const fn default_true() -> bool {
true
}
File renamed without changes.
21 changes: 12 additions & 9 deletions src/main.rs
Original file line number Diff line number Diff line change
@@ -1,24 +1,21 @@
mod bar;
mod collection;
mod config;
mod icon;
mod modules;
mod popup;
mod style;
mod sway;

use crate::bar::create_bar;
use crate::config::Config;
use crate::style::load_css;
use crate::sway::SwayOutput;
use dirs::config_dir;
use gtk::prelude::*;
use gtk::{gdk, Application};
use ksway::client::Client;
use ksway::IpcCommand;
use serde::Deserialize;

#[derive(Deserialize)]
struct SwayOutput {
name: String,
}

#[tokio::main]
async fn main() {
Expand All @@ -27,8 +24,11 @@ async fn main() {
.build();

let mut sway_client = Client::connect().expect("Failed to connect to Sway IPC");
let outputs = sway_client.ipc(IpcCommand::GetOutputs).expect("Failed to get Sway outputs");
let outputs = serde_json::from_slice::<Vec<SwayOutput>>(&outputs).expect("Failed to deserialize outputs message from Sway IPC");
let outputs = sway_client
.ipc(IpcCommand::GetOutputs)
.expect("Failed to get Sway outputs");
let outputs = serde_json::from_slice::<Vec<SwayOutput>>(&outputs)
.expect("Failed to deserialize outputs message from Sway IPC");

app.connect_activate(move |app| {
let config = Config::load().unwrap_or_default();
Expand All @@ -42,7 +42,10 @@ async fn main() {
let num_monitors = display.n_monitors();
for i in 0..num_monitors {
let monitor = display.monitor(i).unwrap();
let monitor_name = &outputs.get(i as usize).expect("GTK monitor output differs from Sway's").name;
let monitor_name = &outputs
.get(i as usize)
.expect("GTK monitor output differs from Sway's")
.name;

let config = config.monitors.as_ref().map_or(&config, |monitor_config| {
monitor_config.get(i as usize).unwrap_or(&config)
Expand Down
7 changes: 6 additions & 1 deletion src/modules/clock/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,12 @@ impl Module<Button> for ClockModule {
fn into_widget(self, info: &ModuleInfo) -> Button {
let button = Button::new();

let popup = Popup::new("popup-clock", info.app, Orientation::Vertical, info.bar_position);
let popup = Popup::new(
"popup-clock",
info.app,
Orientation::Vertical,
info.bar_position,
);
popup.add_clock_widgets();

button.show_all();
Expand Down
97 changes: 97 additions & 0 deletions src/modules/focused.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,97 @@
use crate::icon;
use crate::modules::{Module, ModuleInfo};
use crate::sway::node::get_open_windows;
use crate::sway::WindowEvent;
use glib::Continue;
use gtk::prelude::*;
use gtk::{IconTheme, Image, Label, Orientation};
use ksway::{Client, IpcEvent};
use serde::Deserialize;
use tokio::task::spawn_blocking;

#[derive(Debug, Deserialize, Clone)]
pub struct FocusedModule {
#[serde(default = "crate::config::default_true")]
show_icon: bool,
#[serde(default = "crate::config::default_true")]
show_title: bool,

#[serde(default = "default_icon_size")]
icon_size: i32,
icon_theme: Option<String>,
}

const fn default_icon_size() -> i32 {
32
}

impl Module<gtk::Box> for FocusedModule {
fn into_widget(self, _info: &ModuleInfo) -> gtk::Box {
let icon_theme = IconTheme::new();

if let Some(theme) = self.icon_theme {
icon_theme.set_custom_theme(Some(&theme));
}

let container = gtk::Box::new(Orientation::Horizontal, 5);

let icon = Image::builder().name("icon").build();
let label = Label::builder().name("label").build();

container.add(&icon);
container.add(&label);

let mut sway = Client::connect().unwrap();

let srx = sway.subscribe(vec![IpcEvent::Window]).unwrap();
let (tx, rx) = glib::MainContext::channel(glib::PRIORITY_DEFAULT);

let focused = get_open_windows(&mut sway)
.into_iter()
.find(|node| node.focused);

if let Some(focused) = focused {
tx.send(focused).unwrap();
}

spawn_blocking(move || loop {
while let Ok((_, payload)) = srx.try_recv() {
let payload: WindowEvent = serde_json::from_slice(&payload).unwrap();

let update = match payload.change.as_str() {
"focus" => true,
"title" => payload.container.focused,
_ => false,
};

if update {
tx.send(payload.container).unwrap();
}
}
sway.poll().unwrap();
});

{
rx.attach(None, move |node| {
let value = node
.name
.as_deref()
.unwrap_or_else(|| node.get_id());

let pixbuf = icon::get_icon(&icon_theme, node.get_id(), self.icon_size);

if self.show_icon {
icon.set_pixbuf(pixbuf.as_ref());
}

if self.show_title {
label.set_label(value);
}

Continue(true)
});
}

container
}
}
4 changes: 2 additions & 2 deletions src/modules/launcher/item.rs
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
use crate::collection::Collection;
use crate::modules::launcher::icon::{find_desktop_file, get_icon};
use crate::modules::launcher::node::SwayNode;
use crate::icon::{find_desktop_file, get_icon};
use crate::modules::launcher::popup::Popup;
use crate::modules::launcher::FocusEvent;
use crate::popup::PopupAlignment;
use crate::sway::SwayNode;
use gtk::prelude::*;
use gtk::{Button, IconTheme, Image};
use std::process::{Command, Stdio};
Expand Down
31 changes: 10 additions & 21 deletions src/modules/launcher/mod.rs
Original file line number Diff line number Diff line change
@@ -1,13 +1,12 @@
mod icon;
mod item;
mod node;
mod popup;

use crate::collection::Collection;
use crate::modules::launcher::item::{ButtonConfig, LauncherItem, LauncherWindow};
use crate::modules::launcher::node::{get_open_windows, SwayNode};
use crate::modules::launcher::popup::Popup;
use crate::modules::{Module, ModuleInfo};
use crate::sway::node::get_open_windows;
use crate::sway::{SwayNode, WindowEvent};
use gtk::prelude::*;
use gtk::{IconTheme, Orientation};
use ksway::{Client, IpcEvent};
Expand All @@ -20,28 +19,14 @@ use tokio::task::spawn_blocking;
#[derive(Debug, Deserialize, Clone)]
pub struct LauncherModule {
favorites: Option<Vec<String>>,
#[serde(default = "default_false")]
#[serde(default = "crate::config::default_false")]
show_names: bool,
#[serde(default = "default_true")]
#[serde(default = "crate::config::default_true")]
show_icons: bool,

icon_theme: Option<String>,
}

const fn default_false() -> bool {
false
}

const fn default_true() -> bool {
true
}

#[derive(Debug, Deserialize)]
struct WindowEvent {
change: String,
container: SwayNode,
}

#[derive(Debug)]
pub enum FocusEvent {
AppId(String),
Expand Down Expand Up @@ -181,7 +166,6 @@ impl Launcher {
} else {
windows.get_mut(&window.id).unwrap().name = Some(name);
}

}
}

Expand All @@ -207,7 +191,12 @@ impl Module<gtk::Box> for LauncherModule {

let mut sway = Client::connect().unwrap();

let popup = Popup::new("popup-launcher", info.app, Orientation::Vertical, info.bar_position);
let popup = Popup::new(
"popup-launcher",
info.app,
Orientation::Vertical,
info.bar_position,
);
let container = gtk::Box::new(Orientation::Horizontal, 0);

let (ui_tx, mut ui_rx) = mpsc::channel(32);
Expand Down
5 changes: 3 additions & 2 deletions src/modules/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5,20 +5,21 @@
/// Clicking the widget opens a popup containing the current time
/// with second-level precision and a calendar.
pub mod clock;
pub mod focused;
pub mod launcher;
pub mod mpd;
pub mod script;
pub mod sysinfo;
pub mod tray;
pub mod workspaces;

use crate::config::BarPosition;
/// Shamelessly stolen from here:
/// <https://github.com/zeroeightysix/rustbar/blob/master/src/modules/module.rs>
use glib::IsA;
use gtk::{Application, Widget};
use serde::de::DeserializeOwned;
use serde_json::Value;
use crate::config::BarPosition;

#[derive(Clone)]
pub enum ModuleLocation {
Expand All @@ -31,7 +32,7 @@ pub struct ModuleInfo<'a> {
pub app: &'a Application,
pub location: ModuleLocation,
pub bar_position: &'a BarPosition,
pub output_name: &'a str
pub output_name: &'a str,
}

pub trait Module<W>
Expand Down
Loading

0 comments on commit dc14cb0

Please sign in to comment.