lib/trainbbcode.rb in trainbbcode-1.0.1 vs lib/trainbbcode.rb in trainbbcode-1.1.0

@@ -6,11 +6,11 @@
 require 'trainbbcode/swear_filter.rb'
 require 'rubygems'
 require 'coderay'
 
 #Helper Method
-if defined? RAILS_ROOT then
+if defined? Rails
 	require 'trainbbcode/application_helper.rb'
 	ActionView::Base.send :include, TBBCHelper
 end
 
 class TBBC
@@ -42,12 +42,19 @@
 		input=input.gsub("\r","")
 		scan=input.scan(/\[code lang=(.+?)\](.+?)\[\/code\]/m)
 		scan.each do |splits|
 			parse=splits[1].gsub("&lt;","<").gsub("&gt;",">")
 			lang=splits[0]
-			parsed="[nobbc]" + CodeRay.scan(parse, lang).div(:css => :class, :line_numbers => @config[:syntax_highlighting_line_numbers]) + "[/nobbc]"
+			parsed="[nobbc]" + CodeRay.scan(parse, lang).div(:line_numbers => @config[:syntax_highlighting_line_numbers]) + "[/nobbc]"
 			input=input.gsub("[code lang=#{lang}]#{splits[1]}[/code]",parsed)
 		end
 		input
 	end
-	
+
+	def needs_html_safe?
+		if defined? Rails
+			return Rails.version =~ /^3\./
+		else
+			return false
+		end
+	end
 end