freels-mongrel 1.1.2

data/TODO

@@ -0,0 +1,4 @@
+
+* Rewrite and merge mongrel cluster and mongrel_rails into something small and maintainable.
+* Initial 1.9 compatibility.
+* 1.9 performance (requires removing the thread queue).

data/bin/mongrel_rails

@@ -0,0 +1,284 @@
+# Copyright (c) 2005 Zed A. Shaw
+# You can redistribute it and/or modify it under the same terms as Ruby.
+#
+# Additional work donated by contributors.  See http://mongrel.rubyforge.org/attributions.html
+# for more information.
+
+require 'yaml'
+require 'etc'
+
+$LOAD_PATH.unshift "#{File.dirname(__FILE__)}/../lib"
+require 'mongrel'
+require 'mongrel/rails'
+
+Mongrel::Gems.require 'gem_plugin'
+
+# require 'ruby-debug'
+# Debugger.start
+
+module Mongrel
+  class Start < GemPlugin::Plugin "/commands"
+    include Mongrel::Command::Base
+
+    def configure
+      options [
+        ["-e", "--environment ENV", "Rails environment to run as", :@environment, ENV['RAILS_ENV'] || "development"],
+        ["-d", "--daemonize", "Run daemonized in the background", :@daemon, false],
+        ['-p', '--port PORT', "Which port to bind to", :@port, 3000],
+        ['-a', '--address ADDR', "Address to bind to", :@address, "0.0.0.0"],
+        ['-l', '--log FILE', "Where to write log messages", :@log_file, "log/mongrel.log"],
+        ['-P', '--pid FILE', "Where to write the PID", :@pid_file, "log/mongrel.pid"],
+        ['-n', '--num-procs INT', "Number of processors active before clients denied", :@num_processors, 1024],
+        ['-o', '--timeout TIME', "Time to wait (in seconds) before killing a stalled thread", :@timeout, 60],
+        ['-t', '--throttle TIME', "Time to pause (in hundredths of a second) between accepting clients", :@throttle, 0],
+        ['-m', '--mime PATH', "A YAML file that lists additional MIME types", :@mime_map, nil],
+        ['-c', '--chdir PATH', "Change to dir before starting (will be expanded)", :@cwd, Dir.pwd],
+        ['-r', '--root PATH', "Set the document root (default 'public')", :@docroot, "public"],
+        ['-B', '--debug', "Enable debugging mode", :@debug, false],
+        ['-C', '--config PATH', "Use a config file", :@config_file, nil],
+        ['-S', '--script PATH', "Load the given file as an extra config script", :@config_script, nil],
+        ['-G', '--generate PATH', "Generate a config file for use with -C", :@generate, nil],
+        ['', '--user USER', "User to run as", :@user, nil],
+        ['', '--group GROUP', "Group to run as", :@group, nil],
+        ['', '--prefix PATH', "URL prefix for Rails app", :@prefix, nil]
+      ]
+    end
+
+    def validate
+      if @config_file
+        valid_exists?(@config_file, "Config file not there: #@config_file")
+        return false unless @valid
+        @config_file = File.expand_path(@config_file)
+        load_config
+        return false unless @valid
+      end
+
+      @cwd = File.expand_path(@cwd)
+      valid_dir? @cwd, "Invalid path to change to during daemon mode: #@cwd"
+
+      # Change there to start, then we'll have to come back after daemonize
+      Dir.chdir(@cwd)
+
+      valid?(@prefix[0] == ?/ && @prefix[-1] != ?/, "Prefix must begin with / and not end in /") if @prefix
+      valid_dir? File.dirname(@log_file), "Path to log file not valid: #@log_file"
+      valid_dir? File.dirname(@pid_file), "Path to pid file not valid: #@pid_file"
+      valid_dir? @docroot, "Path to docroot not valid: #@docroot"
+      valid_exists? @mime_map, "MIME mapping file does not exist: #@mime_map" if @mime_map
+      valid_exists? @config_file, "Config file not there: #@config_file" if @config_file
+      valid_dir? File.dirname(@generate), "Problem accessing directory to #@generate" if @generate
+      valid_user? @user if @user
+      valid_group? @group if @group
+
+      return @valid
+    end
+
+    def run
+      if @generate
+        @generate = File.expand_path(@generate)
+        Mongrel.log(:error, "** Writing config to \"#@generate\".")
+        open(@generate, "w") {|f| f.write(settings.to_yaml) }
+        Mongrel.log(:error, "** Finished.  Run \"mongrel_rails start -C #@generate\" to use the config file.")
+        exit 0
+      end
+
+      config = Mongrel::Rails::RailsConfigurator.new(settings) do
+        if defaults[:daemon]
+          if File.exist? defaults[:pid_file]
+            Mongrel.log(:error, "!!! PID file #{defaults[:pid_file]} already exists.  Mongrel could be running already.  Check your #{defaults[:log_file]} for errors.")
+            Mongrel.log(:error, "!!! Exiting with error.  You must stop mongrel and clear the .pid before I'll attempt a start.")
+            exit 1
+          end
+
+          daemonize
+          write_pid_file
+          Mongrel.log("Daemonized, any open files are closed.  Look at #{defaults[:pid_file]} and #{defaults[:log_file]} for info.")
+          Mongrel.log("Settings loaded from #{@config_file} (they override command line).") if @config_file
+        end
+
+        Mongrel.log("Starting Mongrel listening at #{defaults[:host]}:#{defaults[:port]}, further information can be found in log/mongrel-#{defaults[:host]}-#{defaults[:port]}.log")
+
+        listener do
+          mime = {}
+          if defaults[:mime_map]
+            Mongrel.log("Loading additional MIME types from #{defaults[:mime_map]}")
+            mime = load_mime_map(defaults[:mime_map], mime)
+          end
+
+          if defaults[:debug]
+            Mongrel.log("Installing debugging prefixed filters. Look in log/mongrel_debug for the files.")
+            debug "/"
+          end
+
+          Mongrel.log("Starting Rails with #{defaults[:environment]} environment...")
+          Mongrel.log("Mounting Rails at #{defaults[:prefix]}...") if defaults[:prefix]
+          uri defaults[:prefix] || "/", :handler => rails(:mime => mime, :prefix => defaults[:prefix])
+          Mongrel.log("Rails loaded.")
+
+          Mongrel.log("Loading any Rails specific GemPlugins" )
+          load_plugins
+
+          if defaults[:config_script]
+            Mongrel.log("Loading #{defaults[:config_script]} external config script")
+            run_config(defaults[:config_script])
+          end
+
+          setup_rails_signals
+        end
+      end
+
+      config.run
+      Mongrel.log("Mongrel #{Mongrel::Const::MONGREL_VERSION} available at #{@address}:#{@port}")
+
+      unless config.defaults[:daemon]
+        Mongrel.log("Use CTRL-C to stop.")
+      end
+
+      config.join
+
+      if config.needs_restart
+        unless RUBY_PLATFORM =~ /djgpp|(cyg|ms|bcc)win|mingw/
+          cmd = "ruby #{__FILE__} start #{original_args.join(' ')}"
+          Mongrel.log("Restarting with arguments:  #{cmd}")
+          config.stop(false, true)
+          config.remove_pid_file
+
+          if config.defaults[:daemon]
+            system cmd
+          else
+            Mongrel.log(:error, "Can't restart unless in daemon mode.")
+            exit 1
+          end
+        else
+          Mongrel.log("Win32 does not support restarts. Exiting.")
+        end
+      end
+    end
+
+    def load_config
+      settings = {}
+      begin
+        settings = YAML.load_file(@config_file)
+      ensure
+        Mongrel.log(:error, "** Loading settings from #{@config_file} (they override command line).") unless @daemon || settings[:daemon]
+      end
+
+      settings[:includes] ||= ["mongrel"]
+
+      # Config file settings will override command line settings
+      settings.each do |key, value|
+        key = key.to_s
+        if config_keys.include?(key)
+          key = 'address' if key == 'host'
+          key = 'num_processors' if key == 'num_procs'
+          self.instance_variable_set("@#{key}", value)
+        else
+          failure "Unknown configuration setting: #{key}"  
+          @valid = false
+        end
+      end
+    end
+
+    def config_keys
+      @config_keys ||=
+        %w(address host port cwd log_file pid_file environment docroot mime_map daemon debug includes config_script num_processors num_procs timeout throttle user group prefix)
+    end
+
+    def settings
+      config_keys.inject({}) do |hash, key|
+        value = self.instance_variable_get("@#{key}")
+        key = 'host' if key == 'address'
+        key = 'num_processors' if key == 'num_procs'
+        hash[key.to_sym] ||= value
+        hash
+      end
+    end
+  end
+
+  def Mongrel::send_signal(signal, pid_file)
+    pid = open(pid_file).read.to_i
+    print "Sending #{signal} to Mongrel at PID #{pid}..."
+    begin
+      Process.kill(signal, pid)
+    rescue Errno::ESRCH
+      puts "Process does not exist.  Not running."
+    end
+
+    puts "Done."
+  end
+
+
+  class Stop < GemPlugin::Plugin "/commands"
+    include Mongrel::Command::Base
+
+    def configure 
+      options [ 
+        ['-c', '--chdir PATH', "Change to dir before starting (will be expanded).", :@cwd, "."],
+        ['-f', '--force', "Force the shutdown (kill -9).", :@force, false],
+        ['-w', '--wait SECONDS', "Wait SECONDS before forcing shutdown", :@wait, "0"], 
+        ['-P', '--pid FILE', "Where the PID file is located.", :@pid_file, "log/mongrel.pid"]
+      ]
+    end
+
+    def validate
+      @cwd = File.expand_path(@cwd)
+      valid_dir? @cwd, "Invalid path to change to during daemon mode: #@cwd"
+
+      Dir.chdir @cwd
+
+      valid_exists? @pid_file, "PID file #@pid_file does not exist.  Not running?"
+      return @valid
+    end
+
+    def run
+      if @force
+        @wait.to_i.times do |waiting|
+          exit(0) if not File.exist? @pid_file
+          sleep 1
+        end
+
+        Mongrel::send_signal("KILL", @pid_file) if File.exist? @pid_file
+      else
+        Mongrel::send_signal("TERM", @pid_file)
+      end
+    end
+  end
+
+
+  class Restart < GemPlugin::Plugin "/commands"
+    include Mongrel::Command::Base
+
+    def configure 
+      options [ 
+        ['-c', '--chdir PATH', "Change to dir before starting (will be expanded)", :@cwd, '.'],
+        ['-s', '--soft', "Do a soft restart rather than a process exit restart", :@soft, false],
+        ['-P', '--pid FILE', "Where the PID file is located", :@pid_file, "log/mongrel.pid"]
+      ]
+    end
+
+    def validate
+      @cwd = File.expand_path(@cwd)
+      valid_dir? @cwd, "Invalid path to change to during daemon mode: #@cwd"
+
+      Dir.chdir @cwd
+
+      valid_exists? @pid_file, "PID file #@pid_file does not exist.  Not running?"
+      return @valid
+    end
+
+    def run
+      if @soft
+        Mongrel::send_signal("HUP", @pid_file)
+      else
+        Mongrel::send_signal("USR2", @pid_file)
+      end
+    end
+  end
+end
+
+
+GemPlugin::Manager.instance.load "mongrel" => GemPlugin::INCLUDE, "rails" => GemPlugin::EXCLUDE
+
+
+if not Mongrel::Command::Registry.instance.run ARGV
+  exit 1
+end

data/examples/builder.rb

@@ -0,0 +1,29 @@
+require 'mongrel'
+
+class TestPlugin < GemPlugin::Plugin "/handlers"
+  include Mongrel::HttpHandlerPlugin
+
+  def process(request, response)
+    Mongrel.log(:error, "My options are: #{options.inspect}")
+    Mongrel.log(:error, "Request Was:")
+    Mongrel.log(:error, request.params.to_yaml)
+  end
+end
+
+config = Mongrel::Configurator.new :host => "127.0.0.1" do
+  load_plugins :includes => ["mongrel"], :excludes => ["rails"]
+  daemonize :cwd => Dir.pwd, :log_file => "mongrel.log", :pid_file => "mongrel.pid"
+  
+  listener :port => 3000 do
+    uri "/app", :handler => plugin("/handlers/testplugin", :test => "that")
+    uri "/app", :handler => Mongrel::DirHandler.new(".")
+    load_plugins :includes => ["mongrel", "rails"]
+  end
+
+  trap("INT") { stop }
+  run
+end
+
+config.join
+
+

data/examples/camping/README

@@ -0,0 +1,3 @@
+To get these examples running, install Camping.
+
+Instructions here: http://code.whytheluckystiff.net/camping/

data/examples/camping/blog.rb

@@ -0,0 +1,294 @@
+#!/usr/bin/env ruby
+
+$:.unshift File.dirname(__FILE__) + "/../../lib"
+require 'rubygems'
+require_gem 'camping', '>=1.4'
+require 'camping/session'
+  
+Camping.goes :Blog
+
+module Blog
+    include Camping::Session
+end
+
+module Blog::Models
+    def self.schema(&block)
+        @@schema = block if block_given?
+        @@schema
+    end
+  
+    class Post < Base; belongs_to :user; end
+    class Comment < Base; belongs_to :user; end
+    class User < Base; end
+end
+
+Blog::Models.schema do
+    create_table :blog_posts, :force => true do |t|
+      t.column :id,       :integer, :null => false
+      t.column :user_id,  :integer, :null => false
+      t.column :title,    :string,  :limit => 255
+      t.column :body,     :text
+    end
+    create_table :blog_users, :force => true do |t|
+      t.column :id,       :integer, :null => false
+      t.column :username, :string
+      t.column :password, :string
+    end
+    create_table :blog_comments, :force => true do |t|
+      t.column :id,       :integer, :null => false
+      t.column :post_id,  :integer, :null => false
+      t.column :username, :string
+      t.column :body,     :text
+    end
+    execute "INSERT INTO blog_users (username, password) VALUES ('admin', 'camping')"
+end
+
+module Blog::Controllers
+    class Index < R '/'
+        def get
+            @posts = Post.find :all
+            render :index
+        end
+    end
+     
+    class Add
+        def get
+            unless @state.user_id.blank?
+                @user = User.find @state.user_id
+                @post = Post.new
+            end
+            render :add
+        end
+        def post
+            post = Post.create :title => input.post_title, :body => input.post_body,
+                               :user_id => @state.user_id
+            redirect View, post
+        end
+    end
+
+    class Info < R '/info/(\d+)', '/info/(\w+)/(\d+)', '/info', '/info/(\d+)/(\d+)/(\d+)/([\w-]+)'
+        def get(*args)
+            div do
+                code args.inspect; br; br
+                code ENV.inspect; br
+                code "Link: #{R(Info, 1, 2)}"
+            end
+        end
+    end
+
+    class View < R '/view/(\d+)'
+        def get post_id 
+            @post = Post.find post_id
+            @comments = Models::Comment.find :all, :conditions => ['post_id = ?', post_id]
+            render :view
+        end
+    end
+     
+    class Edit < R '/edit/(\d+)', '/edit'
+        def get post_id 
+            unless @state.user_id.blank?
+                @user = User.find @state.user_id
+            end
+            @post = Post.find post_id
+            render :edit
+        end
+     
+        def post
+            @post = Post.find input.post_id
+            @post.update_attributes :title => input.post_title, :body => input.post_body
+            redirect View, @post
+        end
+    end
+     
+    class Comment
+        def post
+            Models::Comment.create(:username => input.post_username,
+                       :body => input.post_body, :post_id => input.post_id)
+            redirect View, input.post_id
+        end
+    end
+     
+    class Login
+        def post
+            @user = User.find :first, :conditions => ['username = ? AND password = ?', input.username, input.password]
+     
+            if @user
+                @login = 'login success !'
+                @state.user_id = @user.id
+            else
+                @login = 'wrong user name or password'
+            end
+            render :login
+        end
+    end
+     
+    class Logout
+        def get
+            @state.user_id = nil
+            render :logout
+        end
+    end
+     
+    class Style < R '/styles.css'
+        def get
+            @headers["Content-Type"] = "text/css; charset=utf-8"
+            @body = %{
+                body {
+                    font-family: Utopia, Georga, serif;
+                }
+                h1.header {
+                    background-color: #fef;
+                    margin: 0; padding: 10px;
+                }
+                div.content {
+                    padding: 10px;
+                }
+            }
+        end
+    end
+end
+
+module Blog::Views
+
+    def layout
+      html do
+        head do
+          title 'blog'
+          link :rel => 'stylesheet', :type => 'text/css', 
+               :href => '/styles.css', :media => 'screen'
+        end
+        body do
+          h1.header { a 'blog', :href => R(Index) }
+          div.content do
+            self << yield
+          end
+        end
+      end
+    end
+
+    def index
+      if @posts.empty?
+        p 'No posts found.'
+        p { a 'Add', :href => R(Add) }
+      else
+        for post in @posts
+          _post(post)
+        end
+      end
+    end
+
+    def login
+      p { b @login }
+      p { a 'Continue', :href => R(Add) }
+    end
+
+    def logout
+      p "You have been logged out."
+      p { a 'Continue', :href => R(Index) }
+    end
+
+    def add
+      if @user
+        _form(post, :action => R(Add))
+      else
+        _login
+      end
+    end
+
+    def edit
+      if @user
+        _form(post, :action => R(Edit))
+      else
+        _login
+      end
+    end
+
+    def view
+        _post(post)
+
+        p "Comment for this post:"
+        for c in @comments
+          h1 c.username
+          p c.body
+        end
+
+        form :action => R(Comment), :method => 'post' do
+          label 'Name', :for => 'post_username'; br
+          input :name => 'post_username', :type => 'text'; br
+          label 'Comment', :for => 'post_body'; br
+          textarea :name => 'post_body' do; end; br
+          input :type => 'hidden', :name => 'post_id', :value => post.id
+          input :type => 'submit'
+        end
+    end
+
+    # partials
+    def _login
+      form :action => R(Login), :method => 'post' do
+        label 'Username', :for => 'username'; br
+        input :name => 'username', :type => 'text'; br
+
+        label 'Password', :for => 'password'; br
+        input :name => 'password', :type => 'text'; br
+
+        input :type => 'submit', :name => 'login', :value => 'Login'
+      end
+    end
+
+    def _post(post)
+      h1 post.title
+      p post.body
+      p do
+        a "Edit", :href => R(Edit, post)
+        a "View", :href => R(View, post)
+      end
+    end
+
+    def _form(post, opts)
+      p do
+        text "You are logged in as #{@user.username} | "
+        a 'Logout', :href => R(Logout)
+      end
+      form({:method => 'post'}.merge(opts)) do
+        label 'Title', :for => 'post_title'; br
+        input :name => 'post_title', :type => 'text', 
+              :value => post.title; br
+
+        label 'Body', :for => 'post_body'; br
+        textarea post.body, :name => 'post_body'; br
+
+        input :type => 'hidden', :name => 'post_id', :value => post.id
+        input :type => 'submit'
+      end
+    end
+end
+ 
+def Blog.create
+    Camping::Models::Session.create_schema
+    unless Blog::Models::Post.table_exists?
+        ActiveRecord::Schema.define(&Blog::Models.schema)
+    end
+end
+
+if __FILE__ == $0
+  require 'mongrel/camping'
+
+  Blog::Models::Base.establish_connection :adapter => 'sqlite3', :database => 'blog.db'
+  Blog::Models::Base.logger = Logger.new('camping.log')
+  Blog::Models::Base.threaded_connections=false
+  Blog.create
+  
+  # Use the Configurator as an example rather than Mongrel::Camping.start
+  config = Mongrel::Configurator.new :host => "0.0.0.0" do
+    listener :port => 3002 do
+      uri "/blog", :handler => Mongrel::Camping::CampingHandler.new(Blog)
+      uri "/favicon", :handler => Mongrel::Error404Handler.new("")
+      trap("INT") { stop }
+      run
+    end
+  end
+
+  puts "** Blog example is running at http://localhost:3002/blog"
+  puts "** Default username is `admin', password is `camping'"
+  config.join
+end