402 lines
11 KiB
Ruby
Executable File
402 lines
11 KiB
Ruby
Executable File
#!/usr/bin/env ruby
|
|
|
|
require_relative "./lib/X-kutu"
|
|
|
|
if X.deploy < 0
|
|
raise "Failed to deploy X"
|
|
end
|
|
|
|
at_exit { X.cleanup }
|
|
|
|
$monitors = {}
|
|
$workspaces = {}
|
|
$windows = {}
|
|
|
|
$keybind_actions = {}
|
|
$mousebind_actions = {}
|
|
|
|
$mouse_data = {}
|
|
|
|
def refresh_monitors!
|
|
$monitors.clear
|
|
xrandr_output = `xrandr --query`
|
|
connected = xrandr_output.each_line.select { |line| line.include?(" connected") }
|
|
connected.sort_by! { |line| line.include?(" primary") ? 0 : 1 }
|
|
connected.first(2).each_with_index do |line, index|
|
|
next unless line =~ /(\d+)x(\d+)\+(\d+)\+(\d+)/
|
|
w, h, x, y = $1.to_i, $2.to_i, $3.to_i, $4.to_i
|
|
key = index.zero? ? :primary : :secondary
|
|
$monitors[key] = { x: x, y: y, width: w, height: h }
|
|
end
|
|
end
|
|
|
|
def fixed_size_or_aspect?(hints)
|
|
return false unless hints
|
|
flags = hints[:flags] || 0
|
|
same_size = (flags & (16 | 256) != 0) && (flags & 32 != 0) &&
|
|
hints[:max_width] == (hints[:min_width] || hints[:base_width]) &&
|
|
hints[:max_height] == (hints[:min_height] || hints[:base_height])
|
|
fixed_aspect = (flags & 128 != 0) &&
|
|
hints[:min_aspect_num] == hints[:max_aspect_num] &&
|
|
hints[:min_aspect_den] == hints[:max_aspect_den] &&
|
|
hints[:min_aspect_num] != 0
|
|
same_size || fixed_aspect
|
|
end
|
|
|
|
refresh_monitors!
|
|
|
|
def keybind(key, &block)
|
|
X.add_keybind key
|
|
$keybind_actions[key] = block
|
|
end
|
|
|
|
def mousebind(btn, &block)
|
|
X.add_mousebind btn
|
|
$mousebind_actions[btn] = block if block
|
|
end
|
|
|
|
class Node
|
|
attr_accessor :size
|
|
|
|
def initialize
|
|
@size = 0
|
|
end
|
|
end
|
|
|
|
class Window < Node
|
|
attr_accessor :window_id, :x, :y, :width, :height, :state, :floating, :workspace
|
|
|
|
def initialize(window_id, workspace)
|
|
@workspace = workspace
|
|
@window_id = window_id
|
|
@name = X.get_wm_name(window_id)
|
|
@wm_n_hints = X.get_wm_n_hints(window_id)
|
|
@floating = false
|
|
if @wm_n_hints
|
|
if fixed_size_or_aspect?(@wm_n_hints)
|
|
@floating = true
|
|
@width = @wm_n_hints[:max_width]
|
|
if @wm_n_hints[:flags] & 128 != 0
|
|
@height = @wm_n_hints[:max_width] * (@wm_n_hints[:min_aspect_num] / @wm_n_hints[:min_aspect_den])
|
|
else
|
|
@height = @wm_n_hints[:max_height]
|
|
end
|
|
@x = @workspace.width / 2 - @width / 2
|
|
@y = @workspace.height / 2 - @height / 2
|
|
apply_geometry!
|
|
end
|
|
end
|
|
@wm_hints = X.get_wm_hints(window_id)
|
|
if @wm_hints
|
|
if @wm_hints[:flags] & 1 != 0 && @wm_hints[:initial_state] == 3
|
|
X.set_wm_state window_id, 1
|
|
end
|
|
end
|
|
@transient_for = X.get_wm_transient_for(window_id)
|
|
@floating = true unless @transient_for.zero?
|
|
super()
|
|
end
|
|
|
|
def delete
|
|
@workspace.remove self
|
|
end
|
|
|
|
def each_leaf(&block)
|
|
block.call(self)
|
|
end
|
|
|
|
def move(x, y)
|
|
return unless @floating
|
|
@x, @y = x, y
|
|
apply_geometry!
|
|
end
|
|
|
|
def resize(width, height)
|
|
return unless @floating
|
|
@x = @workspace.width / 2 - width / 2
|
|
@y = @workspace.height / 2 - height / 2
|
|
@width, @height = width, height
|
|
apply_geometry!
|
|
end
|
|
|
|
def apply_geometry!
|
|
X.move_window @window_id, @x, @y
|
|
X.resize_window @window_id, @width, @height
|
|
end
|
|
end
|
|
|
|
class WindowBlock < Node
|
|
attr_accessor :children, :direction, :size, :x, :y, :width, :height
|
|
|
|
def initialize(direction = :horizontal)
|
|
@children = []
|
|
@direction = direction
|
|
super()
|
|
end
|
|
|
|
def add_node(node)
|
|
@children << node
|
|
end
|
|
|
|
def each_leaf(&block)
|
|
@children.each { |child| child.each_leaf(&block) }
|
|
end
|
|
|
|
def each_node(&block)
|
|
block.call(self)
|
|
@children.each { |child| child.each_node(&block) if child.is_a? WindowBlock }
|
|
end
|
|
|
|
def compute_geometry!
|
|
return if @children.empty?
|
|
|
|
horizontal = @direction == :horizontal
|
|
total_fixed = @children.map { |c| c.size || 0 }.sum
|
|
flex_count = @children.count { |c| c.size.to_i <= 0 }
|
|
total_space = horizontal ? @width : @height
|
|
remaining_space = total_space - total_fixed
|
|
remaining_space = 0 if remaining_space < 0
|
|
flex_size = flex_count > 0 ? remaining_space / flex_count : 0
|
|
pos = horizontal ? @x : @y
|
|
|
|
@children.each do |child|
|
|
child_size = child.size.to_i > 0 ? child.size.to_i : flex_size
|
|
if horizontal
|
|
child.x = pos
|
|
child.y = @y
|
|
child.width = child_size
|
|
child.height = @height
|
|
else
|
|
child.x = @x
|
|
child.y = pos
|
|
child.width = @width
|
|
child.height = child_size
|
|
end
|
|
child.compute_geometry! if child.is_a?(WindowBlock)
|
|
child.apply_geometry! if child.is_a?(Window)
|
|
pos += child_size
|
|
end
|
|
end
|
|
end
|
|
|
|
class RootWindowBlock < WindowBlock
|
|
def initialize(workspace)
|
|
super(:horizontal)
|
|
@x = workspace.x
|
|
@y = workspace.y
|
|
@width = workspace.width
|
|
@height = workspace.height
|
|
end
|
|
end
|
|
|
|
class Workspace
|
|
attr_reader :name, :monitor, :x, :y, :width, :height, :tiled_root_block
|
|
|
|
def initialize(name, monitor = :primary)
|
|
@monitor = monitor
|
|
@x = $monitors[monitor][:x]
|
|
@y = $monitors[monitor][:y]
|
|
@width = $monitors[monitor][:width]
|
|
@height = $monitors[monitor][:height]
|
|
@tiled_root_block = RootWindowBlock.new(self)
|
|
@untiled_windows = []
|
|
@name = name
|
|
end
|
|
|
|
def windows
|
|
windows = []
|
|
@untiled_windows.each { |w| windows << w }
|
|
@tiled_root_block.each_leaf { |w| windows << w }
|
|
windows
|
|
end
|
|
|
|
def add(window_id)
|
|
window = Window.new(window_id, self)
|
|
if window.floating
|
|
@untiled_windows << window
|
|
else
|
|
@tiled_root_block.add_node window
|
|
end
|
|
$windows[window_id] = window
|
|
end
|
|
|
|
def remove(window)
|
|
@untiled_windows.delete window
|
|
@tiled_root_block.each_node { |node| node.children.delete window }
|
|
$windows.delete window.window_id
|
|
end
|
|
|
|
def close_all
|
|
self.windows.each do |window|
|
|
X.kill window.window_id
|
|
remove window.window_id
|
|
end
|
|
end
|
|
|
|
def hide
|
|
self.windows.each { |window| X.hide window.window_id }
|
|
end
|
|
|
|
def show
|
|
self.windows.each { |window| X.show window.window_id }
|
|
end
|
|
end
|
|
|
|
EVENT_TYPES = {
|
|
1 => :create,
|
|
2 => :closed,
|
|
3 => :enter,
|
|
4 => :showed,
|
|
5 => :show_request,
|
|
6 => :mouse_press,
|
|
7 => :mouse_drag,
|
|
8 => :mouse_release,
|
|
9 => :key_press,
|
|
10 => :key_release,
|
|
11 => :closed,
|
|
12 => :configure_request,
|
|
13 => :resize_request
|
|
}.freeze
|
|
|
|
$workspaces[:main] = Workspace.new(:main)
|
|
|
|
keybind(24) do |event|
|
|
X.kill event[:window]
|
|
end
|
|
|
|
keybind(25) do |_event|
|
|
pid = spawn("kitty")
|
|
Process.detach pid
|
|
end
|
|
|
|
keybind(26) do |_event|
|
|
$workspaces.each_value(&:close_all)
|
|
exit 1
|
|
end
|
|
|
|
keybind(27) do |_event|
|
|
pp $workspaces[:main].windows
|
|
end
|
|
|
|
mousebind 1
|
|
mousebind 3
|
|
|
|
def compute_drop_targets!
|
|
targets = []
|
|
|
|
$monitors.each_value do |monitor|
|
|
margin = 40
|
|
targets << { type: :monitor_top, x: monitor[:x], y: monitor[:y], width: monitor[:width], height: margin, monitor: monitor }
|
|
targets << { type: :monitor_bottom, x: monitor[:x], y: monitor[:y] + monitor[:height] - margin, width: monitor[:width], height: margin, monitor: monitor }
|
|
targets << { type: :monitor_left, x: monitor[:x], y: monitor[:y], width: margin, height: monitor[:height], monitor: monitor }
|
|
targets << { type: :monitor_right, x: monitor[:x] + monitor[:width] - margin, y: monitor[:y], width: margin, height: monitor[:height], monitor: monitor }
|
|
end
|
|
|
|
$windows.each do |w|
|
|
# next if w.hidden?
|
|
margin = 60
|
|
targets << { type: :window_top, x: w.x, y: w.y, width: w.width, height: margin, window: w }
|
|
targets << { type: :window_bottom, x: w.x, y: w.y + w.height - margin, width: w.width, height: margin, window: w }
|
|
targets << { type: :window_left, x: w.x, y: w.y, width: margin, height: w.height, window: w }
|
|
targets << { type: :window_right, x: w.x + w.width - margin, y: w.y, width: margin, height: w.height, window: w }
|
|
end
|
|
|
|
targets
|
|
end
|
|
|
|
def drop_target(targets, cx, cy)
|
|
targets.find do |t|
|
|
cx >= t[:x] &&
|
|
cy >= t[:y] &&
|
|
cx < t[:x] + t[:width] &&
|
|
cy < t[:y] + t[:height]
|
|
end
|
|
end
|
|
|
|
loop do
|
|
event = X.wait_for_event
|
|
case EVENT_TYPES[event[:type]]
|
|
when :create
|
|
next unless event[:override_redirect].zero?
|
|
X.subscribe event[:window]
|
|
when :closed
|
|
pp "Deleting window #{event[:window]}"
|
|
$windows[event[:window]]&.delete
|
|
$workspaces[:main].tiled_root_block.compute_geometry!
|
|
when :enter
|
|
X.focus event[:window]
|
|
X.send_to_top event[:window]
|
|
floating_windows = $windows.select { |_i, w| w.floating }
|
|
floating_windows.each { |_i, w| X.send_to_top w.window_id }
|
|
when :showed
|
|
next unless event[:override_redirect].zero?
|
|
X.focus event[:window]
|
|
X.send_to_top event[:window]
|
|
floating_windows = $windows.select { |_i, w| w.floating }
|
|
floating_windows.each { |_i, w| X.send_to_top w.window_id }
|
|
when :show_request
|
|
X.show event[:window]
|
|
$workspaces[:main].add event[:window] if $windows[event[:window]].nil?
|
|
$workspaces[:main].tiled_root_block.compute_geometry!
|
|
when :mouse_press
|
|
next if event[:is_root] != 0
|
|
X.focus event[:window]
|
|
$mouse_data[:btn] = event[:btn]
|
|
$mouse_data[:window] = $windows[event[:window]]
|
|
X.grab_pointer event[:window]
|
|
if $mouse_data[:window].floating
|
|
$mouse_data[:mode] = :floating
|
|
$mouse_data[:pointer] = X.get_pointer
|
|
$mouse_data[:geometry] = X.get_geometry event[:window]
|
|
else
|
|
$mouse_data[:mode] = :tiled
|
|
end
|
|
$mousebind_actions[event[:btn]]&.call(event)
|
|
when :mouse_drag
|
|
mouse_pos = X.get_pointer
|
|
if $mouse_data[:mode] == :floating
|
|
dx = mouse_pos[:x] - $mouse_data[:pointer][:x]
|
|
dy = mouse_pos[:y] - $mouse_data[:pointer][:y]
|
|
if $mouse_data[:btn] == 1
|
|
$mouse_data[:window].move $mouse_data[:geometry][:x] + dx, $mouse_data[:geometry][:y] + dy
|
|
elsif $mouse_data[:btn] == 3
|
|
X.resize_window $mouse_data[:window].window_id,
|
|
[$mouse_data[:geometry][:width] + dx, 50].max,
|
|
[$mouse_data[:geometry][:height] + dy, 50].max
|
|
end
|
|
elsif $mouse_data[:mode] == :tiled
|
|
if $mouse_data[:btn] == 1
|
|
find_targets = compute_drop_targets!
|
|
target = drop_target(find_targets, mouse_pos[:x], mouse_pos[:y])
|
|
X.draw_rectangle target[:x], target[:y], target[:width], target[:height], 0x00ff00
|
|
elsif $mouse_data[:btn] == 3
|
|
# TODO: tile resize . dynamic
|
|
end
|
|
end
|
|
when :mouse_release
|
|
if $mouse_data[:mode] == :floating
|
|
if [1, 3].include?($mouse_data[:btn])
|
|
X.ungrab_pointer
|
|
X.focus $mouse_data[:window].window_id
|
|
X.send_to_top $mouse_data[:window].window_id
|
|
end
|
|
elsif $mouse_data[:mode] == :tiled
|
|
# TODO
|
|
end
|
|
X.focus $mouse_data[:window].window_id if $mouse_data[:window]
|
|
$mouse_data = {}
|
|
when :key_press
|
|
$keybind_actions[event[:btn]]&.call(event)
|
|
when :key_release
|
|
# TODO
|
|
when :configure_request
|
|
$windows[event[:window]]&.resize event[:width], event[:height]
|
|
X.send_to_top event[:window]
|
|
X.focus event[:window]
|
|
when :resize_request
|
|
$windows[event[:window]]&.resize event[:width], event[:height]
|
|
X.send_to_top event[:window]
|
|
X.focus event[:window]
|
|
end
|
|
end
|