3 # Licensed under the Apache License, Version 2.0 (the "License"); you may
4 # not use this file except in compliance with the License. You may obtain
5 # a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
11 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12 # License for the specific language governing permissions and limitations
15 # Shell script tool to run puppet inside of the given docker container image.
16 # Uses the config file at /var/lib/docker-puppet/docker-puppet.json as a source for a JSON
17 # array of [config_volume, puppet_tags, manifest, config_image, [volumes]] settings
18 # that can be used to generate config files or run ad-hoc puppet modules
19 # inside of a container.
26 import multiprocessing
29 # this is to match what we do in deployed-server
31 subproc = subprocess.Popen(['hostname', '-s'],
32 stdout=subprocess.PIPE,
33 stderr=subprocess.PIPE)
34 cmd_stdout, cmd_stderr = subproc.communicate()
35 return cmd_stdout.rstrip()
39 print('Pulling image: %s' % name)
40 subproc = subprocess.Popen(['/usr/bin/docker', 'pull', name],
41 stdout=subprocess.PIPE,
42 stderr=subprocess.PIPE)
43 cmd_stdout, cmd_stderr = subproc.communicate()
48 def rm_container(name):
49 if os.environ.get('SHOW_DIFF', None):
50 print('Diffing container: %s' % name)
51 subproc = subprocess.Popen(['/usr/bin/docker', 'diff', name],
52 stdout=subprocess.PIPE,
53 stderr=subprocess.PIPE)
54 cmd_stdout, cmd_stderr = subproc.communicate()
58 print('Removing container: %s' % name)
59 subproc = subprocess.Popen(['/usr/bin/docker', 'rm', name],
60 stdout=subprocess.PIPE,
61 stderr=subprocess.PIPE)
62 cmd_stdout, cmd_stderr = subproc.communicate()
66 process_count = int(os.environ.get('PROCESS_COUNT',
67 multiprocessing.cpu_count()))
69 config_file = os.environ.get('CONFIG', '/var/lib/docker-puppet/docker-puppet.json')
70 print('docker-puppet')
71 print('CONFIG: %s' % config_file)
72 with open(config_file) as f:
73 json_data = json.load(f)
75 # To save time we support configuring 'shared' services at the same
76 # time. For example configuring all of the heat services
77 # in a single container pass makes sense and will save some time.
78 # To support this we merge shared settings together here.
80 # We key off of config_volume as this should be the same for a
81 # given group of services. We are also now specifying the container
82 # in which the services should be configured. This should match
83 # in all instances where the volume name is also the same.
87 for service in (json_data or []):
90 if isinstance(service, dict):
92 service.get('config_volume'),
93 service.get('puppet_tags'),
94 service.get('step_config'),
95 service.get('config_image'),
96 service.get('volumes', []),
99 config_volume = service[0] or ''
100 puppet_tags = service[1] or ''
101 manifest = service[2] or ''
102 config_image = service[3] or ''
103 volumes = service[4] if len(service) > 4 else []
105 if not manifest or not config_image:
109 print('config_volume %s' % config_volume)
110 print('puppet_tags %s' % puppet_tags)
111 print('manifest %s' % manifest)
112 print('config_image %s' % config_image)
113 print('volumes %s' % volumes)
114 # We key off of config volume for all configs.
115 if config_volume in configs:
116 # Append puppet tags and manifest.
117 print("Existing service, appending puppet tags and manifest\n")
119 configs[config_volume][1] = '%s,%s' % (configs[config_volume][1],
122 configs[config_volume][2] = '%s\n%s' % (configs[config_volume][2],
124 if configs[config_volume][3] != config_image:
125 print("WARNING: Config containers do not match even though"
126 " shared volumes are the same!\n")
128 print("Adding new service\n")
129 configs[config_volume] = service
131 print('Service compilation completed.\n')
133 def mp_puppet_config((config_volume, puppet_tags, manifest, config_image, volumes)):
136 print('config_volume %s' % config_volume)
137 print('puppet_tags %s' % puppet_tags)
138 print('manifest %s' % manifest)
139 print('config_image %s' % config_image)
140 print('volumes %s' % volumes)
141 hostname = short_hostname()
142 sh_script = '/var/lib/docker-puppet/docker-puppet-%s.sh' % config_volume
144 with open(sh_script, 'w') as script_file:
145 os.chmod(script_file.name, 0755)
146 script_file.write("""#!/bin/bash
149 cp -a /tmp/puppet-etc/* /etc/puppet
150 rm -Rf /etc/puppet/ssl # not in use and causes permission errors
151 echo '{"step": %(step)s}' > /etc/puppet/hieradata/docker.json
153 if [ -n "%(puppet_tags)s" ]; then
154 TAGS='--tags "%(puppet_tags)s"'
156 FACTER_hostname=%(hostname)s FACTER_uuid=docker /usr/bin/puppet apply --verbose $TAGS /etc/config.pp
159 if [ -z "%(no_archive)s" ]; then
160 rm -Rf /var/lib/config-data/%(name)s
162 # copying etc should be enough for most services
163 mkdir -p /var/lib/config-data/%(name)s/etc
164 cp -a /etc/* /var/lib/config-data/%(name)s/etc/
166 if [ -d /root/ ]; then
167 cp -a /root/ /var/lib/config-data/%(name)s/root/
169 if [ -d /var/lib/ironic/tftpboot/ ]; then
170 mkdir -p /var/lib/config-data/%(name)s/var/lib/ironic/
171 cp -a /var/lib/ironic/tftpboot/ /var/lib/config-data/%(name)s/var/lib/ironic/tftpboot/
173 if [ -d /var/lib/ironic/httpboot/ ]; then
174 mkdir -p /var/lib/config-data/%(name)s/var/lib/ironic/
175 cp -a /var/lib/ironic/httpboot/ /var/lib/config-data/%(name)s/var/lib/ironic/httpboot/
178 # apache services may files placed in /var/www/
179 if [ -d /var/www/ ]; then
180 mkdir -p /var/lib/config-data/%(name)s/var/www
181 cp -a /var/www/* /var/lib/config-data/%(name)s/var/www/
184 """ % {'puppet_tags': puppet_tags, 'name': config_volume,
185 'hostname': hostname,
186 'no_archive': os.environ.get('NO_ARCHIVE', ''),
187 'step': os.environ.get('STEP', '6')})
189 with tempfile.NamedTemporaryFile() as tmp_man:
190 with open(tmp_man.name, 'w') as man_file:
191 man_file.write('include ::tripleo::packages\n')
192 man_file.write(manifest)
194 rm_container('docker-puppet-%s' % config_volume)
195 pull_image(config_image)
197 dcmd = ['/usr/bin/docker', 'run',
199 '--name', 'docker-puppet-%s' % config_volume,
200 '--volume', '%s:/etc/config.pp:ro' % tmp_man.name,
201 '--volume', '/etc/puppet/:/tmp/puppet-etc/:ro',
202 '--volume', '/usr/share/openstack-puppet/modules/:/usr/share/openstack-puppet/modules/:ro',
203 '--volume', '/var/lib/config-data/:/var/lib/config-data/:rw',
204 '--volume', 'tripleo_logs:/var/log/tripleo/',
205 '--volume', '%s:%s:rw' % (sh_script, sh_script) ]
207 for volume in volumes:
208 dcmd.extend(['--volume', volume])
210 dcmd.extend(['--entrypoint', sh_script])
213 if os.environ.get('NET_HOST', 'false') == 'true':
214 print('NET_HOST enabled')
215 dcmd.extend(['--net', 'host', '--volume',
216 '/etc/hosts:/etc/hosts:ro'])
217 dcmd.append(config_image)
219 subproc = subprocess.Popen(dcmd, stdout=subprocess.PIPE,
220 stderr=subprocess.PIPE, env=env)
221 cmd_stdout, cmd_stderr = subproc.communicate()
224 if subproc.returncode != 0:
225 print('Failed running docker-puppet.py for %s' % config_volume)
226 rm_container('docker-puppet-%s' % config_volume)
227 return subproc.returncode
229 # Holds all the information for each process to consume.
230 # Instead of starting them all linearly we run them using a process
231 # pool. This creates a list of arguments for the above function
235 for config_volume in configs:
237 service = configs[config_volume]
238 puppet_tags = service[1] or ''
239 manifest = service[2] or ''
240 config_image = service[3] or ''
241 volumes = service[4] if len(service) > 4 else []
244 puppet_tags = "file,file_line,concat,%s" % puppet_tags
246 puppet_tags = "file,file_line,concat"
248 process_map.append([config_volume, puppet_tags, manifest, config_image, volumes])
250 for p in process_map:
253 # Fire off processes to perform each configuration. Defaults
254 # to the number of CPUs on the system.
255 p = multiprocessing.Pool(process_count)
256 returncodes = list(p.map(mp_puppet_config, process_map))
257 config_volumes = [pm[0] for pm in process_map]
259 for returncode, config_volume in zip(returncodes, config_volumes):
261 print('ERROR configuring %s' % config_volume)