/*

 * Copyright (c) 2015, 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.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.DataOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.nio.ByteOrder;

import java.nio.file.Path;

import java.util.ArrayList;

import java.util.Collections;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import java.util.stream.Stream;

import jdk.tools.jlink.internal.Archive;

import jdk.tools.jlink.internal.ImageFileCreator;

import jdk.tools.jlink.internal.ImagePluginStack;

import jdk.tools.jlink.plugin.ExecutableImage;

import jdk.tools.jlink.builder.ImageBuilder;

import jdk.tools.jlink.plugin.ModulePool;

/*

 * @test

 * @summary ImageFileCreator class test

 * @author Jean-Francois Denise

 * @modules jdk.jlink/jdk.tools.jlink.internal

 *          jdk.jlink/jdk.tools.jlink.builder

 *          java.base/jdk.internal.jimage

 * @run main/othervm -verbose:gc -Xmx1g ImageFileCreatorTest

 */

public class ImageFileCreatorTest {

    private static class TestArchive implements Archive {

        private final String name;

        private final List<Entry> entries = new ArrayList<>();

        private TestArchive(String name, List<String> entries) {

            this.name = name;

            for (String p : entries) {

                this.entries.add(new TestEntry(p, p));

            }

        }

        @Override

        public String moduleName() {

            return name;

        }

        @Override

        public Stream<Entry> entries() {

            return entries.stream();

        }

        @Override

        public Path getPath() {

            return null;

        }

        @Override

        public void open() throws IOException {

        }

        @Override

        public void close() throws IOException {

        }

        private class TestEntry extends Entry {

            TestEntry(String path, String name) {

                super(TestArchive.this, path, name, Entry.EntryType.CLASS_OR_RESOURCE);

            }

            @Override

            public long size() {

                return 0;

            }

            @Override

            public InputStream stream() throws IOException {

                return new ByteArrayInputStream(new byte[0]);

            }

        }

    }

    public static void main(String[] args) throws Exception {

        {

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

            entries.add("classes/class");

            test(entries);

        }

        {

            // Add an entry that is a directory, that is wrong

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

            entries.add("classes");

            entries.add("classes/class");

            test(entries);

        }

        {

            // Add an entry that is wrongly prefixed by /

            // /bad//classes/class is the resource added

            // /bad/classes/class is the metadata node built.

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

            entries.add("/classes/class");

            test(entries);

        }

        {

            // Trailing '/' is wrong

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

            entries.add("classes/class/");

            test(entries);

        }

        {

            // Too much '/' characters

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

            entries.add("classes//class/");

            test(entries);

        }

        {

            // Too much '/' characters

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

            entries.add("classes/////class/");

            test(entries);

        }

        {

            // Single '/' character

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

            entries.add("/");

            test(entries);

        }

        {

            // 2 '/' characters

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

            entries.add("//");

            test(entries);

        }

        {

            // 3 '/' characters

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

            entries.add("///");

            test(entries);

        }

        {

            // no character

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

            entries.add("");

            test(entries);

        }

        {

            // all together

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

            entries.add("");

            entries.add("///");

            entries.add("//");

            entries.add("/");

            entries.add("classes/////class/");

            entries.add("classes//class/");

            entries.add("classes/class/");

            entries.add("/classes/class");

            entries.add("classes");

            entries.add("classes/class");

            test(entries);

        }

    }

    private static void test(List<String> entries) throws Exception {

        TestArchive arch = new TestArchive("bad", entries);

        Set<Archive> archives = new HashSet<>();

        archives.add(arch);

        ImageBuilder noopBuilder = new ImageBuilder() {

            @Override

            public DataOutputStream getJImageOutputStream() {

                return new DataOutputStream(new ByteArrayOutputStream());

            }

            @Override

            public ExecutableImage getExecutableImage() {

                return null;

            }

            @Override

            public void storeFiles(ModulePool content) {

            }

        };

        ImagePluginStack stack = new ImagePluginStack(noopBuilder, Collections.emptyList(),

                null, Collections.emptyList());

        ImageFileCreator.create(archives, ByteOrder.nativeOrder(), stack);

    }

}