#!/usr/bin/python3
# -*- coding: utf-8 -*-

import argparse
import os
from pkg_resources import parse_version

from spoc import repo_local
from spoc import repo_online
from spoc import 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):
    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() 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()
    for app in apps:
        print(app)

@locked(LOCK_FILE, print_lock)
def install(app_name):
    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()
    for layer in set(required_images):
        if layer not in local_images:
            queue.download_image(Image(layer, False))
    queue.download_app(App(app_name, False))
    queue.process()

@locked(LOCK_FILE, print_lock)
def update(app_name):
    App(app_name, False).update()

@locked(LOCK_FILE, print_lock)
def uninstall(app_name):
    App(app_name, False).uninstall()

def start(app_name):
    App(app_name).start()

def stop(app_name):
    App(app_name).stop()

def status(app_name):
    for container,status in App(app_name).status():
        print(f'{container}: {status}')

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)
        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):
    App(app_name, False).unpublish()

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'), 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')

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)
else:
    parser.print_usage()