joe-merb-core 0.9.8

data/lib/merb-core/plugins.rb

@@ -0,0 +1,67 @@
+module Merb
+
+  module Plugins
+
+    # ==== Returns
+    # Hash::
+    #   The configuration loaded from Merb.root / "config/plugins.yml" or, if
+    #   the load fails, an empty hash whose default value is another Hash.
+    def self.config
+      @config ||= begin
+        # this is so you can do Merb.plugins.config[:helpers][:awesome] = "bar"
+        config_hash = Hash.new {|h,k| h[k] = {}}
+        file = Merb.root / "config" / "plugins.yml"
+
+        if File.exists?(file)
+          require 'yaml'
+          to_merge = YAML.load_file(file)
+        else
+          to_merge = {}
+        end
+        
+        config_hash.merge(to_merge)
+      end
+    end
+
+    # ==== Returns
+    # Array(String):: All Rakefile load paths Merb uses for plugins.
+    def self.rakefiles
+      Merb.rakefiles
+    end
+    
+    # ==== Returns
+    # Array(String):: All Generator load paths Merb uses for plugins.
+    def self.generators
+      Merb.generators
+    end
+
+    # ==== Parameters
+    # *rakefiles:: Rakefiles to add to the list of plugin Rakefiles.
+    #
+    # ==== Notes
+    #
+    # This is a recommended way to register your plugin's Raketasks
+    # in Merb.
+    #
+    # ==== Examples
+    # From merb_sequel plugin:
+    #
+    # if defined(Merb::Plugins)
+    #   Merb::Plugins.add_rakefiles "merb_sequel" / "merbtasks"
+    # end
+    def self.add_rakefiles(*rakefiles)
+      Merb.add_rakefiles(*rakefiles)
+    end
+    
+    # ==== Parameters
+    # *generators:: Generator paths to add to the list of plugin generators.
+    #
+    # ==== Notes
+    #
+    # This is the recommended way to register your plugin's generators
+    # in Merb.
+    def self.add_generators(*generators)
+      Merb.add_generators(*generators)
+    end
+  end
+end

data/lib/merb-core/rack.rb

@@ -0,0 +1,27 @@
+require 'rack'
+module Merb
+  module Rack
+    autoload :Application,         'merb-core/rack/application'
+    autoload :Adapter,             'merb-core/rack/adapter'
+    autoload :AbstractAdapter,     'merb-core/rack/adapter/abstract'
+    autoload :Ebb,                 'merb-core/rack/adapter/ebb'
+    autoload :EventedMongrel,      'merb-core/rack/adapter/evented_mongrel'    
+    autoload :FastCGI,             'merb-core/rack/adapter/fcgi'
+    autoload :Irb,                 'merb-core/rack/adapter/irb'
+    autoload :Middleware,          'merb-core/rack/middleware'
+    autoload :Mongrel,             'merb-core/rack/adapter/mongrel'
+    autoload :Runner,              'merb-core/rack/adapter/runner'    
+    autoload :SwiftipliedMongrel,  'merb-core/rack/adapter/swiftiplied_mongrel'
+    autoload :Thin,                'merb-core/rack/adapter/thin'
+    autoload :ThinTurbo,           'merb-core/rack/adapter/thin_turbo'
+    autoload :WEBrick,             'merb-core/rack/adapter/webrick'
+    autoload :PathPrefix,          'merb-core/rack/middleware/path_prefix'
+    autoload :Static,              'merb-core/rack/middleware/static'
+    autoload :Profiler,            'merb-core/rack/middleware/profiler'
+    autoload :Tracer,              'merb-core/rack/middleware/tracer'
+    autoload :ContentLength,       'merb-core/rack/middleware/content_length'
+    autoload :ConditionalGet,      'merb-core/rack/middleware/conditional_get'
+    autoload :Csrf,                'merb-core/rack/middleware/csrf'
+    autoload :StreamWrapper,       'merb-core/rack/stream_wrapper'
+  end # Rack
+end # Merb

data/lib/merb-core/rack/adapter.rb

@@ -0,0 +1,47 @@
+module Merb
+  
+  module Rack
+    
+    class Adapter
+
+      class << self
+        # ==== Parameters
+        # id<String>:: The identifier of the Rack adapter class to retrieve.
+        #
+        # ==== Returns.
+        # Class:: The adapter class.
+        def get(id)
+          if @adapters[id.to_s]
+            Object.full_const_get(@adapters[id.to_s])
+          else
+            Merb.fatal! "The adapter #{id} did not exist"
+          end
+        end
+
+        # Registers a new Rack adapter.
+        #
+        # ==== Parameters
+        # ids<Array>:: Identifiers by which this adapter is recognized by.
+        # adapter_class<Class>:: The Rack adapter class.
+        def register(ids, adapter_class)
+          @adapters ||= Hash.new
+          ids.each { |id| @adapters[id] = "Merb::Rack::#{adapter_class}" }
+        end
+      end # class << self
+      
+    end # Adapter
+    
+    # Register some Rack adapters
+    Adapter.register %w{ebb},            :Ebb
+    Adapter.register %w{emongrel},       :EventedMongrel
+    Adapter.register %w{fastcgi fcgi},   :FastCGI
+    Adapter.register %w{irb},            :Irb
+    Adapter.register %w{mongrel},        :Mongrel  
+    Adapter.register %w{runner},         :Runner
+    Adapter.register %w{smongrel swift}, :SwiftipliedMongrel
+    Adapter.register %w{thin},           :Thin
+    Adapter.register %w{thin-turbo tt},  :ThinTurbo
+    Adapter.register %w{webrick},        :WEBrick
+    
+  end # Rack
+end # Merb

data/lib/merb-core/rack/adapter/ebb.rb

@@ -0,0 +1,24 @@
+require 'ebb'
+module Merb
+  
+  module Rack
+
+    class Ebb < Merb::Rack::AbstractAdapter
+      # start an Ebb server on given host and port.
+      
+      def self.new_server(port)
+        Merb::Dispatcher.use_mutex = false
+        opts = @opts.merge(:port => port)
+        @th = Thread.new { Thread.current[:server] = ::Ebb.start_server(opts[:app], opts) }
+      end
+      
+      def self.start_server
+        @th.join
+      end
+      
+      def self.stop(status = 0)
+        ::Ebb.stop_server
+      end
+    end
+  end
+end

data/lib/merb-core/rack/adapter/evented_mongrel.rb

@@ -0,0 +1,13 @@
+require 'swiftcore/evented_mongrel'
+require 'merb-core/rack/handler/mongrel'
+module Merb
+  module Rack
+
+    class EventedMongrel < Mongrel
+      def self.new_server(port)
+        Merb::Dispatcher.use_mutex = false
+        super
+      end
+    end
+  end
+end

data/lib/merb-core/rack/adapter/fcgi.rb

@@ -0,0 +1,17 @@
+module Merb
+  module Rack
+    
+    class FastCGI
+      # ==== Parameters
+      # opts<Hash>:: Options for FastCGI (see below).
+      #
+      # ==== Options (opts)
+      # :app<String>>:: The application name.
+      def self.start(opts={})
+        Merb.logger.warn!("Using FastCGI adapter")
+        Merb::Server.change_privilege
+        ::Rack::Handler::FastCGI.run(opts[:app], opts)
+      end
+    end
+  end
+end

data/lib/merb-core/rack/adapter/irb.rb

@@ -0,0 +1,119 @@
+module Merb
+  module Rack
+    class Console
+      # ==== Parameters
+      # name<~to_sym, Hash>:: The name of the route to generate.
+      # params<Hash>:: The params to use in the route generation.
+      #
+      # ==== Returns
+      # String:: The generated URL.
+      #
+      # ==== Alternatives
+      # If name is a hash, it will be merged with params.
+      def url(name, *args)
+        args << {}
+        Merb::Router.url(name, *args)
+      end
+
+      # Reloads classes using Merb::BootLoader::ReloadClasses.
+      def reload!
+        Merb::BootLoader::ReloadClasses.reload
+      end
+
+      # Prints all routes for the application.
+      def show_routes
+        seen = []
+        unless Merb::Router.named_routes.empty?
+          puts "==== Named routes"
+          Merb::Router.named_routes.each do |name,route|
+            # something weird happens when you combine sprintf and irb
+            puts "Helper     : #{name}"
+            meth = $1.upcase if route.conditions[:method].to_s =~ /(get|post|put|delete)/
+            puts "HTTP method: #{meth || 'GET'}"
+            puts "Route      : #{route}"
+            puts "Params     : #{route.params.inspect}"
+            puts
+            seen << route
+          end
+        end
+        puts "==== Anonymous routes"
+        (Merb::Router.routes - seen).each do |route|
+          meth = $1.upcase if route.conditions[:method].to_s =~ /(get|post|put|delete)/
+          puts "HTTP method: #{meth || 'GET'}"
+          puts "Route      : #{route}"
+          puts "Params     : #{route.params.inspect}"
+          puts
+        end
+        nil
+      end
+
+      # Starts a sandboxed session (delegates to any Merb::Orms::* modules).
+      #
+      # An ORM should implement Merb::Orms::MyOrm#open_sandbox! to support this.
+      # Usually this involves starting a transaction.
+      def open_sandbox!
+        puts "Loading #{Merb.environment} environment in sandbox (Merb #{Merb::VERSION})"
+        puts "Any modifications you make will be rolled back on exit"
+        orm_modules.each { |orm| orm.open_sandbox! if orm.respond_to?(:open_sandbox!) }
+      end
+
+      # Ends a sandboxed session (delegates to any Merb::Orms::* modules).
+      #
+      # An ORM should implement Merb::Orms::MyOrm#close_sandbox! to support this.
+      # Usually this involves rolling back a transaction.
+      def close_sandbox!
+        orm_modules.each { |orm| orm.close_sandbox! if orm.respond_to?(:close_sandbox!) }
+        puts "Modifications have been rolled back"
+      end
+
+      # Explictly show logger output during IRB session
+      def trace_log!
+        Merb.logger.auto_flush = true
+      end
+
+      private
+
+      # ==== Returns
+      # Array:: All Merb::Orms::* modules.
+      def orm_modules
+        if Merb.const_defined?('Orms')
+          Merb::Orms.constants.map { |c| Merb::Orms::const_get(c) }
+        else
+          []
+        end
+      end
+
+    end
+
+    class Irb
+      # ==== Parameters
+      # opts<Hash>:
+      #   Options for IRB. Currently this is not used by the IRB adapter.
+      #
+      # ==== Notes
+      # If the +.irbrc+ file exists, it will be loaded into the IRBRC
+      # environment variable.
+      def self.start(opts={})
+        m = Merb::Rack::Console.new
+        m.extend Merb::Test::RequestHelper
+        Object.send(:define_method, :merb) { m }
+        ARGV.clear # Avoid passing args to IRB
+        m.open_sandbox! if sandboxed?
+        require 'irb'
+        require 'irb/completion'
+        if File.exists? ".irbrc"
+          ENV['IRBRC'] = ".irbrc"
+        end
+        IRB.start
+        at_exit do merb.close_sandbox! if sandboxed? end
+        exit
+      end
+
+      private
+
+      def self.sandboxed?
+        Merb::Config[:sandbox]
+      end
+    end
+  end
+end

data/lib/merb-core/rack/adapter/mongrel.rb

@@ -0,0 +1,33 @@
+require 'mongrel'
+require 'merb-core/rack/handler/mongrel'
+module Merb
+
+  module Rack
+
+    class Mongrel < Merb::Rack::AbstractAdapter
+
+      def self.stop(status = 0)
+        if @server
+          begin
+            @server.stop(true)
+          rescue Mongrel::TimeoutError
+            Merb.logger.fatal! "Your process took too long to shut " \
+              "down, so mongrel killed it."
+          end
+          true
+        end
+      end
+      
+      def self.new_server(port)
+        @server = ::Mongrel::HttpServer.new(@opts[:host], port)
+      end
+      
+      def self.start_server
+        @server.register('/', ::Merb::Rack::Handler::Mongrel.new(@opts[:app]))
+        @server.run.join
+      end
+      
+    end
+    
+  end
+end

data/lib/merb-core/rack/adapter/runner.rb

@@ -0,0 +1,28 @@
+module Merb
+  
+  module Rack
+    
+    class Runner
+      # ==== Parameters
+      # opts<Hash>:: Options for the runner (see below).
+      #
+      # ==== Options (opts)
+      # :runner_code<String>:: The code to run.
+      #
+      # ==== Notes
+      # If opts[:runner_code] matches a filename, that file will be read and
+      # the contents executed. Otherwise the code will be executed directly.
+      def self.start(opts={})
+        Merb::Server.change_privilege
+        if opts[:runner_code]
+          if File.exists?(opts[:runner_code])
+            eval(File.read(opts[:runner_code]), TOPLEVEL_BINDING, __FILE__, __LINE__)
+          else
+            eval(opts[:runner_code], TOPLEVEL_BINDING, __FILE__, __LINE__)
+          end
+          exit
+        end  
+      end
+    end
+  end
+end

data/lib/merb-core/rack/adapter/swiftiplied_mongrel.rb

@@ -0,0 +1,14 @@
+require 'swiftcore/swiftiplied_mongrel'
+require 'merb-core/rack/handler/mongrel'
+module Merb
+  module Rack
+
+    class SwiftipliedMongrel < Mongrel
+      def self.new_server(port)
+        Merb::Dispatcher.use_mutex = false
+        super
+      end      
+    end
+    
+  end
+end

data/lib/merb-core/rack/adapter/thin.rb

@@ -0,0 +1,40 @@
+require "thin"
+
+module Merb
+
+  module Rack
+
+    class Thin < Merb::Rack::AbstractAdapter
+      # start a Thin server on given host and port.
+
+      def self.new_server(port)
+        Merb::Dispatcher.use_mutex = false
+        
+        if @opts[:socket] || @opts[:socket_file]
+          socket = port.to_s
+          socket_file = @opts[:socket_file] || "#{Merb.root}/log/merb.#{socket}.sock"
+          Merb.logger.warn!("Using Thin adapter with socket file #{socket_file}.")
+          @server = ::Thin::Server.new(socket_file, @opts[:app], @opts)
+        else
+          Merb.logger.warn!("Using Thin adapter on host #{@opts[:host]} and port #{port}.")
+          if @opts[:host].include?('/')
+            @opts[:host] = "#{@opts[:host]}-#{port}"
+          end
+          @server = ::Thin::Server.new(@opts[:host], port, @opts[:app], @opts)
+        end
+      end
+
+      def self.start_server
+        ::Thin::Logging.silent = true
+        @server.start
+      end
+      
+      def self.stop(status = 0)
+        if @server
+          @server.stop
+          true
+        end
+      end
+    end
+  end
+end