slapos-theia

#!/usr/bin/python3

import os
import sys
import time
import subprocess

OCF_SUCCESS = 0
OCF_ERR_GENERIC = 1
OCF_NOT_RUNNING = 7

action = sys.argv[1]

instance_root = os.getenv('OCF_RESKEY_instance_root') or ''
instance_root_b = instance_root.encode('utf8')
part_id = os.getenv('OCF_RESKEY_part_id')
slappart = 'slappart%s' % part_id
slappart_b = slappart.encode('utf8')
slapuser = 'slapuser%s' % part_id
serious_service_list = []
serious_services_text = os.getenv('OCF_RESKEY_serious_services', None)
if serious_services_text:
  serious_service_list = serious_services_text.split(',')
autostart_service_list = []
autostart_services_text = os.getenv('OCF_RESKEY_autostart_services', None)
if autostart_services_text:
  autostart_service_list = autostart_services_text.split(',')
#kumofs_gateway,kumofs_manager,kumofs_server,mariadb,zeo,zope,haproxy


def splitLine(line):
  return [item for item in line.split(b' ') if item]

def runCommand(*args):
  return subprocess.run(args, capture_output=True)

def runSlapOS(*args):
  return runCommand('slapos', *args)

def checkTheiaStatus(target_status):
  completed_process = runSlapOS('node')
  for line in completed_process.stdout.split(b'\n'):
    if line.startswith(slappart_b):
      split_line = splitLine(line)
      if (split_line[0].startswith(slappart_b + b':theia-instance') and
          split_line[1].decode('utf8') == target_status):
        return True
  return False

def isTheiaRunning():
  return checkTheiaStatus('RUNNING')

def isTheiaStopped():
  return checkTheiaStatus('STOPPED')

def runSlapOSInTheia(*args):
  return runCommand('sudo', '-u', slapuser,
                    '%s/%s/srv/runner/bin/slapos' % (instance_root, slappart),
                    *args)

def checkAppStatus(target_status):
  completed_process = runSlapOSInTheia('node')
  if completed_process.stdout == b'unix://%s/%s/srv/runner/instance/sv.sock no such file\n' % (instance_root_b, slappart_b):
    if target_status == 'STOPPED':
      return True
    else:
      return False
  service_data_list = []
  for line in completed_process.stdout.split(b'\n'):
    if not line:
      continue
    split_line = splitLine(line)
    if split_line[0].decode('utf8') == 'watchdog':
      continue
    service_id = split_line[0].decode('utf8')
    service_name = split_line[0].split(b':')[1].split(b'-')[0].decode('utf8')
    current_status = split_line[1].decode('utf8')
    service_data_list.append((service_id, service_name, current_status))
  for service_id, service_name, current_status in service_data_list:
    if service_name in serious_service_list:
      if current_status != target_status:
        return False
  if target_status == 'RUNNING' and autostart_service_list:
    for service_id, service_name, current_status in service_data_list:
      if current_status == 'EXITED' and service_name in autostart_service_list:
        runSlapOSInTheia('node', 'start', service_id)
  return True

def isAppRunning():
  return checkAppStatus('RUNNING')

def isAppStopped():
  return checkAppStatus('STOPPED')

def monitor():
  result = OCF_ERR_GENERIC
  if not os.path.exists(instance_root):
    result = OCF_NOT_RUNNING
  try:
    if isTheiaRunning() and isAppRunning():
      result = OCF_SUCCESS
    elif isTheiaStopped() and isAppStopped():
      result = OCF_NOT_RUNNING
  except:
    pass
  sys.exit(result)

def start():
  result = OCF_ERR_GENERIC
  try:
    while not isTheiaRunning():
      runSlapOS('node', 'start', slappart + ':*')
      time.sleep(3)
    while not isAppRunning():
      runSlapOSInTheia('node', 'start', 'all')
      time.sleep(3)
    result = OCF_SUCCESS
  except:
    pass
  sys.exit(result)

def stop():
  result = OCF_ERR_GENERIC
  try:
    while not isAppStopped():
      runSlapOSInTheia('node', 'stop', 'all')
      time.sleep(3)
    while not isTheiaStopped():
      runSlapOS('node', 'stop', slappart + ':*')
      time.sleep(3)
    result = OCF_SUCCESS
  except:
    pass
  sys.exit(result)

def metadata():
  print ('''\
<?xml version="1.0"?>
<!DOCTYPE resource-agent SYSTEM "ra-api-1.dtd">
<resource-agent name="slapos-theia">
<version>1.0</version>
<longdesc lang="en">
This is the resource agent for SlapOS Theia.
</longdesc>
<shortdesc lang="en">Manages SlapOS Theia instance</shortdesc>

<parameters>
<parameter name="instance_root" unique="0" required="1">
<longdesc>
Instance Root Directory Path
</longdesc>
<shortdesc>Instance Root</shortdesc>
<content type="string"/>
</parameter>
<parameter name="part_id" unique="1" required="1">
<longdesc>
Computer Partition ID, slappartX
</longdesc>
<shortdesc>Partition ID</shortdesc>
<content type="integer"/>
</parameter>
<parameter name="serious_services" unique="0" required="0">
<longdesc>
Serious service names separated by comma
If serious service fails, then this resource is considered as failed.
</longdesc>
<shortdesc>Serious service names</shortdesc>
<content type="string"/>
</parameter>
<parameter name="autostart_services" unique="0" required="0">
<longdesc>
Auto-start service names separated by comma
If auto-start service becomes `EXITED`, then `slapos node start` is executed
automatically and it is not considered as failed.
</longdesc>
<shortdesc>Auto-start service names</shortdesc>
<content type="string"/>
</parameter>
</parameters>

<actions>
<action name="start"   timeout="60s" />
<action name="stop"    timeout="60s" />
<action name="monitor" depth="0"  timeout="30s" interval="60s" />
<action name="meta-data"  timeout="5s" />
</actions>
</resource-agent>
''')
  sys.exit(OCF_SUCCESS)

if action == 'monitor':
  monitor()
elif action == 'start':
  start()
elif action == 'stop':
  stop()
elif action == 'meta-data':
  metadata()