362 lines
13 KiB
Python
362 lines
13 KiB
Python
#!/usr/bin/python2
|
|
#
|
|
# The Qubes OS Project, http://www.qubes-os.org
|
|
#
|
|
# Copyright (C) 2010 Joanna Rutkowska <joanna@invisiblethingslab.com>
|
|
# Copyright (C) 2013 Marek Marczykowski <marmarek@invisiblethingslab.com>
|
|
#
|
|
# 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; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# 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, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
#
|
|
|
|
import subprocess
|
|
import sys
|
|
import os
|
|
import os.path
|
|
import shutil
|
|
import dbus
|
|
|
|
from qubes.qubes import QubesVm, QubesHVm
|
|
from qubes.qubes import QubesException, QubesHost, QubesVmLabels
|
|
from qubes.qubes import vm_files, system_path, dry_run
|
|
|
|
import qubes.imgconverter
|
|
|
|
vm_files['appmenus_templates_subdir'] = 'apps.templates'
|
|
vm_files['appmenus_template_icons_subdir'] = 'apps.tempicons'
|
|
vm_files['appmenus_icons_subdir'] = 'apps.icons'
|
|
vm_files['appmenus_template_templates_subdir'] = 'apps-template.templates'
|
|
vm_files['appmenus_whitelist'] = 'whitelisted-appmenus.list'
|
|
|
|
system_path['appmenu_start_hvm_template'] = \
|
|
'/usr/share/qubes-appmenus/qubes-start.desktop'
|
|
system_path['appmenu_create_cmd'] = \
|
|
'/usr/libexec/qubes-appmenus/create-apps-for-appvm.sh'
|
|
system_path['appmenu_remove_cmd'] = \
|
|
'/usr/libexec/qubes-appmenus/remove-appvm-appmenus.sh'
|
|
|
|
|
|
def QubesVm_get_attrs_config(self, attrs):
|
|
attrs["appmenus_templates_dir"] = {
|
|
"func": lambda x:
|
|
os.path.join(self.dir_path, vm_files["appmenus_templates_subdir"])
|
|
if self.updateable else
|
|
(self.template.appmenus_templates_dir if self.template is not None
|
|
else None)
|
|
}
|
|
attrs["appmenus_template_icons_dir"] = {
|
|
"func": lambda x:
|
|
os.path.join(self.dir_path,
|
|
vm_files["appmenus_template_icons_subdir"])
|
|
if self.updateable else
|
|
(self.template.appmenus_template_icons_dir
|
|
if self.template is not None else None)
|
|
}
|
|
attrs["appmenus_icons_dir"] = {
|
|
"func": lambda x:
|
|
os.path.join(self.dir_path, vm_files["appmenus_icons_subdir"])}
|
|
return attrs
|
|
|
|
|
|
def QubesTemplateVm_get_attrs_config(self, attrs):
|
|
attrs['appmenus_templates_dir'] = {
|
|
'func': lambda x:
|
|
os.path.join(self.dir_path, vm_files["appmenus_templates_subdir"])}
|
|
attrs['appmenus_template_icons_dir'] = {
|
|
'func': lambda x:
|
|
os.path.join(self.dir_path,
|
|
vm_files["appmenus_template_icons_subdir"])}
|
|
return attrs
|
|
|
|
|
|
def QubesVm_appmenus_create(self, verbose=False, source_template=None):
|
|
if source_template is None:
|
|
source_template = self.template
|
|
|
|
if self.internal:
|
|
return
|
|
if self.is_disposablevm():
|
|
return
|
|
|
|
if self.is_netvm():
|
|
vmtype = 'servicevms'
|
|
elif self.is_template():
|
|
vmtype = 'vm-templates'
|
|
else:
|
|
vmtype = 'appvms'
|
|
|
|
try:
|
|
msgoutput = None if verbose else open(os.devnull, 'w')
|
|
if source_template is not None:
|
|
subprocess.check_call([system_path["appmenu_create_cmd"],
|
|
source_template.appmenus_templates_dir,
|
|
self.name, vmtype, self.label.icon],
|
|
stdout=msgoutput, stderr=msgoutput)
|
|
elif self.appmenus_templates_dir is not None:
|
|
subprocess.check_call([system_path["appmenu_create_cmd"],
|
|
self.appmenus_templates_dir, self.name,
|
|
vmtype, self.label.icon],
|
|
stdout=msgoutput, stderr=msgoutput)
|
|
else:
|
|
# Only add apps to menu
|
|
subprocess.check_call([system_path["appmenu_create_cmd"],
|
|
"none", self.name, vmtype,
|
|
self.label.icon],
|
|
stdout=msgoutput, stderr=msgoutput)
|
|
except subprocess.CalledProcessError:
|
|
print >> sys.stderr, "Ooops, there was a problem creating appmenus " \
|
|
"for {0} VM!".format(self.name)
|
|
|
|
|
|
def QubesVm_appmenus_remove(self):
|
|
if self.is_netvm():
|
|
vmtype = 'servicevms'
|
|
elif self.is_template():
|
|
vmtype = 'vm-templates'
|
|
else:
|
|
vmtype = 'appvms'
|
|
subprocess.check_call([system_path["appmenu_remove_cmd"], self.name,
|
|
vmtype], stderr=open(os.devnull, 'w'))
|
|
|
|
|
|
def QubesVm_appicons_create(self, srcdir=None):
|
|
if srcdir is None:
|
|
srcdir = self.appmenus_template_icons_dir
|
|
if srcdir is None:
|
|
return
|
|
if not os.path.exists(srcdir):
|
|
return
|
|
|
|
if self.internal:
|
|
return
|
|
if self.is_disposablevm():
|
|
return
|
|
|
|
whitelist = os.path.join(self.dir_path, vm_files['appmenus_whitelist'])
|
|
if os.path.exists(whitelist):
|
|
whitelist = [line.strip() for line in open(whitelist)]
|
|
else:
|
|
whitelist = None
|
|
|
|
if not os.path.exists(self.appmenus_icons_dir):
|
|
os.mkdir(self.appmenus_icons_dir)
|
|
elif not os.path.isdir(self.appmenus_icons_dir):
|
|
os.unlink(self.appmenus_icons_dir)
|
|
os.mkdir(self.appmenus_icons_dir)
|
|
|
|
for icon in os.listdir(srcdir):
|
|
desktop = os.path.splitext(icon)[0] + '.desktop'
|
|
if whitelist and desktop not in whitelist:
|
|
continue
|
|
|
|
src_icon = os.path.join(srcdir, icon)
|
|
dst_icon = os.path.join(self.appmenus_icons_dir, icon)
|
|
if not os.path.exists(dst_icon) or \
|
|
os.path.getmtime(src_icon) > os.path.getmtime(dst_icon):
|
|
qubes.imgconverter.tint(src_icon, dst_icon, self.label.color)
|
|
|
|
|
|
def QubesVm_appicons_remove(self):
|
|
if not os.path.exists(self.appmenus_icons_dir):
|
|
return
|
|
for icon in os.listdir(self.appmenus_icons_dir):
|
|
os.unlink(os.path.join(self.appmenus_icons_dir, icon))
|
|
|
|
|
|
def QubesVm_appicons_cleanup(self):
|
|
srcdir = self.appmenus_template_icons_dir
|
|
if srcdir is None:
|
|
return
|
|
if not os.path.exists(srcdir):
|
|
return
|
|
if not os.path.exists(self.appmenus_icons_dir):
|
|
return
|
|
|
|
for icon in os.listdir(self.appmenus_icons_dir):
|
|
if not os.path.exists(os.path.join(srcdir, icon)):
|
|
os.unlink(os.path.join(self.appmenus_icons_dir, icon))
|
|
|
|
|
|
def QubesVm_pre_rename(self, new_name):
|
|
self.appmenus_remove()
|
|
|
|
|
|
def QubesVm_post_rename(self, old_name):
|
|
old_dirpath = os.path.join(os.path.dirname(self.dir_path), old_name)
|
|
if self.appmenus_templates_dir is not None:
|
|
self.appmenus_templates_dir = self.appmenus_templates_dir.replace(
|
|
old_dirpath, self.dir_path)
|
|
|
|
self.appmenus_create()
|
|
|
|
|
|
def QubesVm_create_on_disk(self, verbose, source_template):
|
|
if isinstance(self, QubesHVm) and source_template is None:
|
|
if verbose:
|
|
print >> sys.stderr, "--> Creating appmenus directory: {0}".format(
|
|
self.appmenus_templates_dir)
|
|
os.mkdir(self.appmenus_templates_dir)
|
|
shutil.copy(system_path["appmenu_start_hvm_template"],
|
|
self.appmenus_templates_dir)
|
|
|
|
source_whitelist_filename = 'vm-' + vm_files["appmenus_whitelist"]
|
|
if self.is_netvm():
|
|
source_whitelist_filename = 'netvm-' + vm_files["appmenus_whitelist"]
|
|
if source_template and os.path.exists(
|
|
os.path.join(source_template.dir_path, source_whitelist_filename)):
|
|
if verbose:
|
|
print >> sys.stderr, "--> Creating default whitelisted apps list: {0}". \
|
|
format(self.dir_path + '/' + vm_files["whitelisted_appmenus"])
|
|
shutil.copy(
|
|
os.path.join(source_template.dir_path, source_whitelist_filename),
|
|
os.path.join(self.dir_path, vm_files["whitelisted_appmenus"]))
|
|
|
|
if source_template and self.updateable:
|
|
if verbose:
|
|
print >> sys.stderr, "--> Copying the template's appmenus templates dir:\n{0} ==>\n{1}". \
|
|
format(source_template.appmenus_templates_dir,
|
|
self.appmenus_templates_dir)
|
|
if os.path.isdir(source_template.appmenus_templates_dir):
|
|
shutil.copytree(source_template.appmenus_templates_dir,
|
|
self.appmenus_templates_dir)
|
|
else:
|
|
os.mkdir(self.appmenus_templates_dir)
|
|
if os.path.isdir(source_template.appmenus_template_icons_dir):
|
|
shutil.copytree(source_template.appmenus_template_icons_dir,
|
|
self.appmenus_template_icons_dir)
|
|
else:
|
|
os.mkdir(self.appmenus_template_icons_dir)
|
|
|
|
# Create appmenus
|
|
self.appicons_create()
|
|
self.appmenus_create(verbose=verbose)
|
|
|
|
|
|
def QubesVm_clone_disk_files(self, src_vm, verbose):
|
|
if src_vm.updateable and src_vm.appmenus_templates_dir is not None and \
|
|
self.appmenus_templates_dir is not None:
|
|
if verbose:
|
|
print >> sys.stderr, "--> Copying the template's appmenus templates dir:\n{0} ==>\n{1}". \
|
|
format(src_vm.appmenus_templates_dir,
|
|
self.appmenus_templates_dir)
|
|
shutil.copytree(src_vm.appmenus_templates_dir,
|
|
self.appmenus_templates_dir)
|
|
|
|
if src_vm.updateable and src_vm.appmenus_template_icons_dir is not None \
|
|
and self.appmenus_template_icons_dir is not None and \
|
|
os.path.isdir(src_vm.appmenus_template_icons_dir):
|
|
if verbose:
|
|
print >> sys.stderr, "--> Copying the template's appmenus " \
|
|
"template icons dir:\n{0} ==>\n{1}". \
|
|
format(src_vm.appmenus_template_icons_dir,
|
|
self.appmenus_template_icons_dir)
|
|
shutil.copytree(src_vm.appmenus_template_icons_dir,
|
|
self.appmenus_template_icons_dir)
|
|
|
|
for whitelist in (
|
|
vm_files["appmenus_whitelist"],
|
|
'vm-' + vm_files["appmenus_whitelist"],
|
|
'netvm-' + vm_files["appmenus_whitelist"]):
|
|
if os.path.exists(os.path.join(src_vm.dir_path, whitelist)):
|
|
if verbose:
|
|
print >> sys.stderr, "--> Copying whitelisted apps list: {0}". \
|
|
format(whitelist)
|
|
shutil.copy(os.path.join(src_vm.dir_path, whitelist),
|
|
os.path.join(self.dir_path, whitelist))
|
|
|
|
# Create appmenus
|
|
self.appicons_create()
|
|
self.appmenus_create(verbose=verbose)
|
|
|
|
|
|
def QubesVm_remove_from_disk(self):
|
|
self.appmenus_remove()
|
|
|
|
|
|
def QubesVm_label_setter(self, _):
|
|
self.appicons_create()
|
|
|
|
# Apparently desktop environments heavily caches the icons,
|
|
# see #751 for details
|
|
if os.environ.get("DESKTOP_SESSION", "") == "kde-plasma":
|
|
try:
|
|
os.unlink(os.path.expandvars(
|
|
"$HOME/.kde/cache-$HOSTNAME/icon-cache.kcache"))
|
|
except:
|
|
pass
|
|
try:
|
|
notify_object = dbus.SessionBus().get_object(
|
|
"org.freedesktop.Notifications",
|
|
"/org/freedesktop/Notifications")
|
|
notify_object.Notify(
|
|
"Qubes", 0, self.label.icon, "Qubes",
|
|
"You will need to log off and log in again for the VM icons "
|
|
"to update in the KDE launcher menu",
|
|
[], [], 10000,
|
|
dbus_interface="org.freedesktop.Notifications")
|
|
except:
|
|
pass
|
|
elif os.environ.get("DESKTOP_SESSION", "") == "xfce":
|
|
self.appmenus_remove()
|
|
self.appmenus_create()
|
|
|
|
|
|
def QubesVm_appmenus_recreate(self):
|
|
"""
|
|
Force recreation of all appmenus and icons. For example when VM label
|
|
color was changed
|
|
"""
|
|
self.appmenus_remove()
|
|
self.appicons_remove()
|
|
self.appicons_create()
|
|
self.appmenus_create()
|
|
|
|
|
|
def QubesVm_appmenus_update(self):
|
|
"""
|
|
Similar to appmenus_recreate, but do not touch unchanged files
|
|
"""
|
|
self.appmenus_remove()
|
|
self.appicons_create()
|
|
self.appicons_cleanup()
|
|
self.appmenus_create()
|
|
|
|
|
|
def QubesVm_set_attr(self, name, newvalue, oldvalue):
|
|
if name == 'internal':
|
|
if newvalue and not oldvalue:
|
|
self.appmenus_remove()
|
|
elif not newvalue and oldvalue:
|
|
self.appmenus_create()
|
|
|
|
# new methods
|
|
QubesVm.appmenus_create = QubesVm_appmenus_create
|
|
QubesVm.appmenus_remove = QubesVm_appmenus_remove
|
|
QubesVm.appmenus_recreate = QubesVm_appmenus_recreate
|
|
QubesVm.appmenus_update = QubesVm_appmenus_update
|
|
QubesVm.appicons_create = QubesVm_appicons_create
|
|
QubesVm.appicons_cleanup = QubesVm_appicons_cleanup
|
|
QubesVm.appicons_remove = QubesVm_appicons_remove
|
|
|
|
# hooks for existing methods
|
|
QubesVm.hooks_get_attrs_config.append(QubesVm_get_attrs_config)
|
|
QubesVm.hooks_pre_rename.append(QubesVm_pre_rename)
|
|
QubesVm.hooks_post_rename.append(QubesVm_post_rename)
|
|
QubesVm.hooks_create_on_disk.append(QubesVm_create_on_disk)
|
|
QubesVm.hooks_clone_disk_files.append(QubesVm_clone_disk_files)
|
|
QubesVm.hooks_remove_from_disk.append(QubesVm_remove_from_disk)
|
|
QubesVm.hooks_label_setter.append(QubesVm_label_setter)
|
|
QubesVm.hooks_set_attr.append(QubesVm_set_attr)
|