paritytimeline 0.4

data/lib/timeline/track.rb

@@ -0,0 +1,116 @@
+module Timeline::Track
+  extend ActiveSupport::Concern
+
+  module ClassMethods
+    def track(name, options={})
+      @name = name
+      @actor = options.delete :actor
+      @actor ||= :creator
+      @object = options.delete :object
+      @target = options.delete :target
+      @followers = options.delete :followers
+      @followers ||= :followers
+      @mentionable = options.delete :mentionable
+
+      method_name = "track_#{@name}_after_#{@callback}".to_sym
+      define_activity_method method_name, actor: @actor, object: @object, target: @target, followers: @followers, verb: name, mentionable: @mentionable
+
+      send "after_#{@callback}".to_sym, method_name, if: options.delete(:if)
+    end
+
+    private
+      def define_activity_method(method_name, options={})
+        define_method method_name do
+          @actor = send(options[:actor])
+          @fields_for = {}
+          @object = set_object(options[:object])
+          @target = !options[:target].nil? ? send(options[:target].to_sym) : nil
+          @extra_fields ||= nil
+          @followers = @actor.send(options[:followers].to_sym)
+          @mentionable = options[:mentionable]
+          add_activity activity(verb: options[:verb])
+        end
+      end
+  end
+
+  protected
+    def activity(options={})
+      {
+        verb: options[:verb],
+        actor: options_for(@actor),
+        object: options_for(@object),
+        target: options_for(@target),
+        created_at: Time.now
+      }
+    end
+
+    def add_activity(activity_item)
+      redis_add "global:activity", activity_item
+      add_activity_to_user(activity_item[:actor][:id], activity_item)
+      add_activity_by_user(activity_item[:actor][:id], activity_item)
+      add_mentions(activity_item)
+      add_activity_to_followers(activity_item) if @followers.any?
+    end
+
+    def add_activity_by_user(user_id, activity_item)
+      redis_add "user:id:#{user_id}:posts", activity_item
+    end
+
+    def add_activity_to_user(user_id, activity_item)
+      redis_add "user:id:#{user_id}:activity", activity_item
+    end
+
+    def add_activity_to_followers(activity_item)
+      @followers.each { |follower| add_activity_to_user(follower.id, activity_item) }
+    end
+
+    def add_mentions(activity_item)
+      return unless @mentionable and @object.send(@mentionable)
+      @object.send(@mentionable).scan(/@\w+/).each do |mention|
+        if user = @actor.class.find_by_username(mention[1..-1])
+          add_mention_to_user(user.id, activity_item)
+        end
+      end
+    end
+
+    def add_mention_to_user(user_id, activity_item)
+      redis_add "user:id:#{user_id}:mentions", activity_item
+    end
+
+    def extra_fields_for(object)
+      return {} unless @fields_for.has_key?(object.class.to_s.downcase.to_sym)
+      @fields_for[object.class.to_s.downcase.to_sym].inject({}) do |sum, method|
+        sum[method.to_sym] = @object.send(method.to_sym)
+        sum
+      end
+    end
+
+    def options_for(target)
+      if !target.nil?
+        {
+          id: target.id,
+          class: target.class.to_s,
+          display_name: target.to_s
+        }.merge(extra_fields_for(target))
+      else
+        nil
+      end
+    end
+
+    def redis_add(list, activity_item)
+      Timeline.redis.lpush list, Timeline.encode(activity_item)
+    end
+
+    def set_object(object)
+      case
+      when object.is_a?(Symbol)
+        send(object)
+      when object.is_a?(Array)
+        @fields_for[self.class.to_s.downcase.to_sym] = object
+        self
+      else
+        self
+      end
+    end
+
+end

data/lib/timeline/untitled.txt

@@ -0,0 +1,9 @@
+1.upto(User.last.id).each do |u|
+   list = "user:id:#{u}:notification"
+   Timeline.redis.ltrim list, 0, 30
+   list = "user:id:#{u}:posts"
+   Timeline.redis.ltrim list, 0, 30
+   list = "user:id:#{u}:activity"
+   Timeline.redis.ltrim list, 0, 30
+   Timeline.redis.ltrim "global:activity:new_premium_deal", 0, 200
+end

data/lib/timeline/version.rb

@@ -0,0 +1,3 @@
+module Timeline
+  VERSION = "0.4"
+end

data/spec/activity_spec.rb

@@ -0,0 +1,12 @@
+require File.join(File.dirname(__FILE__), %w[spec_helper])
+
+describe Timeline::Activity do
+  describe "initialized with json" do
+    let(:json) { { id: "1", verb: "new_post" } }
+
+    it "returns a Hashie-fied object" do
+      Timeline::Activity.new(json).id.should == "1"
+      Timeline::Activity.new(json).verb.should == "new_post"
+    end
+  end
+end

data/spec/actor_spec.rb

@@ -0,0 +1,23 @@
+require File.join(File.dirname(__FILE__), %w[spec_helper])
+
+class User
+  include Timeline::Actor
+
+  attr_accessor :id
+end
+
+describe Timeline::Actor do
+  describe "when included" do
+    before { @user = User.new }
+
+    it "defines a timeline association" do
+      @user.should respond_to :timeline
+    end
+
+    describe ".timeline" do
+      subject { @user.timeline }
+
+      it { should == [] }
+    end
+  end
+end