117 lines
4.0 KiB
Python
117 lines
4.0 KiB
Python
"""Amavis related functions."""
|
|
|
|
import os
|
|
|
|
from .. import package
|
|
from .. import utils
|
|
|
|
from . import base
|
|
from . import backup, install
|
|
|
|
|
|
class Amavis(base.Installer):
|
|
|
|
"""Amavis installer."""
|
|
|
|
appname = "amavis"
|
|
packages = {
|
|
"deb": [
|
|
"libdbi-perl", "amavisd-new", "arc", "arj", "cabextract",
|
|
"liblz4-tool", "lrzip", "lzop", "p7zip-full", "rpm2cpio",
|
|
"unrar-free",
|
|
],
|
|
"rpm": [
|
|
"amavisd-new", "arj", "lz4", "lzop", "p7zip",
|
|
],
|
|
}
|
|
with_db = True
|
|
|
|
@property
|
|
def config_dir(self):
|
|
"""Return appropriate config dir."""
|
|
if package.backend.FORMAT == "rpm":
|
|
return "/etc/amavisd"
|
|
return "/etc/amavis"
|
|
|
|
def get_daemon_name(self):
|
|
"""Return appropriate daemon name."""
|
|
if package.backend.FORMAT == "rpm":
|
|
return "amavisd"
|
|
return "amavis"
|
|
|
|
def get_config_files(self):
|
|
"""Return appropriate config files."""
|
|
if package.backend.FORMAT == "deb":
|
|
return [
|
|
"conf.d/05-node_id", "conf.d/15-content_filter_mode",
|
|
"conf.d/50-user"]
|
|
return ["amavisd.conf"]
|
|
|
|
def get_packages(self):
|
|
"""Additional packages."""
|
|
packages = super(Amavis, self).get_packages()
|
|
if package.backend.FORMAT == "deb":
|
|
db_driver = "pg" if self.db_driver == "pgsql" else self.db_driver
|
|
return packages + ["libdbd-{}-perl".format(db_driver)]
|
|
if self.db_driver == "pgsql":
|
|
db_driver = "Pg"
|
|
elif self.db_driver == "mysql":
|
|
db_driver = "MySQL"
|
|
else:
|
|
raise NotImplementedError("DB driver not supported")
|
|
packages += ["perl-DBD-{}".format(db_driver)]
|
|
name, version = utils.dist_info()
|
|
if version.startswith('7'):
|
|
packages += ["cabextract", "lrzip", "unar", "unzoo"]
|
|
elif version.startswith('8'):
|
|
packages += ["perl-IO-stringy"]
|
|
return packages
|
|
|
|
def get_sql_schema_path(self):
|
|
"""Return schema path."""
|
|
version = package.backend.get_installed_version("amavisd-new")
|
|
if version is None:
|
|
# Fallback to amavis...
|
|
version = package.backend.get_installed_version("amavis")
|
|
if version is None:
|
|
raise utils.FatalError("Amavis is not installed")
|
|
path = self.get_file_path(
|
|
"amavis_{}_{}.sql".format(self.dbengine, version))
|
|
if not os.path.exists(path):
|
|
version = ".".join(version.split(".")[:-1]) + ".X"
|
|
path = self.get_file_path(
|
|
"amavis_{}_{}.sql".format(self.dbengine, version))
|
|
if not os.path.exists(path):
|
|
raise utils.FatalError("Failed to find amavis database schema")
|
|
return path
|
|
|
|
def pre_run(self):
|
|
"""Tasks to run first."""
|
|
with open("/etc/mailname", "w") as fp:
|
|
fp.write("{}\n".format(self.config.get("general", "hostname")))
|
|
|
|
def post_run(self):
|
|
"""Additional tasks."""
|
|
install("spamassassin", self.config, self.upgrade, self.restore)
|
|
install("clamav", self.config, self.upgrade, self.restore)
|
|
|
|
def custom_backup(self, path):
|
|
"""Backup custom configuration if any."""
|
|
if package.backend.FORMAT == "deb":
|
|
amavis_custom = f"{self.config_dir}/conf.d/99-custom"
|
|
if os.path.isfile(amavis_custom):
|
|
utils.copy_file(amavis_custom, path)
|
|
utils.success("Amavis custom configuration saved!")
|
|
backup("spamassassin", self.config, os.path.dirname(path))
|
|
|
|
def restore(self):
|
|
"""Restore custom config files."""
|
|
if package.backend.FORMAT != "deb":
|
|
return
|
|
amavis_custom_configuration = os.path.join(
|
|
self.restore, "custom/99-custom")
|
|
if os.path.isfile(amavis_custom_configuration):
|
|
utils.copy_file(amavis_custom_configuration, os.path.join(
|
|
self.config_dir, "conf.d"))
|
|
utils.success("Custom amavis configuration restored.")
|