Module: FriendlySlug::ActiveRecord::Base

Included in:
ActiveRecord::Base
Defined in:
lib/friendly_slug/active_record/base.rb

Instance Method Summary collapse

Instance Method Details

#build_friendly_slug(*attribute_list, use: nil) ⇒ Object



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
# File 'lib/friendly_slug/active_record/base.rb', line 4

def build_friendly_slug(*attribute_list, use: nil)
  instance_variable_set("@_friendly_attribute_list", attribute_list)
  instance_variable_set("@_friendly_use_key", use)

  instance_eval do
    def _friendly_attribute_list
      @_friendly_attribute_list
    end

    _friendly_attribute_list.each do |attribute|
      define_singleton_method :"_friendly_#{attribute.to_s}_key" do
        instance_variable_set("@_friendly_#{attribute.to_s}_key", attribute)
      end
    end

    def _friendly_use_key
      @_friendly_use_key
    end

    def find_slugged(id)
      find(id.split("-").send(_friendly_use_key))
    end
  end

  class_eval do
    before_save :_update_slug

    def to_param
      if self.respond_to?(:slug)
        if self.slug.nil? || _unique_attribute_changed?
          _create_slug
        else
          self.slug
        end
      else
        _create_slug
      end
    end

    private 
    def _lookup_key(k)
      k.is_a?(Symbol)? self.send(k) : k.to_s 
    end

    def _update_slug
      if self.class._friendly_use_key == :database
        current_slug = self.to_param
        unless _slug_exists?(current_slug)
          self.slug = current_slug
        else
          self.slug = [current_slug.to_s, SecureRandom.hex(6)].join("-")
        end
      end
    end

    def _create_slug
      fresh_slug = self.class._friendly_attribute_list.map do |attribute|
        _lookup_key(self.class.send("_friendly_#{attribute.to_s}_key")).to_s
      end.
        join("-").
        gsub(/<\/?[^>]*>|[^\.\w\s-]/, '').
        strip.downcase.gsub(/\s{1,}|\./, '-').
        gsub(/-{2,}/, "-")

      fresh_slug[-1] == "-" ? fresh_slug[0...-1] : fresh_slug
    end

    def _unique_attribute_changed?
      self.send("#{self.class.send("_friendly_attribute_list").first.to_s}_changed?".to_sym)
    end

    def _slug_exists?(current_slug)
      self.class.where("slug = ? AND id != ?", current_slug, self.id.nil? ? "NULL" : self.id).any? 
    end
  end
end