stalkerr 0.0.1 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5e221a6891528b82fcb96f86fd755fc67849b574
-  data.tar.gz: c3b563649487135c4c77e4ce77f3f3f3dee32205
+  metadata.gz: 37afbc1d1e7cc13a267357dbe454fdd58839856d
+  data.tar.gz: 3450f6d52f97e8eb53dbb75767f913ab6ee07a1e
 SHA512:
-  metadata.gz: 80b0e80f8c158bb4081e9426a25d4381343f40b2bfd604318b73bccc7d10b23cefba63535744500632f21044b3a72e1270b971d8b53daaa03086ca8355dae1fa
-  data.tar.gz: 13ae4d63529d31a4e98195f52b5813a6c8b798983ebd2f6561a9fa94fdbf88be2b765df34b88fe10b43701d464796e5c79aa5549bb7b8f95d93fa93bfc6a50ac
+  metadata.gz: 866a38efb3b596514926e829f04265896903cd7f12ec2bd9209935ab19a0633e990860e61b50884b74ea7f4c6755a1c2df4c553dcb1888205d32f68ec31ba21a
+  data.tar.gz: 94d131d97fd63d09e75d7e209b2ef70ea1b662bfa2a48f9e29c38f34efe0c53a9b7565fb0dded8c2db0a9668131dbd7acfda18bf8b9f1573c44296aaf5044d91

data/Gemfile

@@ -3,4 +3,5 @@ source 'https://rubygems.org'
 gem 'net-irc'
 gem 'json'
 gem 'octokit'
+gem 'qiita'
 gem 'string-irc'

data/README.md

@@ -17,6 +17,7 @@ Usage
 ### Connecting to Stalkerr
 
     /join #github <username>:<password>
+    /join #qiita <username>:<password>
 
 Contributing
 ------------
@@ -30,7 +31,7 @@ Contributing
 Author
 ------
 
-- [@linyows](https://github.com/linyows)
+- [linyows](https://github.com/linyows)
 
 License
 -------

data/lib/stalkerr/session.rb

@@ -6,20 +6,24 @@ Dir["#{File.dirname(__FILE__)}/target/*.rb"].each { |p| require p }
 
 class Stalkerr::Session < Net::IRC::Server::Session
 
+  def server_name
+    'Stalkerr'
+  end
+
+  def server_version
+    Stalkerr::VERSION
+  end
+
   def initialize(*args)
     super
     @debug = args.last.debug
-    @channels = {}
+    @channels = @threads = @targets = {}
     Dir["#{File.dirname(__FILE__)}/target/*.rb"].each do |path|
       name = File.basename(path, '.rb')
       @channels.merge!(name.to_sym => "##{name}")
     end
   end
 
-  def on_disconnected
-    @retrieve_thread.kill rescue nil
-  end
-
   def on_user(m)
     super
     @real, *@opts = @real.split(/\s+/)
@@ -36,23 +40,40 @@ class Stalkerr::Session < Net::IRC::Server::Session
 
   def on_join(m)
     super
+    create_worker(m.params)
+  end
 
-    matched = m.params[1].match(/(.*?):(.*)/)
-    channel = m.params[0]
+  def on_part(m)
+    super
+    kill @threads[m.params.first]
+  end
 
-    if !@channels.value?(channel) || !matched
-      @log.error "#{channel} not found."
+  def on_disconnected
+    super
+    kill_all
+  end
+
+  private
+
+  def create_worker(params)
+    channels = params[0].split(',')
+    keys = params[1].split(',')
+    channels.each_with_index.map { |v, i| [v, keys[i]] }.each do |channel, key|
+      guard auth_data(key).merge(channel: channel)
     end
+  end
 
-    @class_name = "Stalkerr::Target::#{@channels.invert[channel].capitalize}"
-    @username = matched[1]
-    @password = matched[2]
-    post @username, JOIN, channel
+  def auth_data(key)
+    id, pw = key.match(/(.*?):(.*)/).to_a.pop(2)
+    { username: id, password: pw }
+  end
 
-    @retrieve_thread = Thread.start do
+  def guard(params)
+    post params[:username], JOIN, params[:channel]
+    @threads[params[:channel]] = Thread.start(target params) do |service|
       loop do
         begin
-          target.stalking do |prefix, command, *params|
+          service.stalking do |prefix, command, *params|
             post(prefix, command, *params)
           end
           sleep Stalkerr::Const::FETCH_INTERVAL
@@ -65,9 +86,20 @@ class Stalkerr::Session < Net::IRC::Server::Session
     end
   end
 
-  private
+  def target(params)
+    ch = params[:channel]
+    class_name = "Stalkerr::Target::#{@channels.invert[ch].capitalize}"
+    unless @targets[ch].is_a?(class_name.constantize)
+      @targets[ch] = class_name.constantize.new(params[:username], params[:password])
+    end
+    @targets[ch]
+  end
+
+  def kill(thread)
+    thread.kill if thread && thread.alive?
+  end
 
-  def target
-    @target ||= @class_name.constantize.new(@username, @password)
+  def kill_all
+    @threads.each { |channel, thread| thread.kill if thread.alive? } rescue nil
   end
 end

data/lib/stalkerr/target/github.rb

@@ -19,12 +19,15 @@ module Stalkerr::Target
     end
 
     def client
-      @client ||= Octokit::Client.new(login: @username, password: @password)
+      if !@client || @client && !@client.authenticated?
+        @client = Octokit.new(login: @username, password: @password)
+      end
+      @client
     end
 
     def stalking(&post)
       @post = post
-      client.received_events(@username).reverse_each { |event|
+      client.received_events(@username).sort_by(&:id).reverse_each { |event|
         if @last_event_id.nil?
           next if Time.now.utc - Stalkerr::Const::ROLLBACK_SEC >= Time.parse(event.created_at).utc
         else
@@ -97,10 +100,11 @@ module Stalkerr::Target
           none_repository = true
           status = "created repository"
           title = event.repo.name
+          title = "#{title}: #{obj.description}" if obj.description
         else
           status = "created #{obj.ref_type}:#{obj.ref}"
+          title = obj.description
         end
-        title = obj.description
         link = "#{HOST}/#{event.repo.name}"
       when 'DeleteEvent'
         status = "deleted #{obj.ref_type}:#{obj.ref}"
@@ -180,6 +184,7 @@ module Stalkerr::Target
     end
 
     def split_for_comment(string)
+      return [] unless string.is_a? String
       string.split(/\r\n|\n/).map { |v| v unless v.eql? '' }.compact
     end
 

data/lib/stalkerr/target/qiita.rb

@@ -0,0 +1,114 @@
+require 'time'
+require 'net/irc'
+require 'qiita'
+require 'net/http'
+require 'string-irc'
+require 'stalkerr'
+
+module Stalkerr::Target
+  class Qiita
+    include Net::IRC::Constants
+
+    HOST = 'http://qiita.com'
+    CHANNEL = '#qiita'
+    INTERVAL = 60 * 10
+
+    def initialize(username, password)
+      @username = username
+      @password = password
+      @last_fetched_at = nil
+      @marker = nil
+    end
+
+    def client
+      @client ||= ::Qiita.new(url_name: @username, password: @password)
+    end
+
+    def stalking(&post)
+      return if @last_fetched_at && Time.now.utc < @last_fetched_at + INTERVAL
+      @last_fetched_at = Time.now.utc
+
+      @post = post
+      stocked_items = posted_items = []
+      stocks = {}
+
+      followings = client.user_following_users(@username).map { |u| u.url_name }.compact
+      followings.each do |user|
+        begin
+          stocks[user] = client.user_stocks(user)
+          stocked_items = stocked_items + stocks[user]
+        rescue => e
+          nil
+        end
+        begin
+          posted_items = posted_items + client.user_items(user)
+        rescue => e
+          nil
+        end
+      end
+
+      tags = client.user_following_tags(@username).map { |t| t.url_name }.compact
+      new_items = tags.inject([]) { |arr, tag| arr + client.tag_items(encoder tag) }
+
+      items = (stocked_items + posted_items + new_items).uniq
+      items[0...30].sort_by(&:id).each do |obj|
+        next if @marker && @marker >= obj.id
+        type = 'new'
+        nick = obj.user.url_name
+        case
+        when stocked_items.include?(obj)
+          type = 'stock'
+          stocks.each { |user, items| nick = user and break if items.include?(obj) }
+        when posted_items.include?(obj)
+          type = 'post'
+        end
+        parse type, [nick, obj]
+        @marker = obj.id
+      end
+    end
+
+    def parse(type, data)
+      nick, obj = data
+      header = status = title = link = ''
+      body = []
+      notice_body = false
+
+      case type
+      when 'stock'
+        status = "stocked entry"
+        color = :pink
+      when 'post'
+        status = "posted entry"
+        color = :yellow
+      when 'new'
+        status = "new entry"
+        color = :aqua
+        notice_body = true
+      end
+      title = "#{obj.title}"
+      body = split_for_body obj.raw_body
+      link = obj.url
+
+      header = StringIrc.new(status).send(color)
+      header = "#{header} #{title}" unless title.eql? ''
+      header = "#{header} - #{StringIrc.new(link).blue}"
+
+      @post.call simple(nick), NOTICE, CHANNEL, header
+      mode = notice_body ? NOTICE : PRIVMSG
+      body.each { |b| @post.call simple(nick), mode, CHANNEL, b } unless body.eql? []
+    end
+
+    def split_for_body(string)
+      return [] unless string.is_a?(String)
+      string.split(/\r\n|\n/).map { |v| v unless v.eql? '' }.compact
+    end
+
+    def simple(string)
+      string.gsub('@github', '')
+    end
+
+    def encoder(string)
+      URI.encode(string).gsub('.', '%2e')
+    end
+  end
+end

data/lib/stalkerr/version.rb

@@ -1,3 +1,3 @@
 module Stalkerr
-  VERSION = "0.0.1"
+  VERSION = "0.1.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stalkerr
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.1.0
 platform: ruby
 authors:
 - linyows
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-03-26 00:00:00.000000000 Z
+date: 2013-03-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: net-irc
@@ -88,6 +88,7 @@ files:
 - lib/stalkerr/session.rb
 - lib/stalkerr/target.rb
 - lib/stalkerr/target/github.rb
+- lib/stalkerr/target/qiita.rb
 - lib/stalkerr/version.rb
 - stalkerr.gemspec
 homepage: https://github.com/linyows/stalkerr