com.shapesecurity.shift.codegen.WebSafeTokenStream Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of shift Show documentation

Shift format ECMAScript 6 AST tooling

There is a newer version: 2.2.0

package com.shapesecurity.shift.codegen;

import com.shapesecurity.shift.utils.Utils;
import org.jetbrains.annotations.NotNull;

class WebSafeTokenStream extends TokenStream {
	public WebSafeTokenStream(@NotNull StringBuilder writer) {
		super(writer);
	}

	@Override
	public void put(@NotNull String tokenStr) {
		if (
			this.lastChar == '<' && (tokenStr.startsWith("script") || tokenStr.startsWith("/script")) ||
            Utils.isIdentifierPart(this.lastChar) && (Utils.isIdentifierPart(tokenStr.charAt(0)) || tokenStr.charAt(0) == '\\')
		) {
			writer.append(this.lastChar = ' ');
		}
		super.put(tokenStr);
	}
}