/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package com.sun.java.util.jar.pack;

import java.io.BufferedInputStream;

import java.io.BufferedOutputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.io.PrintStream;

import java.text.MessageFormat;

import java.nio.file.Files;

import java.nio.file.Path;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.ListIterator;

import java.util.Map;

import java.util.Properties;

import java.util.ResourceBundle;

import java.util.SortedMap;

import java.util.TreeMap;

import java.util.jar.JarFile;

import java.util.jar.JarOutputStream;

import java.util.jar.Pack200;

import java.util.zip.GZIPInputStream;

import java.util.zip.GZIPOutputStream;

/** Command line interface for Pack200.

 */

class Driver {

        private static final ResourceBundle RESOURCE =

                ResourceBundle.getBundle("com.sun.java.util.jar.pack.DriverResource");

    public static void main(String[] ava) throws IOException {

        List<String> av = new ArrayList<>(Arrays.asList(ava));

        boolean doPack   = true;

        boolean doUnpack = false;

        boolean doRepack = false;

        boolean doZip = true;

        String logFile = null;

        String verboseProp = Utils.DEBUG_VERBOSE;

        {

            // Non-standard, undocumented "--unpack" switch enables unpack mode.

            String arg0 = av.isEmpty() ? "" : av.get(0);

            switch (arg0) {

                case "--pack":

                av.remove(0);

                    break;

                case "--unpack":

                av.remove(0);

                doPack = false;

                doUnpack = true;

                    break;

            }

        }

        // Collect engine properties here:

        Map<String,String> engProps = new HashMap<>();

        engProps.put(verboseProp, System.getProperty(verboseProp));

        String optionMap;

        String[] propTable;

        if (doPack) {

            optionMap = PACK200_OPTION_MAP;

            propTable = PACK200_PROPERTY_TO_OPTION;

        } else {

            optionMap = UNPACK200_OPTION_MAP;

            propTable = UNPACK200_PROPERTY_TO_OPTION;

        }

        // Collect argument properties here:

        Map<String,String> avProps = new HashMap<>();

        try {

            for (;;) {

                String state = parseCommandOptions(av, optionMap, avProps);

                // Translate command line options to Pack200 properties:

            eachOpt:

                for (Iterator<String> opti = avProps.keySet().iterator();

                     opti.hasNext(); ) {

                    String opt = opti.next();

                    String prop = null;

                    for (int i = 0; i < propTable.length; i += 2) {

                        if (opt.equals(propTable[1+i])) {

                            prop = propTable[0+i];

                            break;

                        }

                    }

                    if (prop != null) {

                        String val = avProps.get(opt);

                        opti.remove();  // remove opt from avProps

                        if (!prop.endsWith(".")) {

                            // Normal string or boolean.

                            if (!(opt.equals("--verbose")

                                  || opt.endsWith("="))) {

                                // Normal boolean; convert to T/F.

                                boolean flag = (val != null);

                                if (opt.startsWith("--no-"))

                                    flag = !flag;

                                val = flag? "true": "false";

                            }

                            engProps.put(prop, val);

                        } else if (prop.contains(".attribute.")) {

                            for (String val1 : val.split("\0")) {

                                String[] val2 = val1.split("=", 2);

                                engProps.put(prop+val2[0], val2[1]);

                            }

                        } else {

                            // Collection property: pack.pass.file.cli.NNN

                            int idx = 1;

                            for (String val1 : val.split("\0")) {

                                String prop1;

                                do {

                                    prop1 = prop+"cli."+(idx++);

                                } while (engProps.containsKey(prop1));

                                engProps.put(prop1, val1);

                            }

                        }

                    }

                }

                // See if there is any other action to take.

                if ("--config-file=".equals(state)) {

                    String propFile = av.remove(0);

                    Properties fileProps = new Properties();

                    try (InputStream propIn = new FileInputStream(propFile)) {

                        fileProps.load(propIn);

                    }

                    if (engProps.get(verboseProp) != null)

                        fileProps.list(System.out);

                    for (Map.Entry<Object,Object> me : fileProps.entrySet()) {

                        engProps.put((String) me.getKey(), (String) me.getValue());

                    }

                } else if ("--version".equals(state)) {

                    return;

                } else if ("--help".equals(state)) {

                    printUsage(doPack, true, System.out);

                    return;

                } else {

                    break;

                }

            }

        } catch (IllegalArgumentException ee) {

                System.err.println(MessageFormat.format(RESOURCE.getString(DriverResource.BAD_ARGUMENT), ee));

            printUsage(doPack, false, System.err);

            System.exit(2);

            return;

        }

        // Deal with remaining non-engine properties:

        for (String opt : avProps.keySet()) {

            String val = avProps.get(opt);

            switch (opt) {

                case "--repack":

                    doRepack = true;

                    break;

                case "--no-gzip":

                    doZip = (val == null);

                    break;

                case "--log-file=":

                    logFile = val;

                    break;

                default:

                    throw new InternalError(MessageFormat.format(

                            RESOURCE.getString(DriverResource.BAD_OPTION),

                            opt, avProps.get(opt)));

            }

        }

        if (logFile != null && !logFile.equals("")) {

            if (logFile.equals("-")) {

                System.setErr(System.out);

            } else {

                OutputStream log = new FileOutputStream(logFile);

                //log = new BufferedOutputStream(out);

                System.setErr(new PrintStream(log));

            }

        }

        boolean verbose = (engProps.get(verboseProp) != null);

        String packfile = "";

        if (!av.isEmpty())

            packfile = av.remove(0);

        String jarfile = "";

        if (!av.isEmpty())

            jarfile = av.remove(0);

        String newfile = "";  // output JAR file if --repack

        String bakfile = "";  // temporary backup of input JAR

        String tmpfile = "";  // temporary file to be deleted

        if (doRepack) {

            // The first argument is the target JAR file.

            // (Note:  *.pac is nonstandard, but may be necessary

            // if a host OS truncates file extensions.)

            if (packfile.toLowerCase().endsWith(".pack") ||

                packfile.toLowerCase().endsWith(".pac") ||

                packfile.toLowerCase().endsWith(".gz")) {

                System.err.println(MessageFormat.format(

                        RESOURCE.getString(DriverResource.BAD_REPACK_OUTPUT),

                        packfile));

                printUsage(doPack, false, System.err);

                System.exit(2);

            }

            newfile = packfile;

            // The optional second argument is the source JAR file.

            if (jarfile.equals("")) {

                // If only one file is given, it is the only JAR.

                // It serves as both input and output.

                jarfile = newfile;

            }

            tmpfile = createTempFile(newfile, ".pack").getPath();

            packfile = tmpfile;

            doZip = false;  // no need to zip the temporary file

        }

        if (!av.isEmpty()

            // Accept jarfiles ending with .jar or .zip.

            // Accept jarfile of "-" (stdout), but only if unpacking.

            || !(jarfile.toLowerCase().endsWith(".jar")

                 || jarfile.toLowerCase().endsWith(".zip")

                 || (jarfile.equals("-") && !doPack))) {

            printUsage(doPack, false, System.err);

            System.exit(2);

            return;

        }

        if (doRepack)

            doPack = doUnpack = true;

        else if (doPack)

            doUnpack = false;

        Pack200.Packer jpack = Pack200.newPacker();

        Pack200.Unpacker junpack = Pack200.newUnpacker();

        jpack.properties().putAll(engProps);

        junpack.properties().putAll(engProps);

        if (doRepack && newfile.equals(jarfile)) {

            String zipc = getZipComment(jarfile);

            if (verbose && zipc.length() > 0)

                System.out.println(MessageFormat.format(RESOURCE.getString(DriverResource.DETECTED_ZIP_COMMENT), zipc));

            if (zipc.indexOf(Utils.PACK_ZIP_ARCHIVE_MARKER_COMMENT) >= 0) {

                    System.out.println(MessageFormat.format(RESOURCE.getString(DriverResource.SKIP_FOR_REPACKED), jarfile));

                        doPack = false;

                        doUnpack = false;

                        doRepack = false;

            }

        }

        try {

            if (doPack) {

                // Mode = Pack.

                JarFile in = new JarFile(new File(jarfile));

                OutputStream out;

                // Packfile must be -, *.gz, *.pack, or *.pac.

                if (packfile.equals("-")) {

                    out = System.out;

                    // Send warnings, etc., to stderr instead of stdout.

                    System.setOut(System.err);

                } else if (doZip) {

                    if (!packfile.endsWith(".gz")) {

                    System.err.println(MessageFormat.format(RESOURCE.getString(DriverResource.WRITE_PACK_FILE), packfile));

                        printUsage(doPack, false, System.err);

                        System.exit(2);

                    }

                    out = new FileOutputStream(packfile);

                    out = new BufferedOutputStream(out);

                    out = new GZIPOutputStream(out);

                } else {

                    if (!packfile.toLowerCase().endsWith(".pack") &&

                            !packfile.toLowerCase().endsWith(".pac")) {

                        System.err.println(MessageFormat.format(RESOURCE.getString(DriverResource.WRITE_PACKGZ_FILE),packfile));

                        printUsage(doPack, false, System.err);

                        System.exit(2);

                    }

                    out = new FileOutputStream(packfile);

                    out = new BufferedOutputStream(out);

                }

                jpack.pack(in, out);

                //in.close();  // p200 closes in but not out

                out.close();

            }

            if (doRepack && newfile.equals(jarfile)) {

                // If the source and destination are the same,

                // we will move the input JAR aside while regenerating it.

                // This allows us to restore it if something goes wrong.

                File bakf = createTempFile(jarfile, ".bak");

                // On Windows target must be deleted see 4017593

                bakf.delete();

                boolean okBackup = new File(jarfile).renameTo(bakf);

                if (!okBackup) {

                        throw new Error(MessageFormat.format(RESOURCE.getString(DriverResource.SKIP_FOR_MOVE_FAILED),bakfile));

                } else {

                    // Open jarfile recovery bracket.

                    bakfile = bakf.getPath();

                }

            }

            if (doUnpack) {

                // Mode = Unpack.

                InputStream in;

                if (packfile.equals("-"))

                    in = System.in;

                else

                    in = new FileInputStream(new File(packfile));

                BufferedInputStream inBuf = new BufferedInputStream(in);

                in = inBuf;

                if (Utils.isGZIPMagic(Utils.readMagic(inBuf))) {

                    in = new GZIPInputStream(in);

                }

                String outfile = newfile.equals("")? jarfile: newfile;

                OutputStream fileOut;

                if (outfile.equals("-"))

                    fileOut = System.out;

                else

                    fileOut = new FileOutputStream(outfile);

                fileOut = new BufferedOutputStream(fileOut);

                try (JarOutputStream out = new JarOutputStream(fileOut)) {

                    junpack.unpack(in, out);

                    // p200 closes in but not out

                }

                // At this point, we have a good jarfile (or newfile, if -r)

            }

            if (!bakfile.equals("")) {

                        // On success, abort jarfile recovery bracket.

                        new File(bakfile).delete();

                        bakfile = "";

            }

        } finally {

            // Close jarfile recovery bracket.

            if (!bakfile.equals("")) {

                File jarFile = new File(jarfile);

                jarFile.delete(); // Win32 requires this, see above

                new File(bakfile).renameTo(jarFile);

            }

            // In all cases, delete temporary *.pack.

            if (!tmpfile.equals(""))

                new File(tmpfile).delete();

        }

    }

    private static

    File createTempFile(String basefile, String suffix) throws IOException {

        File base = new File(basefile);

        String prefix = base.getName();

        if (prefix.length() < 3)  prefix += "tmp";

        File where = (base.getParentFile() == null && suffix.equals(".bak"))

                ? new File(".").getAbsoluteFile()

                : base.getParentFile();

        Path tmpfile = (where == null)

                ? Files.createTempFile(prefix, suffix)

                : Files.createTempFile(where.toPath(), prefix, suffix);

        return tmpfile.toFile();

    }

    private static

    void printUsage(boolean doPack, boolean full, PrintStream out) {

        String prog = doPack ? "pack200" : "unpack200";

        String[] packUsage = (String[])RESOURCE.getObject(DriverResource.PACK_HELP);

        String[] unpackUsage = (String[])RESOURCE.getObject(DriverResource.UNPACK_HELP);

        String[] usage = doPack? packUsage: unpackUsage;

        for (int i = 0; i < usage.length; i++) {

            out.println(usage[i]);

            if (!full) {

            out.println(MessageFormat.format(RESOURCE.getString(DriverResource.MORE_INFO), prog));

                break;

            }

        }

    }

    private static

        String getZipComment(String jarfile) throws IOException {

        byte[] tail = new byte[1000];

        long filelen = new File(jarfile).length();

        if (filelen <= 0)  return "";

        long skiplen = Math.max(0, filelen - tail.length);

        try (InputStream in = new FileInputStream(new File(jarfile))) {

            in.skip(skiplen);

            in.read(tail);

            for (int i = tail.length-4; i >= 0; i--) {

                if (tail[i+0] == 'P' && tail[i+1] == 'K' &&

                    tail[i+2] ==  5  && tail[i+3] ==  6) {

                    // Skip sig4, disks4, entries4, clen4, coff4, cmt2

                    i += 4+4+4+4+4+2;

                    if (i < tail.length)

                        return new String(tail, i, tail.length-i, "UTF8");

                    return "";

                }

            }

            return "";

        }

    }

    private static final String PACK200_OPTION_MAP =

        (""

         +"--repack                 $ \n  -r +>- @--repack              $ \n"

         +"--no-gzip                $ \n  -g +>- @--no-gzip             $ \n"

         +"--strip-debug            $ \n  -G +>- @--strip-debug         $ \n"

         +"--no-keep-file-order     $ \n  -O +>- @--no-keep-file-order  $ \n"

         +"--segment-limit=      *> = \n  -S +>  @--segment-limit=      = \n"

         +"--effort=             *> = \n  -E +>  @--effort=             = \n"

         +"--deflate-hint=       *> = \n  -H +>  @--deflate-hint=       = \n"

         +"--modification-time=  *> = \n  -m +>  @--modification-time=  = \n"

         +"--pass-file=        *> &\0 \n  -P +>  @--pass-file=        &\0 \n"

         +"--unknown-attribute=  *> = \n  -U +>  @--unknown-attribute=  = \n"

         +"--class-attribute=  *> &\0 \n  -C +>  @--class-attribute=  &\0 \n"

         +"--field-attribute=  *> &\0 \n  -F +>  @--field-attribute=  &\0 \n"

         +"--method-attribute= *> &\0 \n  -M +>  @--method-attribute= &\0 \n"

         +"--code-attribute=   *> &\0 \n  -D +>  @--code-attribute=   &\0 \n"

         +"--config-file=      *>   . \n  -f +>  @--config-file=        . \n"

         // Negative options as required by CLIP:

         +"--no-strip-debug  !--strip-debug         \n"

         +"--gzip            !--no-gzip             \n"

         +"--keep-file-order !--no-keep-file-order  \n"

         // Non-Standard Options

         +"--verbose                $ \n  -v +>- @--verbose             $ \n"

         +"--quiet        !--verbose  \n  -q +>- !--verbose               \n"

         +"--log-file=           *> = \n  -l +>  @--log-file=           = \n"

         //+"--java-option=      *> = \n  -J +>  @--java-option=        = \n"

         +"--version                . \n  -V +>  @--version             . \n"

         +"--help               . \n  -? +> @--help . \n  -h +> @--help . \n"

         // Termination:

         +"--           . \n"  // end option sequence here

         +"-   +?    >- . \n"  // report error if -XXX present; else use stdout

         );

    // Note: Collection options use "\0" as a delimiter between arguments.

    // For Java version of unpacker (used for testing only):

    private static final String UNPACK200_OPTION_MAP =

        (""

         +"--deflate-hint=       *> = \n  -H +>  @--deflate-hint=       = \n"

         +"--verbose                $ \n  -v +>- @--verbose             $ \n"

         +"--quiet        !--verbose  \n  -q +>- !--verbose               \n"

         +"--remove-pack-file       $ \n  -r +>- @--remove-pack-file    $ \n"

         +"--log-file=           *> = \n  -l +>  @--log-file=           = \n"

         +"--config-file=        *> . \n  -f +>  @--config-file=        . \n"

         // Termination:

         +"--           . \n"  // end option sequence here

         +"-   +?    >- . \n"  // report error if -XXX present; else use stdin

         +"--version                . \n  -V +>  @--version             . \n"

         +"--help               . \n  -? +> @--help . \n  -h +> @--help . \n"

         );

    private static final String[] PACK200_PROPERTY_TO_OPTION = {

        Pack200.Packer.SEGMENT_LIMIT, "--segment-limit=",

        Pack200.Packer.KEEP_FILE_ORDER, "--no-keep-file-order",

        Pack200.Packer.EFFORT, "--effort=",

        Pack200.Packer.DEFLATE_HINT, "--deflate-hint=",

        Pack200.Packer.MODIFICATION_TIME, "--modification-time=",

        Pack200.Packer.PASS_FILE_PFX, "--pass-file=",

        Pack200.Packer.UNKNOWN_ATTRIBUTE, "--unknown-attribute=",

        Pack200.Packer.CLASS_ATTRIBUTE_PFX, "--class-attribute=",

        Pack200.Packer.FIELD_ATTRIBUTE_PFX, "--field-attribute=",

        Pack200.Packer.METHOD_ATTRIBUTE_PFX, "--method-attribute=",

        Pack200.Packer.CODE_ATTRIBUTE_PFX, "--code-attribute=",

        //Pack200.Packer.PROGRESS, "--progress=",

        Utils.DEBUG_VERBOSE, "--verbose",

        Utils.COM_PREFIX+"strip.debug", "--strip-debug",

    };

    private static final String[] UNPACK200_PROPERTY_TO_OPTION = {

        Pack200.Unpacker.DEFLATE_HINT, "--deflate-hint=",

        //Pack200.Unpacker.PROGRESS, "--progress=",

        Utils.DEBUG_VERBOSE, "--verbose",

        Utils.UNPACK_REMOVE_PACKFILE, "--remove-pack-file",

    };

    /*-*

     * Remove a set of command-line options from args,

     * storing them in the map in a canonicalized form.

     * <p>

     * The options string is a newline-separated series of

     * option processing specifiers.