1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240
|
module Stringex
module ActsAsUrl
module Adapter
class Base
attr_accessor :base_url, :callback_options, :configuration, :instance, :klass, :settings
def initialize(configuration)
ensure_loadable
self.configuration = configuration
self.settings = configuration.settings
end
def create_callbacks!(klass)
self.klass = klass
self.callback_options = {}
create_method_to_callback
create_callback
end
def ensure_unique_url!(instance)
@url_owners = nil
self.instance = instance
handle_url!
handle_blacklisted_url!
handle_duplicate_url! unless settings.allow_duplicates
end
def initialize_urls!(klass)
self.klass = klass
klass_previous_instances do |instance|
ensure_unique_url_for! instance
end
end
def url_attribute(instance)
# Retrieve from database record if there are errors on attribute_to_urlify
if !is_new?(instance) && is_present?(instance.errors[settings.attribute_to_urlify])
self.instance = instance
read_attribute instance_from_db, settings.url_attribute
else
read_attribute instance, settings.url_attribute
end
end
def self.ensure_loadable
raise "The #{self} adapter cannot be loaded" unless loadable?
Stringex::ActsAsUrl::Adapter.add_loaded_adapter self
end
def self.loadable?
orm_class
rescue NameError
false
end
private
def add_new_record_url_owner_conditions
return if is_new?(instance)
@url_owner_conditions.first << " and #{primary_key} != ?"
@url_owner_conditions << instance.id
end
def add_scoped_url_owner_conditions
[settings.scope_for_url].flatten.compact.each do |scope|
@url_owner_conditions.first << " and #{scope} = ?"
@url_owner_conditions << instance.send(scope)
end
end
def create_callback
klass.send(
klass_callback_method,
:ensure_unique_url,
**callback_options
)
end
def klass_callback_method
settings.sync_url ? klass_sync_url_callback_method : klass_non_sync_url_callback_method
end
def klass_sync_url_callback_method
configuration.settings.callback_method
end
def klass_non_sync_url_callback_method
case configuration.settings.callback_method
when :before_save
:before_create
else # :before_validation
callback_options[:on] = :create
configuration.settings.callback_method
end
end
def create_method_to_callback
klass.class_eval <<-"END"
def #{settings.url_attribute}
acts_as_url_configuration.adapter.url_attribute self
end
END
end
def duplicate_for_base_url(n)
"#{base_url}#{settings.duplicate_count_separator}#{n}"
end
def ensure_loadable
self.class.ensure_loadable
end
# NOTE: The <tt>instance</tt> here is not the cached instance but a block variable
# passed from <tt>klass_previous_instances</tt>, just to be clear
def ensure_unique_url_for!(instance)
instance.send :ensure_unique_url
instance.save
end
def get_base_url_owner_conditions
@url_owner_conditions = ["#{settings.url_attribute} LIKE ?", base_url + '%']
end
def handle_duplicate_url!
return if !url_taken?(base_url)
n = nil
sequence = duplicate_url_sequence.tap(&:rewind)
loop do
n = sequence.next
break unless url_taken?(duplicate_for_base_url(n))
end
write_url_attribute duplicate_for_base_url(n)
end
def duplicate_url_sequence
settings.duplicate_sequence ||
Enumerator.new do |enum|
n = 1
loop do
enum.yield n
n += 1
end
end
end
def url_taken?(url)
if settings.url_taken_method
instance.send(settings.url_taken_method, url)
else
url_owners.any?{|owner| url_attribute_for(owner) == url}
end
end
def handle_url!
self.base_url = instance.send(settings.url_attribute)
modify_base_url if is_blank?(base_url) || !settings.only_when_blank
write_url_attribute base_url
end
def handle_blacklisted_url!
return unless settings.blacklist.to_set.include?(base_url)
self.base_url = settings.blacklist_policy.call(instance, base_url)
write_url_attribute base_url
end
def instance_from_db
instance.class.find(instance.id)
end
def is_blank?(object)
object.blank?
end
def is_new?(object)
object.new_record?
end
def is_present?(object)
object.present?
end
def loadable?
self.class.loadable?
end
def modify_base_url
root = instance.send(settings.attribute_to_urlify).to_s
self.base_url = root.to_url(configuration.string_extensions_settings)
end
def orm_class
self.class.orm_class
end
def primary_key
instance.class.primary_key
end
def read_attribute(instance, attribute)
instance.read_attribute attribute
end
def url_attribute_for(object)
object.send settings.url_attribute
end
def url_owner_conditions
get_base_url_owner_conditions
add_new_record_url_owner_conditions
add_scoped_url_owner_conditions
@url_owner_conditions
end
def url_owners
@url_owners ||= url_owners_class.unscoped.where(url_owner_conditions).to_a
end
def url_owners_class
return instance.class unless settings.enforce_uniqueness_on_sti_base_class
klass = instance.class
while klass.superclass < orm_class
klass = klass.superclass
end
klass
end
def write_attribute(instance, attribute, value)
instance.send :write_attribute, attribute, value
end
def write_url_attribute(value)
write_attribute instance, settings.url_attribute, value
end
end
end
end
end
|