/*
 * 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.  Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

package build.tools.module;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class GenModuleLoaderMap {
    private static final String USAGE =
        "GenModuleLoaderMap -o <output-file> -boot m1[,m2]* -platform m3[,m4]* <original-source>";

    public static void main(String... args) throws Exception {
        // default set of boot modules and ext modules
        Stream<String> bootModules = Stream.empty();
        Stream<String> platformModules = Stream.empty();
        Path outfile = null;
        Path source = null;
        for (int i=0; i < args.length; i++) {
            String option = args[i];
            if (option.startsWith("-")) {
                String arg = args[++i];
                if (option.equals("-boot")) {
                    String[] mns = arg.split(",");
                    bootModules = Stream.concat(bootModules, Arrays.stream(mns));
                } else if (option.equals("-platform")) {
                    String[] mns = arg.split(",");
                    platformModules = Stream.concat(platformModules, Arrays.stream(mns));
                } else if (option.equals("-o")) {
                    outfile = Paths.get(arg);
                } else {
                    throw new IllegalArgumentException("invalid option: " + option);
                }
            } else {
                source = Paths.get(option);
            }
        }

        if (outfile == null) {
            throw new IllegalArgumentException("-o must be specified");
        }
        if (Files.notExists(source)) {
            throw new IllegalArgumentException(source + " not exist");
        }

        boolean needsQuotes = outfile.toString().contains(".java.tmp");

        try (BufferedWriter bw = Files.newBufferedWriter(outfile, StandardCharsets.UTF_8);
             PrintWriter writer = new PrintWriter(bw)) {
            for (String line : Files.readAllLines(source)) {
                if (line.contains("@@BOOT_MODULE_NAMES@@")) {
                    line = patch(line, "@@BOOT_MODULE_NAMES@@", bootModules, needsQuotes);
                } else if (line.contains("@@PLATFORM_MODULE_NAMES@@")) {
                    line = patch(line, "@@PLATFORM_MODULE_NAMES@@", platformModules, needsQuotes);
                }
                writer.println(line);
            }
        }
    }

    private static String patch(String s, String tag, Stream<String> stream, boolean needsQuotes) {
        String mns = null;
        if (needsQuotes) {
            mns = stream.sorted()
                .collect(Collectors.joining("\",\n            \""));
        } else {
            mns = stream.sorted()
                .collect(Collectors.joining("\n"));
        }
        return s.replace(tag, mns);
    }

    /**
     * Reads the contents of the given modules file.
     */
    private static Set<String> readModuleSet(String name) throws IOException {
        try (InputStream is = GenModuleLoaderMap.class.getResourceAsStream(name);
             BufferedReader reader = new BufferedReader(new InputStreamReader(is))) {
            return reader.lines().collect(Collectors.toSet());
        }
    }
}