package com.datastax.oss.driver.internal.core.metadata.schema;
import com.datastax.oss.driver.api.core.CqlIdentifier;
import com.datastax.oss.driver.api.core.metadata.schema.Describable;
import com.datastax.oss.driver.shaded.guava.common.base.Strings;
import java.util.function.Consumer;
import net.jcip.annotations.NotThreadSafe;
@NotThreadSafe
public class ScriptBuilder {
private static final int INDENT_SIZE = 4;
private final boolean pretty;
private final StringBuilder builder = new StringBuilder();
private int indent;
private boolean isAtLineStart;
private boolean isFirstOption = true;
public ScriptBuilder(boolean pretty) {
this.pretty = pretty;
}
public ScriptBuilder append(String s) {
if (pretty && isAtLineStart && indent > 0) {
builder.append(Strings.repeat(" ", indent * INDENT_SIZE));
}
isAtLineStart = false;
builder.append(s);
return this;
}
public ScriptBuilder append(CqlIdentifier id) {
append(id.asCql(pretty));
return this;
}
public ScriptBuilder newLine() {
if (pretty) {
builder.append('\n');
} else {
builder.append(' ');
}
isAtLineStart = true;
return this;
}
public ScriptBuilder forceNewLine(int count) {
builder.append(Strings.repeat("\n", count));
isAtLineStart = true;
return this;
}
public ScriptBuilder increaseIndent() {
indent += 1;
return this;
}
public ScriptBuilder decreaseIndent() {
if (indent > 0) {
indent -= 1;
}
return this;
}
public ScriptBuilder andWith() {
if (isFirstOption) {
append(" WITH ");
isFirstOption = false;
} else {
newLine();
append("AND ");
}
return this;
}
public <E> ScriptBuilder forEach(Iterable<E> iterable, Consumer<E> action) {
for (E e : iterable) {
action.accept(e);
}
return this;
}
public String build() {
return builder.toString();
}
}