FQDN check for mail

Author: gau1991

ee/cli/plugins/stack.py

@@ -12,6 +12,7 @@
 from ee.core.mysql import EEMysql
 from ee.core.addswap import EESwap
 from ee.core.git import EEGit
+from ee.core.checkfqdn import check_fqdn
 from pynginxconfig import NginxConfig
 from ee.core.services import EEService
 import random
@@ -44,6 +45,8 @@ class Meta:
                 dict(help='Install admin tools stack', action='store_true')),
             (['--mail'],
                 dict(help='Install mail server stack', action='store_true')),
+            (['--mailscanner'],
+                dict(help='Install mail scanner stack', action='store_true')),
             (['--nginx'],
                 dict(help='Install Nginx stack', action='store_true')),
             (['--php'],
@@ -588,6 +591,20 @@ def post_pref(self, apt_packages, packages):
                                 out=ee_amavis)
                 ee_amavis.close()
 
+                # Amavis ViMbadmin configuration
+                if os.path.isfile("/etc/postfix/mysql/virtual_alias_maps.cf"):
+                    vm_host = os.popen("grep hosts /etc/postfix/mysql/virtual_"
+                                       "alias_maps.cf | awk \'{ print $3 }\' |"
+                                       " tr -d '\\n'").read()
+                    vm_pass = os.popen("grep password /etc/postfix/mysql/"
+                                       "virtual_alias_maps.cf | awk \'{ print "
+                                       "$3 }\' | tr -d '\\n'").read()
+
+                    data = dict(host=vm_host, password=vm_pass)
+                    vm_config = open('/etc/amavis/conf.d/50-user', 'w')
+                    self.app.render((data), '50-user.mustache', out=vm_config)
+                    vm_config.close()
+
                 # Amavis postfix configuration
                 EEShellExec.cmd_exec(self, "postconf -e \"content_filter = "
                                      "smtp-amavis:[127.0.0.1]:10024\"")
@@ -948,7 +965,8 @@ def install(self, packages=[], apt_packages=[], disp_msg=True):
                (not self.app.pargs.php) and (not self.app.pargs.mysql) and
                (not self.app.pargs.postfix) and (not self.app.pargs.wpcli) and
                (not self.app.pargs.phpmyadmin) and
-               (not self.app.pargs.adminer) and (not self.app.pargs.utils)):
+               (not self.app.pargs.adminer) and (not self.app.pargs.utils) and
+               (not self.app.pargs.mailscanner)):
                 self.app.pargs.web = True
 
             if self.app.pargs.web:
@@ -975,6 +993,8 @@ def install(self, packages=[], apt_packages=[], disp_msg=True):
                 self.app.pargs.postfix = True
 
                 if not EEAptGet.is_installed(self, 'dovecot-core'):
+                    check_fqdn(self,
+                               os.popen("hostname -f | tr -d '\n'").read())
                     Log.debug(self, "Setting apt_packages variable for mail")
                     apt_packages = apt_packages + EEVariables.ee_mail
                     packages = packages + [["https://github.com/opensolutions/"
@@ -990,8 +1010,11 @@ def install(self, packages=[], apt_packages=[], disp_msg=True):
                                             "Roundcube"]]
 
                     if EEVariables.ee_ram > 1024:
-                        apt_packages = (apt_packages +
-                                        EEVariables.ee_mailscanner)
+                        self.app.pargs.mailscanner = True
+                    else:
+                        Log.info(self, "System RAM is less than 1GB\nMail "
+                                 "scanner packages are not going to install"
+                                 " automatically")
                 else:
                     Log.info(self, "Mail server is already installed")
 
@@ -1050,6 +1073,9 @@ def install(self, packages=[], apt_packages=[], disp_msg=True):
                                         "htdocs/db/adminer/index.php",
                                         "Adminer"]]
 
+            if self.app.pargs.mailscanner:
+                apt_packages = (apt_packages + EEVariables.ee_mailscanner)
+
             if self.app.pargs.utils:
                 Log.debug(self, "Setting packages variable for utils")
                 packages = packages + [["http://phpmemcacheadmin.googlecode"
@@ -1149,6 +1175,9 @@ def remove(self):
                 EEMysql.execute(self, "drop database IF EXISTS vimbadmin")
                 EEMysql.execute(self, "drop database IF EXISTS roundcubemail")
 
+        if self.app.pargs.mailscanner:
+            apt_packages = (apt_packages + EEVariables.ee_mailscanner)
+
         if self.app.pargs.nginx:
             Log.debug(self, "Removing apt_packages variable of Nginx")
             apt_packages = apt_packages + EEVariables.ee_nginx
@@ -1225,6 +1254,9 @@ def purge(self):
                 EEMysql.execute(self, "drop database IF EXISTS vimbadmin")
                 EEMysql.execute(self, "drop database IF EXISTS roundcubemail")
 
+        if self.app.pargs.mailscanner:
+            apt_packages = (apt_packages + EEVariables.ee_mailscanner)
+
         if self.app.pargs.nginx:
             Log.debug(self, "Purge apt_packages variable of Nginx")
             apt_packages = apt_packages + EEVariables.ee_nginx

ee/cli/templates/50-user.mustache

@@ -8,7 +8,7 @@ $final_spam_destiny  = D_PASS;
 
 # We need to provide list of domains for which filtering need to be done
 @lookup_sql_dsn = (
-['DBI:mysql:database=vimbadmin;host=127.0.0.1;port=3306',
+['DBI:mysql:database=vimbadmin;host={{host}};port=3306',
 'vimbadmin',
 '{{password}}']);
 

ee/core/checkfqdn.py

@@ -0,0 +1,22 @@
+from ee.core.shellexec import EEShellExec
+from ee.core.variables import EEVariables
+import os
+
+
+def check_fqdn(self, ee_host):
+    #ee_host=os.popen("hostname -f | tr -d '\n'").read()
+    if '.' in ee_host:
+        EEVariables.ee_fqdn = ee_host
+        with open('/etc/hostname', 'w') as hostfile:
+            hostfile.write(ee_host)
+
+        EEShellExec.cmd_exec(self, "sed -i \"1i\\127.0.0.1 {0}\" /etc/hosts"
+                                   .format(ee_host))
+        if EEVariables.ee_platform_distro == 'debian':
+            EEShellExec.cmd_exec(self, "/etc/init.d/hostname.sh start")
+        else:
+            EEShellExec.cmd_exec(self, "service hostname restart")
+
+    else:
+        ee_host = input("Enter hostname [fqdn]:")
+        check_fqdn(self, ee_host)