/*

 * Copyright (c) 2008, 2011, Oracle and/or its affiliates. All rights reserved.

 * 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.

 *

 * 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.

 */

import java.nio.file.attribute.BasicFileAttributes;

import java.nio.file.FileVisitResult;

import java.nio.file.SimpleFileVisitor;

import javax.tools.ToolProvider;

import java.io.BufferedReader;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.PrintStream;

import java.nio.file.Files;

import java.nio.file.Path;

import java.util.ArrayList;

import java.util.List;

import java.util.Map;

import javax.tools.JavaCompiler;

import static java.nio.file.StandardCopyOption.*;

/**

 * This class provides some common utilities for the launcher tests.

 */

public enum TestHelper {

    INSTANCE;

    static final String JAVAHOME = System.getProperty("java.home");

    static final boolean isSDK = JAVAHOME.endsWith("jre");

    static final String javaCmd;

    static final String java64Cmd;

    static final String javacCmd;

    static final JavaCompiler compiler;

    static final boolean debug = Boolean.getBoolean("TestHelper.Debug");

    static final boolean isWindows =

            System.getProperty("os.name", "unknown").startsWith("Windows");

    static final boolean is64Bit =

            System.getProperty("sun.arch.data.model").equals("64");

    static final boolean is32Bit =

            System.getProperty("sun.arch.data.model").equals("32");

    static final boolean isSolaris =

            System.getProperty("os.name", "unknown").startsWith("SunOS");

    static final boolean isLinux =

            System.getProperty("os.name", "unknown").startsWith("Linux");

    static final boolean isDualMode = isSolaris;

    static final boolean isSparc = System.getProperty("os.arch").startsWith("sparc");

    static int testExitValue = 0;

    static {

        if (is64Bit && is32Bit) {

            throw new RuntimeException("arch model cannot be both 32 and 64 bit");

        }

        if (!is64Bit && !is32Bit) {

            throw new RuntimeException("arch model is not 32 or 64 bit ?");

        }

        compiler = ToolProvider.getSystemJavaCompiler();

        File binDir = (isSDK) ? new File((new File(JAVAHOME)).getParentFile(), "bin")

            : new File(JAVAHOME, "bin");

        File javaCmdFile = (isWindows)

                ? new File(binDir, "java.exe")

                : new File(binDir, "java");

        javaCmd = javaCmdFile.getAbsolutePath();

        if (!javaCmdFile.canExecute()) {

        }

        File javacCmdFile = (isWindows)

                ? new File(binDir, "javac.exe")

                : new File(binDir, "javac");

        javacCmd = javacCmdFile.getAbsolutePath();

        if (!javacCmdFile.canExecute()) {

        }

        if (isSolaris) {

            File sparc64BinDir = new File(binDir,isSparc ? "sparcv9" : "amd64");

            File java64CmdFile= new File(sparc64BinDir, "java");

            if (java64CmdFile.exists() && java64CmdFile.canExecute()) {

                java64Cmd = java64CmdFile.getAbsolutePath();

            } else {

                java64Cmd = null;

            }

        } else {

            java64Cmd = null;

        }

    }

    /*

     * is a dual mode available in the test jdk

     */

    static boolean dualModePresent() {

        return isDualMode && java64Cmd != null;

    }

    /*

     * usually the jre/lib/arch-name is the same as os.arch, except for x86.

     */

    static String getJreArch() {

        String arch = System.getProperty("os.arch");

        return arch.equals("x86") ? "i386" : arch;

    }

    /*

     * get the complementary jre arch ie. if sparc then return sparcv9 and

     * vice-versa.

     */

    static String getComplementaryJreArch() {

        String arch = System.getProperty("os.arch");

        if (arch != null) {

            switch (arch) {

                case "sparc":

                    return "sparcv9";

                case "sparcv9":

                    return "sparc";

                case "x86":

                    return "amd64";

                case "amd64":

                    return "i386";

            }

        }

        return null;

    }

    /*

     * A convenience method to create a jar with jar file name and defs

     */

    static void createJar(File jarName, String... mainDefs)

            throws FileNotFoundException{

        createJar(null, jarName, new File("Foo"), mainDefs);

    }

    /*

     * A convenience method to create a java file, compile and jar it up, using

     * the sole class file name in the jar, as the Main-Class attribute value.

     */

    static void createJar(File jarName, File mainClass, String... mainDefs)

            throws FileNotFoundException {

            createJar(null, jarName, mainClass, mainDefs);

    }

    /*

     * A generic jar file creator to create a java file, compile it

     * and jar it up, a specific Main-Class entry name in the

     * manifest can be specified or a null to use the sole class file name

     * as the Main-Class attribute value.

     */

    static void createJar(String mEntry, File jarName, File mainClass,

            String... mainDefs) throws FileNotFoundException {

        if (jarName.exists()) {

            jarName.delete();

        }

        try (PrintStream ps = new PrintStream(new FileOutputStream(mainClass + ".java"))) {

            ps.println("public class Foo {");

            if (mainDefs != null) {

                for (String x : mainDefs) {

                    ps.println(x);

                }

            }

            ps.println("}");

        }

        String compileArgs[] = {

            mainClass + ".java"

        };

        if (compiler.run(null, null, null, compileArgs) != 0) {

            throw new RuntimeException("compilation failed " + mainClass + ".java");

        }

        if (mEntry == null) {

            mEntry = mainClass.getName();

        }

        String jarArgs[] = {

            (debug) ? "cvfe" : "cfe",

            jarName.getAbsolutePath(),

            mEntry,

            mainClass.getName() + ".class"

        };

        createJar(jarArgs);

    }

   static void createJar(String... args) {

        sun.tools.jar.Main jarTool =

                new sun.tools.jar.Main(System.out, System.err, "JarCreator");

        if (!jarTool.run(args)) {

            String message = "jar creation failed with command:";

            for (String x : args) {

                message = message.concat(" " + x);

            }

            throw new RuntimeException(message);

        }

   }

   static void copyFile(File src, File dst) throws IOException {

        Path parent = dst.toPath().getParent();

        if (parent != null) {

            Files.createDirectories(parent);

        }

        Files.copy(src.toPath(), dst.toPath(), COPY_ATTRIBUTES, REPLACE_EXISTING);

    }

    static void recursiveDelete(File target) throws IOException {

        if (!target.exists()) {

            return;

        }

        Files.walkFileTree(target.toPath(), new SimpleFileVisitor<Path>() {

            @Override

            public FileVisitResult postVisitDirectory(Path dir, IOException exc) {

                try {

                    Files.deleteIfExists(dir);

                } catch (IOException ex) {

                    System.out.println("Error: could not delete: " + dir.toString());

                    System.out.println(ex.getMessage());

                    return FileVisitResult.TERMINATE;

                }

                return FileVisitResult.CONTINUE;

            }

            @Override

            public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) {

                try {

                    Files.deleteIfExists(file);

                } catch (IOException ex) {

                    System.out.println("Error: could not delete: " + file.toString());

                    System.out.println(ex.getMessage());

                    return FileVisitResult.TERMINATE;

                }

                return FileVisitResult.CONTINUE;

            }

        });

    }

    static TestResult doExec(String...cmds) {

        return doExec(null, cmds);

    }

    /*

     * A method which executes a java cmd and returns the results in a container

     */

    static TestResult doExec(Map<String, String> envToSet, String...cmds) {

        String cmdStr = "";

        for (String x : cmds) {

            cmdStr = cmdStr.concat(x + " ");

        }

        ProcessBuilder pb = new ProcessBuilder(cmds);

        Map<String, String> env = pb.environment();

        if (envToSet != null) {

            env.putAll(envToSet);

        }

        BufferedReader rdr = null;

        try {

            List<String> outputList = new ArrayList<>();

            pb.redirectErrorStream(true);

            Process p = pb.start();

            rdr = new BufferedReader(new InputStreamReader(p.getInputStream()));

            String in = rdr.readLine();

            while (in != null) {

                outputList.add(in);

                in = rdr.readLine();

            }

            p.waitFor();

            p.destroy();

            return new TestHelper.TestResult(cmdStr, p.exitValue(), outputList,

                    env, new Throwable("current stack of the test"));

        } catch (Exception ex) {

            ex.printStackTrace();

            throw new RuntimeException(ex.getMessage());

        }

    }

    /*

     * A class to encapsulate the test results and stuff, with some ease

     * of use methods to check the test results.

     */

    static class TestResult {

        StringBuilder status;

        int exitValue;

        List<String> testOutput;

        Map<String, String> env;

        Throwable t;

        public TestResult(String str, int rv, List<String> oList,

                Map<String, String> env, Throwable t) {

            status = new StringBuilder("Executed command: " + str + "\n");

            exitValue = rv;

            testOutput = oList;

            this.env = env;

            this.t = t;

        }

        void appendStatus(String x) {

            status = status.append("  " + x + "\n");

        }

        void checkNegative() {

            if (exitValue == 0) {

                appendStatus("Error: test must not return 0 exit value");

                testExitValue++;

            }

        }

        void checkPositive() {

            if (exitValue != 0) {

                appendStatus("Error: test did not return 0 exit value");

                testExitValue++;

            }

        }

        boolean isOK() {

            return exitValue == 0;

        }

        boolean isZeroOutput() {

            if (!testOutput.isEmpty()) {

                appendStatus("Error: No message from cmd please");

                testExitValue++;

                return false;

            }

            return true;

        }

        boolean isNotZeroOutput() {

            if (testOutput.isEmpty()) {

                appendStatus("Error: Missing message");

                testExitValue++;

                return false;

            }

            return true;

        }

        @Override

        public String toString() {

            status.append("++++Begin Test Info++++\n");

            status.append("++++Test Environment++++\n");

            for (String x : env.keySet()) {

                status.append(x).append("=").append(env.get(x)).append("\n");

            }

            status.append("++++Test Output++++\n");

            for (String x : testOutput) {

                appendStatus(x);

            }

            status.append("++++Test Stack Trace++++\n");

            status.append(t.toString());

            for (StackTraceElement e : t.getStackTrace()) {

                status.append(e.toString());

            }

            status.append("++++End of Test Info++++\n");

            return status.toString();

        }

        boolean contains(String str) {

            for (String x : testOutput) {

                if (x.contains(str)) {

                    return true;

                }

            }

            appendStatus("Error: string <" + str + "> not found");

            testExitValue++;

            return false;

        }

        boolean matches(String stringToMatch) {

          for (String x : testOutput) {

                if (x.matches(stringToMatch)) {

                    return true;

                }

            }

            appendStatus("Error: string <" + stringToMatch + "> not found");

            testExitValue++;

            return false;

        }

    }

}