net.oneandone.stool.setup.BinMan Maven / Gradle / Ivy

Go to download
/**
 * Copyright 1&1 Internet AG, https://github.com/1and1/
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package net.oneandone.stool.setup;

import net.oneandone.stool.util.Files;
import net.oneandone.sushi.cli.Console;
import net.oneandone.sushi.fs.Settings;
import net.oneandone.sushi.fs.World;
import net.oneandone.sushi.fs.file.FileNode;

import java.io.IOException;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;

/** Generates bin- and man directories. Used at build time (for Debian package) or runtime (Java installer) */
public class BinMan {
    /** generate file hierarchy for Debian package */
    public static void main(String[] args) throws Exception {
        World world;
        Console console;
        FileNode target;
        FileNode man;
        FileNode bin;

        if (args.length != 1) {
            throw new IllegalArgumentException();
        }
        world = new World();
        console = Console.create(world);
        target = world.file(args[0]);
        target.mkdir();
        bin = world.file("/usr/share/stool");
        man = world.file("/usr/share/man");
        new BinMan(console, true, bin, man, target.join(bin.getName()), target.join(man.getName())).create();
        System.exit(0);
    }

    public static BinMan java(Console console, boolean withJar, FileNode installedBin, FileNode installedMan) {
        return new BinMan(console, withJar, installedBin, installedMan, installedBin, installedMan);
    }

    //--

    private final Console console;

    /** to create bin directory with a stool jar. False in tests, when stool.jar is not in classpath */
    private final boolean withJar;

    // locations when installed on target machine
    private final FileNode installedMan;
    private final FileNode installedBin;
    private final FileNode nowBin;
    private final FileNode nowMan;

    private BinMan(Console console, boolean withJar, FileNode installedBin, FileNode installedMan, FileNode nowBin, FileNode nowMan) {
        this.console = console;
        this.withJar = withJar;
        this.installedBin = installedBin;
        this.installedMan = installedMan;
        this.nowBin = nowBin;
        this.nowMan = nowMan;
    }

    public void remove() throws IOException {
        nowBin.deleteTree();
        nowMan.deleteTree();
    }

    public void create() throws IOException {
        bin();
        man();
    }

    //--

    // CAUTION: does not generate the lib symlink
    private void bin() throws IOException {
        final byte[] marker = "exit $?\n".getBytes(Settings.UTF_8);
        Map variables;
        byte[] bytes;
        int ofs;

        variables = new HashMap<>();
        variables.put("stool.bin", installedBin.getAbsolute());
        variables.put("man.path", "/usr/share/man".equals(installedMan.getAbsolute()) ? "" :
                "# note that the empty entry instructs man to search locations.\n" +
                        "export MANPATH=" + installedMan.getAbsolute() + ":$MANPATH\n");
        Files.createStoolDirectory(console.verbose, nowBin);
        Files.template(console.verbose, console.world.resource("templates/bin"), nowBin, variables);
        if (withJar) {
            // strip launcher from application file
            bytes = console.world.locateClasspathItem(getClass()).readBytes();
            ofs = indexOf(bytes, marker) + marker.length;
            try (OutputStream out = nowBin.join("stool.jar").createOutputStream()) {
                out.write(bytes, ofs, bytes.length - ofs);
            }
        }
    }

    private void man() throws IOException {
        Files.createStoolDirectory(console.verbose, nowMan);
        console.world.resource("templates/man").copyDirectory(nowMan);
        Files.stoolTree(console.verbose, nowMan);
    }

    //--

    private static int indexOf(byte[] array, byte[] sub) {
        int j;

        for (int i = 0; i < array.length - sub.length; i++) {
            for (j = 0; j < sub.length; j++) {
                if (sub[j] != array[i + j]) {
                    break;
                }
            }
            if (j == sub.length) {
                return i;
            }
        }
        throw new IllegalStateException();
    }
}