add possibility to add javascript

1 files changed, 13 insertions, 0 deletions

diff --git a/html.rb b/html.rb
index 5dbb3e6..57f69a6 100644
--- a/html.rb
+++ b/html.rb
@@ -31,6 +31,7 @@ class HTML
 		end
 
 		@body = ""
+		@htmlheader = ''
 		@css = []
 		@atom = []
 	end
@@ -52,6 +53,8 @@ HEAD
 			ret += "<link rel='alternate'  type='application/atom+xml' href='#{href}' />\n"
 		}
 
+		ret += @htmlheader
+
 		ret += "</head>"
 		ret
 	end
@@ -72,9 +75,19 @@ HEAD
 		@header["cookie"] ||= []
 		@header["cookie"] << c
 	end
+	def add_head_script(file)
+		add_html_head("<script type='text/javascript' src='#{file}'></script>")
+	end
+	def add_script(file)
+		self << "<script type='text/javascript' src='#{file}'></script>"
+	end
 	def << (bodycontent)
 		@body += bodycontent.chomp + "\n"
 	end
+	def add_html_head(headercontent)
+		@htmlheader += headercontent.chomp + "\n"
+	end
+
 	def out(cgi)
 		#FIXME: quick and dirty fix for encoding problem
 		{