add supervisor support

Author: k-okada

src/robot_upstart/install_script.py

@@ -62,7 +62,7 @@ def get_argument_parser():
                    help="Specify an a value for ROS_LOG_DIR in the job launch context.")
     p.add_argument("--augment", action='store_true',
                    help="Bypass creating the job, and only copy user files. Assumes the job was previously created.")
-    p.add_argument("--provider", type=str, metavar="[upstart|systemd]",
+    p.add_argument("--provider", type=str, metavar="[upstart|systemd|supervisor]",
                    help="Specify provider if the autodetect fails to identify the correct provider")
     p.add_argument("--symlink", action='store_true',
                    help="Create symbolic link to job launch files instead of copying them.")
@@ -71,6 +71,9 @@ def get_argument_parser():
     p.add_argument("--systemd-after", type=str, metavar="After=",
                    help="Set the string of the After= section"
                         "of the generated Systemd service file")
+    p.add_argument("--supervisor-priority", type=int, metavar="Priority=",
+                   help="Set the value of the priority= section"
+                        "of the generated Supervisor conf file")
 
     return p
 
@@ -89,7 +92,8 @@ def main():
         name=job_name, interface=args.interface, user=args.user,
         workspace_setup=args.setup, rosdistro=args.rosdistro,
         master_uri=args.master, log_path=args.logdir,
-        systemd_after=args.systemd_after)
+        systemd_after=args.systemd_after,
+        supervisor_priority=args.supervisor_priority)
 
     for this_pkgpath in args.pkgpath:
         pkg, pkgpath = this_pkgpath.split('/', 1)
@@ -121,6 +125,8 @@ def main():
         provider = providers.Upstart
     if args.provider == 'systemd':
         provider = providers.Systemd
+    if args.provider == 'supervisor':
+        provider = providers.Supervisor
     if args.symlink:
         j.symlink = True
 

src/robot_upstart/job.py

@@ -41,7 +41,7 @@ class Job(object):
 
     def __init__(self, name="ros", interface=None, user=None, workspace_setup=None,
                  rosdistro=None, master_uri=None, log_path=None,
-                 systemd_after=None):
+                 systemd_after=None, supervisor_priority=None):
         """Construct a new Job definition.
 
         :param name: Name of job to create. Defaults to "ros", but you might
@@ -105,6 +105,10 @@ def __init__(self, name="ros", interface=None, user=None, workspace_setup=None,
         # of the generated Systemd service file
         self.systemd_after = systemd_after or "network.target"
 
+        # Set the value of the "priority=" section
+        # of the generated Supservisor conf file
+        self.supervisor_priority = supervisor_priority or 200
+
         # Set of files to be installed for the job. This is only launchers
         # and other user-specified configs--- nothing related to the system
         # startup job itself. List of strs.

src/robot_upstart/providers.py

@@ -40,6 +40,8 @@ def detect_provider():
     print(os.path.realpath(cmd))
     if b'systemd' in os.path.realpath(cmd):
         return Systemd
+    if b'supervisor' in os.path.realpath(cmd):
+        return Supervisor
     return Upstart
 
 
@@ -228,3 +230,78 @@ def _fill_template(self, template):
             self.interpreter.file(f)
             return self.interpreter.output.getvalue()
         self.set_job_path()
+
+
+class Supervisor(Generic):
+    """ The Supervisor implementation places the user-specified files in ``/etc/ros/DISTRO/NAME.d``,
+    and creates an systemd job configuration in ``/lib/systemd/system/NAME.d``. Two additional
+    helper scripts are created for starting and stopping the job, places in
+    ``/usr/sbin``.
+    To detect which system you're using run: ps -p1 | grep systemd && echo systemd || echo upstart
+    """
+
+    def generate_install(self):
+        # Default is /etc/ros/DISTRO/JOBNAME.d
+        self._set_job_path()
+
+        # User-specified launch files.
+        self._add_job_files()
+
+        # This is optional to support the old --augment flag where a "job" only adds
+        # launch files to an existing configuration.
+        if self.job.generate_system_files:
+            # Share a single instance of the EmPy interpreter.
+            self.interpreter = em.Interpreter(globals=self.job.__dict__.copy())
+
+            self.installation_files[os.path.join(self.root, "etc/supervisor/conf.d", self.job.name + ".conf")] = {
+                "content": self._fill_template("templates/supervisor_job.conf.em"), "mode": 0o644}
+            self.installation_files[os.path.join(self.root, "usr/sbin", self.job.name + "-start")] = {
+                "content": self._fill_template("templates/job-start.em"), "mode": 0o755}
+            self.installation_files[os.path.join(self.root, "usr/sbin", self.job.name + "-stop")] = {
+                "content": self._fill_template("templates/job-stop.em"), "mode": 0o755}
+            self.interpreter.shutdown()
+
+        # Add an annotation file listing what has been installed. This is a union of what's being
+        # installed now with what has been installed previously, so that an uninstall should remove
+        # all of it. A more sophisticated future implementation could track contents or hashes and
+        # thereby warn users when a new installation is stomping a change they have made.
+        self._load_installed_files_set()
+        self.installed_files_set.update(list(self.installation_files.keys()))
+
+        # Remove the job directory. This will fail if it is not empty, and notify the user.
+        self.installed_files_set.add(self.job.job_path)
+
+        # Remove the annotation file itself.
+        self.installed_files_set.add(self.installed_files_set_location)
+
+        self.installation_files[self.installed_files_set_location] = {
+            "content": "\n".join(self.installed_files_set)}
+
+        return self.installation_files
+
+    def post_install(self):
+        print("** To complete installation please run the following command:")
+        print(" sudo supervisorctl reload" +
+              " && sudo supervisorctl status" +
+              " ; browse https://localhost:9001")
+
+    def generate_uninstall(self):
+        self._set_job_path()
+        self._load_installed_files_set()
+
+        for filename in self.installed_files_set:
+            self.installation_files[filename] = {"remove": True}
+
+        return self.installation_files
+
+    def _set_job_path(self):
+        self.job.job_path = os.path.join(
+            self.root, "etc/ros", self.job.rosdistro, self.job.name + ".d")
+
+    def _fill_template(self, template):
+        self.interpreter.output = io.StringIO()
+        self.interpreter.reset()
+        with open(find_in_workspaces(project="robot_upstart", path=template)[0]) as f:
+            self.interpreter.file(f)
+            return self.interpreter.output.getvalue()
+        self.set_job_path()