#!/usr/bin/env python
#
# Copyright 2022 Google LLC
#
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os
import shlex
import subprocess
import sys
import tempfile

batchMinify = False

sksl_minify = sys.argv[1]
shared_module = sys.argv[2]
public_module = sys.argv[3]
rt_shader_module = sys.argv[4]
input_root_dir = sys.argv[5]
output_root_dir = sys.argv[6]
# The last arg is a file containing a space seperated list of filenames
input_file = sys.argv[7]
with open(input_file, 'r') as reader:
    all_inputs = shlex.split(reader.read())

inputs = []
for file in all_inputs:
    if (file.endswith(".rts") or file.endswith(".privrts") or file.endswith(".rtcf") or
        file.endswith(".rtb") or file.endswith(".mfrag") or file.endswith(".mvert")):
        inputs.append(file)

def executeWorklist(input, worklist):
    # Invoke sksl-minify, passing in the worklist.
    worklist.close()
    try:
        output = subprocess.check_output([
            sksl_minify, worklist.name], stderr=subprocess.STDOUT).decode('utf-8', errors='ignore')
    except subprocess.CalledProcessError as err:
        if err.returncode != 1:
            print("### " + input + " sksl-minify error:\n")
            print("\n".join(err.output.decode('utf-8', errors='ignore').splitlines()))
            sys.exit(err.returncode)
        pass  # Compile errors (exit code 1) are expected and normal in test code

    # Delete the worklist file now that execution is complete.
    os.remove(worklist.name)

worklist = tempfile.NamedTemporaryFile(suffix='.worklist', delete=False, mode='w')

# Here we loop over the inputs and convert them into a worklist file for sksl-minify.
for input in inputs:
    # Derive the target path from the input filename and remove the extension so it can
    # end with .minified.sksl
    target = input.replace(input_root_dir, output_root_dir)
    target = os.path.splitext(target)[0]
    target_dir = os.path.dirname(target)
    if not os.path.isdir(target_dir):
        os.mkdir(target_dir)

    noExt, ext = os.path.splitext(input)
    head, tail = os.path.split(noExt)

    if ext == '.rts':
        worklist.write("--shader\n")
    elif ext == '.privrts':
        worklist.write("--privshader\n")
    elif ext == '.rtcf':
        worklist.write("--colorfilter\n")
    elif ext == '.rtb':
        worklist.write("--blender\n")
    elif ext == '.mfrag':
        worklist.write("--meshfrag\n")
    elif ext == '.mvert':
        worklist.write("--meshvert\n")

    worklist.write(target + ".minified.sksl\n")
    worklist.write(input + "\n")
    if ext == '.privrts':
        worklist.write(rt_shader_module + "\n")
    worklist.write(public_module + "\n")
    worklist.write(shared_module + "\n\n")

    # Compile items one at a time.
    if not batchMinify:
        executeWorklist(input, worklist)
        worklist = tempfile.NamedTemporaryFile(suffix='.worklist', delete=False, mode='w')

# Compile everything all in one go.
if batchMinify:
    executeWorklist("", worklist)
else:
    worklist.close()
    os.remove(worklist.name)