Merge remote-tracking branch 'upstream/pull/4407'

[rails.git] / app / models / acl.rb

diff --git a/app/models/acl.rb b/app/models/acl.rb
index 5fb99b9e5bfd7bbfeb0f93a3c64da86ee19846f9..26285cef38ac0825ab237a7ba955a6118761d137 100644 (file)
--- a/app/models/acl.rb
+++ b/app/models/acl.rb
@@ -1,13 +1,60 @@
-class Acl < ActiveRecord::Base
-  def self.find_by_address(address, options)
-    self.with_scope(:find => {:conditions => ["inet_aton(?) & netmask = address", address]}) do
-      return self.find(:first, options)
+# == Schema Information
+#
+# Table name: acls
+#
+#  id      :bigint(8)        not null, primary key
+#  address :inet
+#  k       :string           not null
+#  v       :string
+#  domain  :string
+#  mx      :string
+#
+# Indexes
+#
+#  acls_k_idx             (k)
+#  index_acls_on_address  (address) USING gist
+#  index_acls_on_domain   (domain)
+#  index_acls_on_mx       (mx)
+#
+
+class Acl < ApplicationRecord
+  validates :k, :presence => true
+
+  def self.match(address, options = {})
+    acls = Acl.where("address >>= ?", address)
+
+    if options[:domain]
+      labels = options[:domain].split(".")
+
+      until labels.empty?
+        acls = acls.or(Acl.where(:domain => labels.join(".")))
+        labels.shift
+      end

     end
     end

+
+    acls = acls.or(Acl.where(:mx => options[:mx])) if options[:mx]
+
+    acls

   end
 
   end
 

-  def self.find_all_by_address(address, options)
-    self.with_scope(:find => {:conditions => ["inet_aton(?) & netmask = address", address]}) do
-      return self.find(:all, options)
-    end
+  def self.no_account_creation(address, options = {})
+    match(address, options).exists?(:k => "no_account_creation")
+  end
+
+  def self.allow_account_creation(address, options = {})
+    acls = Acl.where("address >>= ?", address)
+              .and(Acl.where(:k => "allow_account_creation"))
+    acls = acls.and(Acl.where(:domain => options[:domain])) if options[:domain]
+    acls = acls.and(Acl.where(:mx => options[:mx])) if options[:mx]
+
+    !acls.empty?
+  end
+
+  def self.no_note_comment(address, domain = nil)
+    match(address, :domain => domain).exists?(:k => "no_note_comment")
+  end
+
+  def self.no_trace_download(address, domain = nil)
+    match(address, :domain => domain).exists?(:k => "no_trace_download")

   end
 end
   end
 end