friendfeed 0.1.9 → 0.1.10

data/VERSION

@@ -1 +1 @@
-0.1.9
+0.1.10

data/bin/tw2ff

@@ -55,10 +55,6 @@ EOM
   exit 1
 end
 
-def puterror(message)
-  STDERR.puts MYNAME + ': ' + e.to_s
-end
-
 def putinfo(fmt, *args)
   STDERR.puts sprintf(fmt, *args)
 end
@@ -76,6 +72,42 @@ def Status(key)
   end
 end
 
+def agent
+  $agent ||= Mechanize.new
+end
+
+def parse_uri(url)
+  case url
+  when URI
+    url
+  else
+    begin
+      URI.parse(url)
+    rescue URI::InvalidURIError
+      dir, file = File.split(url)
+      URI.parse(File.join(dir, URI.escape(file)))
+    end
+  end
+end
+
+def get_file(url)
+  uri = parse_uri(url)
+  putinfo 'Fetching %s', uri
+  agent.get_file(uri)
+end
+
+class FriendFeed::Client
+  def change_picture_to_url(id, url)
+    t = Tempfile.open("picture")
+    t.write get_file(url)
+    t.close
+    File.open(t.path) { |f|
+      change_picture(id, f)
+    }
+    t.unlink
+  end
+end
+
 def friendfeed_client
   $ff_client ||=
     begin
@@ -89,13 +121,20 @@ end
 class Twitter::Base
   def all_friends
     list = []
-    (1..100).each { |i|
-      slice = friends(:page => i)
-      list.concat(slice)
-      break if slice.size < 100
+    cursor = -1
+    loop {
+      result = friends(:cursor => cursor)
+      break if result.users.nil?
+      list.concat(result.users)
+      cursor = result[:next_cursor]
+      break if cursor.zero?
     }
     list
   end
+
+  def get_profile(name)
+    friends(:screen_name => name).first
+  end
 end
 
 def twitter_client
@@ -216,25 +255,58 @@ EOF
         putinfo 'Skipping a protected user %s', name
       }
 
-      agent = Mechanize.new
-
       to_subscribe.each { |name|
         putinfo 'Creating an imaginary friend for %s', name
         id = ffcli.create_imaginary_friend('(%s)' % name)
         ffcli.add_twitter(id, name)
-        if picture_urls.key?(name)
+        if url = picture_urls[name]
           putinfo 'Setting the picture of %s', name
-          t = Tempfile.open("picture")
-          t.write agent.get_file(picture_urls[name])
-          t.close
-          File.open(t.path) { |f|
-            ffcli.change_picture(id, f)
-          }
+          ffcli.change_picture_to_url(id, url)
         end
       }
     end
   end
 
+  mode 'icons' do
+    description 'Update imaginary friends icons with those of their Twitter accounts'
+
+    def run
+      twcli = twitter_client()
+
+      picture_urls = {}
+
+      twcli.all_friends.each { |friend|
+        name = friend.screen_name.downcase
+        picture_urls[name] = friend.profile_image_url
+      }
+
+      ffcli = friendfeed_client()
+
+      putinfo "Checking imaginary friends in FriendFeed..."
+      ffcli.get_imaginary_friends.each { |profile|
+        profile['services'].each { |service|
+          url = service['profileUrl'] or next
+          if (name = TWITTER_URI.route_to(url).to_s).match(/\A[A-Za-z0-9_]+\z/)
+            name.downcase!
+            if picture_urls.key?(name)
+              url = picture_urls[name]
+            else
+              friend = twcli.get_profile(name) rescue
+                begin
+                  putinfo "Failed to get profile of %s", name
+                  next
+                end
+              url = friend.profile_image_url
+            end
+            id = profile['id']
+            putinfo 'Changing the picture of %s', name
+            ffcli.change_picture_to_url(id, url)
+          end
+        }
+      }
+    end
+  end
+
   mode 'likes' do
     description 'Synchronize Twitter favorites and FriendFeed likes as far as possible'
 

data/friendfeed.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{friendfeed}
-  s.version = "0.1.9"
+  s.version = "0.1.10"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Akinori MUSHA"]
-  s.date = %q{2010-02-09}
+  s.date = %q{2010-02-12}
   s.default_executable = %q{tw2ff}
   s.description = %q{This is a Ruby library to provide access to FriendFeed API's.
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: friendfeed
 version: !ruby/object:Gem::Version 
-  version: 0.1.9
+  version: 0.1.10
 platform: ruby
 authors: 
 - Akinori MUSHA
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-02-09 00:00:00 +09:00
+date: 2010-02-12 00:00:00 +09:00
 default_executable: tw2ff
 dependencies: 
 - !ruby/object:Gem::Dependency