com.dslplatform.compiler.client.Context Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dsl-clc Show documentation
Command line client for interaction with DSL Platform compiler (https://dsl-platform.com)
The newest version!
package com.dslplatform.compiler.client;

import com.dslplatform.compiler.client.parameters.DisableColors;
import com.dslplatform.compiler.client.parameters.LogOutput;
import com.dslplatform.compiler.client.parameters.DisablePrompt;
import org.fusesource.jansi.Ansi;
import org.fusesource.jansi.Ansi.Color;
import org.fusesource.jansi.AnsiConsole;

import java.io.*;
import java.util.HashMap;
import java.util.Map;

public class Context implements Closeable {
	private final Map parameters = new HashMap();
	private final Map cache = new HashMap();

	private PrintStream console;

	private boolean withLog;
	private boolean noPrompt;
	private boolean withColor = true;

	public Context() {
		this(AnsiConsole.out());
	}

	protected Context(PrintStream console) {
		this.console = console;
	}

	public void put(final CompileParameter parameter, final String value) {
		if (parameter instanceof DisablePrompt) {
			noPrompt = true;
		} else if (parameter instanceof LogOutput) {
			withLog = true;
		} else if (parameter instanceof DisableColors) {
			withColor = false;
			console = System.out;
		}
		parameters.put(parameter.getAlias(), value);
	}

	public void put(final String parameter, final String value) {
		parameters.put(parameter.toLowerCase(), value);
	}

	public boolean contains(final CompileParameter parameter) {
		return parameters.containsKey(parameter.getAlias());
	}

	public boolean contains(final String parameter) {
		return parameters.containsKey(parameter);
	}

	public String get(final CompileParameter parameter) {
		return parameters.get(parameter.getAlias());
	}

	public String get(final String parameter) {
		return parameters.get(parameter.toLowerCase());
	}

	public void cache(final String name, final Object value) {
		cache.put(name, value);
	}

	public  T notify(final String action, final T target) {
		log("Notify: " + action + " for " + target);
		return target;
	}

	@SuppressWarnings("unchecked")
	public  T load(final String name) {
		return (T) cache.get(name);
	}

	private static synchronized void write(final PrintStream console, final boolean newLine, final String... values) {
		if (values.length == 0) {
			console.println();
		} else {
			if (newLine) {
				for (final String v : values) {
					console.println(v);
				}
			} else {
				for (final String v : values) {
					console.print(v);
				}
			}
		}
		console.flush();
	}

	public void show(final String... values) {
		write(console, true, values);
	}

	public static String inColor(final Ansi.Color color, final String message) {
		return Ansi.ansi().fg(color).a(message).reset().toString();
	}

	public void log(final String value) {
		if (withLog) {
			write(console, true, withColor ? inColor(Color.YELLOW, value) : value);
		}
	}

	public void log(final char[] value, final int len) {
		if (withLog) {
			final String msg = new String(value, 0, len);
			write(console, false, withColor ? inColor(Color.YELLOW, msg) : msg);
		}
	}


	public void warning(final String value) {
		write(console, true, withColor ? inColor(Color.MAGENTA, value) : value);
	}

	public void warning(final Exception ex) {
		warning(ex.getMessage());
		if (withLog) {
			final StringWriter sw = new StringWriter();
			ex.printStackTrace(new PrintWriter(sw));
			warning(sw.toString());
		}
	}

	public void error(final String value) {
		write(console, true, withColor ? inColor(Color.RED, value) : value);
	}

	public void error(final Exception ex) {
		if (ex instanceof ExitException) return;
		final String description = ex.getMessage();
		if (description == null) error(ex.getClass().getName() + " error without description");
		else error(description);
		if (withLog) {
			final StringWriter sw = new StringWriter();
			ex.printStackTrace(new PrintWriter(sw));
			error(sw.toString());
		}
	}

	public boolean canInteract() {
		return !noPrompt && System.console() != null;
	}

	private void askSafe(final String question, final Color color) {
		if (withColor) {
			try {
				write(console, false, Ansi.ansi().fgBright(color).bold().a(question + " ").boldOff().reset().toString());
				return;
			} catch (NoSuchMethodError ignore) {
				warning("Incompatible jansi found on classpath. Reverting to no-color");
				withColor = false;
			}
		}
		write(console, false, question + " ");
	}

	public String ask(final String question) {
		askSafe(question, Color.DEFAULT);
		return System.console().readLine();
	}

	public char[] askSecret(final String question) {
		askSafe(question, Color.CYAN);
		return System.console().readPassword();
	}

	@Override
	public void close() {
		for (Object it : cache.values()) {
			if (it instanceof Closeable) {
				try {
					((Closeable) it).close();
				} catch (IOException e) {
					error(e);
				}
			}
		}
	}
}