1 files changed, 84 insertions, 0 deletions
diff --git a/yarns/lib.py b/yarns/lib.py
index 290512c..5fbd5ab 100644
--- a/yarns/lib.py
+++ b/yarns/lib.py
@@ -19,6 +19,7 @@ import errno
 import json
 import os
 import random
+import signal
 import socket
 import sys
 import time
@@ -26,6 +27,7 @@ import urllib
 
 import cliapp
 import requests
+import yaml
 
 from yarnutils import *
 
@@ -35,6 +37,88 @@ datadir = os.environ['DATADIR']
 V = Variables(datadir)
 
 
+def start_controller():
+    port = V['port'] = random_free_port()
+
+    V['url'] = 'http://127.0.0.1:{}'.format(V['port'])
+
+    filename = 'ick_controller.yaml'
+    env = dict(os.environ)
+    env['ICK_CONTROLLER_CONFIG'] = filename
+    write_yaml(filename, {
+        'token-issuer': V['issuer'],
+        'token-audience': V['audience'],
+        'token-public-key': cat('token.key.pub'),
+        'log': [
+            {
+                'filename': 'ick_controller.log',
+            },
+        ],
+        'statedir': V['statedir'],
+        'apt-server': 'localhost',
+        'artifact-store': V['artifact_store'],
+        'auth-url': V['auth_url'],
+        'notify-url': V['notify_url'],
+    })
+
+    V['pid'] = gunicorn('ick_controller', 'app', port, env)
+
+
+def stop_controller():
+    if V['pid'] is not None:
+        os.kill(int(V['pid']), signal.SIGTERM)
+
+
+def start_artifact_store():
+    port = V['bsport'] = random_free_port()
+
+    V['bsurl'] = 'http://127.0.0.1:{}'.format(V['bsport'])
+
+    filename = 'artifact_store.yaml'
+    env = dict(os.environ)
+    env['ARTIFACT_STORE_CONFIG'] = filename
+    write_yaml(filename, {
+        'token-issuer': V['issuer'],
+        'token-audience': V['audience'],
+        'token-public-key': cat('token.key.pub'),
+        'log': [
+            {
+                'filename': 'artifact_store.log',
+            },
+        ],
+        'blobdir': V['blobdir'],
+    })
+
+    V['bspid'] = gunicorn('artifact_store', 'app', port, env)
+
+
+def stop_artifact_store():
+    if V['pid'] is not None:
+        os.kill(int(V['bspid']), signal.SIGTERM)
+
+
+def write_yaml(filename, obj):
+    yaml.safe_dump(obj, open(filename, 'w'))
+
+
+def gunicorn(module_name, var_name, port, env):
+    log_filename = '{}.gunicorn.log'.format(module_name)
+    pid_filename = '{}.pid'.format(module_name)
+
+    argv = [
+        'gunicorn3',
+        '--daemon',
+        '--bind', '127.0.0.1:{}'.format(port),
+        '--log-file', log_filename,
+        '--log-level', 'debug',
+        '-p', pid_filename,
+        '{}:{}'.format(module_name, var_name),
+    ]
+    cliapp.runcmd(argv, env=env)
+    wait_for_port(port)
+    return int(cat(pid_filename))
+
+
 def random_free_port():
     MAX = 1000
     for i in range(MAX):