/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 *  contributor license agreements.  See the NOTICE file distributed with
 *  this work for additional information regarding copyright ownership.
 *  The ASF licenses this file to You 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.
 */
package org.apache.openejb.maven.plugin.runner;

import org.apache.openejb.config.RemoteServer;
import org.apache.openejb.loader.Files;
import org.apache.openejb.loader.IO;
import org.apache.openejb.loader.Zips;
import org.apache.openejb.util.Pipe;
import org.apache.tomee.util.QuickServerXmlParser;

import java.io.File;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Locale;
import java.util.Properties;

import static java.util.Arrays.asList;

public class ExecRunner {
    private static final String SH_BAT_AUTO = "[.sh|.bat]";

    public static void main(final String[] rawArgs) throws Exception {
        final String[] args;
        if (rawArgs == null || rawArgs.length == 0) {
            args = new String[]{"run"};
        } else {
            args = rawArgs;
        }

        final Properties config = new Properties();

        final ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
        final InputStream is = contextClassLoader.getResourceAsStream("configuration.properties");
        if (is != null) {
            config.load(new InputStreamReader(is, "UTF-8"));
            is.close();
        } else {
            throw new IllegalArgumentException("Config not found");
        }

        final String distrib = config.getProperty("distribution");
        final String workingDir = config.getProperty("workingDir");
        final InputStream distribIs = contextClassLoader.getResourceAsStream(distrib);
        File distribOutput = new File(workingDir);
        final File timestampFile = new File(distribOutput, "timestamp.txt");
        final boolean forceDelete = Boolean.getBoolean("tomee.runner.force-delete");
        if (forceDelete
                || !timestampFile.exists()
                || Long.parseLong(IO.slurp(timestampFile).replace(System.getProperty("line.separator"), "")) < Long.parseLong(config.getProperty("timestamp"))) {
            if (forceDelete || timestampFile.exists()) {
                System.out.println("Deleting " + distribOutput.getAbsolutePath());
                Files.delete(distribOutput);
            }
            System.out.println("Extracting tomee to " + distribOutput.getAbsolutePath());
            Zips.unzip(distribIs, distribOutput, false);
            IO.writeString(timestampFile, config.getProperty("timestamp", Long.toString(System.currentTimeMillis())));
        }

        final File[] scripts = new File(distribOutput, "bin").listFiles();
        if (scripts != null) { // dont use filefilter to avoid dependency issue
            for (final File f : scripts) {
                if (f.getName().endsWith(".sh") && !f.canExecute()) {
                    if (!f.setExecutable(true, true)) {
                        System.err.println("Failed make file executable: " + f);
                    }
                }
            }
        }

        String cmd = config.getProperty("command");
        if (cmd.endsWith(SH_BAT_AUTO)) {
            final int lastSlash = cmd.lastIndexOf('/');
            if (lastSlash > 0) {
                final String dir = cmd.substring(0, lastSlash);
                final boolean isWin = System.getProperty("os.name").toLowerCase(Locale.ENGLISH).contains("win");
                final String script = cmd.substring(lastSlash + 1, cmd.length() - SH_BAT_AUTO.length()).replace('/', isWin ? '\\' : '/')
                        + (isWin ? ".bat" : ".sh");
                final File scriptFile = new File(distribOutput, dir + File.separator + script);
                if (!scriptFile.exists()) {
                    throw new IllegalArgumentException("Can't find  " + cmd);
                }
                cmd = scriptFile.getAbsolutePath();
                if (cmd.endsWith(".sh")) {
                    if (!scriptFile.setExecutable(true)) {
                        System.err.println("Failed make script file executable: " + scriptFile);
                    }
                }
            }
        }

        final String additionalArgs = System.getProperty("additionalSystemProperties");

        final Collection<String> params = new ArrayList<String>();
        if ("java".equals(cmd)) {
            final File[] extracted = distribOutput.listFiles();
            if (extracted != null) {
                File newRoot = null;
                for (final File e : extracted) {
                    if (e.isDirectory()) {
                        if (newRoot == null) {
                            newRoot = e;
                        } else {
                            newRoot = null;
                            break;
                        }
                    }
                }
                if (newRoot != null) {
                    distribOutput = newRoot;
                }
            }

            final QuickServerXmlParser parser = QuickServerXmlParser.parse(new File(distribOutput, "conf/server.xml"));

            System.setProperty("openejb.home", distribOutput.getAbsolutePath());
            System.setProperty("server.shutdown.port", parser.stop());
            System.setProperty("server.shutdown.command", config.getProperty("shutdownCommand"));

            final RemoteServer server = new RemoteServer();
            server.setPortStartup(Integer.parseInt(parser.http()));

            if (config.containsKey("additionalClasspath")) {
                server.setAdditionalClasspath(config.getProperty("additionalClasspath"));
            }

            final List<String> jvmArgs = new LinkedList<String>();
            if (additionalArgs != null) {
                Collections.addAll(jvmArgs, additionalArgs.split(" "));
            }
            for (final String k : config.stringPropertyNames()) {
                if (k.startsWith("jvmArg.")) {
                    jvmArgs.add(config.getProperty(k));
                }
            }

            if ("run".equals(args[0])) {
                args[0] = "start";
            }
            try {
                server.start(jvmArgs, args[0], true);
                server.getServer().waitFor();
            } catch (final Exception e) {
                server.destroy();
            }
        } else {
            // TODO: split cmd correctly to support multiple inlined segments in cmd
            if (cmd.endsWith(".bat") && !cmd.startsWith("cmd.exe")) {
                params.add("cmd.exe");
                params.add("/c");
            } // else suppose the user knows what he does
            params.add(cmd);
            params.addAll(asList(args));

            final ProcessBuilder builder = new ProcessBuilder(params.toArray(new String[params.size()])).directory(distribOutput);

            final String existingOpts = System.getenv("CATALINA_OPTS");
            final String catalinaOpts = config.getProperty("catalinaOpts");
            if (catalinaOpts != null || existingOpts != null || additionalArgs != null) { // inherit from existing env
                builder.environment().put("CATALINA_OPTS", identityOrEmpty(catalinaOpts) + " " + identityOrEmpty(existingOpts) + " " + identityOrEmpty(additionalArgs));
            }

            boolean redirectOut = false;
            try { // java >= 7
                ProcessBuilder.class.getDeclaredMethod("inheritIO").invoke(builder);
            } catch (final Throwable th) { // java 6
                redirectOut = true;
            }

            final Process process = builder.start();
            if (redirectOut) {
                Pipe.pipe(process);
            }

            process.waitFor();
        }

        System.out.flush();
        System.err.flush();
        System.out.println("Exited Successfully!");
    }

    private static String identityOrEmpty(final String value) {
        return (value != null ? value : "");
    }

    private ExecRunner() {
        // no-op
    }
}
