#

# Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

# All rights reserved.

# This component and the accompanying materials are made available

# under the terms of "Eclipse Public License v1.0"

# which accompanies this distribution, and is available

# at the URL "http://www.eclipse.org/legal/epl-v10.html".

#

# Initial Contributors:

# Nokia Corporation - initial contribution.

#

# Contributors:

#

# Description: 

#

import sys, os, shutil, imp

import logging

from optparse import OptionParser

log = logging.getLogger()

ROOT_PATH = os.path.abspath(os.path.dirname(__file__))

SOURCE_ROOT = os.path.normpath(os.path.join(ROOT_PATH, '../source'))

SCRIPTS_SOURCE_ROOT = os.path.normpath(os.path.join(SOURCE_ROOT, 'scripts'))

PLUGIN_SOURCE_ROOT = os.path.normpath(os.path.join(SOURCE_ROOT, 'plugins'))

TESTAUTOMATION_ROOT = os.path.normpath(os.path.join(SOURCE_ROOT, 'testautomation'))

assert os.path.exists(SOURCE_ROOT)

assert os.path.exists(SCRIPTS_SOURCE_ROOT)

assert os.path.exists(PLUGIN_SOURCE_ROOT)

assert os.path.exists(TESTAUTOMATION_ROOT)

sys.path.append(TESTAUTOMATION_ROOT)

import testautomation

from testautomation.copy_dir import copy_dir

from testautomation import plugin_utils

import utils

utils.setup_logging('export_bat.log')

def read_export_function_from_file(file_path):

    if not os.path.exists(file_path):

        return None

    m = imp.load_source(

        file_path.replace('\\', '__')

                 .replace('/', '__')

                 .replace(':', '_')

                 .replace('.', '_')

                 .replace(' ', '_'),

        file_path)

    try:

        return m.export_standalone

    except AttributeError:

        return None

def find_egg_file(dir, name, python_version):

    """

    Returns the name of an egg file in the given directory that starts with the

    given name and is for the given Python version.

    >>> find_egg_file('dep-eggs', 'simplejson', '2.5')

    'simplejson-2.0.9-py2.5-win32.egg'

    >>> find_egg_file('dep-eggs', 'simplejson', '2.6')

    'simplejson-2.0.9-py2.6-win32.egg'

    """

    for filename in os.listdir(dir):

        if filename.startswith(name) and 'py' + python_version in filename:

            return filename

    return None

def main(argv):

    # -----------

    # Parse args

    # -----------

    parser = OptionParser()

    parser.add_option("-t", "--target-dir",

                      help="The directory where the test are to be exported.")

    parser.add_option("-p", "--plugin-package",

                      help="The plug-in package for exporting plug-in integration tests.",

                      default=None)

    (options, args) = parser.parse_args()

    if options.target_dir is None:

        parser.error("Target directory must be given")

    PYTHON_VERSION = utils.get_python_version()

    TARGET_PATH = options.target_dir

    PLUGIN_PACKAGE = options.plugin_package

    log.info("Target directory: %s" % TARGET_PATH)

    log.info("Plug-in package:  %r" % PLUGIN_PACKAGE)

    log.info("Python version:   %s" % PYTHON_VERSION)

    log.info("Cleaning target directory...")

    utils.recreate_dir(TARGET_PATH)

    # -------------------------

    # Export script test files

    # -------------------------

    log.info("Copying script test files...")

    SCRIPT_TESTS_DIR = os.path.join(SCRIPTS_SOURCE_ROOT, 'tests')

    assert os.path.exists(SCRIPT_TESTS_DIR)

    copy_dir(source_dir             = SCRIPT_TESTS_DIR,

             target_dir             = os.path.join(TARGET_PATH, 'tests'),

             dir_ignore_functions   = [lambda d: d in ('.svn', 'temp', 'export_standalone')],

             file_ignore_functions  = [lambda f: f == 'cone.log' or f.endswith('.pyc')])

    log.info("Copying script test overlay files...")

    copy_dir(source_dir = os.path.join(ROOT_PATH, "export-bat/scripts-tests-overlay"),

             target_dir = TARGET_PATH,

             dir_ignore_functions = [lambda d: d  == '.svn'])

    # --------------------------------------

    # Export plug-in integration test files

    # --------------------------------------

    log.info("Exporting plug-in integration test files...")

    subpaths_by_package = plugin_utils.find_plugin_package_subpaths(PLUGIN_SOURCE_ROOT, 'integration-test', PLUGIN_PACKAGE)

    for package_name, tests_path in subpaths_by_package:

        log.debug("  Package: %s" % package_name)

        log.debug("  Path:    %s" % tests_path)

        log.debug("  Copying test files...")

        target_path = os.path.join(TARGET_PATH, 'plugin-tests', package_name + '_tests')

        copy_dir(source_dir             = tests_path,

                 target_dir             = target_path,

                 dir_ignore_functions   = [lambda d: d in ('.svn', 'temp')],

                 file_ignore_functions  = [lambda f: f in ('cone.log', 'export_standalone.py') or f.endswith('.pyc')])

        log.debug("  Copying overlay files...")

        overlay_path = os.path.join('export-bat/plugin-integration-test-overlay')

        copy_dir(source_dir             = overlay_path,

                 target_dir             = target_path,

                 dir_ignore_functions   = [lambda d: d == '.svn'])

        log.debug("  Exporting extra data...")

        func = read_export_function_from_file(os.path.join(tests_path, 'export_standalone.py'))

        if func:

            log.debug("  Executing export function...")

            func(target_path)

    TARGET_EGGS_DIR = os.path.join(TARGET_PATH, 'eggs')

    # ---------------------------

    # Copy needed dependency eggs

    # ---------------------------

    log.info("Copying library eggs...")

    DEP_EGGS_DIR = os.path.normpath(os.path.join(ROOT_PATH, '../dep-eggs'))

    assert os.path.isdir(DEP_EGGS_DIR)

    DEPENDENCIES = ['simplejson']

    for dep in DEPENDENCIES:

        egg_file_name = find_egg_file(DEP_EGGS_DIR, dep, PYTHON_VERSION)

        if egg_file_name is None:

            log.critical("Could not find egg file for dependency '%s' from '%s'" % (dep, DEP_EGGS_DIR))

            return 1

        source_path = os.path.join(DEP_EGGS_DIR, egg_file_name)

        target_path = os.path.join(TARGET_EGGS_DIR, egg_file_name)

        utils.copy_file(source_path, target_path)

    # ------------------

    # Build needed eggs

    # ------------------

    log.info("Building eggs...")

    utils.build_egg(os.path.join(SOURCE_ROOT), TARGET_EGGS_DIR)

    utils.build_egg(os.path.join(SOURCE_ROOT, 'testautomation'), TARGET_EGGS_DIR)

    return 0

if __name__ == "__main__":

    sys.exit(main(sys.argv))