/*

 * Copyright (c) 2016, 2017, 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.

 */

/*

 * @test

 * @bug 8167237

 * @summary test that both old style command line options and new gnu style

 *          command line options work with the --release option whether or

 *          not the --release option is preceded by a file name.

 * @library /test/lib

 * @modules jdk.jartool/sun.tools.jar

 * @build jdk.test.lib.Platform

 *        jdk.test.lib.util.FileUtils

 * @run testng ReleaseBeforeFiles

 */

import org.testng.Assert;

import org.testng.annotations.AfterMethod;

import org.testng.annotations.BeforeMethod;

import org.testng.annotations.Test;

import java.io.IOException;

import java.io.UncheckedIOException;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.Arrays;

import java.util.stream.Stream;

import jdk.test.lib.util.FileUtils;

public class ReleaseBeforeFiles {

    private Runnable onCompletion;

    @BeforeMethod

    public void reset() {

        onCompletion = null;

    }

    @AfterMethod

    public void run() {

        if (onCompletion != null) {

            onCompletion.run();

        }

    }

    @Test  // passes before bug fix

    public void test1() throws IOException {

        mkdir("test1");

        touch("test1/testfile1");

        jar("cf test.jar --release 9 test1");

        jar("tf test.jar");

        rm("test.jar test1");

    }

    @Test  // fails before bug fix

    public void test2() throws IOException {

        System.out.println("=====");

        mkdir("test1");

        touch("test1/testfile1");

        onCompletion = () -> rm("test.jar test1");

        jar("--create --file=test.jar --release 9 test1");

        jar("tf test.jar");

    }

    @Test  // passes before bug fix

    public void test3() throws IOException {

        System.out.println("=====");

        mkdir("test1");

        touch("test1/testfile1");

        jar("-cf test.jar -C test1 .");

        jar("-uf test.jar --release 9 -C test1 .");

        jar("tf test.jar");

        rm("test.jar test1");

    }

    @Test  // fails before bug fix

    public void test4() throws IOException {

        System.out.println("=====");

        mkdir("test1");

        touch("test1/testfile1");

        onCompletion = () -> rm("test.jar test1");

        jar("--create --file=test.jar -C test1 .");

        jar("--update --file=test.jar --release 9 -C test1 .");

        jar("tf test.jar");

    }

    @Test  // passes before bug fix since test2 precedes --release 9

    public void test5() throws IOException {

        System.out.println("=====");

        mkdir("test1 test2");

        touch("test1/testfile1 test2/testfile2");

        jar("--create --file=test.jar -C test1 .");

        jar("--update --file=test.jar test2 --release 9 -C test1 .");

        jar("tf test.jar");

        rm("test.jar test1 test2");

    }

    private Stream<Path> mkpath(String... args) {

        return Arrays.stream(args).map(d -> Paths.get(".", d.split("/")));

    }

    private void mkdir(String cmdline) {

        System.out.println("mkdir -p " + cmdline);

        mkpath(cmdline.split(" +")).forEach(p -> {

            try {

                Files.createDirectories(p);

            } catch (IOException x) {

                throw new UncheckedIOException(x);

            }

        });

    }

    private void touch(String cmdline) {

        System.out.println("touch " + cmdline);

        mkpath(cmdline.split(" +")).forEach(p -> {

            try {

                Files.createFile(p);

            } catch (IOException x) {

                throw new UncheckedIOException(x);

            }

        });

    }

    private void rm(String cmdline) {

        System.out.println("rm -rf " + cmdline);

        mkpath(cmdline.split(" +")).forEach(p -> {

            try {

                if (Files.isDirectory(p)) {

                    FileUtils.deleteFileTreeWithRetry(p);

                } else {

                    FileUtils.deleteFileIfExistsWithRetry(p);

                }

            } catch (IOException x) {

                throw new UncheckedIOException(x);

            }

        });

    }

    private void jar(String cmdline) throws IOException {

        System.out.println("jar " + cmdline);

        boolean ok = new sun.tools.jar.Main(System.out, System.err, "jar")

                .run(cmdline.split(" +"));

        Assert.assertTrue(ok);

    }

}