maiha-hosts_access 0.1 → 0.2

data/README

@@ -6,7 +6,8 @@ HostsAccess
 
 Usage
 =====
-  config/environment.rb:
+  config/environment.rb (in Rails)
+  config/init.rb        (in Merb)
 
     # specify hosts to allow access
     HostsAccess.allow = ['127.0.0.1', 'ac.wota.jp']
@@ -20,6 +21,14 @@ Note
   All accesses are allowed if no rules are given
 
 
+Environment
+===========
+
+  tested on
+
+  * Rails 1.2
+  * Merb 1.0.3
+
 
 Copyright (c) 2008 maiha@wota.jp, released under the MIT license
 

data/init.rb

@@ -1,4 +1 @@
-
-ActionController::Base.class_eval do
-  include HostsAccess
-end
+require File.dirname(__FILE__) + '/lib/hosts_access'

data/lib/hosts_access.rb

@@ -1,44 +1,89 @@
 require 'resolv'
 
-module HostsAccess
-  def self.included(base)
-    base.class_eval do
-      include InstanceMethods
-      before_filter :hosts_access
-    end
+######################################################################
+### core_ext
+unless Module.respond_to?(:mattr_accessor)
+  require File.dirname(__FILE__) + '/../core_ext/module/attribute_accessors'
+end
+
+unless [].respond_to?(:extract_options!)
+  require File.dirname(__FILE__) + '/../core_ext/array/extract_options'      
+  class Array #:nodoc:
+    include ActiveSupport::CoreExtensions::Array::ExtractOptions
   end
+end
+
+
 
+######################################################################
+### HostsAccess
+
+module HostsAccess
   mattr_accessor :allow
   mattr_accessor :allow_controller
 
-  module InstanceMethods
   private
     def hosts_access
-      if allow.blank?
-        logger.info "[HostsAccess] OK: not restricted (no rules)"
-        return true
+      if HostsAccess.allow.blank?
+        return hosts_access_allowed("not restricted (no rules)")
       end
 
       if allow_controller
-        Array(allow_controller).each do |name|
-          if name == controller_name
-            logger.info "[HostsAccess] OK: #{controller_name} controller is not restricted"
-            return true
-          end
+        Array(HostsAccess.allow_controller).each do |name|
+          return hosts_access_allowed("#{controller_name} controller is not restricted") if name == controller_name
         end
       end
 
-      Array(allow).each do |host|
+      Array(HostsAccess.allow).each do |host|
         address = host
         address = Resolv.getaddress(host) rescue host unless /\A[\d\.]+\Z/ === host
-        if request.remote_ip == address
-          logger.info "[HostsAccess] OK: #{host} is allowed"
-          return true
-        end
+        return hosts_access_allowed("#{host} is allowed") if request.remote_ip == address
       end
 
-      logger.info "[HostsAccess] NG: #{request.remote_ip} is denied"
+      return hosts_access_denied("#{request.remote_ip} is denied")
+    end
+
+    def hosts_access_allowed(reason)
+      logger.info "[HostsAccess] OK: #{reason}"
+      return true
+    end
+
+    def hosts_access_denied(reason)
+      logger.info "[HostsAccess] NG: #{reason}"
       return false
     end
+end
+
+
+######################################################################
+### for Merb
+if defined?(Merb::Plugins)
+  module HostsAccess
+    def logger
+      Merb.logger
+    end
+  end
+
+  Merb::BootLoader.after_app_loads do
+    Application.class_eval do
+      include HostsAccess
+      before :hosts_access
+
+      private
+        def hosts_access_denied(reason)
+          super
+          throw :halt
+        end
+    end
+  end
+end
+
+######################################################################
+### for Rails
+
+if defined?(ActionController::Base)
+  ActionController::Base.class_eval do
+    include HostsAccess
+    before_filter :hosts_access
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: maiha-hosts_access
 version: !ruby/object:Gem::Version 
-  version: "0.1"
+  version: "0.2"
 platform: ruby
 authors: 
 - maiha
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2008-11-25 00:00:00 -08:00
+date: 2008-12-09 00:00:00 -08:00
 default_executable: 
 dependencies: []