Continued script to set domain (refs #21163).

set_app_domain.py

@@ -64,13 +64,14 @@ class SetAppDomain():
             sys.exit(1)
         if args:
             app = args.pop()
-        # Change domain
+        # Get Nginx conf path and instance user
         instance = None
         if app == 'cm':
             nginx_conf = '/etc/nginx/sites-available/skyreach.conf'
-            self.change_cm_domain(new_domain)
+            key = 'CM_SERVER_NAME'
         elif app == 'mon':
             nginx_conf = '/etc/nginx/sites-available/msmonitor.conf'
+            key = 'MONITOR_SERVER_NAME'
         elif app == 'ms' or app.startswith('ms-'):
             if app.startswith('ms-'):
                 instance = app[3:].strip('. -\t\n')
@@ -80,13 +81,25 @@ class SetAppDomain():
             else:
                 instance = 'msuser'
             nginx_conf = '/etc/nginx/sites-available/mediaserver-%s.conf' % instance
-            self.change_ms_domain(new_domain, instance)
+            key = 'MS_SERVER_NAME'
         else:
             log('Invalid app name requested.')
             log(self.USAGE)
             sys.exit(1)
+        # Change domain
         self.change_nginx_domain(nginx_conf, new_domain)
         self.change_hosts_file(new_domain)
+        self.change_envsetup_conf(key, new_domain)
+        if app == 'cm':
+            warning = self.change_cm_domain(new_domain)
+        elif app == 'mon':
+            warning = None
+        else:
+            warning = self.change_ms_domain(new_domain, instance)
+        log('\033[92mDone\033[0m')
+        if warning:
+            log('\033[93mWarning:\033[0m')
+            log(warning)
         sys.exit(0)
 
     def change_nginx_domain(self, path, new_domain):
@@ -100,10 +113,9 @@ class SetAppDomain():
         changed_lines = 0
         for line in vhost.split('\n'):
             if re.match(r'\s*server_name\s+([\w\-\_\.\ ]+);', line):
-                new_vhost += re.sub(r'server_name\s+([\w\-\_\.\ ]+);', line, 'server_name %s;' % new_domain) + '\n'
+                line = re.sub(r'server_name\s+([\w\-\_\.\ ]+);', 'server_name %s;' % new_domain, line)
                 changed_lines += 1
-            else:
-                new_vhost += line + '\n'
+            new_vhost += line + '\n'
         if changed_lines != 2:
             log('Warning the number of server_name occurence changed in Nginx configuration is not the expected number (2) but is %s.' % changed_lines)
             if not self.force:
@@ -128,6 +140,7 @@ class SetAppDomain():
     def change_hosts_file(self, new_domain):
         with open('/etc/hosts', 'r') as fo:
             hosts = fo.read()
+        ori_hosts = hosts
         if ' ' + new_domain in hosts:
             hosts = hosts.replace(' ' + new_domain, '')
         elif '\t' + new_domain in hosts:
@@ -138,17 +151,39 @@ class SetAppDomain():
             new_hosts += '\n' + line
             if line.startswith('127.0.0.1'):
                 new_hosts += ' ' + new_domain
-        if new_hosts != hosts:
+        if new_hosts != ori_hosts:
             if self.debug:
                 log('New hosts:')
                 log(new_hosts)
             else:
                 with open('/etc/hosts', 'w') as fo:
                     fo.write(new_hosts)
-                log('The hosts file has been update.')
+                log('The "/etc/hosts" file has been update.')
                 utils.run_commands(['service nscd restart'])
         else:
-            log('The hosts file is already up to date.')
+            log('The "/etc/hosts" file is already up to date.')
+
+    def change_envsetup_conf(self, key, new_domain):
+        confs = (
+            'conf.sh',
+            'auto-generated-conf.sh',
+        )
+        replaced = False
+        for path in confs:
+            if os.path.exists(path):
+                with open(path, 'r') as fo:
+                    content = fo.read()
+                if key in content:
+                    content = re.sub(r'%s=.*' % key, '%s=\'%s\'' % (key, new_domain), content)
+                    with open(path, 'w') as fo:
+                        fo.write(content)
+                    replaced = True
+                    log('Envsetup configration file "%s" updated.' % path)
+                    break
+        if not replaced:
+            with open(confs[0], 'a') as fo:
+                fo.write('\n%s=\'%s\'' % (key, new_domain))
+            log('Envsetup configration file "%s" updated.' % confs[0])
 
     def change_ms_domain(self, new_domain, instance):
         try:
@@ -158,24 +193,29 @@ class SetAppDomain():
             log('Assuming that the new url is using HTTPS: "%s"' % new_url)
             cmds = [
                 # set site url in site settings
-                'python3', os.path.join(ms_path, 'scripts', 'mssiteconfig.py'), instance, 'site_url="%s"' % new_url,
+                'python3 %s %s site_url="%s"' % (os.path.join(ms_path, 'scripts', 'mssiteconfig.py'), instance, new_url),
                 # reset all local resources managers
-                'python3', os.path.join(ms_path, 'scripts', 'reset_service_resources.py'), instance, 'local',
-                # change configuration of worker on MS side
-                'sed -i "s@\'url\': \'\\(http[s]*\\)://.*\'@\'url\': \'\\1://%s\'@" /etc/celerity/config.py' % new_domain,
+                'python3 %s %s local' % (os.path.join(ms_path, 'scripts', 'reset_service_resources.py'), instance),
+                # change configuration of celerity on MS side
+                'sed -i "s@\'url\': \'\\(http[s]*\\)://.*\',@\'url\': \'\\1://%s\',@" /etc/celerity/config.py' % new_domain,
+                'sed -i "s@SERVER_URL.*@SERVER_URL = \'%s:6200\'@" /etc/celerity/config.py' % new_url,
+                # restart ms
+                'mscontroller.py restart -u %s' % instance,
             ]
             worker_ips = utils.get_conf('CELERITY_WORKER_IP') or '127.0.1.1'
-            # change configuration of worker on worker side
+            # change configuration of celerity on worker side
             for worker_ip in worker_ips.split(','):
-                cmds.append('ssh root@%s sed -i "s@\'url\': \'\\(http[s]*\\)://.*\'@\'url\': \'\\1://%s\'@" /etc/celerity/config.py' % (worker_ip, new_domain))
+                cmds.append('rsync -avz /etc/celerity/config.py root@%s:/etc/celerity/config.py' % worker_ip)
+                cmds.append('ssh root@%s service celerity-workers restart' % worker_ip)
             utils.run_commands(cmds)
         except Exception as e:
-            log('Unable to set MS domain:\n%s' % e)
+            log('Unable to set domain in MS database and Celerity config:\n%s' % e)
             sys.exit(1)
         else:
-            log('MS domain changed')
-        log('Some steps to change the domain should be done manually:')
-        log('  - Change the domain of MS used in CM stations configuration.')
+            log('Domain changed in MS database and Celerity config.')
+        msg = 'Some steps to change the domain should be done manually:'
+        msg += '\n  - Change the domain of MS used in CM stations configuration.'
+        return msg
 
     def change_cm_domain(self, new_domain):
         try:
@@ -183,16 +223,17 @@ class SetAppDomain():
             log('Assuming that the new url is using HTTPS: "%s"' % new_url)
             cmds = [
                 # set site url in site settings
-                'echo \'from skyreach_site.base.models import SiteSettings; ss = SiteSettings.get_singleton(); ss.url = "%s"; ss.save(); print("site settings saved")\' | su skyreach -c "python3 /home/skyreach/htdocs/skyreach_site/manage.py shell --plain --no-startup"' % new_url,
+                'echo \'from skyreach_site.base.models import SiteSettings; ss = SiteSettings.get_singleton(); ss.url = "%s"; ss.save(); print("Site settings saved.")\' | su skyreach -c "python3 /home/skyreach/htdocs/skyreach_site/manage.py shell --plain --no-startup"' % new_url,
             ]
             utils.run_commands(cmds)
         except Exception as e:
-            log('Unable to set CM domain:\n%s' % e)
+            log('Unable to set domain in CM database:\n%s' % e)
             sys.exit(1)
         else:
-            log('CM domain changed')
-        log('Some steps to change the domain should be done manually:')
-        log('  - Change the url of CM in the related MS.')
+            log('Domain changed in CM database.')
+        msg = 'Some steps to change the domain should be done manually:'
+        msg += '\n  - Change the url of CM in the related MS.'
+        return msg
 
 
 if __name__ == '__main__':