Improve workspace script

1 files changed, 9 insertions, 2 deletions

diff --git a/bin/workspace b/bin/workspace
index 9082f39..f735014 100755
--- a/bin/workspace
+++ b/bin/workspace
@@ -1,5 +1,12 @@
 #!/bin/zsh
 
-workspaces=$(i3-msg -t get_workspaces | grep -Po '"name":.*?[^\\]"' | sed 's/"name":"\([0-9a-zA-Z]\+\)"/\1/g')
-target=$(echo $workspaces | dmenu -i -p "Go to workspace:")
+workspaces=$(i3-msg -t get_workspaces | grep -Po '"name":.*?[^\\]"' | sed 's/"name":"\([^"]\+\)"/\1/g')
+used_numbers=$(echo $workspaces | awk 'BEGIN { FS = ":" } ; { print $1 }')
+max_number=$(echo $used_numbers | tail -1)
+largest_gap=$(echo $used_numbers | awk '$1!=p+1{print p+1}{p=$1}')
+[[ -z $largest_gap ]] && new_number=$(($max_number +  1)) || new_number=$largest_gap
+
+target=$(echo $workspaces | dmenu -i -sf green -p "Go to workspace:")
+[[ ! $target = [0-9]:* ]] && target=$new_number:$target
+
 exec i3-msg workspace $target