atig 0.0.1

data/lib/atig/command/destroy.rb

@@ -0,0 +1,44 @@
+# -*- mode:ruby; coding:utf-8 -*-
+require 'time'
+require 'atig/command/command'
+require 'atig/command/info'
+
+module Atig
+  module Command
+    class Destroy < Atig::Command::Command
+      def initialize(*args); super end
+      def command_name; %w(destroy remove rm) end
+
+      def action(target, mesg, command, args)
+        if args.empty?
+          yield "/me #{command} <ID1> <ID2> ..."
+          return
+        end
+        args.each do|tid|
+          if entry = Info.find_status(db, tid)
+            if entry.user.id == db.me.id
+              api.delay(0) do|t|
+                res = t.post("statuses/destroy/#{entry.status.id}")
+                yield "Destroyed: #{entry.status.text}"
+
+                db.statuses.transaction do|d|
+                  xs = d.find_by_screen_name db.me.screen_name,:limit=>1
+                  d.remove_by_id entry.id
+                  ys = d.find_by_screen_name db.me.screen_name,:limit=>1
+
+                  unless xs.map{|x| x.id} == ys.map{|y| y.id} then
+                    gateway.topic ys.first
+                  end
+                end
+              end
+            else
+              yield "The status you specified by the ID tid is not yours."
+            end
+          else
+            yield "No such ID tid"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/atig/command/dm.rb

@@ -0,0 +1,31 @@
+# -*- mode:ruby; coding:utf-8 -*-
+require 'atig/command/command'
+begin
+  require 'jcode'
+rescue LoadError
+end
+
+module Atig
+  module Command
+    class Dm < Atig::Command::Command
+      def initialize(*args); super end
+      def command_name; %w(d dm dms) end
+
+      def action(target, mesg, command, args)
+        if args.empty?
+          yield "/me #{command} <SCREEN_NAME> blah blah"
+          return
+        end
+        user = args.first
+        text = mesg.split(" ", 3)[2]
+        api.delay(0) do|t|
+          t.post("direct_messages/new",{
+                   :user => user,
+                   :text => text
+                 })
+          yield "Sent message to #{user}: #{text}"
+        end
+      end
+    end
+  end
+end

data/lib/atig/command/favorite.rb

@@ -0,0 +1,27 @@
+# -*- mode:ruby; coding:utf-8 -*-
+require 'atig/command/command'
+require 'atig/command/info'
+
+module Atig
+  module Command
+    class Favorite < Atig::Command::Command
+      def initialize(*args); super end
+      def command_name; %w(fav unfav) end
+
+      def action(target, mesg, command, args)
+        method   = { 'fav' => 'create', 'unfav' => 'destroy' }[command]
+
+        args.each do|tid|
+          if entry = Info.find_status(db, tid)
+            api.delay(0){|t|
+              res = t.post("favorites/#{method}/#{entry.status.id}")
+              yield "#{command.upcase}: #{entry.user.screen_name}: #{entry.status.text}"
+            }
+          else
+            yield "No such ID : #{tid}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/atig/command/info.rb

@@ -0,0 +1,50 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+module Atig
+  module Command
+    module Info
+      def user(db, api, name, &f)
+        if user = db.followings.find_by_screen_name(name) then
+          f.call user
+        else
+          api.delay(0) do|t|
+            user = t.get "users/show",:screen_name=>name
+            f.call user
+          end
+        end
+      end
+
+      def status(db, api, id, &f)
+        if status = db.statuses.find_by_status_id(id) then
+          f.call status
+        else
+          api.delay(0) do|t|
+            status = t.get "statuses/show/#{id}"
+            db.statuses.transaction do|d|
+              d.add :status => status, :user => status.user, :source => :thread
+              f.call d.find_by_status_id(id)
+            end
+          end
+        end
+      end
+
+      def find_status(db, tid_or_screen_name)
+        find = lambda do|x|
+          xs = db.statuses.find_by_screen_name(x, :limit=>1)
+          unless xs.empty? then
+            xs.first
+          else
+            nil
+          end
+        end
+
+        (db.statuses.find_by_tid(tid_or_screen_name) ||
+         db.statuses.find_by_sid(tid_or_screen_name) ||
+         find.call(tid_or_screen_name) ||
+         find.call(tid_or_screen_name.sub(/\A@/,'')))
+      end
+
+      module_function :user,:status, :find_status
+    end
+  end
+end

data/lib/atig/command/limit.rb

@@ -0,0 +1,15 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require 'atig/command/command'
+
+module Atig
+  module Command
+    class Limit < Atig::Command::Command
+      def command_name; %w(rls limit limits) end
+
+      def action(target, mesg, command, args)
+        yield "#{api.remain} / #{api.limit} (reset at #{::Time.at(api.reset).strftime('%Y-%m-%d %H:%M:%S')})"
+      end
+    end
+  end
+end

data/lib/atig/command/location.rb

@@ -0,0 +1,23 @@
+# -*- mode:ruby; coding:utf-8 -*-
+require 'atig/command/command'
+
+module Atig
+  module Command
+    class Location < Atig::Command::Command
+      def command_name; %w(in location loc) end
+
+      def action(target, mesg, command, args)
+        api.delay(0) do|t|
+          location = mesg.split(" ", 2)[1] || ""
+          t.post('account/update_profile',:location=>location)
+
+          if location.empty? then
+            yield "You are nowhere now."
+          else
+            yield "You are in #{location} now."
+          end
+        end
+      end
+    end
+  end
+end

data/lib/atig/command/name.rb

@@ -0,0 +1,18 @@
+# -*- mode:ruby; coding:utf-8 -*-
+require 'atig/command/command'
+
+module Atig
+  module Command
+    class Name < Atig::Command::Command
+      def command_name; %w(name) end
+
+      def action(target, mesg, command, args)
+        api.delay(0) do|t|
+          name = mesg.split(" ", 2)[1] || ""
+          t.post('account/update_profile',:name=>name)
+          yield "You are named #{name}."
+        end
+      end
+    end
+  end
+end

data/lib/atig/command/option.rb

@@ -0,0 +1,37 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require 'atig/command/command'
+require 'atig/option'
+
+module Atig
+  module Command
+    class Option < Atig::Command::Command
+      def initialize(*args)
+        super
+        @methods = OpenStruct.instance_methods
+      end
+
+      def command_name; %w(opt opts option options) end
+
+      def action(target, mesg, command, args)
+        if args.empty?
+          @opts.fields.
+            map{|x| x.to_s }.
+            sort.each do|name|
+              yield "#{name} => #{@opts[name]}"
+          end
+        else
+          _,name,value = mesg.split ' ', 3
+          unless value then
+            # show the value
+            yield "#{name} => #{@opts.send name}"
+          else
+            # set the value
+            @opts.send "#{name}=",::Atig::Option.parse_value(value)
+            yield "#{name} => #{@opts.send name}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/atig/command/refresh.rb

@@ -0,0 +1,18 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+# -*- mode:ruby; coding:utf-8 -*-
+require 'atig/command/info'
+require 'time'
+module Atig
+  module Command
+    class Refresh < Atig::Command::Command
+      def command_name; %w(refresh) end
+
+      def action(target, mesg, command,args)
+        db.followings.invalidate
+        db.lists.invalidate :all
+        yield "refresh followings/lists..."
+      end
+    end
+  end
+end

data/lib/atig/command/reply.rb

@@ -0,0 +1,37 @@
+# -*- mode:ruby; coding:utf-8 -*-
+require 'atig/command/command'
+require 'atig/command/info'
+
+module Atig
+  module Command
+    class Reply < Atig::Command::Command
+      def initialize(*args); super end
+      def command_name; %w(mention re reply rp) end
+
+      def action(target, mesg, command, args)
+        if args.empty?
+          yield "/me #{command} <ID_or_SCREEN_NAME> blah blah"
+          return
+        end
+
+        tid = args.first
+        if entry = Info.find_status(db,tid) then
+          text = mesg.split(" ", 3)[2]
+          name = entry.user.screen_name
+
+          text = "@#{name} #{text}" if text.nil? or not text.include?("@#{name}")
+
+          q = gateway.output_message(:status => text,
+                                     :in_reply_to_status_id => entry.status.id)
+
+          api.delay(0) do|t|
+            ret = t.post("statuses/update", q)
+            gateway.update_status ret, target, "In reply to #{name}: #{entry.status.text}"
+          end
+        else
+          yield "No such ID : #{tid}"
+        end
+      end
+    end
+  end
+end

data/lib/atig/command/retweet.rb

@@ -0,0 +1,63 @@
+# -*- mode:ruby; coding:utf-8 -*-
+
+require 'atig/bitly'
+require 'atig/command/command'
+begin
+  require 'jcode'
+rescue LoadError
+end
+
+module Atig
+  module Command
+    class Retweet < Atig::Command::Command
+      def initialize(*args)
+        super
+        @bitly = Bitly.no_login @log
+      end
+
+      def command_name; %w(ort rt retweet qt) end
+
+      def rt_with_comment(target, comment, entry)
+        screen_name = "@#{entry.user.screen_name}"
+        text = "#{comment.strip} RT #{screen_name}: #{entry.status.text}"
+
+        chars = text.each_char.to_a
+        if chars.size > 140 then
+          url = @bitly.shorten "http://twitter.com/#{entry.user.screen_name}/status/#{entry.status.id}"
+          text = chars[0,140-url.size-1].join('') + ' ' + url
+        end
+        q = gateway.output_message(:status => text)
+        api.delay(0) do|t|
+          ret = t.post("statuses/update", q)
+          gateway.update_status ret,target, "RT to #{entry.user.screen_name}: #{entry.status.text}"
+        end
+      end
+
+      def rt_with_no_comment(target, entry)
+        api.delay(0) do|t|
+          ret = t.post("statuses/retweet/#{entry.status.id}")
+          gateway.update_status ret,target, "RT to #{entry.user.screen_name}: #{entry.status.text}"
+        end
+      end
+
+      def action(target, mesg, command, args)
+        if args.empty?
+          yield "/me #{command} <ID_or_SCREEN_NAME> blah blah"
+          return
+        end
+
+        tid = args.first
+        if status = Info.find_status(db, tid) then
+          if args.size >= 2
+            comment = mesg.split(" ", 3)[2] + " "
+            rt_with_comment(target, comment, status)
+          else
+            rt_with_no_comment(target, status)
+          end
+        else
+          yield "No such ID : #{tid}"
+        end
+      end
+    end
+  end
+end

data/lib/atig/command/search.rb

@@ -0,0 +1,51 @@
+#!/usr/bin/env ruby
+# -*- mode:ruby; coding:utf-8 -*-
+require 'atig/command/command'
+require 'atig/search'
+
+# originally developped by xeres
+# http://blog.xeres.jp/2010/06/04/atig_rb-tweet-search/
+module Atig
+  module Command
+    class Search < Atig::Command::Command
+      def command_name; %w(search s) end
+
+      def action(target, mesg, command, args)
+        if args.empty?
+          yield "/me #{command} [option...] blah blah"
+          return
+        end
+
+        q = mesg.sub(/^#{command}\s+/, '')
+        opts = { :q => q }
+        while /^:(?:(lang)=(\w+))/ =~ args.first
+           opts[$1] = $2
+           q.sub!(/^#{args.first}\W+/, "")
+           args.shift
+        end
+
+        statuses = api.search.get('search', opts).results
+
+        if statuses.empty?
+          yield "\"#{q}\": not found. options=#{opts.inspect} (#{res['completed_in']} sec.)"
+          return
+        end
+
+        statuses.reverse_each do|status|
+          db.statuses.transaction do|d|
+            user = TwitterStruct.make('id'          => status.from_user_id,
+                                      'screen_name' => status.from_user)
+            d.add :status => status, :user => user, :source => :user
+          end
+        end
+
+        statuses.reverse_each do|status|
+          entry = db.statuses.find_by_status_id(status.id)
+          entry.status = entry.status.merge('text' =>
+                                            "#{entry.status.text} (#{entry.status.created_at})")
+          gateway[target].message entry, Net::IRC::Constants::NOTICE
+        end
+      end
+    end
+  end
+end

data/lib/atig/command/spam.rb

@@ -0,0 +1,26 @@
+# -*- mode:ruby; coding:utf-8 -*-
+require 'atig/command/command'
+require 'atig/command/info'
+
+module Atig
+  module Command
+    class Spam < Atig::Command::Command
+      def initialize(*args); super end
+      def command_name; %w(spam SPAM) end
+
+      def action(target, mesg, command, args)
+        if args.empty?
+          yield "/me #{command} <SCREEN_NAME1> <SCREEN_NAME2> ..."
+          return
+        else
+          args.each do|screen_name|
+            api.delay(0) do|t|
+              res = t.post("report_spam",:screen_name => screen_name)
+              yield "Report #{res.screen_name} as SPAMMER"
+            end
+          end
+        end
+      end
+    end
+  end
+end