is

#!/usr/bin/python2.7
# -*- coding: utf-8 -*-
# Started 30/06/2011 by Seblu <seblu@seblu.net>

'''
InstallSystems Command line Tool
'''

import os
import time
import datetime
import argparse
import installsystems
import installsystems.tools as istools
from installsystems.printer import *
from installsystems.repository import Repository
from installsystems.repository import RepositoryManager
from installsystems.repository import RepositoryConfig
from installsystems.image import PackageImage, SourceImage
from installsystems.config import MainConfigFile, RepoConfigFile

def load_repositories(args):
    '''
    Load repositories on a repository manager
    '''
    # remove cache is asked
    if args.no_cache:
        args.cache = None
    # init repo cache object
    repoman = RepositoryManager(args.cache, timeout=args.timeout)
    # register repositories (order matter)
    # load repo configs
    if args.repo_path is not None:
        # from command line
        repoman.register(RepositoryConfig(None, path=args.repo_path))
    else:
        # from config
        for repoconf in RepoConfigFile(args.repo_config).repos:
            # filtering on repository name if present
            if args.repo_filter is not None and repoconf.name != args.repo_filter:
                continue
            repoman.register(repoconf)
    return repoman

def c_new(parser, args):
    '''
    Create a new source image
    '''
    simg = SourceImage.create(args.path)

def c_build(parser, args):
    '''
    Build an image source
    '''
    # build start time
    t0 = time.time()
    # load source image
    simg = SourceImage(args.path)
    # do the job
    simg.build(force=args.force, check=not args.no_check)
    # compute building time
    t1 = time.time()
    dt = int(t1 - t0)
    arrow("Build time: %s" % datetime.timedelta(seconds=dt))

def c_init(parser, args):
    '''
    Create a empty repository
    '''
    # we cannot use load_repositories because repo doesn't exists
    if args.repo_path is not None:
        # from command line
        Repository.create(RepositoryConfig(None, path=args.repo_path))
    else:
        # from config
        for repoconf in RepoConfigFile(args.repo_config).repos:
            # filtering on repository name if present
            if args.repo_filter is not None and repoconf.name != args.repo_filter:
                continue
            Repository.create(repoconf)

def c_add(parser, args):
    '''
    Add an image package into a repository
    '''
    repoman = load_repositories(args)
    if len(repoman) == 0:
        raise Exception("No repository selected")
    elif len(repoman) > 1:
        raise Exception("Please select only one repository")
    repo = repoman[0]
    pkg = PackageImage(args.path)
    repo.add(pkg, delete=not args.preserve)

def c_del(parser, args):
    '''
    Remove an image package from a repository
    '''
    repoman = load_repositories(args)
    if len(repoman) == 0:
        raise Exception("No repository selected")
    elif len(repoman) > 1:
        raise Exception("Please select only one repository")
    repo = repoman[0]
    repo.delete(args.image_name, args.image_version)

def c_install(parser, args):
    '''
    Install an image
    '''
    # looks if arguments is a file or image name
    if istools.pathtype(args.image) == "file" and os.path.isfile(args.image):
        pkg = PackageImage(istools.abspath(args.image))
    elif PackageImage.check_image_name(args.image):
        # get image package
        repoman = load_repositories(args)
        pkg = repoman.get(args.image, args.image_version)
    # add default show help options
    args.subparser.add_argument("-h", "--help", action="help", default=argparse.SUPPRESS,
                                help="show program's version number and exit")
    # Print setup information
    arrow("Installing %s v%s" % (pkg.name, pkg.version))
    # install start time
    t0 = time.time()
    # run parser scripts with parser parser argument
    pkg.run_parser(parser=args.subparser)
    # call parser again, with extended attributes
    arrow("Parsing arguments")
    args = parser.parse_args(namespace=args)
    # run setup scripts
    pkg.run_setup(namespace=args)
    # compute building time
    t1 = time.time()
    dt = int(t1 - t0)
    arrow("Install time: %s" % datetime.timedelta(seconds=dt))

def c_list(parser, args):
    '''
    List images in repository or image content
    '''
    args.repo_filter = None
    if len(args.list_args) == 0:
        # List available repository
        repoman = load_repositories(args)
        arrow('#light##green#[Repos]#reset#')
        for repo in repoman:
            out(repo.config.name)
    # <repo> | <image> | <image file>
    elif len(args.list_args) == 1:
        # <image file>
        if istools.pathtype(args.list_args[0]) == 'file' and\
                os.path.isfile(args.list_args[0]):
            pkg = PackageImage(istools.abspath(args.list_args[0]))
            arrow('#light##green#[Image]#reset# %s' % pkg.name)
            pkg.show(verbose=args.verbose_list, list=True)
        else:
            args.repo_filter = args.list_args[0]
            repoman = load_repositories(args)
            # <repo>
            if len(repoman) != 0:
                for repo in repoman:
                    arrow('#light##green#[Repo]#reset# %s' % repo.config.name)
                    repo.show(args.verbose_list)
            # <image>
            else:
                args.repo_filter = None
                args.image = args.list_args[0]
                args.image_version = None
    # <repo> <image> | <image> <version>
    elif len(args.list_args) == 2:
        # <repo> <image>
        if not PackageImage.check_image_version(args.list_args[1]):
            args.repo_filter = args.list_args[0]
            args.image = args.list_args[1]
            args.image_version = None
        # <image> <version>
        else:
            args.image = args.list_args[0]
            args.image_version = args.list_args[1]
    # <repo> <image> <version>
    elif len(args.list_args) == 3:
        args.repo_filter = args.list_args[0]
        args.image = args.list_args[1]
        args.image_version = args.list_args[2]

    if hasattr(args, 'image'):
        repoman = load_repositories(args)
        pkg = repoman.get(args.image, args.image_version)
        arrow('#light##green#[Image]#reset# %s' % pkg.name)
        pkg.show(verbose=args.verbose_list, list=True)

def c_search(parser, args):
    '''
    Search in repository
    '''
    raise NotImplementedError("Not yet implemented")

def c_get(parser, args):
    '''
    Get a remove image in current directory
    '''
    raise NotImplementedError("Not yet implemented")

def c_clean(parser, args):
    '''
    Clean a repository
    '''
    raise NotImplementedError("Not yet implemented")

def c_help(parser, args):
    '''
    Show help
    '''
    parser.print_help()

def c_version(parser, args):
    '''
    Print installsystems version
    '''
    out(installsystems.version)

# Top level argument parsing
p_main = argparse.ArgumentParser()
p_main.add_argument("-V", "--version", action="version",
                    version=installsystems.version,
                    help="show installsystems version")
# exclusive group on debug/quiet
ex_group = p_main.add_mutually_exclusive_group()
ex_group.add_argument('-d', "--debug", action="store_true",
                      help="active debug mode")
ex_group.add_argument('-q', "--quiet", action="store_true",
                      help="active quiet mode")
# common options
p_main.add_argument("-c", "--config", default="installsystems",
                    help="config file path")
p_main.add_argument("-R", "--repo-config", default="repository",
                    help="repository config file path")
p_main.add_argument("-f", "--repo-filter", default=None,
                    help="select repository by name in config files")
p_main.add_argument("-r", "--repo-path", default=None,
                    help="repository path")
p_main.add_argument("-C", "--cache", default=None,
                    help="path of the repository cache")
p_main.add_argument("--no-cache", action="store_true", default=False,
                    help="Not use persistent db caching")
p_main.add_argument("-t", "--timeout", dest="timeout", type=int, default=3,
                    help="download timeout (default 3)")

# create a subparsers for each command
subparsers = p_main.add_subparsers()

# new command parser
p_new = subparsers.add_parser("new", help=c_new.__doc__.lower())
p_new.add_argument("path", help="Path of new image directory")
p_new.set_defaults(func=c_new)

# build command parser
p_build = subparsers.add_parser("build", help=c_build.__doc__.lower())
p_build.add_argument('-f', "--force", action="store_true", default=False,
                     help="overwrite existing image")
p_build.add_argument('-c', "--no-check", action="store_true", default=False,
                     help="do not check compilation before adding scripts")
p_build.add_argument("path", nargs="?", default=".")
p_build.set_defaults(func=c_build)

# init command parser
p_init = subparsers.add_parser("init", help=c_init.__doc__.lower())
p_init.add_argument("repo_filter", nargs="?", default=argparse.SUPPRESS,
                    help="Name or path of a repository to init")
p_init.set_defaults(func=c_init)

# add command parser
p_add =  subparsers.add_parser("add", help=c_add.__doc__.lower())
p_add.add_argument('-p', "--preserve", action="store_true", default=False,
                   help="don't remove image after adding to database")
p_add.add_argument("path")
p_add.set_defaults(func=c_add)

# del command parser
p_del =  subparsers.add_parser("del", help=c_del.__doc__.lower())
p_del.add_argument("image_name")
p_del.add_argument("image_version")
p_del.set_defaults(func=c_del)

# install command parser
p_install = subparsers.add_parser("install", add_help=False,
                                  help=c_install.__doc__.lower())
p_install.add_argument('-f', "--force", action="store_true", default=False,
                       help="overwrite existing image")
p_install.add_argument("-v", "--image-version", type=int, default=None,
                       help="image version")
p_install.add_argument("image", help="image to install (path or name)")
p_install.set_defaults(func=c_install, subparser=p_install)

# get command parser
p_get = subparsers.add_parser("get", help=c_get.__doc__.lower())
p_get.set_defaults(func=c_get)

# list command parser
p_list = subparsers.add_parser("list", help=c_list.__doc__.lower())
p_list.add_argument("-l", action="store_true", dest='verbose_list',
                       default=False, help='Detailled list output')
p_list.add_argument("list_args", nargs="*", help="Repo or image to list")
p_list.set_defaults(func=c_list)

# search command parser
p_search = subparsers.add_parser("search", help=c_search.__doc__.lower())
p_search.set_defaults(func=c_search)

# clean command parser
p_clean = subparsers.add_parser("clean", help=c_clean.__doc__.lower())
p_clean.set_defaults(func=c_clean)

# help command parser
p_help = subparsers.add_parser("help", help=c_help.__doc__.lower())
p_help.set_defaults(func=c_help)

# version command parser
p_version = subparsers.add_parser("version", help=c_version.__doc__.lower())
p_version.set_defaults(func=c_version)

try:
    # Parse and run
    args = p_main.parse_known_args()[0]
    # set debug and quiet mode before merge
    installsystems.debug = args.debug
    installsystems.quiet = args.quiet
    # load isinstall config
    config = MainConfigFile(args.config, "installsystems")
    config.merge(args)
    # set debug and quiet mode after merge
    installsystems.debug = args.debug
    installsystems.quiet = args.quiet
    if args.func is not c_install:
        args = p_main.parse_args(namespace=args)
    # let's go
    args.func(p_main, args)
except Exception as e:
    error(e)
except KeyboardInterrupt:
    warn("Keyboard Interrupted")