# 
# Copyright (c) 2022 DB Netz AG and others.
#  
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Eclipse Public License v2.0
# which accompanies this distribution, and is available at
# http://www.eclipse.org/legal/epl-v20.html
#
import zipfile
import sys
import subprocess
import os
import re

if len(sys.argv) < 3:
    print("Usage: process.py in.zip outfile")
    sys.exit(-1)

def process_dependency(dep):
    result = re.search(r"p2.eclipse.plugin:maven\.([a-z0-9-.]+)\.artifact\.([a-z0-9-.]+):eclipse-plugin:([0-9\.]+):system", dep)
    if not result:
        return dep 
    elif result.group(1) == 'org.apache.xmlgraphics' and result.group(2) == 'fop':
        # Special case for fop: There is no third component in the version, however maven generates one
        result = re.search(r"p2.eclipse.plugin:maven\.([a-z-.]+)\.artifact\.([a-z-.]+):eclipse-plugin:([0-9\.]+)\.0:system", dep)
        return f"{result.group(1)}:{result.group(2)}:{result.group(3)}:compile"
    else:
        return f"{result.group(1)}:{result.group(2)}:{result.group(3)}:compile"

def process_output(output): 
    return re.sub(r"(org.eclipse.set.*)/(\d+\.\d+\.\d+)\.[0-9]+,", r"\1/\2,", output)

with zipfile.ZipFile(sys.argv[1], 'r') as zf:
    for filename in zf.namelist():
        print("Processing", filename)

        data = zf.read(filename)
        data = "\n".join(map(process_dependency, data.decode("utf-8").splitlines()))

        # npm lock files must be called exactly package-lock.json for Dash to process
        fname = 'package-lock.json' if filename.endswith('.json') else filename
        with open(fname, 'w') as f:
            f.write(data)
        args = 'java -Djava.net.useSystemProxies=true -jar dash.jar -summary DEPENDENCIES ' + fname
        subprocess.run(args, shell=True, check=False)
        os.remove(fname)

        # append to outfile 
        with open('DEPENDENCIES', 'r') as infile:
            with open(sys.argv[2], 'a+') as outfile:
                outfile.write(process_output(infile.read()))
        os.remove("DEPENDENCIES")