/*

 * Copyright 2007 Jaroslav Tulach  All Rights Reserved.

 *

 * 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.  Sun designates this

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

 * by Sun 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.com if you need additional information or

 * have any questions.

 */

package org.apidesign.infra.ant;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.FileDescriptor;

import java.io.PrintStream;

import java.io.PrintWriter;

import java.net.InetAddress;

import java.security.Permission;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import junit.framework.AssertionFailedError;

import org.junit.Assert;

/**

 *

 * @author Jaroslav Tulach

 */

final class ExecuteUtils {

    private ExecuteUtils() {

    }

/*    

    final static void execute (String res, String[] args) throws Exception {

        execute (extractResource (res), args);

    }

  */  

    private static ByteArrayOutputStream out;

    private static ByteArrayOutputStream err;

    final static String getStdOut() {

        return out.toString();

    }

    final static String getStdErr() {

        return err.toString();

    }

    final static void execute(File f, String... args) throws Exception {

        // we need security manager to prevent System.exit

        if (! (System.getSecurityManager () instanceof MySecMan)) {

            out = new java.io.ByteArrayOutputStream ();

            err = new java.io.ByteArrayOutputStream ();

            System.setOut (new java.io.PrintStream (out));

            System.setErr (new java.io.PrintStream (err));

            System.setSecurityManager (new MySecMan ());

        }

        MySecMan sec = (MySecMan)System.getSecurityManager();

        // Jesse claims that this is not the right way how the execution

        // of an ant script should be invoked:

        //

        // better IMHO to just run the task directly

        // (setProject() and similar, configure its bean properties, and call

        // execute()), or just make a new Project and initialize it.

        // ant.Main.main is not intended for embedded use. Then you could get rid

        // of the SecurityManager stuff, would be cleaner I think.

        //

        // If I had to write this once again, I would try to follow the

        // "just make a new Project and initialize it", but as this works

        // for me now, I leave it for the time when somebody really

        // needs that...

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

        arr.add ("-f");

        arr.add (f.toString ());

        arr.addAll(Arrays.asList(args));

        arr.add ("-verbose");

        out.reset ();

        err.reset ();

        try {

            sec.setActive(true);

            org.apache.tools.ant.Main.main (arr.toArray(new String[0]));

        } catch (MySecExc ex) {

            Assert.assertNotNull ("The only one to throw security exception is MySecMan and should set exitCode", sec.exitCode);

            ExecutionError.assertExitCode (

                "Execution has to finish without problems",

                sec.exitCode.intValue ()

            );

        } finally {

            sec.setActive(false);

        }

    }

    static class ExecutionError extends AssertionFailedError {

        public final int exitCode;

        public ExecutionError (String msg, int e) {

            super (msg);

            this.exitCode = e;

        }

        public static void assertExitCode (String msg, int e) {

            if (e != 0) {

                throw new ExecutionError (

                    msg + " was: " + e + "\nOutput: " + out.toString () +

                    "\nError: " + err.toString (),

                    e

                );

            }

        }

    }

    private static class MySecExc extends SecurityException {

        @Override

        public void printStackTrace() {

        }

        @Override

        public void printStackTrace(PrintStream ps) {

        }

        public void printStackTrace(PrintWriter ps) {

        }

    }

    private static class MySecMan extends SecurityManager {

        public Integer exitCode;

        private boolean active;

        public void checkExit (int status) {

            if (active) {

                exitCode = new Integer (status);

                throw new MySecExc ();

            }

        }

        public void checkPermission(Permission perm, Object context) {

        }

        public void checkPermission(Permission perm) {

        /*

            if (perm instanceof RuntimePermission) {

                if (perm.getName ().equals ("setIO")) {

                    throw new MySecExc ();

                }

            }

         */

        }

        public void checkMulticast(InetAddress maddr) {

        }

        public void checkAccess (ThreadGroup g) {

        }

        public void checkWrite (String file) {

        }

        public void checkLink (String lib) {

        }

        public void checkExec (String cmd) {

        }

        public void checkDelete (String file) {

        }

        public void checkPackageAccess (String pkg) {

        }

        public void checkPackageDefinition (String pkg) {

        }

        public void checkPropertyAccess (String key) {

        }

        public void checkRead (String file) {

        }

        public void checkSecurityAccess (String target) {

        }

        public void checkWrite(FileDescriptor fd) {

        }

        public void checkListen (int port) {

        }

        public void checkRead(FileDescriptor fd) {

        }

        public void checkMulticast(InetAddress maddr, byte ttl) {

        }

        public void checkAccess (Thread t) {

        }

        public void checkConnect (String host, int port, Object context) {

        }

        public void checkRead (String file, Object context) {

        }

        public void checkConnect (String host, int port) {

        }

        public void checkAccept (String host, int port) {

        }

        public void checkMemberAccess (Class clazz, int which) {

        }

        public void checkSystemClipboardAccess () {

        }

        public void checkSetFactory () {

        }

        public void checkCreateClassLoader () {

        }

        public void checkAwtEventQueueAccess () {

        }

        public void checkPrintJobAccess () {

        }

        public void checkPropertiesAccess () {

        }

        void setActive(boolean b) {

            active = b;

        }

    } // end of MySecMan

}