/**
    * Copyright 2005-2016 The Kuali Foundation
    *
    * Licensed under the Educational Community 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.opensource.org/licenses/ecl2.php
    *
   * 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 org.kuali.rice.krad.uif.freemarker;
  
  import java.io.IOException;
  import java.io.Writer;
  import java.util.Collection;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  import org.apache.commons.lang.StringEscapeUtils;
  import org.kuali.rice.krad.uif.UifConstants;
  import org.kuali.rice.krad.uif.component.Component;
  import org.kuali.rice.krad.uif.container.CollectionGroup;
  import org.kuali.rice.krad.uif.container.Group;
  import org.kuali.rice.krad.uif.layout.LayoutManager;
  import org.kuali.rice.krad.uif.layout.StackedLayoutManager;
  import org.kuali.rice.krad.uif.util.ScriptUtils;
  import org.kuali.rice.krad.uif.widget.Disclosure;
  import org.kuali.rice.krad.uif.element.Pager;
  import org.kuali.rice.krad.uif.widget.Tooltip;
  import org.springframework.util.StringUtils;
  
  import freemarker.core.Environment;
  import freemarker.core.InlineTemplateUtils;
  import freemarker.core.Macro;
  import freemarker.ext.beans.BeansWrapper;
  import freemarker.template.ObjectWrapper;
  import freemarker.template.TemplateException;
  import freemarker.template.TemplateModel;
  import freemarker.template.TemplateModelException;
  
  /**
   * Inline FreeMarker rendering utilities.
   *
   * @author Kuali Rice Team (rice.collab@kuali.org)
   */
  public class FreeMarkerInlineRenderUtils {
  
      /**
       * Resolve a FreeMarker environment variable as a Java object.
       *
       * @param env The FreeMarker environment.
       * @param name The name of the variable.
       * @return The FreeMarker variable, resolved as a Java object.
       * @see #resolve(Environment, String, Class) for the preferred means to resolve variables for
       * inline rendering.
       */
      @SuppressWarnings("unchecked")
      public static <T> T resolve(Environment env, String name) {
          TemplateModel tm = resolveModel(env, name);
          try {
              return (T) getBeansWrapper(env).unwrap(tm);
          } catch (TemplateModelException e) {
              throw new IllegalArgumentException("Failed to unwrap " + name + ", template model " + tm, e);
          }
      }
  
      /**
       * Resolve a FreeMarker environment variable as a Java object, with type enforcement.
       *
       * <p>
       * This method is the preferred means to resolve variables for inline rendering.
       * </p>
       *
       * @param env The FreeMarker environment.
       * @param name The name of the variable.
       * @param type The expected type of the variable.
       * @return The FreeMarker variable, resolved as a Java object of the given type.
       */
      public static <T> T resolve(Environment env, String name, Class<T> type) {
          Object rv = resolve(env, name);
  
          if ((rv instanceof Collection) && !Collection.class.isAssignableFrom(type)) {
              Collection<?> rc = (Collection<?>) rv;
              if (rc.isEmpty()) {
                  return null;
              } else {
                  rv = rc.iterator().next();
              }
          }
  
          if ("".equals(rv) && !String.class.equals(type)) {
              return null;
          } else {
             return type.cast(rv);
         }
     }
 
     /**
      * Get the object wrapper from the FreeMarker environment, as a {@link BeansWrapper}.
      *
      * @param env The FreeMarker environment.
      * @return The object wrapper from the FreeMarker environment, type-cast as {@link BeansWrapper}
      * .
      */
     public static BeansWrapper getBeansWrapper(Environment env) {
         ObjectWrapper wrapper = env.getObjectWrapper();
 
         if (!(wrapper instanceof BeansWrapper)) {
             throw new UnsupportedOperationException("FreeMarker environment uses unsupported ObjectWrapper " + wrapper);
         }
 
         return (BeansWrapper) wrapper;
     }
 
     /**
      * Resovle a FreeMarker variable as a FreeMarker template model object.
      *
      * @param env The FreeMarker environment.
      * @param name The name of the variable.
      * @return The FreeMarker variable, resolved as a FreeMarker template model object.
      * @see #resolve(Environment, String, Class) for the preferred means to resolve variables for
      * inline rendering.
      */
     public static TemplateModel resolveModel(Environment env, String name) {
         try {
             return env.getVariable(name);
         } catch (TemplateModelException e) {
             throw new IllegalArgumentException("Failed to resolve " + name + " in current freemarker environment", e);
         }
     }
 
     /**
      * Render a KRAD component template inline.
      *
      * <p>
      * This method originated as template.ftl, and supercedes the previous content of that template.
      * </p>
      *
      * @param env The FreeMarker environment.
      * @param component The component to render a template for.
      * @param body The nested body.
      * @param componentUpdate True if this is an update, false for full view.
      * @param includeSrc True to include the template source in the environment when rendering,
      * false to skip inclusion.
      * @param tmplParms Additional parameters to pass to the template macro.
      * @throws TemplateException If FreeMarker rendering fails.
      * @throws IOException If rendering is interrupted due to an I/O error.
      */
     public static void renderTemplate(Environment env, Component component, String body, boolean componentUpdate,
             boolean includeSrc, Map<String, TemplateModel> tmplParms) throws TemplateException, IOException {
         String dataJsScripts = "";
         String templateJsScripts = "";
 
         if (component == null) {
             return;
         }
 
         String s;
         Writer out = env.getOut();
 
         if ((component.isRender() && (!component.isRetrieveViaAjax() || componentUpdate)) || (component
                 .getProgressiveRender() != null && !component.getProgressiveRender().equals("") && !component
                 .isProgressiveRenderViaAJAX() && !component.isProgressiveRenderAndRefresh())) {
 
             if (StringUtils.hasText(s = component.getPreRenderContent())) {
                 out.write(s);
             }
 
             if (component.isSelfRendered()) {
                 out.write(component.getRenderedHtmlOutput());
             } else {
                 if (includeSrc) {
                     env.include(component.getTemplate(), env.getTemplate().getEncoding(), true);
                 }
 
                 Macro fmMacro = component.getTemplateName() == null ? null : (Macro) env.getMainNamespace().get(
                         component.getTemplateName());
 
                 if (fmMacro == null) {
                     // force inclusion of the source to see if we can get the macro
                     env.include(component.getTemplate(), env.getTemplate().getEncoding(), true);
                     fmMacro = component.getTemplateName() == null ? null : (Macro) env.getCurrentNamespace().get(
                             component.getTemplateName());
 
                     // if still missing throw an exception
                     if (fmMacro == null) {
                         throw new TemplateException("No macro found using " + component.getTemplateName(), env);
                     }
                 }
 
                 Map<String, Object> args = new java.util.HashMap<String, Object>();
                 args.put(component.getComponentTypeName(), component);
 
                 if (tmplParms != null) {
                     args.putAll(tmplParms);
                 }
 
                 if (StringUtils.hasText(body)) {
                     args.put("body", body);
                 }
 
                 InlineTemplateUtils.invokeMacro(env, fmMacro, args, null);
             }
 
             if (StringUtils.hasText(s = component.getEventHandlerScript())) {
                 templateJsScripts += s;
             }
 
             if (StringUtils.hasText(s = component.getScriptDataAttributesJs())) {
                 dataJsScripts += s;
             }
 
             if (StringUtils.hasText(s = component.getPostRenderContent())) {
                 out.append(s);
             }
 
         }
 
         if (componentUpdate || UifConstants.ViewStatus.RENDERED.equals(component.getViewStatus())) {
             renderScript(dataJsScripts, component, UifConstants.RoleTypes.DATA_SCRIPT, out);
             renderScript(templateJsScripts, component, null, out);
             return;
         }
 
         String methodToCallOnRefresh = component.getMethodToCallOnRefresh();
         if (!StringUtils.hasText(methodToCallOnRefresh)) {
             methodToCallOnRefresh = "";
         }
 
         if ((!component.isRender() && (component.isProgressiveRenderViaAJAX() || component
                 .isProgressiveRenderAndRefresh() || component.isDisclosedByAction() || component.isRefreshedByAction()))
                 || component.isRetrieveViaAjax()) {
             out.write("<span id=\"");
             out.write(component.getId());
             out.write("\" data-role=\"placeholder\" class=\"uif-placeholder "
                     + component.getStyleClassesAsString()
                     + "\"></span>");
         }
 
         if (StringUtils.hasText(component.getProgressiveRender())) {
             for (String cName : component.getProgressiveDisclosureControlNames()) {
                 templateJsScripts += "var condition = function(){return ("
                         + component.getProgressiveDisclosureConditionJs()
                         + ");};setupProgressiveCheck('"
                         + StringEscapeUtils.escapeJavaScript(cName)
                         + "', '"
                         + component.getId()
                         + "', condition,"
                         + component.isProgressiveRenderAndRefresh()
                         + ", '"
                         + methodToCallOnRefresh
                         + "', "
                         + ScriptUtils.translateValue(component.getFieldsToSendOnRefresh())
                         + ");";
             }
 
             templateJsScripts += "hiddenInputValidationToggle('" + component.getId() + "');";
         }
 
         if (StringUtils.hasText(component.getConditionalRefresh())) {
             for (String cName : component.getConditionalRefreshControlNames()) {
                 templateJsScripts += "var condition = function(){return ("
                         + component.getConditionalRefreshConditionJs()
                         + ");};setupRefreshCheck('"
                         + StringEscapeUtils.escapeJavaScript(cName)
                         + "', '"
                         + component.getId()
                         + "', condition,'"
                         + methodToCallOnRefresh
                         + "', "
                         + ScriptUtils.translateValue(component.getFieldsToSendOnRefresh())
                         + ");";
             }
         }
 
         List<String> refreshWhenChanged = component.getRefreshWhenChangedPropertyNames();
         if (refreshWhenChanged != null) {
             for (String cName : refreshWhenChanged) {
                 templateJsScripts += "setupOnChangeRefresh('"
                         + StringEscapeUtils.escapeJavaScript(cName)
                         + "', '"
                         + component.getId()
                         + "','"
                         + methodToCallOnRefresh
                         + "', "
                         + ScriptUtils.translateValue(component.getFieldsToSendOnRefresh())
                         + ");";
             }
         }
 
         renderScript(dataJsScripts, component, UifConstants.RoleTypes.DATA_SCRIPT, out);
         renderScript(templateJsScripts, component, null, out);
 
         renderTooltip(component, out);
     }
 
     /**
      * Render a KRAD tooltip component.
      *
      * <p>
      * This method originated as template.ftl, and supercedes the previous content of that template.
      * </p>
      *
      * @param component The component to render a tooltip for.
      * @param out The output writer to render to, typically from {@link Environment#getOut()}.
      * @throws IOException If rendering is interrupted due to an I/O error.
      */
     public static void renderTooltip(Component component, Writer out) throws IOException {
         Tooltip tt = component.getToolTip();
         String script = "";
         if (tt != null && StringUtils.hasText(tt.getTooltipContent())) {
             String templateOptionsJSString = tt.getTemplateOptionsJSString();
             script += "createTooltip('"
                     + component.getId()
                     + "', '"
                     + tt.getTooltipContent()
                     + "', "
                     + (templateOptionsJSString == null ? "''" : templateOptionsJSString)
                     + ", "
                     + tt.isOnMouseHover()
                     + ", "
                     + tt.isOnFocus()
                     + ");";
 
             renderScript(script, component, null, out);
         }
     }
 
     /**
      * Render a KRAD script component.
      *
      * <p>
      * This method originated as script.ftl, and supercedes the previous content of that template.
      * </p>
      *
      * @param script The script to render.
      * @param component The component the script is related to.
      * @param out The output writer to render to, typically from {@link Environment#getOut()}.
      * @throws IOException If rendering is interrupted due to an I/O error.
      */
     public static void renderScript(String script, Component component, String role, Writer out) throws IOException {
         if (script == null || "".equals(script.trim())) {
             return;
         }
         out.write("<input name=\"script\" type=\"hidden\" data-role=\"");
         out.write(role == null ? "script" : role);
         out.write("\" ");
 
         if (component != null && component.getId() != null) {
             out.write("data-for=\"");
             out.write(component.getId());
             out.write("\" ");
         }
 
         out.write("value=\"");
         out.write(StringEscapeUtils.escapeHtml(script));
         out.write("\" />");
     }
 
     /**
      * Render common attributes for a KRAD component.
      *
      * <p>
      * NOTICE: By KULRICE-10353 this method duplicates, but does not replace,
      * krad/WEB-INF/ftl/lib/attrBuild.ftl. When updating this method, also update that template.
      * </p>
      *
      * @param component The component to open a render attributes for.
      * @param out The output writer to render to, typically from {@link Environment#getOut()}.
      * @throws IOException If rendering is interrupted due to an I/O error.
      */
     public static void renderAttrBuild(Component component, Writer out) throws IOException {
         String s = component.getStyleClassesAsString();
         if (StringUtils.hasText(s)) {
             out.write(" class=\"");
             out.write(s);
             out.write("\"");
         }
 
         s = component.getStyle();
         if (StringUtils.hasText(s)) {
             out.write(" style=\"");
             out.write(s);
             out.write("\"");
         }
 
         s = component.getTitle();
         if (StringUtils.hasText(s)) {
             out.write(" title=\"");
             out.write(s);
             out.write("\"");
         }
 
         s = component.getRole();
         if (StringUtils.hasText(s)) {
             out.write(" role=\"");
             out.write(s);
             out.write("\"");
         }
 
         s = component.getAriaAttributesAsString();
         if (StringUtils.hasText(s)) {
             out.write(s);
         }
     }
 
     /**
      * Render an open div tag for a component.
      *
      * <p>
      * NOTE: Inline rendering performance is improved by *not* passing continuations for nested body
      * content, so the open div and close div methods are implemented separately. Always call
      * {@link #renderCloseDiv(Writer)} after rendering the &lt;div&gt; body related to this open
      * tag.
      * </p>
      *
      * <p>
      * NOTICE: By KULRICE-10353 this method duplicates, but does not replace,
      * krad/WEB-INF/ftp/lib/div.ftl. When updating this method, also update that template.
      * </p>
      *
      * @param component The component to render a wrapper div for.
      * @param out The output writer to render to, typically from {@link Environment#getOut()}.
      * @throws IOException If rendering is interrupted due to an I/O error.
      */
     public static void renderOpenDiv(Component component, Writer out) throws IOException {
         out.write("<div id=\"");
         out.write(component.getId());
         out.write("\"");
         renderAttrBuild(component, out);
         out.write(component.getSimpleDataAttributes());
         out.write(">");
     }
 
     /**
      * Render a close div tag for a component.
      *
      * <p>
      * NOTE: Inline rendering performance is improved by *not* passing continuations for nested body
      * content, so the open div and close div methods are implemented separately. Always call this
      * method after rendering the &lt;div&gt; body related to and open tag rendered by
      * {@link #renderOpenDiv(Component, Writer)}.
      * </p>
      *
      * <p>
      * NOTICE: By KULRICE-10353 this method duplicates, but does not replace,
      * krad/WEB-INF/ftp/lib/div.ftl. When updating this method, also update that template.
      * </p>
      *
      * @param out The output writer to render to, typically from {@link Environment#getOut()}.
      * @throws IOException If rendering is interrupted due to an I/O error.
      */
     public static void renderCloseDiv(Writer out) throws IOException {
         out.write("</div>");
     }
 
     /**
      * Render open tags wrapping a group component.
      *
      * <p>
      * NOTE: Inline rendering performance is improved by *not* passing continuations for nested body
      * content, so the open and close methods are implemented separately. Always call
      * {@link #renderCloseGroupWrap(Environment, Group)} after rendering the body related to a call to
      * {@link #renderOpenGroupWrap(Environment, Group)}.
      * </p>
      *
      * <p>
      * This method originated as groupWrap.ftl, and supercedes the previous content of that
      * template.
      * </p>
      *
      * @param env The FreeMarker environment to use for rendering.
      * @param group The group to render open wrapper tags for.
      * @throws IOException If rendering is interrupted due to an I/O error.
      * @throws TemplateException If FreeMarker rendering fails.
      */
     public static void renderOpenGroupWrap(Environment env, Group group) throws IOException, TemplateException {
         Writer out = env.getOut();
         renderTemplate(env, group.getHeader(), null, false, false, null);
 
         if (group.isRenderLoading()) {
             out.write("<div id=\"");
             out.write(group.getId());
             out.write("_disclosureContent\" data-role=\"placeholder\"> Loading... </div>");
         } else {
             Disclosure disclosure = group.getDisclosure();
             if (disclosure != null && disclosure.isRender()) {
                 out.write("<div id=\"");
                 out.write(group.getId() + UifConstants.IdSuffixes.DISCLOSURE_CONTENT);
                 out.write("\" data-role=\"disclosureContent\" data-open=\"");
                 out.write(Boolean.toString(disclosure.isDefaultOpen()));
                 out.write("\" class=\"uif-disclosureContent\">");
             }
             renderTemplate(env, group.getInstructionalMessage(), null, false, false, null);
         }
     }
 
     /**
      * Render close tags wrapping a group component.
      *
      * <p>
      * NOTE: Inline rendering performance is improved by *not* passing continuations for nested body
      * content, so the open and close methods are implemented separately. Always call
      * {@link #renderCloseGroupWrap(Environment, Group)} after rendering the body related to a call to
      * {@link #renderOpenGroupWrap(Environment, Group)}.
      * </p>
      *
      * <p>
      * This method originated as groupWrap.ftl, and supercedes the previous content of that
      * template.
      * </p>
      *
      * @param env The FreeMarker environment to use for rendering.
      * @param group The group to render open wrapper tags for.
      * @throws IOException If rendering is interrupted due to an I/O error.
      * @throws TemplateException If FreeMarker rendering fails.
      */
     public static void renderCloseGroupWrap(Environment env, Group group) throws IOException, TemplateException {
         Writer out = env.getOut();
 
         boolean renderLoading = group.isRenderLoading();
         if (!renderLoading) {
             renderTemplate(env, group.getFooter(), null, false, false, null);
         }
 
         Disclosure disclosure = group.getDisclosure();
         if (disclosure != null && disclosure.isRender()) {
             if (!renderLoading) {
                 out.write("</div>");
             }
             Map<String, TemplateModel> tmplParms = new HashMap<String, TemplateModel>();
             tmplParms.put("parent", env.getObjectWrapper().wrap(group));
             renderTemplate(env, disclosure, null, false, false, tmplParms);
         }
     }
 
     /**
      * Render a collection group inline.
      *
      * <p>
      * This method originated as collectionGroup.ftl, and supercedes the previous content of that
      * template.
      * </p>
      *
      * @param group The collection group to render.
      * @throws IOException If rendering is interrupted due to an I/O error.
      * @throws TemplateException If FreeMarker rendering fails.
      */
     public static void renderCollectionGroup(Environment env,
             CollectionGroup group) throws IOException, TemplateException {
         renderOpenGroupWrap(env, group);
 
         Map<String, TemplateModel> tmplParms = new HashMap<String, TemplateModel>();
         tmplParms.put("componentId", env.getObjectWrapper().wrap(group.getId()));
         renderTemplate(env, group.getCollectionLookup(), null, false, false, tmplParms);
 
         if ("TOP".equals(group.getAddLinePlacement())) {
             if (group.isRenderAddBlankLineButton()) {
                 renderTemplate(env, group.getAddBlankLineAction(), null, false, false, null);
             }
 
             if (group.isAddWithDialog()) {
                 renderTemplate(env, group.getAddWithDialogAction(), null, false, false, null);
             }
         }
 
         LayoutManager layoutManager = group.getLayoutManager();
         String managerTemplateName = layoutManager.getTemplateName();
         List<? extends Component> items = group.getItems();
 
         if ("uif_stacked".equals(managerTemplateName)) {
             renderStacked(env, items, (StackedLayoutManager) layoutManager, group);
         } else {
             Macro fmMacro = (Macro) env.getMainNamespace().get(layoutManager.getTemplateName());
 
             if (fmMacro == null) {
                 throw new TemplateException("No macro found using " + layoutManager.getTemplateName(), env);
             }
 
             Map<String, Object> args = new java.util.HashMap<String, Object>();
             args.put("items", items);
             args.put("manager", group.getLayoutManager());
             args.put("container", group);
             InlineTemplateUtils.invokeMacro(env, fmMacro, args, null);
         }
 
         if ("BOTTOM".equals(group.getAddLinePlacement())) {
             if (group.isRenderAddBlankLineButton()) {
                 renderTemplate(env, group.getAddBlankLineAction(), null, false, false, null);
             }
 
             if (group.isAddWithDialog()) {
                 renderTemplate(env, group.getAddWithDialogAction(), null, false, false, null);
             }
         }
 
         if (group.isAddWithDialog()) {
             renderTemplate(env, group.getAddLineDialog(), null, false, false, null);
         }
 
         renderCloseGroupWrap(env, group);
     }
 
     /**
      * Render a stacked collection inline.
      *
      * <p>
      * This method originated as stacked.ftl, and supercedes the previous content of that
      * template.
      * </p>
      *
      * @param env The FreeMarker environment
      * @param items List of items to render in a stacked layout
      * @param manager Layout manager for the container
      * @param container Container to render
      * @throws IOException If rendering is interrupted due to an I/O error.
      * @throws TemplateException If FreeMarker rendering fails.
      */
     public static void renderStacked(Environment env, List<? extends Component> items, StackedLayoutManager manager,
             CollectionGroup container) throws IOException, TemplateException {
         String s;
         Writer out = env.getOut();
 
         Pager pager = manager.getPagerWidget();
         Map<String, TemplateModel> pagerTmplParms = null;
         if (pager != null && container.isUseServerPaging()) {
             pagerTmplParms = new HashMap<String, TemplateModel>();
             renderTemplate(env, pager, null, false, false, pagerTmplParms);
         }
 
 /*
         out.write("<div id=\"");
         out.write(manager.getId());
         out.write("\"");
 
         s = manager.getStyle();
         if (StringUtils.hasText(s)) {
             out.write(" style=\"");
             out.write(s);
             out.write("\"");
         }
 
         s = manager.getStyleClassesAsString();
         if (StringUtils.hasText(s)) {
             out.write(" class=\"");
             out.write(s);
             out.write("\"");
         }
 
         out.write(">");
 */
 
         Group wrapperGroup = manager.getWrapperGroup();
         if (wrapperGroup != null) {
             renderTemplate(env, wrapperGroup, null, false, false, null);
         } else {
             for (Group item : manager.getStackedGroups()) {
                 renderTemplate(env, item, null, false, false, null);
             }
         }
 
         /*out.write("</div>");*/
 
         if (pager != null && container.isUseServerPaging()) {
             pagerTmplParms = new HashMap<String, TemplateModel>();
             renderTemplate(env, pager, null, false, false, pagerTmplParms);
         }
     }
 
 }