org.jspringbot.syntax.KeywordLoggerLifeCycleHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jspringbot-pretty-logger Show documentation

JSpringBot Pretty Logger

There is a newer version: 1.9

Show newest version

package org.jspringbot.syntax;

import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.jspringbot.lifecycle.LifeCycleAdapter;

import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.Map;

public class KeywordLoggerLifeCycleHandler extends LifeCycleAdapter {
    @Override
    public void startKeyword(String name, Map attributes) {
        HighlightKeywordLogger.createAppender(name);
    }

    @Override
    public void endKeyword(String name, Map attributes) {
        if(HighlightKeywordLogger.appender() != null) {
            if(StringUtils.containsIgnoreCase(name, "JSpringBot")) {
                HighlightKeywordLogger.appender().log();
            }

            HighlightKeywordLogger.clear();
        }
    }

    @Override
    public void endJSpringBotKeyword(String name, Map attributes) {
        if(HighlightKeywordLogger.appender() == null) {
            return;
        }

        if(StringUtils.equals((String) attributes.get("status"), "PASS")) {
            HighlightKeywordLogger.appender().log();
            HighlightKeywordLogger.clear();

            return;
        }

        try {
            Object[] params = (Object[]) attributes.get("args");
            HighlightKeywordLogger.appender().appendArguments(params);

            if(attributes.containsKey("exception")) {
                Exception e = (Exception) attributes.get("exception");
                StringWriter sw = new StringWriter();
                PrintWriter pw = new PrintWriter(sw);
                try {
                    e.printStackTrace(pw);
                    pw.flush();
                    sw.flush();

                    HighlightKeywordLogger.appender().appendCode(sw.toString());
                } finally {
                    IOUtils.closeQuietly(pw);
                    IOUtils.closeQuietly(sw);
                }
            }
        } finally {
            HighlightKeywordLogger.appender().log();
            HighlightKeywordLogger.clear();
        }
    }
}