pick_or_create

filter out floats (or any other window configs)

window id filter

fix: center hint should be drawn always

I dont know why I thought it doesnt

Cleaner code with nvim_win_call()

Author: IndianBoy42

README.md

@@ -39,9 +39,11 @@ use {
 
 ```lua
 local picked_window_id = require('window-picker').pick_window()
+local picked_window_id = require('window-picker').pick_or_create()
 ```
 
-You can put the picked window id to good use
+You can put the picked window id to good use.
+`pick_or_create` will create a new split if you use the prefix keys shown on the edges of windows.
 
 ## Configuration
 

lua/window-picker/config.lua

@@ -9,7 +9,10 @@ local config = {
 
 	-- when you go to window selection mode, status bar will show one of
 	-- following letters on them so you can use that letter to select the window
-	selection_chars = 'FJDKSLA;CMRUEIWOQP',
+	selection_chars = 'FDSACMRUEIWOQP',
+
+	-- when creation mode is enabled, the following will be shown (only in floating window mode for now)
+	create_chars = 'HJKL',
 
 	-- This section contains picker specific configurations
 	picker_config = {
@@ -79,6 +82,11 @@ local config = {
 		-- filter using window options
 		wo = {},
 
+		-- filter using window config
+		window_configs = {
+			relative = true, -- checks if this key exists and nonempty (checks for floating windows)
+		},
+
 		-- if the file path contains one of following names, the window
 		-- will be ignored
 		file_path_contains = {},

lua/window-picker/configurer.lua

@@ -18,6 +18,7 @@ end
 
 function M._basic_config_manipulations(config)
 	config.chars = M._str_to_char_list(config.selection_chars)
+	config.create_chars = M._str_to_char_list(config.create_chars)
 	return config
 end
 

lua/window-picker/filters/default-window-filter.lua

@@ -3,6 +3,9 @@ local util = require('window-picker.util')
 --- @class DefaultWindowFilter
 --- @field filter_stack function[]
 --- @field window_options table<string, any> window options to filter
+--- @field window_configs table<string, any> window configs to filter
+--- @field window_ids table<string, any> window ids to include
+--- @field not_window_ids table<string, any> window ids to exclude
 --- @field buffer_options table<string, any> buffer options to filter
 --- @field file_name_contains string[] file names to filter
 --- @field file_path_contains string[] file paths to filter
@@ -15,10 +18,13 @@ function M:new()
 	self.__index = self
 
 	o.filter_stack = {
+		o._window_config_filter,
 		o._window_option_filter,
 		o._buffer_options_filter,
 		o._file_path_contains_filter,
 		o._file_path_contains_filter,
+		o._window_id_filter,
+		o._not_window_id_filter,
 		o._current_window_filter,
 	}
 
@@ -28,8 +34,11 @@ end
 function M:set_config(config)
 	self.window_options = config.wo or {}
 	self.buffer_options = config.bo or {}
+	self.window_configs = config.window_configs or {}
 	self.file_name_contains = config.file_name_contains or {}
 	self.file_path_contains = config.file_path_contains or {}
+	self.window_ids = config.window_ids or {}
+	self.not_window_ids = config.not_window_ids or {}
 	self.include_current_win = config.include_current_win
 end
 
@@ -43,6 +52,33 @@ function M:filter_windows(windows)
 	return filtered_windows
 end
 
+function M:_window_config_filter(windows)
+	if self.window_configs and vim.tbl_count(self.window_configs) > 0 then
+		return util.tbl_filter(windows, function(winid)
+			local config = vim.api.nvim_win_get_config(winid)
+			for cfg_key, cfg_values in pairs(self.window_configs) do
+				local actual_opt = config[cfg_key]
+
+				local has_value = false
+
+				if type(cfg_values) == 'table' then
+					has_value = vim.tbl_contains(cfg_values, actual_opt)
+				elseif cfg_values == true then
+					has_value = actual_opt ~= '' and actual_opt ~= nil
+				end
+
+				if has_value then
+					return false
+				end
+			end
+
+			return true
+		end)
+	else
+		return windows
+	end
+end
+
 function M:_window_option_filter(windows)
 	if self.window_options and vim.tbl_count(self.window_options) > 0 then
 		return util.tbl_filter(windows, function(winid)
@@ -63,6 +99,24 @@ function M:_window_option_filter(windows)
 	end
 end
 
+function M:_window_id_filter(windows)
+	if self.window_ids and vim.tbl_count(self.window_ids) > 0 then
+		windows = util.tbl_filter(windows, function(winid)
+			return vim.tbl_contains(self.window_ids, winid)
+				and not vim.tbl_contains(self.not_window_ids, winid)
+		end)
+	end
+	return windows
+end
+function M:_not_window_id_filter(windows)
+	if self.not_window_ids and vim.tbl_count(self.not_window_ids) > 0 then
+		windows = util.tbl_filter(windows, function(winid)
+			return not vim.tbl_contains(self.not_window_ids, winid)
+		end)
+	end
+	return windows
+end
+
 function M:_buffer_options_filter(windows)
 	if self.buffer_options and vim.tbl_count(self.buffer_options) > 0 then
 		return util.tbl_filter(windows, function(winid)

lua/window-picker/hints/floating-big-letter-hint.lua

@@ -3,6 +3,7 @@ local utf8 = require('window-picker.utf-8')
 --- @class FloatingBigLetterHint
 --- @field win { width: number, height: number } window options
 --- @field windows number[] list of window ids
+--- @field prefix_windows number[][] list of window ids for the prefix characters
 local M = {}
 
 local border = {
@@ -30,6 +31,7 @@ function M:new()
 			height = 8,
 		},
 		windows = {},
+		prefix_windows = { {}, {}, {}, {} },
 	}
 
 	setmetatable(o, self)
@@ -40,6 +42,7 @@ end
 
 function M:set_config(config)
 	self.chars = config.chars
+	self.create_chars = config.create_chars
 	local font = config.picker_config.floating_big_letter.font
 
 	if type(font) == 'string' then
@@ -51,13 +54,25 @@ function M:set_config(config)
 	end
 end
 
-function M:_get_float_win_pos(window)
+local positioning_params = {
+	m = { 0.5, 0.5, 0.5, 0.5, 'NW' }, -- Middle
+	h = { 0, 0, 0.5, 0.5, 'NW' }, -- Middle-left
+	j = { 0.5, 0.5, 1, 0, 'SW' }, -- Bottom-middle
+	k = { 0.5, 0.5, 0, 0, 'NW' }, -- Top-middle
+	l = { 1, 0, 0.5, 0.5, 'NE' }, -- Middle-right
+}
+local create_win_positions = { 'h', 'j', 'k', 'l' }
+
+function M:_get_float_win_pos(window, position)
+	position = position or 'm'
 	local width = vim.api.nvim_win_get_width(window)
 	local height = vim.api.nvim_win_get_height(window)
+	local params = positioning_params[position]
 
 	local point = {
-		x = ((width - self.win.width) / 2),
-		y = ((height - self.win.height) / 2),
+		x = width * params[1] - self.win.width * params[2],
+		y = height * params[3] - self.win.height * params[4],
+		anchor = params[5],
 	}
 
 	return point
@@ -79,7 +94,7 @@ function M._add_big_char_margin(lines)
 	--bottom padding
 	table.insert(lines, #lines + 1, '')
 
-	--left & right padding
+	--left & right paddin
 
 	for _, line in ipairs(lines) do
 		local new_line = string.format(
@@ -95,8 +110,8 @@ function M._add_big_char_margin(lines)
 	return centered_lines
 end
 
-function M:_show_letter_in_window(window, char)
-	local point = self:_get_float_win_pos(window)
+function M:_show_letter_in_window(window, char, position)
+	local point = self:_get_float_win_pos(window, position)
 
 	local lines = self._add_big_char_margin(vim.split(char, '\n'))
 
@@ -107,37 +122,67 @@ function M:_show_letter_in_window(window, char)
 	local window_id = vim.api.nvim_open_win(buffer_id, false, {
 		relative = 'win',
 		win = window,
-		focusable = true,
+		focusable = false,
 		row = point.y,
 		col = point.x,
 		width = width,
 		height = height,
 		style = 'minimal',
 		border = border,
+		anchor = point.anchor,
 	})
 
 	vim.api.nvim_buf_set_lines(buffer_id, 0, 0, true, lines)
 
 	return window_id
 end
 
-function M:draw(windows)
+function M:draw(windows, or_create)
 	for index, window in ipairs(windows) do
-		local char = self.chars[index]
-		local big_char = self.big_chars[char:lower()]
-		local window_id = self:_show_letter_in_window(window, big_char)
-		table.insert(self.windows, window_id)
+		do
+			local char = self.chars[index]
+			local big_char = self.big_chars[char:lower()]
+			local window_id = self:_show_letter_in_window(window, big_char)
+			table.insert(self.windows, window_id)
+		end
+
+		if or_create then
+			for i = 1, 4 do
+				local char = self.create_chars[i]
+				local big_char = self.big_chars[char:lower()]
+				local dir = create_win_positions[i]
+				local window_id =
+					self:_show_letter_in_window(window, big_char, dir)
+				table.insert(self.prefix_windows[i], window_id)
+			end
+		end
 	end
 end
 
-function M:clear()
-	for _, window in ipairs(self.windows) do
+local function clear_list_of_windows(windows)
+	for _, window in ipairs(windows) do
 		if vim.api.nvim_win_is_valid(window) then
 			local buffer = vim.api.nvim_win_get_buf(window)
 			vim.api.nvim_win_close(window, true)
 			vim.api.nvim_buf_delete(buffer, { force = true })
 		end
 	end
+end
+
+function M:clear_prefixes(index)
+	for prefix, windows in ipairs(self.prefix_windows) do
+		-- Clear all prefixes except the one that was chosen
+		if prefix ~= index then
+			clear_list_of_windows(windows)
+			self.prefix_windows[prefix] = {}
+		end
+	end
+end
+
+function M:clear()
+	clear_list_of_windows(self.windows)
+
+	self:clear_prefixes()
 
 	self.windows = {}
 end

lua/window-picker/hints/statusline-hint.lua

@@ -20,7 +20,8 @@ end
 
 function M:set_config(config)
 	self.chars = config.chars
-	self.selection_display = config.picker_config.statusline_winbar_picker.selection_display
+	self.selection_display =
+		config.picker_config.statusline_winbar_picker.selection_display
 
 	vim.api.nvim_set_hl(0, 'WindowPickerStatusLine', {
 		fg = '#ededed',
@@ -69,7 +70,6 @@ function M:draw(windows)
 	for index, window in ipairs(windows) do
 		local char = self.chars[index]
 
-
 		local display_text = self.selection_display
 				and self.selection_display(char, window)
 			or '%=' .. char .. '%='
@@ -83,6 +83,8 @@ function M:draw(windows)
 	vim.cmd.redraw()
 end
 
+function M:clear_prefixes(char) end
+
 --- clear the screen after print
 function M:clear()
 	for window, options in pairs(self.window_options) do

lua/window-picker/init.lua

@@ -3,6 +3,8 @@ local dconfig = require('window-picker.config')
 
 local M = {}
 
+local default_pick, default_pick_or_create
+
 function M.pick_window(custom_config)
 	local config = dconfig
 
@@ -39,6 +41,14 @@ function M.pick_window(custom_config)
 	return builder:new():set_config(config):set_hint(hint):build():pick_window()
 end
 
+function M.pick_or_create(custom_config)
+	custom_config = custom_config or {}
+	custom_config.or_create = true
+	custom_config.include_current_win = custom_config.include_current_win
+		or true
+
+	return M.pick_window(custom_config)
+end
 function M.setup() end
 
 return M