From 80eaf741bb89cf5a87bfde39b5a968de5e2dcb5f Mon Sep 17 00:00:00 2001
From: Usman Nasir <usman@cyberpersons.com>
Date: Tue, 18 Feb 2020 20:15:07 +0500
Subject: [PATCH] make mail child domain optional

---
 plogical/virtualHostUtilities.py              | 86 ++++++++++---------
 .../websiteFunctions/websiteFunctions.js      | 12 ++-
 .../websiteFunctions/createWebsite.html       |  9 ++
 websiteFunctions/website.py                   |  9 +-
 4 files changed, 71 insertions(+), 45 deletions(-)

diff --git a/plogical/virtualHostUtilities.py b/plogical/virtualHostUtilities.py
index b9666d9e4..654df45bc 100755
--- a/plogical/virtualHostUtilities.py
+++ b/plogical/virtualHostUtilities.py
@@ -58,7 +58,7 @@ class virtualHostUtilities:
     @staticmethod
     def createVirtualHost(virtualHostName, administratorEmail, phpVersion, virtualHostUser, ssl,
                           dkimCheck, openBasedir, websiteOwner, packageName, apache,
-                          tempStatusPath='/home/cyberpanel/fakePath'):
+                          tempStatusPath='/home/cyberpanel/fakePath', mailDomain = None):
         try:
 
             logging.CyberCPLogFileWriter.statusWriter(tempStatusPath, 'Running some checks..,0')
@@ -206,56 +206,57 @@ class virtualHostUtilities:
 
             ### For autodiscover of mail clients.
 
-            logging.CyberCPLogFileWriter.statusWriter(tempStatusPath, 'Creating mail child domain..,80')
-            childDomain = 'mail.%s' % (virtualHostName)
-            childPath = '/home/%s/public_html/%s' % (virtualHostName, childDomain)
+            if mailDomain:
+                logging.CyberCPLogFileWriter.statusWriter(tempStatusPath, 'Creating mail child domain..,80')
+                childDomain = 'mail.%s' % (virtualHostName)
+                childPath = '/home/%s/public_html/%s' % (virtualHostName, childDomain)
 
-            virtualHostUtilities.createDomain(virtualHostName, childDomain, 'PHP 7.2', childPath, 1, 0, 0, admin.userName, 0, "/home/cyberpanel/" + str(randint(1000, 9999)))
+                virtualHostUtilities.createDomain(virtualHostName, childDomain, 'PHP 7.2', childPath, 1, 0, 0, admin.userName, 0, "/home/cyberpanel/" + str(randint(1000, 9999)))
 
-            ## update dovecot conf to enable auto-discover
+                ## update dovecot conf to enable auto-discover
 
-            dovecotPath = '/etc/dovecot/dovecot.conf'
+                dovecotPath = '/etc/dovecot/dovecot.conf'
 
-            if os.path.exists(dovecotPath):
-                dovecotContent = open(dovecotPath, 'r').read()
+                if os.path.exists(dovecotPath):
+                    dovecotContent = open(dovecotPath, 'r').read()
 
-                if dovecotContent.find(childDomain) == -1:
-                    content = """\nlocal_name %s {
-      ssl_cert = </etc/letsencrypt/live/%s/fullchain.pem
-      ssl_key = </etc/letsencrypt/live/%s/privkey.pem
-    }\n""" % (childDomain, childDomain, childDomain)
+                    if dovecotContent.find(childDomain) == -1:
+                        content = """\nlocal_name %s {
+          ssl_cert = </etc/letsencrypt/live/%s/fullchain.pem
+          ssl_key = </etc/letsencrypt/live/%s/privkey.pem
+        }\n""" % (childDomain, childDomain, childDomain)
 
-                    writeToFile = open(dovecotPath, 'a')
-                    writeToFile.write(content)
+                        writeToFile = open(dovecotPath, 'a')
+                        writeToFile.write(content)
+                        writeToFile.close()
+
+                        command = 'systemctl restart dovecot'
+                        ProcessUtilities.executioner(command)
+
+                    ### Update postfix configurations
+
+                    postFixPath = '/etc/postfix/main.cf'
+
+                    postFixContent = open(postFixPath, 'r').read()
+
+                    if postFixContent.find('tls_server_sni_maps') == -1:
+                        writeToFile = open(postFixPath, 'a')
+                        writeToFile.write('\ntls_server_sni_maps = hash:/etc/postfix/vmail_ssl.map\n')
+                        writeToFile.close()
+
+                    postfixMapFile = '/etc/postfix/vmail_ssl.map'
+
+                    mapContent = '%s /etc/letsencrypt/live/%s/privkey.pem /etc/letsencrypt/live/%s/fullchain.pem\n' % (childDomain, childDomain, childDomain)
+
+                    writeToFile = open(postfixMapFile, 'a')
+                    writeToFile.write(mapContent)
                     writeToFile.close()
 
-                    command = 'systemctl restart dovecot'
+                    command = 'postmap -F hash:/etc/postfix/vmail_ssl.map'
                     ProcessUtilities.executioner(command)
 
-                ### Update postfix configurations
-
-                postFixPath = '/etc/postfix/main.cf'
-
-                postFixContent = open(postFixPath, 'r').read()
-
-                if postFixContent.find('tls_server_sni_maps') == -1:
-                    writeToFile = open(postFixPath, 'a')
-                    writeToFile.write('\ntls_server_sni_maps = hash:/etc/postfix/vmail_ssl.map\n')
-                    writeToFile.close()
-
-                postfixMapFile = '/etc/postfix/vmail_ssl.map'
-
-                mapContent = '%s /etc/letsencrypt/live/%s/privkey.pem /etc/letsencrypt/live/%s/fullchain.pem\n' % (childDomain, childDomain, childDomain)
-
-                writeToFile = open(postfixMapFile, 'a')
-                writeToFile.write(mapContent)
-                writeToFile.close()
-
-                command = 'postmap -F hash:/etc/postfix/vmail_ssl.map'
-                ProcessUtilities.executioner(command)
-
-                command = 'systemctl restart postfix'
-                ProcessUtilities.executioner(command)
+                    command = 'systemctl restart postfix'
+                    ProcessUtilities.executioner(command)
 
             ###
 
@@ -1391,6 +1392,7 @@ def main():
 
     parser.add_argument('--openBasedirValue', help='open_base dir protection value!')
     parser.add_argument('--tempStatusPath', help='Temporary Status file path.')
+    parser.add_argument('--mailDomain', help='To create or not to create mail domain.')
 
     ## Switch Server
 
@@ -1421,7 +1423,7 @@ def main():
 
         virtualHostUtilities.createVirtualHost(args.virtualHostName, args.administratorEmail, args.phpVersion,
                                                args.virtualHostUser, int(args.ssl), dkimCheck, openBasedir,
-                                               args.websiteOwner, args.package, apache, tempStatusPath)
+                                               args.websiteOwner, args.package, apache, tempStatusPath, int(args.mailDomain))
     elif args.function == "deleteVirtualHostConfigurations":
         vhost.deleteVirtualHostConfigurations(args.virtualHostName)
     elif args.function == "createDomain":
diff --git a/websiteFunctions/static/websiteFunctions/websiteFunctions.js b/websiteFunctions/static/websiteFunctions/websiteFunctions.js
index ee016d01d..dc4eea503 100755
--- a/websiteFunctions/static/websiteFunctions/websiteFunctions.js
+++ b/websiteFunctions/static/websiteFunctions/websiteFunctions.js
@@ -45,7 +45,7 @@ app.controller('createWebsite', function ($scope, $http, $timeout, $window) {
 
         $scope.currentStatus = "Starting creation..";
 
-        var ssl, dkimCheck, openBasedir;
+        var ssl, dkimCheck, openBasedir, mailDomain;
 
         if ($scope.sslCheck === true) {
             ssl = 1;
@@ -65,6 +65,13 @@ app.controller('createWebsite', function ($scope, $http, $timeout, $window) {
             openBasedir = 0
         }
 
+        if ($scope.mailDomain === true) {
+            mailDomain = 1;
+        } else {
+            mailDomain = 0
+        }
+
+
         url = "/websites/submitWebsiteCreation";
 
         var package = $scope.packageForWebsite;
@@ -82,7 +89,8 @@ app.controller('createWebsite', function ($scope, $http, $timeout, $window) {
             ssl: ssl,
             websiteOwner: websiteOwner,
             dkimCheck: dkimCheck,
-            openBasedir: openBasedir
+            openBasedir: openBasedir,
+            mailDomain: mailDomain
         };
 
         var config = {
diff --git a/websiteFunctions/templates/websiteFunctions/createWebsite.html b/websiteFunctions/templates/websiteFunctions/createWebsite.html
index 435b2453b..5c6403205 100755
--- a/websiteFunctions/templates/websiteFunctions/createWebsite.html
+++ b/websiteFunctions/templates/websiteFunctions/createWebsite.html
@@ -105,6 +105,15 @@
                                     </label>
                                 </div>
                             </div>
+                            <label class="col-sm-3 control-label"></label>
+                            <div class="col-sm-9">
+                                <div class="checkbox">
+                                    <label>
+                                        <input ng-model="mailDomain" type="checkbox" value="">
+                                        Create Mail Domain
+                                    </label>
+                                </div>
+                            </div>
                         </div>
 
                         <div ng-hide="installationDetailsForm" class="form-group">
diff --git a/websiteFunctions/website.py b/websiteFunctions/website.py
index f7f47abee..6604340e4 100755
--- a/websiteFunctions/website.py
+++ b/websiteFunctions/website.py
@@ -199,6 +199,8 @@ class WebsiteManager:
             except:
                 externalApp = "".join(re.findall("[a-zA-Z]+", domain))[:7]
 
+
+
             try:
                 counter = 0
                 while 1:
@@ -215,6 +217,11 @@ class WebsiteManager:
             except:
                 apacheBackend = "0"
 
+            try:
+                mailDomain = str(data['mailDomain'])
+            except:
+                mailDomain = "1"
+
             import pwd
             counter = 0
             while 1:
@@ -232,7 +239,7 @@ class WebsiteManager:
                        " --administratorEmail " + adminEmail + " --phpVersion '" + phpSelection + \
                        "' --virtualHostUser " + externalApp + " --ssl " + str(data['ssl']) + " --dkimCheck " \
                        + str(data['dkimCheck']) + " --openBasedir " + str(data['openBasedir']) + \
-                       ' --websiteOwner ' + websiteOwner + ' --package ' + packageName + ' --tempStatusPath ' + tempStatusPath + " --apache " + apacheBackend
+                       ' --websiteOwner ' + websiteOwner + ' --package ' + packageName + ' --tempStatusPath ' + tempStatusPath + " --apache " + apacheBackend  + " --mailDomain %s" % (mailDomain)
 
             ProcessUtilities.popenExecutioner(execPath)
             time.sleep(2)