lib/rack/dev-mark.rb in rack-dev-mark-0.0.4 vs lib/rack/dev-mark.rb in rack-dev-mark-0.0.5

@@ -10,7 +10,23 @@
 module Rack
   module DevMark
     def self.env
       ENV['RAILS_ENV'] || ENV['RACK_ENV']
     end
+
+    def self.production_env
+      @production_env ||= ['production']
+    end
+
+    def self.production_env=(*production_env)
+      @production_env = production_env.flatten.map(&:to_s)
+    end
+
+    def self.theme
+      @theme ||= :github_fork_ribbon
+    end
+
+    def self.theme=(theme)
+      @theme = theme
+    end
   end
 end