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

import os.path
import time
import tarfile

from metafile import Metafile, MalformedMetafileError

# --------------------------------------
def init(parser):

    parser.add_argument( "apps"                                           ,
                         type=str                                         ,
                         nargs="*"                                        ,
                         default=[]                                       ,
                         help="The apps to package ; defaults to all "    \
                              "apps" )

    parser.add_argument( "--pkgtype", "-t"                                ,
                         type=str                                         ,
                         nargs="?"                                        ,
                         default="tgz"                                    ,
                         choices = [ "tgz" ]                              ,
                         help="The type of package to use" )

    parser.add_argument( "--pkgname", "-p"                                ,
                         type=str                                         ,
                         nargs="?"                                        ,
                         default="persoconf." + (time.strftime("%Y%m%d")) ,
                         help="The name of the package to create" )

# --------------------------------------
def run(args, persoconf, logger):

    pkgname = args.pkgname

    # If no app is given: package all apps
    if args.apps == []:
        logger.info("Packaging all apps")
        args.apps = persoconf.list_apps(logger=logger)

    # Some apps were specified, so we list them with their files
    appmetas = []
    for app in args.apps :
        # Load app META file
        try:
            appmetas.append(
                    Metafile(json_path = os.path.join(
                        persoconf.path    ,
                        app               ,
                        persoconf.metafile)
                        )
                    )

        except FileNotFoundError:
            logger.warning( "Metafile %s does not exist for app %s; ignoring"
                            % (persoconf.metafile, app) )
            continue

        except MalformedMetafileError:
            logger.warning( "Malformed metafile %s in app %s; ignoring"
                            % (persoconf.metafile, app) )
            continue

    # TODO check that the app exists

    # Adds the app name if there is only one packaged
    if len(appmetas) == 1:
        pkgname += ("." + appmetas[0].name)
    
    # Switch according to the pkgtype
    if args.pkgtype == "tgz":

        pkgname += ".tgz"

        # Create a tar containing the files in the right place
        pkg = tarfile.open(pkgname, "w:gz")
        
        for appmeta in appmetas:
            for f in appmeta.files:
                filename = persoconf.path +"/"+ appmeta.name +"/"+ f
                filedest = appmeta.files[f]["dest"]

                # Remove possible 'home'. The final targz should be extracted
                # at $HOME.
                if filedest.startswith("~/"):
                    filedest = filedest[2:]

                # TODO save and restore owners and permissions
                try:
                    pkg.add(filename, arcname=filedest)
                    logger.info("Adding %s to package" % f)

                except Exception as e:
                    logger.warning( "Failed to add %s to tar package %s: %s"
                                    % (f, pkgname, str(e)) )


        pkg.close()