has_unique_slug 0.1.0 → 0.1.1

data/lib/has_unique_slug/version.rb

@@ -1,3 +1,3 @@
 module HasUniqueSlug
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/has_unique_slug.rb

@@ -7,27 +7,20 @@ module HasUniqueSlug
   
   module ClassMethods
     
-    def has_unique_slug(*args)
-      
-      # Setup options Hash
-      options = {:generates_with => :title, :slug_column => :slug}
-      options.merge! args.pop if args.last.kind_of? Hash
-      
-      # Standaridize implementation
-      title, slug = options[:generates_with], options[:slug_column]
+    def has_unique_slug(title_col = "title", slug_col = "slug")
       
       # Add ActiveRecord Callback
       before_create do |record|
         
         # Add a slug if slug doesn't exist
-        if record[slug].blank?
-          record[slug] = record[title].parameterize
+        if record[slug_col].blank?
+          record[slug_col] = record[title_col].parameterize
         end
         
         # Ensure the current slug is unique in the database, if not, make it unqiue
         i = 2
-        while not record.class.where(slug => record[slug]).count.zero? 
-          record[slug] = "#{record[slug]}-#{i}"
+        while not record.class.where("#{slug_col} = ?", record[slug_col]).count.zero? 
+          record[slug_col] = "#{record[slug_col]}-#{i}"
           i += 1
         end  
       end
@@ -37,13 +30,24 @@ module HasUniqueSlug
         include HasUniqueSlug::InstanceMethods
       end
       
-      # Add class methods to objects using has_unique_slug
+      # Add configuration mechanism
+      instance_eval <<-EOV
+        def slug_column
+          '#{slug_col}'
+        end
+        
+        def title_column
+          '#{title_col}'
+        end
+      EOV
+      
+      # Add find method to override ActiveRecord::Base.find
       instance_eval do
         def find(*args)
           if args.length == 1
-            where(slug => args.first).first
+            where("#{slug_column} = ?", args.first).first
           else
-            where(slug => args)
+            where("#{slug_column} IN (?)", args)
           end
         end
       end
@@ -54,7 +58,7 @@ module HasUniqueSlug
   module InstanceMethods
 
     def to_param
-      slug
+      self.send(self.class.slug_column)
     end
 
   end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: has_unique_slug
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: 
   segments: 
   - 0
   - 1
-  - 0
-  version: 0.1.0
+  - 1
+  version: 0.1.1
 platform: ruby
 authors: 
 - Brendan Stennett