# Copyright 2024 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import os
import glob
import subprocess
import sys
import argparse
import logging
import shutil
from generate_pdcs import pdc_gen


def find_pdc2png():
    if os.path.exists(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../build/pdc2png/pdc2png'))):
        pdc2png = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../build/pdc2png/pdc2png'))
    elif os.path.exists(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../build/tools/pdc2png'))):
        pdc2png = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../build/tools/pdc2png'))
    elif os.path.exists(os.path.abspath('./pdc2png')):
        pdc2png = os.path.abspath('./pdc2png')
    else:
        pdc2png = None
        logging.warning("Can't find pdc2png")

    return pdc2png


def set_path_to_svg2pdc():
    root_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
    if os.path.exists(os.path.join(root_dir, 'generate_pdcs/pdc_gen.py')):
        sys.path.insert(0, root_dir)
    elif os.path.exists(os.path.abspath('./generate_pdcs/pdc_gen.py')):
        sys.path.insert(0, os.path.abspath('.'))
    else:
        logging.warning("Can't find generate_pdcs/pdc_gen.py")


def log_exception(filename, exc_type, exc_value, exc_traceback):
    import traceback
    lines = traceback.format_exception(exc_type, exc_value, exc_traceback)
    s = "Exception while processing {}\n".format(filename)
    s += ''.join(lines)
    logging.error(s)


# process_svg_files finds all the SVGs in the root and converts them to individual PDCs and then converts them to PNGs.
# It treats all the subdirectories as sequences, creating one PDC from all the SVGs in that directory and then generates
# PDCs from all the frames in the sequence
def process_svg_files(path):
    set_path_to_svg2pdc()

    files = glob.glob(path + "/*.svg")
    dirs = glob.glob(path + "/*/")

    error_files = []
    for f in files:
        try:
            error_files += pdc_gen.create_pdc_from_path(f,
                                                        None,
                                                        viewbox_size=(0, 0),
                                                        verbose=True,
                                                        duration=0,
                                                        play_count=0,
                                                        precise=True,
                                                        raise_error=True)
        except:
            exc_type, exc_value, exc_traceback = sys.exc_info()
            log_exception(f, exc_type, exc_value, exc_traceback)

    for d in dirs:
        try:
            error_files += pdc_gen.create_pdc_from_path(d,
                                                        None,
                                                        viewbox_size=(0, 0),
                                                        verbose=True,
                                                        duration=33,
                                                        play_count=1,
                                                        precise=True,
                                                        raise_error=True)
        except:
            exc_type, exc_value, exc_traceback = sys.exc_info()
            log_exception(f, exc_type, exc_value, exc_traceback)

    return error_files


# convert all PDC files to PNGs
def process_pdc_files(path):
    pdc2png = find_pdc2png()
    if pdc2png is None:
        return

    pdc_files = glob.glob(path + "/*.pdc") + glob.glob(path + "/*/*.pdc")
    if pdc_files:
        try:
            p = subprocess.Popen([pdc2png] + pdc_files, cwd=os.path.abspath(path), stdout=subprocess.PIPE,
                                 stderr=subprocess.PIPE)
            stdout, stderr = p.communicate()
            if stdout:
                logging.info(stdout)
            if stderr:
                logging.error(stderr)
        except OSError:
            logging.warning("Can't find pdc2png executable")

        for f in pdc_files:
            os.remove(f)
    else:
        print logging.info("No .pdc files found in " + path)


# If any files contain invalid points, the images are moved to the 'failed' subdirectory to highlight this for the
# designers
def copy_error_files_to_failed_subdir(path, error_files):
    if len(error_files) == 0:
        return
    fail_dir = os.path.join(path, "failed")
    if not os.path.exists(fail_dir):
        os.makedirs(fail_dir)
    for f in error_files:
        base = os.path.basename(f)
        dir_name = os.path.dirname(f)
        png_base = '.'.join(base.split('.')[:-1]) + '.png'
        png_error = os.path.join(dir_name, png_base)
        png_copy = os.path.join(fail_dir, png_base)
        logging.debug(png_error + " => " + png_copy)
        try:
            shutil.copy(png_error, png_copy)
        except IOError as e:
            logging.error("Failed to copy {} to failed directory".format(f))


def main(path=None):
    # detect whether the script is running within svg2png.app
    is_app = os.path.dirname(os.path.abspath('../')).split('/')[-1] == 'svg2png.app'

    if is_app:
        # if this is running as the app, the SVG files will be placed in the same directory as 'svg2png.app'
        path = '../../../'
        # output errors to log file when running as an app
        logging.basicConfig(format='%(asctime)s - %(levelname)s - %(message)s', filename=os.path.join(path, 'log.log'),
                            level=logging.DEBUG)
    else:
        logging.basicConfig(format='%(asctime)s - %(levelname)s - %(message)s', level=logging.DEBUG)

    if path:
        path = os.path.abspath(path)
        error_files = process_svg_files(path)
        process_pdc_files(path)
        copy_error_files_to_failed_subdir(path, error_files)
    else:
        logging.warning('No path specified')

if __name__ == '__main__':
    # path argument is provided so that this can be used as the standalone script (if no path is provided nothing will
    # happen when invoked as script
    parser = argparse.ArgumentParser()
    parser.add_argument('path', type=str, nargs='?',
                        help="Path to svg file or directory (with multiple svg files)")
    args = parser.parse_args()
    main(args.path)