org.rapidoid.widget.SnippetWidget Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rapidoid-widget Show documentation
Rapidoid Widget
The newest version!
package org.rapidoid.widget;

/*
 * #%L
 * rapidoid-widget
 * %%
 * Copyright (C) 2014 - 2015 Nikolche Mihajlovski and contributors
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import org.rapidoid.annotation.Authors;
import org.rapidoid.annotation.Since;
import org.rapidoid.html.Tag;
import org.rapidoid.lambda.Mapper;
import org.rapidoid.util.U;
import org.rapidoid.util.UTILS;

@Authors("Nikolche Mihajlovski")
@Since("2.3.0")
public class SnippetWidget extends AbstractWidget {

	private static final String JAVA_KEYWORDS = "abstract|continue|for|new|switch|assert|default|goto|package|synchronized|boolean|do|if|private|this|break|double|implements|protected|throw|byte|else|import|public|throws|case|enum|instanceof|return|transient|catch|extends|int|short|try|char|final|interface|static|void|class|finally|long|strictfp|volatile|const|float|native|super|while";

	private static final String tab = "\\t";
	private static final String str1 = "(\"[^\"]*?\")";
	private static final String str2 = "('[^']*?')";
	private static final String num = "(\\d+)";
	private static final String kw = "\\b(" + JAVA_KEYWORDS + ")\\b";
	private static final String cls = "\\b([A-Z]\\w+?)\\b";
	private static final String regex = "(?:" + U.join("|", str1, str2, num, tab, kw, cls) + ")";

	protected String code;

	public SnippetWidget(String code) {
		this.code = code;
	}

	@Override
	protected Tag render() {
		return hardcoded("" + prettify() + "");
	}

	public static String prettify(String sourceCode, boolean escape) {
		// ignoring "\"" => ""
		String snippet = escape ? sourceCode.replace("&", "&").replace("<", "<").replace(">", ">")
				: sourceCode;

		snippet = UTILS.replace(snippet, regex, new Mapper() {
			@Override
			public String map(String[] src) throws Exception {
				String s = src[0];
				char ch = s.charAt(0);

				// FIXME @Annotations

				if (Character.isUpperCase(ch)) {
					return "" + s + "";
				} else if (ch == '"' || ch == "'".charAt(0)) {
					return "" + s + "";
				} else if (Character.isDigit(ch)) {
					return "" + s + "";
				} else if (s.equals("\t")) {
					return "    ";
				} else {
					return "" + s + "";
				}
			}
		});

		snippet = snippet.replaceAll("\n(\\s*)(.*)\\s//\\shere", "\n$1$2");

		return snippet.trim();
	}

	public String prettify() {
		return prettify(code, true);
	}

}