Signal-Server/src/main/java/org/whispersystems/textsecuregcm/liquibase/DbMigrateCommand.java

package org.whispersystems.textsecuregcm.liquibase;


import com.google.common.base.Charsets;
import com.google.common.base.Joiner;
import net.sourceforge.argparse4j.impl.Arguments;
import net.sourceforge.argparse4j.inf.Namespace;
import net.sourceforge.argparse4j.inf.Subparser;

import java.io.OutputStreamWriter;
import java.util.List;

import io.dropwizard.Configuration;
import io.dropwizard.db.DatabaseConfiguration;
import liquibase.Liquibase;

public class DbMigrateCommand<T extends Configuration> extends AbstractLiquibaseCommand<T> {

  public DbMigrateCommand(String migration, DatabaseConfiguration<T> strategy, Class<T> configurationClass) {
    super("migrate", "Apply all pending change sets.", migration, strategy, configurationClass);
  }

  @Override
  public void configure(Subparser subparser) {
    super.configure(subparser);

    subparser.addArgument("-n", "--dry-run")
             .action(Arguments.storeTrue())
             .dest("dry-run")
             .setDefault(Boolean.FALSE)
             .help("output the DDL to stdout, don't run it");

    subparser.addArgument("-c", "--count")
             .type(Integer.class)
             .dest("count")
             .help("only apply the next N change sets");

    subparser.addArgument("-i", "--include")
             .action(Arguments.append())
             .dest("contexts")
             .help("include change sets from the given context");
  }

  @Override
  @SuppressWarnings("UseOfSystemOutOrSystemErr")
  public void run(Namespace namespace, Liquibase liquibase) throws Exception {
    final String context = getContext(namespace);
    final Integer count = namespace.getInt("count");
    final Boolean dryRun = namespace.getBoolean("dry-run");
    if (count != null) {
      if (dryRun) {
        liquibase.update(count, context, new OutputStreamWriter(System.out, Charsets.UTF_8));
      } else {
        liquibase.update(count, context);
      }
    } else {
      if (dryRun) {
        liquibase.update(context, new OutputStreamWriter(System.out, Charsets.UTF_8));
      } else {
        liquibase.update(context);
      }
    }
  }

  private String getContext(Namespace namespace) {
    final List<Object> contexts = namespace.getList("contexts");
    if (contexts == null) {
      return "";
    }
    return Joiner.on(',').join(contexts);
  }
}