398 lines
10 KiB
Python
Executable File
398 lines
10 KiB
Python
Executable File
#! /usr/bin/python3
|
||
#
|
||
# config.py
|
||
# Copyright (C) 2008 Canonical Ltd.
|
||
# Copyright (C) 2008-2014 Dustin Kirkland <kirkland@byobu.org>
|
||
|
||
#
|
||
# Authors: Nick Barcet <nick.barcet@ubuntu.com>
|
||
# Dustin Kirkland <kirkland@byobu.org>
|
||
#
|
||
# This program is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU General Public License as published by
|
||
# the Free Software Foundation, version 3 of the License.
|
||
#
|
||
# This program is distributed in the hope that it will be useful,
|
||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
# GNU General Public License for more details.
|
||
#
|
||
# You should have received a copy of the GNU General Public License
|
||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
||
# If you change any strings, please generate localization information with:
|
||
# ./debian/rules get-po
|
||
|
||
from __future__ import print_function
|
||
import sys
|
||
import os
|
||
import os.path
|
||
import time
|
||
import string
|
||
import subprocess
|
||
import gettext
|
||
import glob
|
||
|
||
|
||
def error(msg):
|
||
print("ERROR: %s" % msg)
|
||
sys.exit(1)
|
||
|
||
|
||
try:
|
||
import snack
|
||
from snack import *
|
||
except Exception:
|
||
error("Could not import the python snack module")
|
||
|
||
|
||
PKG = "byobu"
|
||
HOME = os.getenv("HOME")
|
||
USER = os.getenv("USER")
|
||
BYOBU_CONFIG_DIR = os.getenv("BYOBU_CONFIG_DIR", HOME + "/.byobu")
|
||
BYOBU_RUN_DIR = os.getenv("BYOBU_RUN_DIR", HOME + "/.cache/byobu")
|
||
BYOBU_BACKEND = os.getenv("BYOBU_BACKEND", "tmux")
|
||
BYOBU_SOCKETDIR = os.getenv("SOCKETDIR", "/var/run/screen")
|
||
BYOBU_PREFIX = os.getenv("BYOBU_PREFIX", "/usr")
|
||
SHARE = BYOBU_PREFIX + '/share/' + PKG
|
||
DOC = BYOBU_PREFIX + '/share/doc/' + PKG
|
||
if not os.path.exists(SHARE):
|
||
SHARE = BYOBU_CONFIG_DIR + "/" + SHARE
|
||
if not os.path.exists(DOC):
|
||
DOC = BYOBU_PREFIX + '/share/doc/packages/' + PKG
|
||
if not os.path.exists(DOC):
|
||
DOC = BYOBU_CONFIG_DIR + "/" + DOC
|
||
DEF_ESC = "A"
|
||
RELOAD = "If you are using the default set of keybindings, press\n<F5> or <ctrl-a-R> to activate these changes.\n\nOtherwise, exit this session and start a new one."
|
||
RELOAD_FLAG = "%s/reload-required" % (BYOBU_RUN_DIR)
|
||
ESC = ''
|
||
snack.hotkeys[ESC] = ord(ESC)
|
||
snack.hotkeys[ord(ESC)] = ESC
|
||
gettext.bindtextdomain(PKG, SHARE + '/po')
|
||
gettext.textdomain(PKG)
|
||
_ = gettext.gettext
|
||
|
||
|
||
def ioctl_GWINSZ(fd):
|
||
# Discover terminal width
|
||
try:
|
||
import fcntl
|
||
import termios
|
||
import struct
|
||
import os
|
||
cr = struct.unpack('hh', fcntl.ioctl(fd, termios.TIOCGWINSZ, '1234'))
|
||
except Exception:
|
||
return None
|
||
return cr
|
||
|
||
|
||
def reload_required():
|
||
try:
|
||
if not os.path.exists(BYOBU_CONFIG_DIR):
|
||
# 493 (decimal) is 0755 (octal)
|
||
# Use decimal for portability across all python versions
|
||
os.makedirs(BYOBU_CONFIG_DIR, 493)
|
||
f = open(RELOAD_FLAG, 'w')
|
||
f.close()
|
||
if BYOBU_BACKEND == "screen":
|
||
subprocess.call([BYOBU_BACKEND, "-X", "at", "0", "source", "%s/profile" % BYOBU_CONFIG_DIR])
|
||
except Exception:
|
||
True
|
||
|
||
|
||
def terminal_size():
|
||
# decide on some terminal size
|
||
cr = ioctl_GWINSZ(0) or ioctl_GWINSZ(1) or ioctl_GWINSZ(2)
|
||
# try open fds
|
||
if not cr:
|
||
# ...then ctty
|
||
try:
|
||
fd = os.open(os.ctermid(), os.O_RDONLY)
|
||
cr = ioctl_GWINSZ(fd)
|
||
os.close(fd)
|
||
except Exception:
|
||
pass
|
||
if not cr:
|
||
# env vars or finally defaults
|
||
try:
|
||
cr = (env['LINES'], env['COLUMNS'])
|
||
except Exception:
|
||
cr = (25, 80)
|
||
# reverse rows, cols
|
||
return int(cr[1] - 5), int(cr[0] - 5)
|
||
|
||
|
||
def menu(snackScreen, size, isInstalled):
|
||
if isInstalled:
|
||
installtext = _("Byobu currently launches at login (toggle off)")
|
||
else:
|
||
installtext = _("Byobu currently does not launch at login (toggle on)")
|
||
li = Listbox(height=6, width=60, returnExit=1)
|
||
li.append(_("Help -- Quick Start Guide"), 1)
|
||
li.append(_("Toggle status notifications"), 2)
|
||
li.append(_("Change escape sequence"), 3)
|
||
li.append(installtext, 4)
|
||
bb = ButtonBar(snackScreen, (("Exit", "exit", ESC),), compact=1)
|
||
g = GridForm(snackScreen, _(" Byobu Configuration Menu"), 1, 2)
|
||
g.add(li, 0, 0, padding=(4, 2, 4, 2))
|
||
g.add(bb, 0, 1, padding=(1, 1, 0, 0))
|
||
if bb.buttonPressed(g.runOnce()) == "exit":
|
||
return 0
|
||
else:
|
||
return li.current()
|
||
|
||
|
||
def messagebox(snackScreen, width, height, title, text, scroll=0, buttons=((_("Okay"), "okay"), (_("Cancel"), "cancel", ESC))):
|
||
t = Textbox(width, height, text, scroll=scroll)
|
||
bb = ButtonBar(snackScreen, buttons, compact=1)
|
||
g = GridForm(snackScreen, title, 1, 2)
|
||
g.add(t, 0, 0, padding=(0, 0, 0, 0))
|
||
g.add(bb, 0, 1, padding=(1, 1, 0, 0))
|
||
return bb.buttonPressed(g.runOnce())
|
||
|
||
|
||
def help(snackScreen, size):
|
||
f = open(DOC + '/help.' + BYOBU_BACKEND + '.txt')
|
||
text = f.read()
|
||
f.close()
|
||
text = text.replace("<esckey>", getesckey(), 1)
|
||
t = Textbox(67, 16, text, scroll=1, wrap=1)
|
||
bb = ButtonBar(snackScreen, ((_("Menu"), "menu", ESC),), compact=1)
|
||
g = GridForm(snackScreen, _("Byobu Help"), 2, 4)
|
||
g.add(t, 1, 0)
|
||
g.add(bb, 1, 1, padding=(1, 1, 0, 0))
|
||
button = bb.buttonPressed(g.runOnce())
|
||
return 100
|
||
|
||
|
||
def readstatus():
|
||
status = {}
|
||
glo = {}
|
||
loc = {}
|
||
for f in [SHARE + '/status/status', BYOBU_CONFIG_DIR + '/status']:
|
||
if os.path.exists(f):
|
||
try:
|
||
exec(open(f).read(), glo, loc)
|
||
except Exception:
|
||
error("Invalid configuration [%s]" % f)
|
||
if BYOBU_BACKEND == "tmux":
|
||
items = "%s %s" % (loc["tmux_left"], loc["tmux_right"])
|
||
else:
|
||
items = "%s %s %s %s" % (loc["screen_upper_left"], loc["screen_upper_right"], loc["screen_lower_left"], loc["screen_lower_right"])
|
||
for i in items.split():
|
||
if i.startswith("#"):
|
||
i = i.replace("#", "")
|
||
status[i] = "0"
|
||
else:
|
||
status[i] = "1"
|
||
li = []
|
||
keys = list(status.keys())
|
||
for i in sorted(keys):
|
||
window = [int(status[i]), i]
|
||
li.append(window)
|
||
return li
|
||
|
||
|
||
def genstatusstring(s, status):
|
||
new = ""
|
||
glo = {}
|
||
loc = {}
|
||
exec(open(SHARE + '/status/status').read(), glo, loc)
|
||
for i in loc[s].split():
|
||
if i.startswith("#"):
|
||
i = i.replace("#", "")
|
||
if status[i] == 1:
|
||
new += " " + i
|
||
else:
|
||
new += " #" + i
|
||
return new
|
||
|
||
|
||
def writestatus(items):
|
||
status = {}
|
||
path = BYOBU_CONFIG_DIR + '/status'
|
||
for i in items:
|
||
status[i[1]] = i[0]
|
||
for key in ["tmux_left", "tmux_right", "screen_upper_left", "screen_upper_right", "screen_lower_left", "screen_lower_right"]:
|
||
if key.startswith(BYOBU_BACKEND):
|
||
try:
|
||
f = open(path, "r")
|
||
except Exception:
|
||
f = open(SHARE + '/status/status', "r")
|
||
lines = f.readlines()
|
||
f.close()
|
||
try:
|
||
f = open(path, "w")
|
||
except Exception:
|
||
f = open(path, "a+")
|
||
for l in lines:
|
||
if l.startswith("%s=" % key):
|
||
val = genstatusstring(key, status)
|
||
f.write("%s=\"%s\"\n" % (key, val))
|
||
else:
|
||
f.write(l)
|
||
f.close
|
||
|
||
|
||
def togglestatus(snackScreen, size):
|
||
itemlist = readstatus()
|
||
rl = Label("")
|
||
r = CheckboxTree(12, scroll=1)
|
||
count = 0
|
||
for item in itemlist:
|
||
if item[0] != -1:
|
||
r.append(item[1], count, selected=item[0])
|
||
count = count + 1
|
||
bb = ButtonBar(snackScreen, ((_("Apply"), "apply"), (_("Cancel"), "cancel", ESC)), compact=1)
|
||
g = GridForm(snackScreen, _("Toggle status notifications"), 2, 4)
|
||
g.add(rl, 0, 0, anchorLeft=1, anchorTop=1, padding=(4, 0, 0, 1))
|
||
g.add(r, 1, 0)
|
||
g.add(bb, 1, 1, padding=(4, 1, 0, 0))
|
||
if bb.buttonPressed(g.runOnce()) != "cancel":
|
||
count = 0
|
||
for item in itemlist:
|
||
if item[0] != -1:
|
||
item[0] = r.getEntryValue(count)[1]
|
||
count = count + 1
|
||
writestatus(itemlist)
|
||
reload_required()
|
||
return 100
|
||
|
||
|
||
def install(snackScreen, size, isInstalled):
|
||
out = ""
|
||
if isInstalled:
|
||
if subprocess.call(["byobu-launcher-uninstall"]) == 0:
|
||
out = _("Byobu will not be launched next time you login.")
|
||
button = messagebox(snackScreen, 60, 2, _("Message"), out, buttons=((_("Menu"), )))
|
||
return 101
|
||
else:
|
||
if subprocess.call(["byobu-launcher-install"]) == 0:
|
||
out = _("Byobu will be launched automatically next time you login.")
|
||
button = messagebox(snackScreen, 60, 2, "Message", out, buttons=((_("Menu"), )))
|
||
return 100
|
||
|
||
|
||
def appendtofile(p, s):
|
||
f = open(p, 'a')
|
||
try:
|
||
f.write(s)
|
||
except IOError:
|
||
f.close()
|
||
return
|
||
f.close()
|
||
return
|
||
|
||
|
||
def getesckey():
|
||
line = ""
|
||
if BYOBU_BACKEND == "tmux":
|
||
path = BYOBU_CONFIG_DIR + '/keybindings.tmux'
|
||
if os.path.exists(path):
|
||
for l in open(path):
|
||
if l.startswith("set -g prefix "):
|
||
line = l
|
||
else:
|
||
return DEF_ESC
|
||
else:
|
||
path = BYOBU_CONFIG_DIR + '/keybindings'
|
||
if os.path.exists(path):
|
||
for l in open(path):
|
||
if l.startswith("escape "):
|
||
line = l
|
||
else:
|
||
return DEF_ESC
|
||
if line == "":
|
||
return DEF_ESC
|
||
esc = line[line.find('^') + 1]
|
||
if esc == "`":
|
||
esc = " "
|
||
return esc
|
||
|
||
|
||
def setesckey(key):
|
||
if key.isalpha():
|
||
# throw away outputs in order that the view isn't broken
|
||
nullf = open(os.devnull, "w")
|
||
subprocess.call(["byobu-ctrl-a", "screen", key], stdout=nullf)
|
||
nullf.close()
|
||
|
||
|
||
def chgesc(snackScreen, size):
|
||
esc = Entry(2, text=getesckey(), returnExit=1)
|
||
escl = Label(_("Escape key: ctrl-"))
|
||
bb = ButtonBar(snackScreen, ((_("Apply"), "apply"), (_("Cancel"), "cancel", ESC)), compact=1)
|
||
g = GridForm(snackScreen, _("Change escape sequence"), 2, 4)
|
||
g.add(escl, 0, 0, anchorLeft=1, padding=(1, 0, 0, 1))
|
||
g.add(esc, 1, 0, anchorLeft=1)
|
||
g.add(bb, 1, 1)
|
||
g.setTimer(100)
|
||
loop = 1
|
||
while loop:
|
||
which = g.run()
|
||
if which == "TIMER":
|
||
val = esc.value()
|
||
if len(val) > 1:
|
||
esc.set(val[1])
|
||
# Ensure that escape sequence is not \ or /
|
||
if val == '/' or val == '\\':
|
||
esc.set(DEF_ESC)
|
||
# Ensure that the escape sequence is not set to a number
|
||
try:
|
||
dummy = int(esc.value())
|
||
esc.set(DEF_ESC)
|
||
except Exception:
|
||
# do nothing
|
||
dummy = "foo"
|
||
else:
|
||
loop = 0
|
||
snackScreen.popWindow()
|
||
button = bb.buttonPressed(which)
|
||
if button != "cancel":
|
||
setesckey(esc.value())
|
||
reload_required()
|
||
if button == "exit":
|
||
return 0
|
||
return 100
|
||
|
||
|
||
def autolaunch():
|
||
if os.path.exists(BYOBU_CONFIG_DIR + "/disable-autolaunch"):
|
||
return 0
|
||
try:
|
||
for line in open("%s/.profile" % HOME):
|
||
if "byobu-launch" in line:
|
||
return 1
|
||
except Exception:
|
||
return 0
|
||
if os.path.exists("/etc/profile.d/Z97-%s.sh" % PKG):
|
||
return 1
|
||
return 0
|
||
|
||
|
||
def main():
|
||
"""This is the main loop of our utility"""
|
||
size = terminal_size()
|
||
snackScreen = SnackScreen()
|
||
snackScreen.drawRootText(1, 0, _('Byobu Configuration Menu'))
|
||
snackScreen.pushHelpLine(_('<Tab> between elements | <Enter> selects | <Esc> exits'))
|
||
isInstalled = autolaunch()
|
||
tag = 100
|
||
while tag > 0:
|
||
tag = menu(snackScreen, size, isInstalled)
|
||
if tag == 1:
|
||
tag = help(snackScreen, size)
|
||
elif tag == 2:
|
||
tag = togglestatus(snackScreen, size)
|
||
elif tag == 3:
|
||
tag = chgesc(snackScreen, size)
|
||
elif tag == 4:
|
||
tag = install(snackScreen, size, isInstalled)
|
||
isInstalled = autolaunch()
|
||
snackScreen.finish()
|
||
sys.exit(0)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|