moulinette/yunohost

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import sys
import argparse
import gettext
import getpass
try:
    import yaml
except ImportError:
    sys.stderr.write('Error: Yunohost CLI Require yaml lib\n')
    sys.stderr.write('apt-get install python-yaml\n')
    sys.exit(1)
import json
if not __debug__:
    import traceback

gettext.install('YunoHost')

try:
    from yunohost import YunoHostError, YunoHostLDAP, str_to_func, colorize, pretty_print_dict, display_error, validate, win, parse_dict
except ImportError:
    sys.stderr.write('Error: Yunohost CLI Require YunoHost lib\n')
    sys.exit(1)


def main():
    """
    Main instructions

    Parse the action_dict and execute the action-specific function,
    then print json or pretty result if executed in a tty :)

    Returns:
        int -- 0 or error code

    """

    if len(sys.argv) < 2:
        sys.argv.append('-h')

    with open('action_map.yml') as f:
        action_map = yaml.load(f)

    admin_password_provided = False
    json_print = False
    write_ldap = True
    postinstall = False

    for key, arg in enumerate(sys.argv):
        if arg == '--admin-password':
            admin_password_provided = True
            admin_password = sys.argv[key+1]
            sys.argv.pop(key)
            sys.argv.pop(key)
        if arg == '--no-ldap':
            write_ldap = False
            sys.argv.pop(key)
        if arg == '--json':
            json_print = True
            sys.argv.pop(key)

    try:
        try:
            with open('/etc/yunohost/installed') as f: pass
        except IOError:
            postinstall = True
            if len(sys.argv) < 3 or sys.argv[1] != 'tools' or sys.argv[2] != 'postinstall':
                raise YunoHostError(17, _("YunoHost is not correctly installed, please execute 'yunohost tools postinstall'"))

        args = parse_dict(action_map)
        args_dict = vars(args).copy()
        for key in args_dict.keys():
            sanitized_key = key.replace('-', '_')
            if sanitized_key is not key:
                args_dict[sanitized_key] = args_dict[key]
                del args_dict[key]
        del args_dict['func']
        try:
            with open('/etc/yunohost/passwd') as f:
                admin_password = f.read()
                admin_password_provided = True
        except IOError: pass
        if postinstall:
            result = args.func(**args_dict)
        elif admin_password_provided:
            with YunoHostLDAP(password=admin_password):
                result = args.func(**args_dict)
        elif os.isatty(1) and write_ldap:
            admin_password = getpass.getpass(colorize(_('Admin Password: '), 'yellow'))
            with YunoHostLDAP(password=admin_password):
                try:
                    with open('/var/run/yunohost.pid', 'r'):
                        raise YunoHostError(1, _("A YunoHost command is already running"))
                except IOError:
                    with open('/var/run/yunohost.pid', 'w') as f:
                        f.write('ldap')
                        os.system('chmod 400 /var/run/yunohost.pid')
                    with open('/etc/yunohost/passwd', 'w') as f:
                        f.write(admin_password)
                        os.system('chmod 400 /etc/yunohost/passwd')
                    try:
                        result = args.func(**args_dict)
                    except KeyboardInterrupt, EOFError:
                        raise YunoHostError(125, _("Interrupted"))
                    finally:
                        os.remove('/etc/yunohost/passwd')
                        os.remove('/var/run/yunohost.pid')
        else:
            with YunoHostLDAP(anonymous=True):
                result = args.func(**args_dict)
    #except TypeError, error:
        #if not __debug__ :
            #traceback.print_exc()
        #print(_("Not (yet) implemented function"))
        #return 1
    except YunoHostError, error:
        display_error(error, json_print)
        return error.code
    else:
        if json_print or not os.isatty(1) and result is not None:
            if len(win) > 0:
                result['success'] = win
            print(json.dumps(result))
        elif result is not None:
            pretty_print_dict(result)
        else:
            pass

    return 0

if __name__ == '__main__':
    sys.exit(main())