/*

 * Copyright (c) 2014, 2019, 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 jdk.jpackage.internal;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.OutputStreamWriter;

import java.io.PrintWriter;

import java.io.Writer;

import java.nio.file.Files;

import java.nio.file.Path;

import java.text.MessageFormat;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Optional;

import java.util.ResourceBundle;

import javax.xml.stream.XMLOutputFactory;

import javax.xml.stream.XMLStreamException;

import javax.xml.stream.XMLStreamWriter;

import static jdk.jpackage.internal.StandardBundlerParam.*;

import static jdk.jpackage.internal.MacBaseInstallerBundler.SIGNING_KEYCHAIN;

import static jdk.jpackage.internal.MacBaseInstallerBundler.SIGNING_KEY_USER;

import static jdk.jpackage.internal.MacAppImageBuilder.MAC_CF_BUNDLE_IDENTIFIER;

public class MacPkgBundler extends MacBaseInstallerBundler {

    private static final ResourceBundle I18N = ResourceBundle.getBundle(

            "jdk.jpackage.internal.resources.MacResources");

    private static final String DEFAULT_BACKGROUND_IMAGE = "background_pkg.png";

    private static final String TEMPLATE_PREINSTALL_SCRIPT =

            "preinstall.template";

    private static final String TEMPLATE_POSTINSTALL_SCRIPT =

            "postinstall.template";

    private static final BundlerParamInfo<File> PACKAGES_ROOT =

            new StandardBundlerParam<>(

            "mac.pkg.packagesRoot",

            File.class,

            params -> {

                File packagesRoot =

                        new File(TEMP_ROOT.fetchFrom(params), "packages");

                packagesRoot.mkdirs();

                return packagesRoot;

            },

            (s, p) -> new File(s));

    protected final BundlerParamInfo<File> SCRIPTS_DIR =

            new StandardBundlerParam<>(

            "mac.pkg.scriptsDir",

            File.class,

            params -> {

                File scriptsDir =

                        new File(CONFIG_ROOT.fetchFrom(params), "scripts");

                scriptsDir.mkdirs();

                return scriptsDir;

            },

            (s, p) -> new File(s));

    public static final

            BundlerParamInfo<String> DEVELOPER_ID_INSTALLER_SIGNING_KEY =

            new StandardBundlerParam<>(

            "mac.signing-key-developer-id-installer",

            String.class,

            params -> {

                    String result = MacBaseInstallerBundler.findKey(

                            "Developer ID Installer: "

                            + SIGNING_KEY_USER.fetchFrom(params),

                            SIGNING_KEYCHAIN.fetchFrom(params),

                            VERBOSE.fetchFrom(params));

                    if (result != null) {

                        MacCertificate certificate = new MacCertificate(

                                result, VERBOSE.fetchFrom(params));

                        if (!certificate.isValid()) {

                            Log.error(MessageFormat.format(

                                    I18N.getString("error.certificate.expired"),

                                    result));

                        }

                    }

                    return result;

                },

            (s, p) -> s);

    public static final BundlerParamInfo<String> MAC_INSTALL_DIR =

            new StandardBundlerParam<>(

            "mac-install-dir",

            String.class,

             params -> {

                 String dir = INSTALL_DIR.fetchFrom(params);

                 return (dir != null) ? dir : "/Applications";

             },

            (s, p) -> s

    );

    public static final BundlerParamInfo<String> INSTALLER_SUFFIX =

            new StandardBundlerParam<> (

            "mac.pkg.installerName.suffix",

            String.class,

            params -> "",

            (s, p) -> s);

    public File bundle(Map<String, ? super Object> params,

            File outdir) throws PackagerException {

        Log.verbose(MessageFormat.format(I18N.getString("message.building-pkg"),

                APP_NAME.fetchFrom(params)));

        IOUtils.writableOutputDir(outdir.toPath());

        try {

            File appImageDir = prepareAppBundle(params);

            if (appImageDir != null && prepareConfigFiles(params)) {

                File configScript = getConfig_Script(params);

                if (configScript.exists()) {

                    Log.verbose(MessageFormat.format(I18N.getString(

                            "message.running-script"),

                            configScript.getAbsolutePath()));

                    IOUtils.run("bash", configScript);

                }

                return createPKG(params, outdir, appImageDir);

            }

            return null;

            Log.verbose(ex);

            throw new PackagerException(ex);

        }

    }

    private File getPackages_AppPackage(Map<String, ? super Object> params) {

        return new File(PACKAGES_ROOT.fetchFrom(params),

                APP_NAME.fetchFrom(params) + "-app.pkg");

    }

    private File getConfig_DistributionXMLFile(

            Map<String, ? super Object> params) {

        return new File(CONFIG_ROOT.fetchFrom(params), "distribution.dist");

    }

    private File getConfig_BackgroundImage(Map<String, ? super Object> params) {

        return new File(CONFIG_ROOT.fetchFrom(params),

                APP_NAME.fetchFrom(params) + "-background.png");

    }

    private File getConfig_BackgroundImageDarkAqua(Map<String, ? super Object> params) {

        return new File(CONFIG_ROOT.fetchFrom(params),

                APP_NAME.fetchFrom(params) + "-background-darkAqua.png");

    }

    private File getScripts_PreinstallFile(Map<String, ? super Object> params) {

        return new File(SCRIPTS_DIR.fetchFrom(params), "preinstall");

    }

    private File getScripts_PostinstallFile(

            Map<String, ? super Object> params) {

        return new File(SCRIPTS_DIR.fetchFrom(params), "postinstall");

    }

    private String getAppIdentifier(Map<String, ? super Object> params) {

        return MAC_CF_BUNDLE_IDENTIFIER.fetchFrom(params);

    }

    private void preparePackageScripts(Map<String, ? super Object> params)

            throws IOException {

        Log.verbose(I18N.getString("message.preparing-scripts"));

        Map<String, String> data = new HashMap<>();

        Path appLocation = Path.of(MAC_INSTALL_DIR.fetchFrom(params),

                         APP_NAME.fetchFrom(params) + ".app", "Contents", "app");

        data.put("INSTALL_LOCATION", MAC_INSTALL_DIR.fetchFrom(params));

        data.put("APP_LOCATION", appLocation.toString());

        try (Writer w = Files.newBufferedWriter(

                getScripts_PreinstallFile(params).toPath())) {

            String content = preprocessTextResource(

                    getScripts_PreinstallFile(params).getName(),

                    I18N.getString("resource.pkg-preinstall-script"),

                    TEMPLATE_PREINSTALL_SCRIPT,

                    data,

                    VERBOSE.fetchFrom(params),

                    RESOURCE_DIR.fetchFrom(params));

            w.write(content);

        }

        getScripts_PreinstallFile(params).setExecutable(true, false);

        try (Writer w = Files.newBufferedWriter(

                getScripts_PostinstallFile(params).toPath())) {

            String content = preprocessTextResource(

                    getScripts_PostinstallFile(params).getName(),

                    I18N.getString("resource.pkg-postinstall-script"),

                    TEMPLATE_POSTINSTALL_SCRIPT,

                    data,

                    VERBOSE.fetchFrom(params),

                    RESOURCE_DIR.fetchFrom(params));

            w.write(content);

        }

        getScripts_PostinstallFile(params).setExecutable(true, false);

    }

    private void prepareDistributionXMLFile(Map<String, ? super Object> params)

        File f = getConfig_DistributionXMLFile(params);

        Log.verbose(MessageFormat.format(I18N.getString(

                "message.preparing-distribution-dist"), f.getAbsolutePath()));

        XMLOutputFactory xmlFactory = XMLOutputFactory.newInstance();

        try (Writer w = new OutputStreamWriter(new FileOutputStream(f), "UTF-8")) {

            XMLStreamWriter xml = xmlFactory.createXMLStreamWriter(w);

            xml.writeStartDocument("UTF-8", "1.0");

            xml.writeStartElement("installer-gui-script");

            xml.writeAttribute("minSpecVersion", "1");

            xml.writeStartElement("title");

            xml.writeCharacters(APP_NAME.fetchFrom(params));

            xml.writeEndElement();

            xml.writeStartElement("background");

            xml.writeAttribute("file", getConfig_BackgroundImage(params).getName());

            xml.writeAttribute("mime-type", "image/png");

            xml.writeAttribute("alignment", "bottomleft");

            xml.writeAttribute("scaling", "none");

            xml.writeEndElement();

            xml.writeStartElement("background-darkAqua");

            xml.writeAttribute("file", getConfig_BackgroundImageDarkAqua(params).getName());

            xml.writeAttribute("mime-type", "image/png");

            xml.writeAttribute("alignment", "bottomleft");

            xml.writeAttribute("scaling", "none");

            xml.writeEndElement();

            String licFileStr = LICENSE_FILE.fetchFrom(params);

            if (licFileStr != null) {

                File licFile = new File(licFileStr);

                xml.writeStartElement("license");

                xml.writeAttribute("file", licFile.getAbsolutePath());

                xml.writeAttribute("mime-type", "text/rtf");

                xml.writeEndElement();

            }

            /*

             * Note that the content of the distribution file

             * below is generated by productbuild --synthesize

             */

            String appId = getAppIdentifier(params);

            xml.writeStartElement("pkg-ref");

            xml.writeAttribute("id", appId);

            xml.writeEndElement(); // </pkg-ref>

            xml.writeStartElement("options");

            xml.writeAttribute("customize", "never");

            xml.writeAttribute("require-scripts", "false");

            xml.writeEndElement(); // </options>

            xml.writeStartElement("choices-outline");

            xml.writeStartElement("line");

            xml.writeAttribute("choice", "default");

            xml.writeStartElement("line");

            xml.writeAttribute("choice", appId);

            xml.writeEndElement(); // </line>

            xml.writeEndElement(); // </line>

            xml.writeEndElement(); // </choices-outline>

            xml.writeStartElement("choice");

            xml.writeAttribute("id", "default");

            xml.writeEndElement(); // </choice>

            xml.writeStartElement("choice");

            xml.writeAttribute("id", appId);

            xml.writeAttribute("visible", "false");

            xml.writeStartElement("pkg-ref");

            xml.writeAttribute("id", appId);

            xml.writeEndElement(); // </pkg-ref>

            xml.writeEndElement(); // </choice>

            xml.writeStartElement("pkg-ref");

            xml.writeAttribute("id", appId);

            xml.writeAttribute("version", VERSION.fetchFrom(params));

            xml.writeAttribute("onConclusion", "none");

            xml.writeEndElement(); // </pkg-ref>

            xml.writeEndElement(); // </installer-gui-script>

            xml.writeEndDocument();

            xml.flush();

            xml.close();

        } catch (XMLStreamException ex) {

            Log.verbose(ex);

            throw new IOException(ex);

        }

    }

    private boolean prepareConfigFiles(Map<String, ? super Object> params)

        File imageTarget = getConfig_BackgroundImage(params);

        fetchResource(imageTarget.getName(),

                I18N.getString("resource.pkg-background-image"),

                DEFAULT_BACKGROUND_IMAGE,

                imageTarget,

                VERBOSE.fetchFrom(params),

                RESOURCE_DIR.fetchFrom(params));

        imageTarget = getConfig_BackgroundImageDarkAqua(params);

        fetchResource(imageTarget.getName(),

                I18N.getString("resource.pkg-background-image"),

                DEFAULT_BACKGROUND_IMAGE,

                imageTarget,

                VERBOSE.fetchFrom(params),

                RESOURCE_DIR.fetchFrom(params));

        prepareDistributionXMLFile(params);

        fetchResource(getConfig_Script(params).getName(),

                I18N.getString("resource.post-install-script"),

                (String) null,

                getConfig_Script(params),

                VERBOSE.fetchFrom(params),

                RESOURCE_DIR.fetchFrom(params));

        return true;

    }

    // name of post-image script

    private File getConfig_Script(Map<String, ? super Object> params) {

        return new File(CONFIG_ROOT.fetchFrom(params),

                APP_NAME.fetchFrom(params) + "-post-image.sh");

    }

    private void patchCPLFile(File cpl) throws IOException {

        String cplData = Files.readString(cpl.toPath());

        String[] lines = cplData.split("\n");

        try (PrintWriter out = new PrintWriter(Files.newBufferedWriter(

                cpl.toPath()))) {

            int skip = 0;

            // Used to skip Java.runtime bundle, since

            // pkgbuild with --root will find two bundles app and Java runtime.

            // We cannot generate component proprty list when using

            // --component argument.

            for (int i = 0; i < lines.length; i++) {

                if (lines[i].trim().equals("<key>BundleIsRelocatable</key>")) {

                    out.println(lines[i]);

                    out.println("<false/>");

                    i++;

                } else if (lines[i].trim().equals("<key>ChildBundles</key>")) {

                    ++skip;

                } else if ((skip > 0) && lines[i].trim().equals("</array>")) {

                    --skip;

                } else {

                    if (skip == 0) {

                        out.println(lines[i]);

                    }

                }

            }

        }

    }

    // pkgbuild includes all components from "--root" and subfolders,

    // so if we have app image in folder which contains other images, then they

    // will be included as well. It does have "--filter" option which use regex

    // to exclude files/folder, but it will overwrite default one which excludes

    // based on doc "any .svn or CVS directories, and any .DS_Store files".

    // So easy aproach will be to copy user provided app-image into temp folder

    // if root path contains other files.

    private String getRoot(Map<String, ? super Object> params,

            File appLocation) throws IOException {

        String root = appLocation.getParent() == null ?

                "." : appLocation.getParent();

        File rootDir = new File(root);

        File[] list = rootDir.listFiles();

        if (list != null) { // Should not happend

            // We should only have app image and/or .DS_Store

            if (list.length == 1) {

                return root;

            } else if (list.length == 2) {

                // Check case with app image and .DS_Store

                if (list[0].toString().toLowerCase().endsWith(".ds_store") ||

                    list[1].toString().toLowerCase().endsWith(".ds_store")) {

                    return root; // Only app image and .DS_Store

                }

            }

        }

        // Copy to new root

        Path newRoot = Files.createTempDirectory(

                TEMP_ROOT.fetchFrom(params).toPath(),

                "root-");

        IOUtils.copyRecursive(appLocation.toPath(),

                newRoot.resolve(appLocation.getName()));

        return newRoot.toString();

    }

    private File createPKG(Map<String, ? super Object> params,

            File outdir, File appLocation) {

        // generic find attempt

        try {

            File appPKG = getPackages_AppPackage(params);

            String root = getRoot(params, appLocation);

            // Generate default CPL file

            File cpl = new File(CONFIG_ROOT.fetchFrom(params).getAbsolutePath()

                    + File.separator + "cpl.plist");

            ProcessBuilder pb = new ProcessBuilder("pkgbuild",

                    "--root",

                    root,

                    "--install-location",

                    MAC_INSTALL_DIR.fetchFrom(params),

                    "--analyze",

                    cpl.getAbsolutePath());

            IOUtils.exec(pb);

            patchCPLFile(cpl);

            preparePackageScripts(params);

            // build application package

            pb = new ProcessBuilder("pkgbuild",

                    "--root",

                    root,

                    "--install-location",

                    MAC_INSTALL_DIR.fetchFrom(params),

                    "--component-plist",

                    cpl.getAbsolutePath(),

                    "--scripts",

                    SCRIPTS_DIR.fetchFrom(params).getAbsolutePath(),

                    appPKG.getAbsolutePath());

            IOUtils.exec(pb);

            // build final package

            File finalPKG = new File(outdir, INSTALLER_NAME.fetchFrom(params)

                    + INSTALLER_SUFFIX.fetchFrom(params)

                    + ".pkg");

            outdir.mkdirs();

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

            commandLine.add("productbuild");

            commandLine.add("--resources");

            commandLine.add(CONFIG_ROOT.fetchFrom(params).getAbsolutePath());

            // maybe sign

            if (Optional.ofNullable(MacAppImageBuilder.

                    SIGN_BUNDLE.fetchFrom(params)).orElse(Boolean.TRUE)) {

                if (Platform.getMajorVersion() > 10 ||

                    (Platform.getMajorVersion() == 10 &&

                    Platform.getMinorVersion() >= 12)) {

                    // we need this for OS X 10.12+

                    Log.verbose(I18N.getString("message.signing.pkg"));

                }

                String signingIdentity =

                        DEVELOPER_ID_INSTALLER_SIGNING_KEY.fetchFrom(params);

                if (signingIdentity != null) {

                    commandLine.add("--sign");

                    commandLine.add(signingIdentity);

                }

                String keychainName = SIGNING_KEYCHAIN.fetchFrom(params);