surok/surok.py

#!/usr/bin/python3

from time import sleep
from os import listdir
from os.path import isfile, join 
import json
from surok.templates import gen
from surok.discovery import resolve
from surok.system import reload_conf


# Load base configurations
surok_conf = '/etc/surok/conf/surok.json'

# Command line arguments
parser = argparse.ArgumentParser()
parser.add_argument('-c', '--config')
args = parser.parse_args()
if args.config:
        surok_conf = args.config

f = open(surok_conf, 'r')
conf = json.loads(f.read())
print(conf)
f.close()


# Get app configurations 
def get_configs():
    confs = [f for f in listdir(conf['confd']) if isfile(join(conf['confd'], f))]
    return confs


# Get Surok App configuration
def load_app_conf(app):
    f = open(conf['confd'] + '/' + app)
    c = json.loads(f.read())
    f.close()

    return c


# Main loop
while 1:
    confs = get_configs()
    for app in confs:
        app_conf = load_app_conf(app)
        app_hosts = resolve(app_conf, conf)
        my = { 'services': app_hosts }
        service_conf = gen(my, app_conf['template'])

        print(reload_conf(service_conf, app_conf))


    sleep(conf['wait_time'])