Contents

module Firefly
  class Url
    include DataMapper::Resource
    
    VALID_URL_REGEX = /^(http|https):\/\/[a-z0-9]+([\-\.]{1}[a-z0-9]+)*\.[a-z]{2,5}(:[0-9]{1,5})?(\/.*)?$/ix
    
    property :id,           Serial
    property :url,          String,     :index => true, :length => 255
    property :code,         String,     :index => true, :length => 16
    property :clicks,       Integer,    :default => 0
    property :created_at,   DateTime,   :default => Proc.new{Time.now}
    
    # Increase the visits counter by 1
    def register_click!
      self.update(:clicks => self.clicks + 1)
    end
    
    # Shorten a long_url and return a new FireFly::Url
    def self.shorten(long_url)
      return nil unless valid_url?(long_url)
      
      long_url = normalize_url(long_url)
      
      the_url = Firefly::Url.first(:url => long_url) || Firefly::Url.create(:url => long_url)
      the_url.update(:code => Firefly::Base62.encode(the_url.id.to_i)) if the_url.code.nil?
      the_url
    end  
    
    private
      # Normalize the URL
      def self.normalize_url(url)
        URI.parse(url).normalize.to_s
      end
      
      # Validates the URL to be a valid http or https one. 
      def self.valid_url?(url)
        url.match(Firefly::Url::VALID_URL_REGEX)
      end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
firefly-0.4.5	lib/firefly/url.rb
firefly-0.4.4	lib/firefly/url.rb
firefly-0.4.3	lib/firefly/url.rb
firefly-0.4.2	lib/firefly/url.rb
firefly-0.4.1	lib/firefly/url.rb