geppetto 0.9.2

data/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2011 Ternary Labs. All Rights Reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.textile

@@ -0,0 +1,17 @@
+Geppetto is a simple command line tool to help manage Facebook test users and the generation of content.
+
+h2. Installation
+
+@gem install geppetto@
+
+h2. Usage
+
+@geppetto help@
+
+h2. Dependencies
+
+A working version of ImageMagick library is required.
+
+To install on OS X via port
+
+@port install tiff -macosx imagemagick +q8 +gs +wmf@

data/bin/geppetto

@@ -0,0 +1,37 @@
+#!/usr/bin/env ruby
+
+require 'geppetto'
+require 'geppetto/cli'
+
+name = File.basename(__FILE__)
+settings_path = ".#{name}/settings.yaml"
+
+if File.exist?(settings_path)
+  ::Settings = YAML.load_file(settings_path)
+elsif File.exists?(File.expand_path("~/#{settings_path}"))
+  ::Settings = YAML.load_file(File.expand_path("~/#{settings_path}"))
+else
+  puts "You need to create a settings file in ~/#{settings_path}. Sample content:"
+  sample = <<HERE
+dev:  &defaults
+  app_id: <-- YOUR APP ID -->
+  secret: <-- YOUR APP SECRET -->
+  # http://developers.facebook.com/docs/authentication/permissions/
+  default_permissions: [read_stream, offline_access, publish_stream, friends_hometown, friends_location]
+qa:
+    app_id: <-- YOUR APP ID -->
+    secret: <-- YOUR APP SECRET -->
+    <<: *defaults
+HERE
+  puts sample
+  exit
+end
+
+begin
+  Geppetto::CLI.start
+rescue Koala::Facebook::APIError => e
+  puts e
+rescue Interrupt => e
+  puts "\nQuitting..."
+  exit 1
+end

data/lib/geppetto/cli.rb

@@ -0,0 +1,271 @@
+require 'thor'
+require 'koala'
+require 'ostruct'
+require 'yaml'
+require 'progressbar'
+require 'rmagick'
+require 'tempfile'
+
+module Geppetto
+  class CLI < Thor 
+    include Koala
+
+    method_option :env, :type => :string, :default => 'dev', :desc => "App environment"  
+    def initialize(*args)
+      super
+      unless Settings[options.env]
+        puts "Environement #{options.env} is not defined in the settings.yaml file"
+        exit
+      end
+      @test_users = Facebook::TestUsers.new(:app_id => Settings[options.env]['app_id'], :secret => Settings[options.env]['secret'])
+    end
+
+    desc "version", "Show program version"
+    def version
+      say "Version: #{::Geppetto::VERSION}", :white
+    end
+  
+    desc "create [NB]", "Create one or many test users"
+    method_option :connected, :type => :boolean, :default => true, :alias => :string, :desc => "Whether the user has 'installed' the FB application"
+    method_option :networked, :type => :boolean, :default => false, :alias => :string, :desc => "Automatically befriend every user with each other"
+    def create(nb=1)
+      create_users(nb, options.connected?, options.networked?, Settings[options.env]['default_permissions'])
+    end
+  
+    desc "list", "List all test users"
+    def list
+      users_hash = @test_users.list
+      say "Listing #{users_hash.size} test users", :white
+      users_hash.each{|user_hash|
+        dump(user_hash)
+      }
+    end
+
+    desc "delete ID", "Delete a test user"
+    def delete(id)
+      if yes? "Are you sure you want to delete user #{id}?", :red
+        @test_users.delete(id)
+        say "Deleted 1 user", :white
+      end
+    end
+
+    desc "delete_all", "Delete all test users"
+    def delete_all
+      if yes? "Are you sure you want to delete ALL existing test users?", :red
+        size = get_test_users.size
+        say "Deleting #{size} users", :white
+        progress = ProgressBar.new("Deleting", size)
+        get_test_users.each{|user_hash|
+          @test_users.delete(user_hash)
+          progress.inc
+        }
+        progress.finish
+      end
+    end
+  
+    desc "about ID", "Display information about the given user"
+    def about(id)
+      @graph = Facebook::GraphAPI.new(get_token_for_user_id(id))
+      account = @graph.get_object("me")    
+      dump(account)
+    end
+
+    desc "befriend ID1 ID2", "Friend two users"
+    def befriend(id1, id2)
+      @test_users.befriend(get_user_hash(id1), get_user_hash(id2))
+    end
+  
+    desc "wall_post ID TEXT", "Post to the user's wall"
+    def wall_post(id, text)
+      @graph = Facebook::GraphAPI.new(get_token_for_user_id(id))
+      @graph.put_wall_post(text)
+    end
+
+    desc "generate_status", "Every user will post a new status update"
+    def generate_status
+      users_hash = get_test_users
+      say "Posting status for #{users_hash.size} users", :white
+      progress = ProgressBar.new("Posting", users_hash.size)
+      users_hash.each{|user_hash|
+        # Post on the user's wall
+        @graph = Facebook::GraphAPI.new(get_token_for_user_id(user_hash['id']))
+        @graph.put_wall_post("Status update created at #{Time.now}")
+        progress.inc
+      }
+      progress.finish
+    end
+
+    desc "generate_posts", "Every user will post on their friend's wall"
+    def generate_posts
+      users_hash = get_test_users
+      say "Posting for #{users_hash.size} users", :white
+      users_hash.each{|user_hash|
+        @graph = Facebook::GraphAPI.new(get_token_for_user_id(user_hash['id']))
+
+        # Post on friend's wall
+        friends = @graph.get_connections(user_hash['id'], 'friends')
+        progress = ProgressBar.new("Posting", friends.size)
+        friends.each{|friend_hash|
+          @graph.put_wall_post("Post created at #{Time.now}", {}, friend_hash['id'])
+          progress.inc
+        }
+        progress.finish
+      }
+    end
+
+    desc "generate_likes", "Every user will like all their friend's posts"
+    def generate_likes
+      users_hash = get_test_users
+      say "Posting likes for #{users_hash.size} users", :white
+      users_hash.each{|user_hash|
+        # Get this user's feed
+        token = get_token_for_user_id(user_hash['id'])
+        @graph = Facebook::GraphAPI.new(token)
+        feed = @graph.get_connections('me', 'home')
+        progress = ProgressBar.new("Liking", feed.size)
+        feed.each {|item|
+          @graph.put_like(item['id'])
+          progress.inc
+        }
+        progress.finish
+      }
+    end
+
+    desc "generate_comments", "Every user will add a comment to each post in their feed"
+    def generate_comments
+      users_hash = get_test_users
+      say "Posting comments for #{users_hash.size} users", :white
+      users_hash.each{|user_hash|
+        # Get this user's feed
+        @graph = Facebook::GraphAPI.new(get_token_for_user_id(user_hash['id']))
+        feed = @graph.get_connections('me', 'home')
+        progress = ProgressBar.new("Comments", feed.size)
+        feed.each {|item|
+          @graph.put_comment(item['id'], "Comment created at #{Time.now}")
+          progress.inc
+        }
+        progress.finish
+      }
+    end
+
+    desc "generate_images", "Every user will post a picture to their feed"
+    def generate_images
+      users_hash = get_test_users
+      say "Posting images for #{users_hash.size} users", :white
+      progress = ProgressBar.new("Images", users_hash.size)
+      users_hash.each{|user_hash|
+        # Get this user's feed
+        @graph = Facebook::GraphAPI.new(get_token_for_user_id(user_hash['id']))
+        file = Tempfile.new('geppetto')
+        begin
+          create_image(file.path)
+          @graph.put_picture(file.path, 'image/jpeg', {:message => "Image created at #{Time.now}"})
+        ensure
+          file.close
+          file.unlink
+        end
+        progress.inc
+      }
+      progress.finish
+    end
+
+    desc "build", "Built test user network and generate posts, likes, images"
+    def build
+      say "Build test setup", :white
+      delete_all
+      create_users(10, true, true, Settings[options.env]['default_permissions'])
+      generate_status
+      generate_posts
+      generate_images
+      generate_comments
+      generate_likes
+    end
+
+    desc "frenzy", "Genera posts, likes, images, etc. for all users in a random fashion. CTRL-C to cancel."
+    def frenzy
+      say "Frenzy! (CTRL-C to stop)", :white
+      actions = %w(generate_status generate_posts generate_images generate_comments generate_likes)
+      while (true) 
+        send actions[rand(actions.size)]
+        say "Sleeping"
+        sleep 2.0
+      end
+    end
+  
+    private
+    def dump(hash)
+      say "-" * 80
+      max_key_width = hash.keys.collect{|key| key.size}.max
+      format_string = "%#{max_key_width}s: %s"
+      hash.each{|key,value| say format(format_string, key, value)}
+      say "-" * 80
+    end
+
+    # Returned a sanitized version of the test users list. For some reason, some test users
+    # are now returned without oauth access token.
+    def get_test_users
+      @test_users.list.delete_if {|user_hash| !user_hash.include?('access_token') }
+    end
+  
+    def get_user_hash(user_id)
+      if user_id
+        get_test_users.each {|user_hash| return user_hash if user_hash['id'] == user_id}
+      end
+      raise Facebook::APIError.new({"type" => 'Error', 'message' => "This user id is unknown"})
+    end
+  
+    def get_token_for_user_id(user_id)
+      return get_user_hash(user_id)['access_token']
+    end
+
+    # Crate an image containing text
+    def create_image(file)
+      image = Magick::Image.new(320, 240, Magick::HatchFill.new("##{'%02X' % rand(255)}#{'%02X' % rand(255)}#{'%02X' % rand(255)}"))
+      image.format = "jpg"
+      image.write(file)
+    end
+  
+    def create_users(nb, connected, networked, permissions)
+      if networked
+        say "Creating #{nb} networked test users", :white
+        users_hash = create_network(nb.to_i, connected, permissions.join(","))
+        users_hash.each{|user_hash|
+          dump(user_hash)
+        }      
+      else
+        say "Creating #{nb} test users", :white  
+        while(nb.to_i > 0)      
+          user_hash = @test_users.create(connected, permissions.join(","))
+          dump(user_hash)
+          nb = nb.to_i-1
+        end
+      end    
+    end  
+  
+    def create_network(network_size, installed = true, permissions = '')
+      network_size = 100 if network_size > 100
+      progress = ProgressBar.new("Creating", network_size)
+      users = (0...network_size).collect { 
+        progress.inc
+        @test_users.create(installed, permissions) 
+      }
+      progress.finish
+
+      friends = users.clone
+      users.each do |user|
+        # Remove this user from list of friends
+        friends.delete_at(0)
+        # befriend all the others
+        friends.each do |friend|
+          say "Befriending #{user['id']} #{friend['id']}"
+          begin
+            @test_users.befriend(user, friend)
+          rescue Facebook::APIError => e
+            say "Problem befriending: #{e}", :red
+          end
+        end
+      end
+      return users
+    end  
+  end
+end

data/lib/geppetto/version.rb

@@ -0,0 +1,3 @@
+module Geppetto
+  VERSION = '0.9.2'
+end

data/lib/geppetto.rb

@@ -0,0 +1 @@
+require 'geppetto/version'

metadata

@@ -0,0 +1,131 @@
+--- !ruby/object:Gem::Specification 
+name: geppetto
+version: !ruby/object:Gem::Version 
+  hash: 63
+  prerelease: 
+  segments: 
+  - 0
+  - 9
+  - 2
+  version: 0.9.2
+platform: ruby
+authors: 
+- Georges Auberger
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-03-26 00:00:00 -07:00
+default_executable: geppetto
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: koala
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 62196359
+        segments: 
+        - 1
+        - 0
+        - 0
+        - beta
+        - 2
+        version: 1.0.0.beta2
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: thor
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: progressbar
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: rmagick
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id004
+description: A simple command line tool to help manage Facebook test users and the generation of content.
+email: georges@ternarylabs.com
+executables: 
+- geppetto
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- bin/geppetto
+- lib/geppetto/cli.rb
+- lib/geppetto/version.rb
+- lib/geppetto.rb
+- LICENSE
+- README.textile
+has_rdoc: true
+homepage: https://github.com/ternarylabs/geppetto
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: 
+- ImageMagick
+rubyforge_project: 
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: A simple command line tool to help manage Facebook test users and the generation of content.
+test_files: []
+