wmtools/base.nim

276 lines
6.8 KiB
Nim
Raw Normal View History

import std/[os,osproc,strutils,json,rdstdin,marshal]
2022-02-06 21:10:06 +01:00
type
2022-04-28 21:58:40 +02:00
Info* = object
2022-04-28 13:50:07 +02:00
title*: string
selected_fg*: string
selected_bg*: string
unselected_fg*: string
unselected_bg*: string
2022-02-06 21:10:06 +01:00
full_text*: string
# next few are for i3bar use
2022-02-06 21:10:06 +01:00
border*: string
2022-02-12 16:19:43 +01:00
background*: string
color*: string
html_text*: string
short_text*: string
2022-05-24 22:11:16 +02:00
args*: seq[string]
2022-07-05 14:00:40 +02:00
Menu = object
2022-05-06 12:12:13 +02:00
command: string
bottom: string
grab_kb: string
i_case: string
lines_shown: string
monitor: string
prompt: string
font: string
norm_bg: string
norm_fg: string
sel_bg: string
sel_fg: string
2022-07-05 14:00:40 +02:00
extra_cmd: string
i3BarInput* = object
button*: int
2022-02-12 16:19:43 +01:00
x*: int
y*: int
const WM_TOOLS_DIR* = getHomeDir() & "Nextcloud/.wm_tools/"
2022-02-06 21:10:06 +01:00
const background* = "#000000"
const backgroundalt* = "#bb222222"
const backgroundalt2* = "#bb333333"
const foreground* = "#dfdfdf"
const foregroundalt* = "#777"
const foregroundalt2* = "#ccc"
2022-02-12 16:19:43 +01:00
const black* = "#000000"
2022-02-27 15:57:40 +01:00
const white* = "#FFFFFF"
2022-02-06 21:10:06 +01:00
const yellow* = "#ffb52a"
const red* = "#e60053"
const purple* = "#9f78e1"
const blue* = "#0a6cf5"
const lightblue* = "#7296EF"
const lighterblue* = "#B5DDF7"
const green* = "#4b9901"
const lightgreen* = "#00ff00"
const grey* = "#dfdfdf"
const darkgrey* = "#444"
const primary* = yellow
const secondary* = red
const alert* = "#bd2c40"
2022-05-06 12:12:13 +02:00
const MAX_LINES = 20
const font = "Hermit-12"
const WL_DMENU = "dmenu"
const WL_ROFI = "wofi --dmenu"
var loop* = false
var stoploop* = true
var tool* = "dmenu"
var passmenu* = false
var command_wrapper* = false
var run_command* = ""
var wayland* = false
2022-02-06 21:10:06 +01:00
2022-05-10 21:47:15 +02:00
proc newInfo*(str: string = "Info"): Info =
var title = str
if tool == "rofi":
2022-05-10 21:47:15 +02:00
title = title & " : "
2022-04-28 21:58:40 +02:00
return Info(
title: title,
selected_fg: black,
selected_bg: white,
unselected_fg: white,
unselected_bg: black,
# next few are for i3bar use
2022-02-27 15:57:40 +01:00
border: white,
background: black,
color: foreground,
2022-02-27 15:57:40 +01:00
)
2022-05-06 12:12:13 +02:00
proc newMenuConfig(cmd: string = "dmenu"): Menu =
var run = cmd
if wayland and cmd == "dmenu":
run = WL_DMENU
2022-07-05 14:00:40 +02:00
var menu = Menu()
menu.command = run
2022-07-05 14:00:40 +02:00
menu.prompt = "-p"
menu.i_case = "-i"
menu.lines_shown = "-l"
return menu
proc newRofiConfig(cmd: string = "rofi -dmenu"): Menu =
var run = cmd
if wayland and cmd == "rofi -dmenu":
run = WL_ROFI
var menu = newMenuConfig(run)
2022-07-05 14:00:40 +02:00
menu.extra_cmd = "-markup-rows -kb-row-select \"Tab\" -kb-row-tab \"\""
return menu
proc newDmenuConfig(cmd: string = "dmenu"): Menu =
var run = cmd
if wayland and cmd == "dmenu":
run = WL_DMENU
var menu = newMenuConfig(run)
2022-07-05 14:00:40 +02:00
menu.bottom = "-b"
menu.grabkb = "-f"
menu.monitor = "-m"
menu.font = "-fn"
menu.norm_bg = "-nb"
menu.norm_fg = "-nf"
menu.sel_bg = "-sb"
menu.sel_fg = "-sf"
return menu
proc newMenu(): Menu =
if passmenu:
return newDmenuConfig("passmenu")
elif command_wrapper:
return newDmenuConfig(run_command)
elif tool == "rofi":
2022-05-06 12:12:13 +02:00
return newRofiConfig()
elif tool == "dmenu":
2022-07-05 14:00:40 +02:00
return newDmenuConfig()
return newMenuConfig()
2022-05-06 12:12:13 +02:00
proc debugLog*(str: string) =
let f = open("/tmp/debug.txt",fmAppend)
defer: f.close()
f.writeLine(str)
proc switchTwmMode*(mode: string = "default") =
# I intend to add support for more twm as time goes on (I switch around a lot)
# Switch out of an i3 bindsym mode if set
if wayland:
discard execCmd("sway mode \"" & mode & "\"")
else:
discard execCmd("i3-msg mode \"" & mode & "\"")
proc checkWayland() =
if getEnv("XDG_SESSION_TYPE") == "wayland":
wayland = true
proc parseInput*(): i3BarInput =
let input = readLineFromStdin("")
try:
let jsonNode = parseJson(input)
let i3input = to(jsonNode, i3BarInput)
return i3input
except:
return i3BarInput()
2022-04-28 13:34:09 +02:00
2022-02-12 14:20:21 +01:00
proc clearInput*(count: int = 1) =
2022-02-12 16:19:43 +01:00
for x in countup(1, count):
2022-02-12 14:20:21 +01:00
discard readLineFromStdin("")
2022-02-06 21:10:06 +01:00
proc getArguments*(): seq[string] =
let args = commandLineParams()
return args
2022-05-06 12:12:13 +02:00
proc stripQuotes*(str: string): string =
return replace(str,"\"",""")
proc quote*(str: string): string =
var text = str
# May need to put some further work to escape some special chars here
text = stripQuotes(text)
# Put leading and ending quote marks in
return " \"" & text & "\" "
# ^ Add a spaces ^ so the previous flag isn't touching
2022-07-05 14:00:40 +02:00
#
proc markup(str: string): string =
var text = str
return text
2022-05-06 12:12:13 +02:00
2022-07-05 14:00:40 +02:00
proc genMenuCmd*(data: Info, opts: varargs[string], rofi: bool = false): string =
# Build dmenu/rofi command
2022-05-06 12:12:13 +02:00
var cmd = ""
# if the text is empty, we don't want to create a menu item of it
2022-05-06 12:12:13 +02:00
if data.full_text != "":
2022-07-05 14:00:40 +02:00
let text = markup(data.full_text)
cmd &= text & "\n"
for opt in opts:
2022-07-05 14:00:40 +02:00
let text = markup(opt)
cmd = cmd & text & "\n"
2022-04-28 21:58:40 +02:00
cmd.removeSuffix("\n")
2022-05-06 12:12:13 +02:00
cmd = "echo -e" & quote(cmd) & " | "
2022-07-05 14:00:40 +02:00
var menu = newMenu()
cmd = cmd & menu.command & " "
cmd = cmd & menu.extra_cmd & " "
cmd = cmd & menu.i_case & " "
cmd = cmd & menu.lines_shown & " " & $MAX_LINES & " "
cmd = cmd & menu.prompt & quote(data.title)
cmd = cmd & menu.norm_bg & quote(data.unselected_bg)
cmd = cmd & menu.norm_fg & quote(data.unselected_fg)
cmd = cmd & menu.sel_bg & quote(data.selected_bg)
cmd = cmd & menu.sel_fg & quote(data.selected_fg)
cmd = cmd & menu.font & quote(font)
echo cmd
return cmd
2022-05-06 12:12:13 +02:00
2022-07-05 14:00:40 +02:00
proc runMenu*(data: Info, opts: varargs[string], rofi: bool = false): string =
let cmd = genMenuCmd(data, opts, rofi)
2022-05-06 12:12:13 +02:00
#echo cmd
#
# Run command and get output
2022-05-09 21:12:05 +02:00
var output = execCmdEx(cmd)
output.output.stripLineEnd()
return output.output
2022-05-09 22:08:54 +02:00
proc copyToClipboard*(str: string) =
if wayland:
discard execCmd("wl-copy " & str)
else:
discard execCmd("echo -n " & quote(str) & " | xclip -selection clipboard")
2022-05-14 16:44:20 +02:00
proc outputData*(data: Info, args: varargs[string]): string {.discardable.} =
var output = ""
if tool == "rofi":
2022-07-05 14:00:40 +02:00
output = runMenu(data,args, rofi = true)
elif loop:
# mainly for i3bar/i3blocks compatible output
var j_data = data
if j_data.html_text != "":
j_data.full_text = j_data.html_text
echo $$j_data
else:
# if all else fails, use dmenu (default)
2022-07-05 14:00:40 +02:00
output = runMenu(data,args)
return output
proc checkCacheDir() =
if not dirExists(WM_TOOLS_DIR):
createDir(WM_TOOLS_DIR)
# At Start up:
checkCacheDir()
checkWayland()
# Switch bindsym mode back to default as it could be being used.
switchTwmMode()
let args* = getArguments()
for idx, arg in args:
case arg:
of "noloop":
stoploop = true
of "i3bar":
# I've kind of changed from using an i3bar to using #nobar so i3bar
# isn't really supported any more but this is here for backwards compatibility
loop = true
stoploop = false
of "dmenu":
stoploop = true
tool = "dmenu"
of "rofi":
stoploop = true
tool = "rofi"
of ["pass","passmenu"]:
passmenu = true
break
2022-04-28 13:50:07 +02:00