From 0742efdfa8ea8bf8dcdc5af438819ad999866e32 Mon Sep 17 00:00:00 2001
From: Ian Gulliver <flamingcow@gmail.com>
Date: Thu, 31 Mar 2016 23:24:11 -0700
Subject: [PATCH] Wait for service to come live before imaging

---
 server/modules/autoimage.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/server/modules/autoimage.py b/server/modules/autoimage.py
index e47aa37..7d9f64f 100755
--- a/server/modules/autoimage.py
+++ b/server/modules/autoimage.py
@@ -4,6 +4,7 @@ import argparse
 import os
 import shutil
 import subprocess
+from urllib import parse
 
 
 parser = argparse.ArgumentParser(description='iconograph autoimage')
@@ -63,6 +64,8 @@ def main():
       FLAGS.ca_cert,
       os.path.join(FLAGS.chroot_path, 'autoimage', 'config', 'ca.cert.pem'))
 
+  parsed = parse.urlparse(FLAGS.base_url)
+
   init = os.path.join(FLAGS.chroot_path, 'etc', 'init', 'autoimage.conf')
   with open(init, 'w') as fh:
     fh.write("""
@@ -72,9 +75,12 @@ start on net-device-up
 
 script
   chvt 7
+  /autoimage/imager/wait_for_service.py --host=%(host)s --service=%(service)s </dev/tty7 >/dev/tty7
   /autoimage/imager/image.py --device=%(device)s --persistent-percent=%(persistent_percent)d --ca-cert=/autoimage/config/ca.cert.pem --base-url=%(base_url)s </dev/tty7 >/dev/tty7
 end script
 """ % {
+      'host': parsed.host,
+      'service': parsed.port or parsed.scheme,
       'device': FLAGS.device,
       'persistent_percent': FLAGS.persistent_percent,
       'base_url': FLAGS.base_url,