io.bdeploy.bhive.cli.ImportTool Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of api Show documentation
Public API including dependencies, ready to be used for integrations and plugins.
There is a newer version: 7.3.6
package io.bdeploy.bhive.cli;

import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Map;
import java.util.TreeMap;

import io.bdeploy.bhive.BHive;
import io.bdeploy.bhive.BHiveTransactions.Transaction;
import io.bdeploy.bhive.cli.ImportTool.ImportConfig;
import io.bdeploy.bhive.model.Manifest;
import io.bdeploy.bhive.op.ImportOperation;
import io.bdeploy.common.cfg.Configuration.EnvironmentFallback;
import io.bdeploy.common.cfg.Configuration.Help;
import io.bdeploy.common.cfg.Configuration.Validator;
import io.bdeploy.common.cfg.ExistingPathValidator;
import io.bdeploy.common.cfg.PathOwnershipValidator;
import io.bdeploy.common.cli.ToolBase.CliTool.CliName;
import io.bdeploy.common.cli.ToolBase.ConfiguredCliTool;
import io.bdeploy.common.cli.ToolCategory;
import io.bdeploy.common.cli.data.RenderableResult;

/**
 * Import a source directory recursively into a hive and create a manifest for
 * it using the given values.
 */
@Help("Imports a source directory into a BHive")
@ToolCategory(BHiveCli.FS_TOOLS)
@CliName("import")
public class ImportTool extends ConfiguredCliTool {

    public @interface ImportConfig {

        @Help("The path to the directory to import")
        @Validator(ExistingPathValidator.class)
        String source();

        @Help("The target BHive to import into")
        @EnvironmentFallback("BHIVE")
        @Validator({ ExistingPathValidator.class, PathOwnershipValidator.class })
        String hive();

        @Help("Manifest(s) to create. Format is 'name:tag'")
        String manifest();

        @Help("Can appear multiple times. ':' separated key value pair of label to add")
        String[] label() default {};

        @Help("Parallelism - how many threads to use to import. Default: 4")
        int jobs() default 4;
    }

    public ImportTool() {
        super(ImportConfig.class);
    }

    @Override
    protected RenderableResult run(ImportConfig config) {
        helpAndFailIfMissing(config.source(), "Missing --source");
        helpAndFailIfMissing(config.hive(), "Missing --hive");
        helpAndFailIfMissing(config.manifest(), "Missing --manifest");

        Path source = Paths.get(config.source());
        Path target = Paths.get(config.hive());

        if (!Files.isDirectory(source)) {
            helpAndFail("Source path must be a directory (" + source + ")");
        }

        Map labels = new TreeMap<>();
        for (String lbl : config.label()) {
            if (!lbl.contains(":")) {
                helpAndFail("Label must use ':' as key/value separator (" + lbl + ")");
            }
            int idefix = lbl.indexOf(':');
            String k = lbl.substring(0, idefix);
            String v = lbl.substring(idefix + 1);

            labels.put(k, v);
        }

        try (BHive hive = new BHive(target.toUri(), getAuditorFactory().apply(target), getActivityReporter());
                Transaction t = hive.getTransactions().begin()) {
            hive.setParallelism(config.jobs());

            ImportOperation op = new ImportOperation().setSourcePath(source).setManifest(Manifest.Key.parse(config.manifest()));
            labels.forEach(op::addLabel);
            Manifest.Key result = hive.execute(op);

            return createSuccess().addField("Key", result.toString());
        }
    }

}