ru.org.linux.util.formatter.ToLorCodeTexFormatter.java Source code

Introduction

Here is the source code for ru.org.linux.util.formatter.ToLorCodeTexFormatter.java

Source

/*
 * Copyright 1998-2016 Linux.org.ru
 *    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.
 */

package ru.org.linux.util.formatter;

import com.google.common.base.Strings;
import org.springframework.stereotype.Service;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 *  ? ? TeX ? ? ??  
 * ?? : ? ? code   
 */
@Service
public class ToLorCodeTexFormatter {
    private static final Pattern QUOTE_PATTERN = Pattern.compile("^(\\>+)");
    private static final Pattern CODE_PATTERN = Pattern.compile("(?:[^\\[]|^)\\[code(:?=[\\w\\s]+)?]");
    private static final Pattern CODE_END_PATTERN = Pattern.compile("\\[/code]");
    private static final Pattern NL_REGEXP = Pattern.compile("\r?\n");

    public static String quote(String text, String newLine) {
        StringBuilder buf = new StringBuilder();
        String[] lines = NL_REGEXP.split(text);
        int globalNestingLevel = 0;
        int currentLine = 0;

        boolean isCode = false;

        for (String line : lines) {
            currentLine += 1;

            if (line.isEmpty()) {
                if (globalNestingLevel > 0) {
                    buf.append(Strings.repeat("[/quote]", globalNestingLevel));
                    globalNestingLevel = 0;
                } else {
                    if (isCode) {
                        buf.append('\n');
                    } else {
                        buf.append(newLine);
                    }
                }
                continue;
            }

            Matcher m = QUOTE_PATTERN.matcher(line);
            if (!isCode && m.find()) {
                int nestingLevel = m.group(1).length();
                if (globalNestingLevel == 0) {
                    buf.append(Strings.repeat("[quote]", nestingLevel));
                    globalNestingLevel = nestingLevel;
                } else if (nestingLevel < globalNestingLevel) {
                    buf.append(Strings.repeat("[/quote]", globalNestingLevel - nestingLevel));
                    globalNestingLevel = nestingLevel;
                } else if (nestingLevel > globalNestingLevel) {
                    buf.append(Strings.repeat("[quote]", nestingLevel - globalNestingLevel));
                    globalNestingLevel = nestingLevel;
                }
                buf.append(escapeCode(line.substring(nestingLevel)));
                if (currentLine < lines.length) {
                    buf.append("[br]");
                }
            } else {
                if (globalNestingLevel > 0) {
                    buf.append(Strings.repeat("[/quote]", globalNestingLevel));
                    globalNestingLevel = 0;
                }

                Matcher codeMatcher = CODE_PATTERN.matcher(line);

                if (codeMatcher.find()) {
                    isCode = true;
                }

                Matcher codeEndMatcher = CODE_END_PATTERN.matcher(line);
                if (codeEndMatcher.find()) {
                    if (globalNestingLevel > 0) {
                        line = escapeCode(line);
                    } else {
                        isCode = false;
                    }
                }

                buf.append(line);

                if (currentLine < lines.length) {
                    if (isCode) {
                        buf.append('\n');
                    } else {
                        buf.append(newLine);
                    }
                }
            }
        }

        if (globalNestingLevel > 0) {
            buf.append(Strings.repeat("[/quote]", globalNestingLevel));
        }

        return buf.toString();
    }

    /**
     *  ?
     * @param text ?
     * @return  ?
     */
    public String format(String text) {
        return quote(text, "\n");
    }

    private static final Pattern CODE_ESCAPE_REGEXP = Pattern.compile("([^\\[]|^)\\[(/?code(:?=[\\w\\s]+)?)]");

    public static String escapeCode(String text) {
        return CODE_ESCAPE_REGEXP.matcher(text).replaceAll("$1[[$2]]");
    }
}