atig 0.0.1

data/lib/atig/bitly.rb

@@ -0,0 +1,52 @@
+# -*- mode:ruby; coding:utf-8 -*-
+require 'rubygems'
+require 'json'
+require 'atig/http'
+
+module Atig
+  class Bitly
+    class << self
+      def no_login(logger)
+        self.new logger, nil, nil
+      end
+
+      def login(logger, login, key)
+        self.new logger, login, key
+      end
+    end
+
+    def initialize(logger, login, key)
+      @log   = logger
+      @login = login
+      @key   = key
+      @http  = Http.new logger
+    end
+
+    def shorten(url)
+      return url if url =~ /bit\.ly/
+      bitly = URI("http://api.bit.ly/v3/shorten")
+      if @login and @key
+        bitly.path  = "/shorten"
+        bitly.query = {
+          :format => "json", :longUrl => url, :login => @login, :apiKey => @key,
+        }.to_query_str(";")
+        req = @http.req(:get, bitly, {})
+        res = @http.http(bitly, 5, 10).request(req)
+
+        res = JSON.parse(res.body)
+
+        if res['statusCode'] == "ERROR" then
+          @log.error res['errorMessage']
+          url
+        else
+          res["results"][url]['shortUrl']
+        end
+      else
+        url
+      end
+    rescue Errno::ETIMEDOUT, JSON::ParserError, IOError, Timeout::Error, Errno::ECONNRESET => e
+      @log.error e
+      url
+    end
+  end
+end

data/lib/atig/channel.rb

@@ -0,0 +1,5 @@
+require 'atig/channel/timeline'
+require 'atig/channel/mention'
+require 'atig/channel/dm'
+require 'atig/channel/list'
+require 'atig/channel/retweet'

data/lib/atig/channel/channel.rb

@@ -0,0 +1,17 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+module Atig
+  module Channel
+    class Channel
+      def initialize(context, gateway, db)
+        @db      = db
+        @channel = gateway.channel channel_name, :handler=>self
+        @channel.join_me
+
+        db.statuses.listen do|entry|
+          @channel.topic entry if entry.user.id == db.me.id
+        end
+      end
+    end
+  end
+end

data/lib/atig/channel/dm.rb

@@ -0,0 +1,14 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+module Atig
+  module Channel
+    class Dm
+      def initialize(context, gateway, db)
+        channel = gateway.channel db.me.screen_name
+        db.dms.listen do|dm|
+          channel.message(dm)
+        end
+      end
+    end
+  end
+end

data/lib/atig/channel/list.rb

@@ -0,0 +1,76 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+module Atig
+  module Channel
+    class List
+      class Handler
+        def initialize(db, name)
+          @db   = db
+          @name = name
+        end
+
+        def on_invite(api, nick)
+          return if @name.include? '^'
+
+          api.post("#{@db.me.screen_name}/#{@name}/members", :id => nick )
+          @db.lists.invalidate @name
+        end
+
+        def on_kick(api, nick)
+          return if @name.include? '^'
+
+          api.delete("#{@db.me.screen_name}/#{@name}/members", :id => nick )
+          @db.lists.invalidate @name
+        end
+
+        def on_who(&f)
+          return unless f
+          @db.lists[@name].users.each(&f)
+        end
+      end
+
+      def initialize(context, gateway, db)
+        @channels = Hash.new do|hash,name|
+          channel = gateway.channel "##{name}", :handler => Handler.new(db, name)
+          channel.join_me
+          hash[name] = channel
+        end
+
+        db.statuses.listen do|entry|
+          if entry.source == :list then
+            @channels[entry.list].message entry
+          else
+            lists = db.lists.find_by_screen_name(entry.user.screen_name)
+            lists.each{|name|
+              @channels[name].message entry
+            }
+          end
+        end
+
+        db.statuses.listen do|entry|
+          if entry.user.id == db.me.id
+            @channels.each{|_,channel|
+              channel.topic entry
+            }
+          end
+        end
+
+        db.lists.listen do|kind, name, users|
+          case kind
+          when :new
+            @channels[name].join_me
+            @channels[name].join db.lists[name].users
+          when :del
+            @channels[name].part_me "No longer follow the list #{name}"
+          when :join
+            @channels[name].join users
+          when :part
+            @channels[name].part users
+          when :mode
+          end
+        end
+      end
+    end
+  end
+end
+

data/lib/atig/channel/mention.rb

@@ -0,0 +1,20 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require 'atig/channel/channel'
+module Atig
+  module Channel
+    class Mention < Atig::Channel::Channel
+      def initialize(context, gateway, db)
+        super
+
+        db.statuses.listen do|entry|
+          if entry.status.text.include?("@#{db.me.screen_name}")
+            @channel.message(entry)
+          end
+        end
+      end
+
+      def channel_name; "#mention" end
+    end
+  end
+end

data/lib/atig/channel/retweet.rb

@@ -0,0 +1,28 @@
+# -*- mode:ruby; coding:utf-8 -*-
+require 'atig/channel/channel'
+
+module Atig
+  module Channel
+    class Retweet < Atig::Channel::Channel
+      def initialize(context, gateway, db)
+        super
+
+        db.statuses.find_all(:limit=>50).reverse_each {|entry|
+          message entry
+        }
+
+        db.statuses.listen {|entry|
+          message entry
+        }
+      end
+
+      def channel_name; "#retweet" end
+
+      def message(entry)
+        if entry.status.retweeted_status then
+          @channel.message entry
+        end
+      end
+    end
+  end
+end

data/lib/atig/channel/timeline.rb

@@ -0,0 +1,74 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require 'atig/channel/channel'
+require 'atig/util'
+require 'atig/update_checker'
+
+module Atig
+  module Channel
+    class Timeline < Atig::Channel::Channel
+      include Util
+
+      def initialize(context, gateway, db)
+        super
+        @log = context.log
+
+        @channel.notify "Client options: #{context.opts.marshal_dump.inspect}"
+
+        # つないだときに発言がないとさみしいので
+        db.statuses.find_all(:limit=>50).reverse_each do|entry|
+          case entry.source
+          when :timeline, :me
+            @channel.message entry
+          end
+        end
+
+        # 最新版のチェック
+        daemon do
+          log :info,"check update"
+          messages = UpdateChecker.latest
+          unless messages.empty?
+            @channel.notify "\002New version is available.\017 run 'git pull'."
+            messages[0, 3].each do |m|
+              @channel.notify "  \002#{m[/.+/]}\017"
+            end
+            @channel.notify("  ... and more. check it: http://mzp.github.com/atig/") if messages.size > 3
+          end
+          sleep (3*60*60)
+        end
+
+        db.statuses.listen do|entry|
+          if db.followings.include?(entry.user) or
+              entry.source == :timeline or
+              entry.source == :user_stream or
+              entry.source == :me then
+            @channel.message entry
+          end
+        end
+
+        @channel.send :join, db.followings.users
+
+        db.followings.listen do|kind, users|
+          @channel.send(kind, users) if @channel.respond_to?(kind)
+        end
+      end
+
+      def on_invite(api, nick)
+        api.post("friendships/create/#{nick}")
+        @db.followings.invalidate
+      end
+
+      def on_kick(api, nick)
+        api.post("friendships/destroy/#{nick}")
+        @db.followings.invalidate
+      end
+
+      def on_who(&f)
+        return unless f
+        @db.followings.users.each(&f)
+      end
+
+      def channel_name; "#twitter" end
+    end
+  end
+end

data/lib/atig/command.rb

@@ -0,0 +1,21 @@
+require 'atig/command/retweet'
+require 'atig/command/reply'
+require 'atig/command/user'
+require 'atig/command/favorite'
+require 'atig/command/uptime'
+require 'atig/command/destroy'
+require 'atig/command/status'
+require 'atig/command/thread'
+require 'atig/command/time'
+require 'atig/command/version'
+require 'atig/command/user_info'
+require 'atig/command/whois'
+require 'atig/command/option'
+require 'atig/command/location'
+require 'atig/command/name'
+require 'atig/command/autofix'
+require 'atig/command/limit'
+require 'atig/command/search'
+require 'atig/command/refresh'
+require 'atig/command/spam'
+require 'atig/command/dm'

data/lib/atig/command/autofix.rb

@@ -0,0 +1,58 @@
+# -*- mode:ruby; coding:utf-8 -*-
+require 'atig/command/command'
+require 'atig/levenshtein'
+
+begin
+  require 'jcode'
+rescue LoadError
+end
+
+module Atig
+  module Command
+    class Autofix < Atig::Command::Command
+      def initialize(*args); super end
+      def command_name; /(?:autofix|topic|overwwrite)!?/ end
+
+      def distance(s1, s2)
+        c1 = s1.split(//)
+        c2 = s2.split(//)
+        distance = Atig::Levenshtein.levenshtein c1, c2
+        distance.to_f / [ c1.size, c2.size ].max
+      end
+
+      def fix?(command, text, prev)
+        command[-1,1] == '!' or distance(text, prev.status.text) < 0.5
+      end
+
+      def action(target, mesg, command, args)
+        if args.empty?
+          yield "/me #{command} blah blah"
+          return
+        end
+        text = mesg.split(" ", 2)[1]
+        q = gateway.output_message(:status => text)
+
+        prev,*_ = db.statuses.find_by_user( db.me, :limit => 1)
+
+        unless fix?(command, q[:status], prev) then
+          api.delay(0, :retry=>3) do|t|
+            ret = t.post("statuses/update", q)
+            gateway.update_status ret, target
+          end
+        else
+          api.delay(0, :retry=>3) do|t|
+            yield "Similar update in previous. Conclude that it has error."
+            yield "And overwrite previous as new status: #{q[:status]}"
+
+            ret = t.post("statuses/update", q)
+            gateway.update_status ret, target
+            t.post("statuses/destroy/#{prev.status.id}")
+            db.statuses.transaction{|d|
+              d.remove_by_id prev.id
+            }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/atig/command/command.rb

@@ -0,0 +1,24 @@
+# -*- mode:ruby; coding:utf-8 -*-
+module Atig
+  module Command
+    class Command
+      attr_reader :gateway, :api, :db, :opts
+      def initialize(context, gateway, api, db)
+        @log     = context.log
+        @opts    = context.opts
+        @gateway = gateway
+        @api     = api
+        @db      = db
+	@gateway.ctcp_action(*command_name) do |target, mesg, command, args|
+          action(target, mesg, command, args){|m|
+            gateway[target].notify m
+          }
+	end
+      end
+
+      def find_by_tid(tid)
+        @db.statuses.find_by_tid tid
+      end
+    end
+  end
+end

data/lib/atig/command/command_helper.rb

@@ -0,0 +1,95 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+class FakeGateway
+  attr_reader :names,:action,:filtered,:updated, :notified
+
+  def initialize(channel)
+    @channel = channel
+  end
+
+  def ctcp_action(*names, &action)
+    @names  = names
+    @action = action
+  end
+
+  def output_message(m); @filtered = m end
+
+  def update_status(*args); @updated = args end
+
+  def [](name)
+    @notified = name
+    @channel
+  end
+
+  def server_name; "server-name" end
+end
+
+class FakeScheduler
+  def initialize(api)
+    @api = api
+  end
+
+  def delay(interval,opt={},&f)
+    f.call @api
+  end
+
+  def limit; @api.limit end
+  def remain; @api.remain end
+  def reset; @api.reset end
+end
+
+class FakeDb
+  attr_reader :statuses, :followings,:lists, :me
+  def initialize(statuses, followings,lists, me)
+    @statuses = statuses
+    @followings = followings
+    @lists = lists
+    @me = me
+  end
+
+  def transaction(&f)
+    f.call self
+  end
+end
+
+class FakeDbEntry
+  def initialize(name)
+    @name = name
+  end
+
+  def transaction(&f)
+    f.call self
+  end
+end
+
+module CommandHelper
+  def init(klass)
+    @log    = mock 'log'
+    @opts   = Atig::Option.new({})
+    context = OpenStruct.new :log=>@log, :opts=>@opts
+
+    @channel    = mock 'channel'
+    @gateway    = FakeGateway.new @channel
+    @api        = mock 'api'
+    @statuses   = FakeDbEntry.new 'status DB'
+    @followings = FakeDbEntry.new 'followings DB'
+    @lists      = {
+      "A" =>  FakeDbEntry.new('list A'),
+      "B" =>  FakeDbEntry.new('list B')
+    }
+
+    @me         = user 1,'me'
+    @db         = FakeDb.new @statuses, @followings, @lists, @me
+    @command = klass.new context, @gateway, FakeScheduler.new(@api), @db
+  end
+
+  def call(channel, command, args)
+    @gateway.action.call channel, "#{command} #{args.join(' ')}", command, args
+  end
+
+  def stub_status(key, hash)
+    @statuses.stub!(key).and_return{|arg,*_|
+      hash.fetch(arg, hash[:default])
+    }
+  end
+end