"""
All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
its licensors.

For complete copyright and license terms please see the LICENSE at the root of this
distribution (the "License"). All use of this software is governed by the License,
or, if provided, by the license below or the license accompanying this file. Do not
remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
"""

import argparse
import logging
import os
from run_launcher_tests import PlatformDriver
import subprocess

logger = logging.getLogger(__name__)


class IOSDriver(PlatformDriver):

    def __init__(self, project_json_path, project_launcher_tests_folder, test_names, screenshots_folder, screenshots_interval, device_udid):
        self.project_json_path = project_json_path
        self.project_launcher_tests_folder = project_launcher_tests_folder
        self.test_names = test_names
        self.screenshots_folder = screenshots_folder
        self.screenshots_interval = screenshots_interval
        self.device_udid = device_udid

    def run_test(self, test_name, map, dynamic_slice, timeout, pass_string, fail_string):

        project = self.read_project_name()
        bundle_id = "com.amazon.lumberyard.{}".format(project)
            
        # Start the process and pass in the test args        
        command_line = ['idevicedebug', '-u', self.device_udid, 'run', bundle_id]
        command_line += ['-ltest_map', map]
        command_line += ['-ltest_slice', dynamic_slice]        
        test_result = self.monitor_process_output(test_name, command_line, pass_string, fail_string, timeout)

        # TODO: Figure out some way to kill the running app. Because we dont know how to do this,
        # we currently have a limitation on iOS we can only run one test at a time.

        # Stop here if we failed.
        if not test_result:
            raise Exception("Test failed.")
    
    def take_screenshot(self, output_path_no_ext):
        # Create the output folder if it is not there
        output_folder = os.path.dirname(output_path_no_ext)
        if not os.path.exists(output_folder):
            os.makedirs(output_folder)
            
        # idevicescreenshot to take a screenshot and save to output path.
        p = subprocess.Popen(['idevicescreenshot', "{}.tiff".format(output_path_no_ext)])
        p.communicate()            
        
def discover_device():
    """ Discover the connected device and get the UDID."""
    logger.info("Getting the connected device UDID ...")
    p = subprocess.Popen(['idevice_id', '--list'], stdout=subprocess.PIPE)
    out, err = p.communicate()
    if not out:
        raise Exception("No output.\nout:{}\nerr:{}\n".format(out, err))
            
    lines = out.splitlines()
    if not len(lines):
        raise Exception("No devices connected.\nout:{}\nerr:{}\n".format(out, err))
    if len(lines) != 1:
        raise Exception("More than one device connected. Use --device-udid\nout:{}\nerr:{}\n".format(out, err))    
    return lines[0]

def main():

    parser = argparse.ArgumentParser(description='Sets up and runs iOS Launcher Tests.')
    parser.add_argument('--project-json-path', required=True, help='Path to the project.json project settings file.')
    parser.add_argument('--project-launcher-tests-folder', required=True, help='Path to the LauncherTests folder in a Project.')
    parser.add_argument('--device-udid', help='The UDID of the iOS device. Will auto detect if just one device is attached.')
    parser.add_argument('--test-names', nargs='+', help='A list of test names to run, default runs all tests.')
    parser.add_argument('--screenshots-folder', default="./temp/iOS/screenshots", help='Output folder for screenshots.')
    parser.add_argument('--screenshots-interval', default=5, help='Time interval between taking screenshots in seconds.')
    args = parser.parse_args()

    logging.basicConfig(level=logging.DEBUG)
    
    device_udid = args.device_udid
    if not device_udid:
        device_udid = discover_device()
    
    logger.info("Running Launcher tests at {} ...".format(args.project_launcher_tests_folder))
    driver = IOSDriver(args.project_json_path, args.project_launcher_tests_folder, args.test_names, args.screenshots_folder, args.screenshots_interval, device_udid)
    driver.run_launcher_tests()
    
if __name__== "__main__":
    main()