spoc/usr/bin/spoc-app

200 lines
7.1 KiB
Python
Executable File

#!/usr/bin/python3
# -*- coding: utf-8 -*-
import argparse
import os
from pkg_resources import parse_version
from spoc import repo_local, repo_online, repo_publish
from spoc.app import App
from spoc.cli import ActionQueue, print_lock, readable_size
from spoc.config import LOCK_FILE
from spoc.flock import locked
from spoc.image import Image
def listing(list_type):
# Lists applications in particular state
if list_type == 'installed':
apps = repo_local.get_apps()
elif list_type == 'online':
apps = repo_online.get_apps()
elif list_type == 'updates':
online_apps = repo_online.get_apps()
apps = [a for a,d in repo_local.get_apps().items() if a in online_apps and parse_version(online_apps[a]['version']) > parse_version(d['version'])]
elif list_type == 'published':
apps = repo_publish.get_apps()
elif list_type == 'running':
apps = [app for app in repo_local.get_apps() if App(app).is_running()]
elif list_type == 'stopped':
apps = [app for app in repo_local.get_apps() if App(app).is_stopped()]
for app in apps:
print(app)
@locked(LOCK_FILE, print_lock)
def install(app_name):
# Install application from online repository
queue = ActionQueue()
required_images = []
for container in repo_online.get_app(app_name)['containers'].values():
required_images.extend(repo_online.get_image(container['image'])['layers'])
local_images = repo_local.get_images()
# Layers need to be downloaded in correct order
for layer in list(dict.fromkeys(required_images)):
if layer not in local_images:
queue.download_image(Image(layer, False))
queue.install_app(App(app_name, False, False))
queue.process()
@locked(LOCK_FILE, print_lock)
def update(app_name):
# Update application from online repository
queue = ActionQueue()
required_images = []
for container in repo_online.get_app(app_name)['containers'].values():
required_images.extend(repo_online.get_image(container['image'])['layers'])
local_images = repo_local.get_images()
# Layers need to be downloaded in correct order
for layer in list(dict.fromkeys(required_images)):
if layer not in local_images:
queue.download_image(Image(layer, False))
queue.update_app(App(app_name, False))
queue.process()
@locked(LOCK_FILE, print_lock)
def uninstall(app_name):
# Remove application and its containers from local repository
queue = ActionQueue()
queue.uninstall_app(App(app_name, False))
queue.process()
def start(app_name):
# Start all application containers
queue = ActionQueue()
queue.start_app(App(app_name))
queue.process()
def stop(app_name):
# Stop all application containers
queue = ActionQueue()
queue.stop_app(App(app_name))
queue.process()
def status(app_name):
# Print status of all application containers
for container,status in sorted(App(app_name).status().items()):
print(f'{container}: {status.value}')
def publish(filename, force):
app_name = os.path.basename(os.path.dirname(os.path.abspath(filename)))
# Check if publishing is needed and attempt to publish the application
if force or app_name not in repo_publish.get_apps():
app = App(app_name, False, False)
print(f'Publishing application {app_name} from file {os.path.abspath(filename)}')
app.unpublish()
size, dlsize = app.publish(filename)
print(f'Application {app_name} compressed from {readable_size(size)} to {readable_size(dlsize)} and published successfully')
else:
print(f'Application {app_name} already published, skipping publish task')
def unpublish(app_name):
# Remove the application from publish repo
App(app_name, False, False).unpublish()
def autostart(app_name, value):
# Set if the application should be autostarted on boot
value = value.lower() in ('1', 'on', 'enable', 'true')
App(app_name, False).set_autostart(value)
def start_autostarted():
# Start all applications (resp. their containers) which are set to be autoostarted on boot
apps = [App(a) for a,d in repo_local.get_apps().items() if d['autostart']]
for app in apps:
app.start()
def stop_all():
# Stop all applications (resp. their containers)
apps = [App(a) for a,d in repo_local.get_apps().items()]
for app in apps:
app.stop()
parser = argparse.ArgumentParser(description='SPOC application manager')
parser.set_defaults(action=None)
subparsers = parser.add_subparsers()
parser_list = subparsers.add_parser('list')
parser_list.set_defaults(action=listing)
parser_list.add_argument('type', choices=('installed', 'online', 'updates', 'published', 'running', 'stopped'), default='installed', const='installed', nargs='?')
parser_install = subparsers.add_parser('install')
parser_install.set_defaults(action=install)
parser_install.add_argument('app')
parser_update = subparsers.add_parser('update')
parser_update.set_defaults(action=update)
parser_update.add_argument('app')
parser_uninstall = subparsers.add_parser('uninstall')
parser_uninstall.set_defaults(action=uninstall)
parser_uninstall.add_argument('app')
parser_start = subparsers.add_parser('start')
parser_start.set_defaults(action=start)
parser_start.add_argument('app')
parser_stop = subparsers.add_parser('stop')
parser_stop.set_defaults(action=stop)
parser_stop.add_argument('app')
parser_status = subparsers.add_parser('status')
parser_status.set_defaults(action=status)
parser_status.add_argument('app')
parser_publish = subparsers.add_parser('publish')
parser_publish.set_defaults(action=publish)
parser_publish.add_argument('-f', '--force', action='store_true', help='Force republish already published application')
parser_publish.add_argument('filename')
parser_unpublish = subparsers.add_parser('unpublish')
parser_unpublish.set_defaults(action=unpublish)
parser_unpublish.add_argument('app')
parser_autostart = subparsers.add_parser('autostart')
parser_autostart.set_defaults(action=autostart)
parser_autostart.add_argument('app')
parser_autostart.add_argument('value', choices=('1', 'on', 'enable', 'true', '0', 'off', 'disable', 'false'), help='Sets the applications to be automatically started after the host boots up')
parser_start_autostarted = subparsers.add_parser('start-autostarted')
parser_start_autostarted.set_defaults(action=start_autostarted)
parser_stop_all = subparsers.add_parser('stop-all')
parser_stop_all.set_defaults(action=stop_all)
args = parser.parse_args()
if args.action is listing:
listing(args.type)
elif args.action is install:
install(args.app)
elif args.action is update:
update(args.app)
elif args.action is uninstall:
uninstall(args.app)
elif args.action is start:
start(args.app)
elif args.action is stop:
stop(args.app)
elif args.action is status:
status(args.app)
elif args.action is publish:
publish(args.filename, args.force)
elif args.action is unpublish:
unpublish(args.app)
elif args.action is autostart:
autostart(args.app, args.value)
elif args.action is start_autostarted:
start_autostarted()
elif args.action is stop_all:
stop_all()
else:
parser.print_usage()