discourse/app/models/topic_link_click.rb

require_dependency 'discourse'
require 'ipaddr'

class TopicLinkClick < ActiveRecord::Base
  belongs_to :topic_link, counter_cache: :clicks
  belongs_to :user

  validates_presence_of :topic_link_id
  validates_presence_of :ip_address

  # Create a click from a URL and post_id
  def self.create_from(args={})

    # If the URL is absolute, allow HTTPS and HTTP versions of it
    if args[:url] =~ /^http/
      http_url = args[:url].sub(/^https/, 'http')
      https_url = args[:url].sub(/^http\:/, 'https:')
      link = TopicLink.select([:id, :user_id]).where('url = ? OR url = ?', http_url, https_url)
    else
      link = TopicLink.select([:id, :user_id]).where(url: args[:url])
    end

    # Find the forum topic link
    link = link.where(post_id: args[:post_id]) if args[:post_id].present?

    # If we don't have a post, just find the first occurance of the link
    link = link.where(topic_id: args[:topic_id]) if args[:topic_id].present?
    link = link.first

    # If no link is found, return the url for relative links
    unless link.present?
      return args[:url] if args[:url] =~ /^\//

      begin
        uri = URI.parse(args[:url])
        return args[:url] if uri.host == URI.parse(Discourse.base_url).host
      rescue
      end

      # If we have it somewhere else on the site, just allow the redirect. This is
      # likely due to a onebox of another topic.
      link = TopicLink.find_by(url: args[:url])
      return link.present? ? link.url : nil
    end

    return args[:url] if (args[:user_id] && (link.user_id == args[:user_id]))

    # Rate limit the click counts to once in 24 hours
    rate_key = "link-clicks:#{link.id}:#{args[:user_id] || args[:ip]}"
    if $redis.setnx(rate_key, "1")
      $redis.expire(rate_key, 1.day.to_i)
      create!(topic_link_id: link.id, user_id: args[:user_id], ip_address: args[:ip])
    end

    args[:url]
  end

end

# == Schema Information
#
# Table name: topic_link_clicks
#
#  id            :integer          not null, primary key
#  topic_link_id :integer          not null
#  user_id       :integer
#  created_at    :datetime         not null
#  updated_at    :datetime         not null
#  ip_address    :inet             not null
#
# Indexes
#
#  by_link  (topic_link_id)
#