170 lines
5.0 KiB
Python
170 lines
5.0 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# pyinotifyd 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 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# pyinotifyd 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 pyinotifyd. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
import filecmp
|
|
import logging
|
|
import os
|
|
import shutil
|
|
import sys
|
|
|
|
|
|
SYSTEMD_PATH = "/lib/systemd/system"
|
|
OPENRC = "/sbin/openrc"
|
|
|
|
|
|
def _systemd_files(pkg_dir, name):
|
|
return [
|
|
(f"{pkg_dir}/misc/systemd/{name}-milter.service",
|
|
f"{SYSTEMD_PATH}/{name}-milter.service", True)]
|
|
|
|
|
|
def _openrc_files(pkg_dir, name):
|
|
return [
|
|
(f"{pkg_dir}/misc/openrc/{name}-milter.initd", f"/etc/init.d/{name}-milter", True),
|
|
(f"{pkg_dir}/misc/openrc/{name}-milter.confd", f"/etc/conf.d/{name}-milter", False)]
|
|
|
|
|
|
def _config_files(pkg_dir, name):
|
|
return [
|
|
(f"{pkg_dir}/misc/{name}.conf.default", f"/etc/{name}/{name}.conf.default", False),
|
|
(f"{pkg_dir}/misc/templates/removed.png", f"/etc/{name}/templates/removed.png", False),
|
|
(f"{pkg_dir}/misc/templates/disclaimer_html.template", f"/etc/{name}/templates/disclaimer_html.template", False),
|
|
(f"{pkg_dir}/misc/templates/disclaimer_text.template", f"/etc/{name}/templates/disclaimer_text.template", False),
|
|
(f"{pkg_dir}/misc/templates/notification.template", f"/etc/{name}/templates/notification.template", False)]
|
|
|
|
|
|
def _install_files(files):
|
|
for src, dst, force in files:
|
|
if os.path.exists(dst):
|
|
if os.path.isdir(dst):
|
|
logging.error(
|
|
" => unable to copy file, destination path is a directory")
|
|
continue
|
|
elif not force:
|
|
logging.info(f" => file {dst} already exists")
|
|
continue
|
|
|
|
try:
|
|
logging.info(f" => install file {dst}")
|
|
shutil.copy2(src, dst)
|
|
except Exception as e:
|
|
logging.error(f" => unable to install file {dst}: {e}")
|
|
|
|
|
|
def _uninstall_files(files):
|
|
for src, dst, force in files:
|
|
if not os.path.isfile(dst):
|
|
continue
|
|
|
|
if not force and not filecmp.cmp(src, dst, shallow=True):
|
|
logging.warning(
|
|
f" => keep modified file {dst}, "
|
|
f"you have to remove it manually")
|
|
continue
|
|
|
|
try:
|
|
logging.info(f" => uninstall file {dst}")
|
|
os.remove(dst)
|
|
except Exception as e:
|
|
logging.error(f" => unable to uninstall file {dst}: {e}")
|
|
|
|
|
|
def _create_dir(path):
|
|
if os.path.isdir(path):
|
|
logging.info(f" => directory {path} already exists")
|
|
else:
|
|
try:
|
|
logging.info(f" => create directory {path}")
|
|
os.mkdir(path)
|
|
except Exception as e:
|
|
logging.error(f" => unable to create directory {path}: {e}")
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
def _delete_dir(path):
|
|
if os.path.isdir(path):
|
|
if not os.listdir(path):
|
|
try:
|
|
logging.info(f" => delete directory {path}")
|
|
os.rmdir(path)
|
|
except Exception as e:
|
|
logging.error(f" => unable to delete directory {path}: {e}")
|
|
else:
|
|
logging.warning(f" => keep non-empty directory {path}")
|
|
|
|
|
|
def _check_root():
|
|
if os.getuid() != 0:
|
|
logging.error("you need to have root privileges, please try again")
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
def _check_systemd():
|
|
systemd = os.path.isdir(SYSTEMD_PATH)
|
|
if systemd:
|
|
logging.info("systemd detected")
|
|
|
|
return systemd
|
|
|
|
|
|
def _check_openrc():
|
|
openrc = os.path.isfile(OPENRC) and os.access(OPENRC, os.X_OK)
|
|
if openrc:
|
|
logging.info("openrc detected")
|
|
|
|
return openrc
|
|
|
|
|
|
def install(name):
|
|
if not _check_root():
|
|
sys.exit(2)
|
|
|
|
pkg_dir = os.path.dirname(__file__)
|
|
|
|
if _check_systemd():
|
|
_install_files(_systemd_files(pkg_dir, name))
|
|
|
|
if _check_openrc():
|
|
_install_files(_openrc_files(pkg_dir, name))
|
|
|
|
for d in [f"/etc/{name}", f"/etc/{name}/templates"]:
|
|
if not _create_dir(d):
|
|
logging.error(" => unable to create config dir, giving up ...")
|
|
sys.exit(3)
|
|
_install_files(_config_files(pkg_dir, name))
|
|
|
|
logging.info(f"{name} successfully installed")
|
|
|
|
|
|
def uninstall(name):
|
|
if not _check_root():
|
|
sys.exit(2)
|
|
|
|
pkg_dir = os.path.dirname(__file__)
|
|
|
|
_uninstall_files(_systemd_files(pkg_dir, name))
|
|
_uninstall_files(_openrc_files(pkg_dir, name))
|
|
_uninstall_files(_config_files(pkg_dir, name))
|
|
|
|
_delete_dir(f"/etc/{name}/templates")
|
|
_delete_dir(f"/etc/{name}")
|
|
|
|
logging.info(f"{name} successfully uninstalled")
|