io.paradoxical.cassieq.commands.ConfigDumpCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cassieq-core Show documentation
The core CassieQ server
The newest version!
package io.paradoxical.cassieq.commands;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializationFeature;
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
import com.fasterxml.jackson.dataformat.yaml.YAMLGenerator;
import com.google.common.base.Charsets;
import com.google.common.io.CharSource;
import com.google.common.io.Files;
import io.dropwizard.cli.ConfiguredCommand;
import io.dropwizard.configuration.ConfigurationFactory;
import io.dropwizard.setup.Bootstrap;
import io.paradoxical.cassieq.ServiceConfiguration;
import net.sourceforge.argparse4j.impl.Arguments;
import net.sourceforge.argparse4j.inf.Namespace;
import net.sourceforge.argparse4j.inf.Subparser;

import java.io.File;
import java.io.IOException;

public class ConfigDumpCommand extends ConfiguredCommand {

    private static final String DefaultConfigFile = "/data/conf/configuration.yml";

    public ConfigDumpCommand() {
        super("config", "Prints out the config settings for use in creating custom config files");
    }

    @Override
    public void run(final Bootstrap wildcardBootstrap, final Namespace namespace) throws Exception {

        // skip validation if dumping full (because its not complete)
        if (namespace.getBoolean("full")) {
            wildcardBootstrap.setConfigurationFactoryFactory(
                    (klass, validator, objectMapper, propertyPrefix) ->
                            new ConfigurationFactory<>(klass, null, objectMapper, propertyPrefix));
        }

        super.run(wildcardBootstrap, namespace);
    }

    @Override
    public void configure(final Subparser subparser) {
        setupDefaultConfigFile(subparser);

        subparser.addArgument("-f", "--full")
                 .action(Arguments.storeTrue())
                 .dest("full")
                 .help("Dumps the full configuration object as YAML");
    }

    public static void setupDefaultConfigFile(final Subparser subparser) {
        subparser.addArgument("file")
                 .nargs("?")
                 .setDefault(DefaultConfigFile)
                 .help("application configuration file");
    }

    @Override
    protected void run(
            final Bootstrap bootstrap,
            final Namespace namespace,
            final ServiceConfiguration configuration) throws Exception {

        if (namespace.getBoolean("full")) {
            dumpFullConfig(configuration);
        }
        else {
            dumpConfigFile(namespace);
        }

        System.out.println();
    }

    private void dumpConfigFile(final Namespace namespace) throws IOException {
        final File configFile = new File(namespace.getString("file"));

        final CharSource configSource = Files.asCharSource(configFile, Charsets.UTF_8);

        configSource.copyTo(System.out);
    }

    private void dumpFullConfig(final ServiceConfiguration configuration) throws JsonProcessingException {
        final YAMLFactory yamlFactory = new YAMLFactory();

        yamlFactory.configure(YAMLGenerator.Feature.USE_NATIVE_TYPE_ID, false);

        final ObjectMapper objectMapper = new ObjectMapper(yamlFactory);

        objectMapper.configure(SerializationFeature.INDENT_OUTPUT, true);

        System.out.println(objectMapper.writeValueAsString(configuration));
    }
}