support for systemd in sliver space for user-provided initscripts
[nodemanager.git] / initscript.py
1 import os, os.path
2 import tools
3
4 import logger
5
6 class Initscript:
7
8     def __init__ (self, name):
9         self.name=name
10         self.initscript = ''
11
12     def configure (self, rec):
13 #        logger.log("Initscript.configure")
14         new_initscript = rec['initscript']
15         if new_initscript != self.initscript:
16             self.initscript = new_initscript
17             # not used anymore, we always check against the installed script
18             #self.initscriptchanged = True
19             self.refresh_slice_vinit()
20
21     def install_and_enable_vinit (self):
22         # the fact that systemd attempts to run old-style services 
23         # says we should do either or or the other and not both
24         # but actually if that was true we could just do it for init and be fine
25         # which is not what we've seen starting with f18
26         # so for now let's do it for both systems unconditionnally
27         self.install_and_enable_vinit_for_init ()
28         self.install_and_enable_vinit_for_systemd ()
29
30     # unconditionnally install and enable the generic vinit script
31     # mimicking chkconfig for enabling the generic vinit script
32     # this is hardwired for runlevel 3
33     def install_and_enable_vinit_for_init (self):
34         "suitable for init-based VMs"
35         vinit_source="/usr/share/NodeManager/sliver-initscripts/vinit"
36         vinit_script="/vservers/%s/etc/rc.d/init.d/vinit"%self.name
37         enable_link="/vservers/%s/etc/rc.d/rc3.d/S99vinit"%self.name
38         enable_target="../init.d/vinit"
39         # install in sliver
40         code=file(vinit_source).read()
41         if tools.replace_file_with_string(vinit_script,code,chmod=0755):
42             logger.log("Initscript: %s: installed generic vinit rc script"%self.name)
43         # create symlink for runlevel 3
44         if not os.path.islink(enable_link):
45             try:
46                 logger.log("Initscript: %s: creating runlevel3 symlink %s"%(self.name,enable_link))
47                 os.symlink(enable_target,enable_link)
48             except:
49                 logger.log_exc("Initscript: %s: failed to create runlevel3 symlink %s"%enable_link)
50
51     # very similar but with systemd unit files - we target 'multi-user' in this context
52     def install_and_enable_vinit_for_systemd(self):
53         "suitable for systemd-based VMs"
54         vinit_source="/usr/share/NodeManager/sliver-systemd/vinit.service"
55         vinit_unit_file="/vservers/%s/usr/lib/systemd/system/vinit.service"%self.name
56         enable_link="/vservers/%s/etc/systemd/system/multi-user.target.wants/vinit.service"%self.name
57         enable_target="/usr/lib/systemd/system/vinit.service"
58         # install in sliver
59         code=file(vinit_source).read()
60         if tools.replace_file_with_string(vinit_unit_file,code,chmod=0755):
61             logger.log("Initscript: %s: installed vinit.service unit file"%self.name)
62         # create symlink for enabling this unit
63         if not os.path.islink(enable_link):
64             try:
65                 logger.log("Initscript: %s: creating enabling symlink %s"%(self.name,enable_link))
66                 os.symlink(enable_target,enable_link)
67             except:
68                 logger.log_exc("Initscript: %s: failed to create enabling symlink %s"%enable_link)
69
70
71 #ln -s '/usr/lib/systemd/system/vinit.service' '/etc/systemd/system/multi-user.target.wants/vinit.service'
72
73
74     # install or remove the slice inistscript, as instructed by the initscript tag
75     def refresh_slice_vinit(self):
76         code=self.initscript
77         sliver_initscript="/vservers/%s/etc/rc.d/init.d/vinit.slice"%self.name
78         if tools.replace_file_with_string(sliver_initscript,code,remove_if_empty=True,chmod=0755):
79             if code:
80                 logger.log("Initscript: %s: Installed new initscript in %s"%(self.name,sliver_initscript))
81                 if self.is_running():
82                     # Only need to rerun the initscript if the vserver is
83                     # already running. If the vserver isn't running, then the
84                     # initscript will automatically be started by
85                     # /etc/rc.d/vinit when the vserver is started.
86                     self.rerun_slice_vinit()
87             else:
88                 logger.log("Initscript: %s: Removed obsolete initscript %s"%(self.name,sliver_initscript))
89