[Yanel-commits] rev 41528 - public/yanel/trunk/src/webapp/src/java/org/wyona/yanel/servlet

Wed Feb 11 14:51:00 CET 2009

Author: michi
Date: 2009-02-11 14:51:00 +0100 (Wed, 11 Feb 2009)
New Revision: 41528

Modified:
   public/yanel/trunk/src/webapp/src/java/org/wyona/yanel/servlet/YanelHTMLUI.java
Log:
check for head tag and if no head tag exists, then add one

Modified: public/yanel/trunk/src/webapp/src/java/org/wyona/yanel/servlet/YanelHTMLUI.java
===================================================================

--- public/yanel/trunk/src/webapp/src/java/org/wyona/yanel/servlet/YanelHTMLUI.java	2009-02-11 13:50:15 UTC (rev 41527)
+++ public/yanel/trunk/src/webapp/src/java/org/wyona/yanel/servlet/YanelHTMLUI.java	2009-02-11 13:51:00 UTC (rev 41528)
@@ -188,6 +188,7 @@
         int c;
         int state = OUTSIDE_TAG;
         StringBuffer tagBuf = null;
+        boolean headExists = false;
         int headcount = 0;
         int bodycount = 0;
         while ((c = reader.read()) != -1) {
@@ -207,6 +208,7 @@
                     tagBuf.append((char)c);
                     String tag = tagBuf.toString();
                     if (tag.startsWith("<head")) {
+                        headExists = true;
                         if (headcount == 0) {
                             writer.write(tag, 0, tag.length());
                             String toolbarString = getToolbarHeader(resource, request);
@@ -216,6 +218,16 @@
                         }
                         headcount++;
                     } else if (tag.startsWith("<body")) {
+                        if (!headExists) {
+                            log.warn("No <head> exists. Hence <head> will be added dynamically.");
+                            String headStartTag = "<head>";
+                            writer.write(headStartTag, 0, headStartTag.length());
+                            String toolbarString = getToolbarHeader(resource, request);
+                            writer.write(toolbarString, 0, toolbarString.length());
+                            String headEndTag = "</head>";
+                            writer.write(headEndTag, 0, headEndTag.length());
+                        }
+
                         if (bodycount == 0) {
                             writer.write(tag, 0, tag.length());
                             String toolbarString = getToolbarBodyStart(resource, request);

