Module: BelongsToCity::ClassMethods

Defined in:
lib/belongs_to_city.rb

Instance Method Summary collapse

Instance Method Details

#belongs_to_city(args = {}) ⇒ Object 



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
 
# File 'lib/belongs_to_city.rb', line 9

def belongs_to_city(args = {})
  city_attribute = args[:as].nil? ? :city : args[:as]
  city_attribute_helper = (city_attribute.to_s + '_name').to_sym
  city_collector = city_attribute.to_s.pluralize.to_sym
  
  attr_accessor city_collector
  
  #Getter and Setter Methods
  define_method "#{city_attribute}_name" do
    city = send(city_attribute)
    return read_attribute("#{city_attribute}_name") if read_attribute("#{city_attribute}_name")
    return "#{city.name}, #{city.state.short2}" if city
  end

  define_method "#{city_attribute}_name=" do |city_name|
    write_attribute("#{city_attribute}_name", city_name)
    write_attribute(city_collector, City.find_by_full_name(city_name) || [])
    if read_attribute(city_collector).size == 1
      send "#{city_attribute}=", read_attribute(city_collector).first
    else
      send "#{city_attribute}=", nil
    end
  end
  
  #Validation Methods
  if args[:required]
    if args[:messages]
      too_many_cities_message = args[:messages][:on_many_matches]
      no_cities_message = args[:messages][:on_no_matches]
    end
    too_many_cities_message ||="Too many cities found in your request, please be more specific"
    no_cities_message ||= "We couldn't find any city in your request"
    validate do |record|
      cities = record.read_attribute(city_collector)
      city = record.read_attribute(city_attribute)
      if cities
        record.errors.add(city_attribute_helper, too_many_cities_message) if (cities.size > 1 rescue nil)
        record.errors.add(city_attribute_helper, no_cities_message) if (cities.nil? or cities.size == 0)
      end
    end
  end
  
end