ProblemStatementTag.java

package com.topcoder.web.common.tag;

import com.topcoder.shared.language.*;
import com.topcoder.shared.problem.Problem;
import com.topcoder.shared.problem.ProblemComponent;
import com.topcoder.shared.problemParser.ProblemComponentFactory;
import com.topcoder.web.common.render.ProblemRenderer;

import javax.servlet.jsp.JspException;
import javax.servlet.jsp.tagext.TagSupport;
import java.io.StringReader;


public class ProblemStatementTag extends TagSupport {

    private String styleClass = null;
    private String text;
    private Language language;
    private Problem problem;


    public void setProblem(Problem p) {
        this.problem = p;
    }

    public void setProblem(String p) {
        this.problem = (Problem) pageContext.findAttribute(p);
    }

    /**
     * Sets the value of <code>textColor</code>.
     *
     * @param val
     */
    public void setStyleClass(String val) {
        styleClass = val;
    }

    /**
     * Sets the value of <code>text</code>.
     *
     * @param val
     */
    public void setText(String val) {
        text = val;
    }

    /**
     * Sets the value of <code>language</code>.
     *
     * @param val
     */
    public void setLanguage(String val) {
        if (val != null) {
            if (val.equals(JavaLanguage.DESCRIPTION)) {
                language = JavaLanguage.JAVA_LANGUAGE;
            } else if (val.equals(CPPLanguage.DESCRIPTION)) {
                language = CPPLanguage.CPP_LANGUAGE;
            } else if (val.equals(CSharpLanguage.DESCRIPTION)) {
                language = CSharpLanguage.CSHARP_LANGUAGE;
            } else if (val.equals(VBLanguage.DESCRIPTION)) {
                language = VBLanguage.VB_LANGUAGE;
            } else if (val.equals(PythonLanguage.DESCRIPTION)) {
                language = PythonLanguage.PYTHON_LANGUAGE;
            }
        }
    }

    public void setLanguage(Language l) {
        this.language = l;
    }

    /**
     * Method specific to JSP Tags.  Validates inputs and writes out property
     * if possible.  If it is not possible because bean does not exist, exits
     * quietly
     *
     * @return The JSP Tag specific return specifying the next action
     *         (Should always return SKIP_BODY)
     * @throws JspException Thrown if name and property are not set or if
     *                      there is IO trouble writing out the tag.
     */
    public int doStartTag() throws JspException {
        if (text == null && problem == null) {
            throw new JspException("Text or Problem must be set");
        }
        if (language == null) {
            throw new JspException(
                    "Language is not set or set to invalid value");
        }
        if (problem == null) {
            problem = new Problem();
            StringReader reader = new StringReader(text);
            ProblemComponent arrProblemComponent[] = new ProblemComponent[1];
            arrProblemComponent[0] =
                    new ProblemComponentFactory().buildFromXML(reader, true);
            problem.setProblemComponents(arrProblemComponent);
        }

        ProblemRenderer pr = new ProblemRenderer(problem);
        pr.setTdClass(styleClass);

        try {
            pageContext.getOut().println(pr.toHTML(language));
        } catch (Exception e) {
            //e.printStackTrace();
            throw new JspException(e.getMessage());
        }

        return SKIP_BODY;
    }

    /**
     * Just in case the app server is caching tag (jboss!!!)
     * we have to clear out all the instance variables at the
     * end of execution
     */
    public int doEndTag() throws JspException {
        this.styleClass = null;
        this.text = null;
        this.language = null;
        this.problem = null;
        return super.doEndTag();
    }
}