lib/milksteak.rb in milksteak-0.0.1 vs lib/milksteak.rb in milksteak-0.0.2

@@ -1,5 +1,56 @@
 require "milksteak/version"
+require "sinatra/base"
+require "models/user"
+require "models/page"
+require "liquid"
+require "bluecloth"
 
 module Milksteak
-  # Your code goes here...
+  class Admin < Sinatra::Base
+    set :views, File.join(File.dirname(__FILE__), "views")
+    set :public_folder, File.join(File.dirname(__FILE__), "public")
+
+    helpers do
+      def h(s); ERB::Util.h(s); end
+
+      def erb(*args)
+        args[0] = args[0].to_sym
+        if args[1]
+          args[1][:layout] = "layouts/#{args[1][:layout]}".to_sym if args[1][:layout]
+        end
+        super
+      end
+
+      def coffee(*args)
+        args[0] = args[0].to_sym
+        super
+      end
+  
+      def partial(template, *args)
+        template_array = template.to_s.split('/')
+        template = template_array[0..-2].join('/') + "/_#{template_array[-1]}"
+        options = args.last.is_a?(Hash) ? args.pop : {}
+        erb(:"#{template}", options)
+      end
+
+      def flash
+        session[:flash] = {} if session[:flash] && session[:flash].class != Hash
+        session[:flash] ||= {}
+      end
+
+      def flash_messages # prints flash messages nicely and consistently
+        return if !session[:flash] or session[:flash].empty?
+        msgs = []
+        [:success, :error, :info].each do |notice|
+          msgs << "<div class=\"alert alert-#{notice}\"><a class=\"close\" data-dismiss=\"alert\">x</a>#{flash[notice]}</div>" if flash[notice]
+        end
+        flash.clear
+        msgs.join
+      end
+    end
+    
+    get "/milksteak" do
+      erb "login", :layout => "admin"
+    end
+  end
 end