prompto.utils.CodeWriter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Core Show documentation
The newest version!
package prompto.utils;

import prompto.grammar.Identifier;
import prompto.parser.Dialect;
import prompto.runtime.Context;
import prompto.type.CategoryType;

public class CodeWriter {
	
	class Indenter {
		
		public String indents = "";
		public boolean isStartOfLine = true;
		
		public void appendTabsIfRequired(String s) {
			appendTabsIfRequired(s.charAt(s.length()-1));
		}

		public void appendTabsIfRequired(char c) {
			if(isStartOfLine)
				sb.append(indents);
			isStartOfLine = c=='\n';
		}
		
		public void indent() {
			indents += '\t';
		}

		public void dedent() {
			if(indents.isEmpty())
				throw new RuntimeException("Illegal dedent!");
			indents = indents.substring(1);
		}
	}
	
	Dialect dialect;
	Context context;
	StringBuilder sb;
	Indenter indenter;
	int escapeMode;
	
	public CodeWriter(Dialect dialect, Context context) {
		this(dialect, context, 0);
	}
	
	public CodeWriter(Dialect dialect, Context context, int escapeMode) {
		this.dialect = dialect;
		this.context = context;
		this.escapeMode = escapeMode;
		this.sb = new StringBuilder();
		indenter = new Indenter();
	}
	
	public CodeWriter(Dialect dialect, Context context, int escapeMode, StringBuilder sb, Indenter indenter) {
		this.dialect = dialect;
		this.context = context;
		this.escapeMode = escapeMode;
		this.sb = sb;
		this.indenter = indenter;
	}

	public Context getContext() {
		return context;
	}
	
	public int getEscapeMode() {
		return escapeMode;
	}
	
	public CodeWriter append(Identifier id) {
		append(id.toString());
		return this;
	}
	
	public CodeWriter append(String s) {
		indenter.appendTabsIfRequired(s);
		sb.append(s);
		return this;
	}

	public CodeWriter appendRaw(String s) {
		sb.append(s);
		return this;
	}

	public CodeWriter append(char c) {
		indenter.appendTabsIfRequired(c);
		sb.append(c);
		return this;
	}

	public CodeWriter trimLast(int count) {
		sb.setLength(sb.length()-count);
		return this;
	}

	public CodeWriter indent() {
		indenter.indent();
		return this;
	}

	public CodeWriter dedent() {
		indenter.dedent();
		return this;
	}

	public CodeWriter newLine() {
		append('\n');
		return this;
	}

	@Override
	public String toString() {
		return sb.toString();
	}

	public Dialect getDialect() {
		return dialect;
	}


	public boolean isGlobalContext() {
		return context.isGlobalsContext();
	}

	public CodeWriter newLocalWriter() {
		return new CodeWriter(dialect, context.newLocalContext(), escapeMode, sb, indenter);
	}

	public CodeWriter newChildWriter() {
		return newChildWriter(context.newChildContext());
	}

	public CodeWriter newChildWriter(Context context) {
		return new CodeWriter(dialect, context, escapeMode, sb, indenter);
	}

	public CodeWriter newMemberWriter() {
		Context context = this.context.newLocalContext();
		context.setParentContext(this.context);
		return new CodeWriter(dialect, context, escapeMode, sb, indenter);
	}

	public CodeWriter newInstanceWriter(CategoryType type) {
		return new CodeWriter(dialect, context.newInstanceContext(type, false), escapeMode, sb, indenter);
	}

	public CodeWriter newDocumentWriter() {
		return new CodeWriter(dialect, context.newDocumentContext(false), escapeMode, sb, indenter);
	}


}