1 /*

     2  * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.  Oracle designates this

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

     9  * by Oracle in the LICENSE file that accompanied this code.

    10  *

    11  * This code is distributed in the hope that it will be useful, but WITHOUT

    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    14  * version 2 for more details (a copy is included in the LICENSE file that

    15  * accompanied this code).

    16  *

    17  * You should have received a copy of the GNU General Public License version

    18  * 2 along with this work; if not, write to the Free Software Foundation,

    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    20  *

    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    22  * or visit www.oracle.com if you need additional information or have any

    23  * questions.

    24  */

    25 

    26 package jdk.jpackage.internal;

    27 

    28 import java.io.IOException;

    29 import java.nio.file.Files;

    30 import java.nio.file.Path;

    31 import java.util.*;

    32 import java.util.function.UnaryOperator;

    33 import java.util.stream.Stream;

    34 

    35 /**

    36  * WiX pipeline. Compiles and links WiX sources.

    37  */

    38 public class WixPipeline {

    39     WixPipeline() {

    40         sources = new ArrayList<>();

    41         lightOptions = new ArrayList<>();

    42     }

    43 

    44     WixPipeline setToolset(Map<WixTool, Path> v) {

    45         toolset = v;

    46         return this;

    47     }

    48 

    49     WixPipeline setWixVariables(Map<String, String> v) {

    50         wixVariables = v;

    51         return this;

    52     }

    53 

    54     WixPipeline setWixObjDir(Path v) {

    55         wixObjDir = v;

    56         return this;

    57     }

    58 

    59     WixPipeline setWorkDir(Path v) {

    60         workDir = v;

    61         return this;

    62     }

    63 

    64     WixPipeline addSource(Path source, Map<String, String> wixVariables) {

    65         WixSource entry = new WixSource();

    66         entry.source = source;

    67         entry.variables = wixVariables;

    68         sources.add(entry);

    69         return this;

    70     }

    71 

    72     WixPipeline addLightOptions(String ... v) {

    73         lightOptions.addAll(List.of(v));

    74         return this;

    75     }

    76 

    77     void buildMsi(Path msi) throws IOException {

    78         List<Path> wixObjs = new ArrayList<>();

    79         for (var source : sources) {

    80             wixObjs.add(compile(source));

    81         }

    82 

    83         List<String> lightCmdline = new ArrayList<>(List.of(

    84                 toolset.get(WixTool.Light).toString(),

    85                 "-nologo",

    86                 "-spdb",

    87                 "-ext", "WixUtilExtension",

    88                 "-out", msi.toString()

    89         ));

    90 

    91         lightCmdline.addAll(lightOptions);

    92         wixObjs.stream().map(Path::toString).forEach(lightCmdline::add);

    93 

    94         Files.createDirectories(msi.getParent());

    95         execute(lightCmdline);

    96     }

    97 

    98     private Path compile(WixSource wixSource) throws IOException {

    99         UnaryOperator<Path> adjustPath = path -> {

   100             return workDir != null ? path.toAbsolutePath() : path;

   101         };

   102 

   103         Path wixObj = adjustPath.apply(wixObjDir).resolve(IOUtils.replaceSuffix(

   104                 wixSource.source.getFileName(), ".wixobj"));

   105 

   106         List<String> cmdline = new ArrayList<>(List.of(

   107                 toolset.get(WixTool.Candle).toString(),

   108                 "-nologo",

   109                 adjustPath.apply(wixSource.source).toString(),

   110                 "-ext", "WixUtilExtension",

   111                 "-arch", "x64",

   112                 "-out", wixObj.toAbsolutePath().toString()

   113         ));

   114 

   115         Map<String, String> appliedVaribales = new HashMap<>();

   116         Stream.of(wixVariables, wixSource.variables)

   117                 .filter(Objects::nonNull)

   118                 .forEachOrdered(appliedVaribales::putAll);

   119 

   120         appliedVaribales.entrySet().stream().map(wixVar -> String.format("-d%s=%s",

   121                 wixVar.getKey(), wixVar.getValue())).forEachOrdered(

   122                 cmdline::add);

   123 

   124         execute(cmdline);

   125 

   126         return wixObj;

   127     }

   128 

   129     private void execute(List<String> cmdline) throws IOException {

   130         Executor.of(new ProcessBuilder(cmdline).directory(

   131                 workDir != null ? workDir.toFile() : null)).executeExpectSuccess();

   132     }

   133 

   134     private final static class WixSource {

   135         Path source;

   136         Map<String, String> variables;

   137     }

   138 

   139     private Map<WixTool, Path> toolset;

   140     private Map<String, String> wixVariables;

   141     private List<String> lightOptions;

   142     private Path wixObjDir;

   143     private Path workDir;

   144     private List<WixSource> sources;

   145 }