jnunemaker-twitter 0.2.8

data/lib/twitter/cli.rb

@@ -0,0 +1,297 @@
+require 'rubygems'
+gem 'main', '>= 2.8.2'
+gem 'highline', '>= 1.4.0'
+gem 'activerecord', '>= 2.1.0'
+gem 'sqlite3-ruby', '>= 1.2.2'
+require 'main'
+require 'highline/import'
+require 'activerecord'
+require 'sqlite3'
+
+HighLine.track_eof = false
+CLI_ROOT = File.expand_path(File.join(File.dirname(__FILE__), 'cli'))
+require CLI_ROOT + '/config'
+require CLI_ROOT + '/helpers'
+Dir[CLI_ROOT + '/models/*.rb'].each { |m| require m }
+
+include Twitter::CLI::Helpers
+
+Main {
+  def run
+    puts "twitter [command] --help for usage instructions."
+    puts "The available commands are: \n   install, uninstall, add, remove, list, change, post, befriend, defriend, follow, leave, d and timeline."
+  end
+  
+  mode 'install' do
+    description 'Creates the sqlite3 database and runs the migrations.'
+    def run
+      migrate
+      say 'Twitter installed.'
+    end
+  end
+  
+  mode 'uninstall' do
+    description 'Removes the sqlite3 database. There is no undo for this.'
+    def run
+      FileUtils.rm(Twitter::CLI::Config[:database])
+      say 'Twitter gem uninstalled.'
+    end
+  end
+  
+  mode 'add' do
+    description 'Adds a new twitter account to the database. Prompts for username and password.'
+    def run
+      account = Hash.new
+      say "Add New Account:"
+
+      account[:username] = ask('Username: ') do |q|
+        q.validate = /\S+/
+      end
+
+      account[:password] = ask("Password (won't be displayed): ") do |q|
+        q.echo = false
+        q.validate = /\S+/
+      end
+      
+      do_work do
+        base(account[:username], account[:password]).verify_credentials
+        account[:current] = Account.current.count > 0 ? false : true
+        Account.create(account)
+        say 'Account added.'
+      end
+    end
+  end
+  
+  mode 'remove' do
+    description 'Removes a twitter account from the database. If username provided it removes that username else it prompts with list and asks for which one you would like to remove.'
+    argument( 'username' ) { 
+      optional
+      description 'username of account you would like to remove' 
+    }
+    
+    def run
+      do_work do
+        if params['username'].given?
+          account = Account.find_by_username(params['username'].value)
+        else
+          Account.find(:all, :order => 'username').each do |a|
+            say "#{a.id}. #{a}"
+          end
+          account_id = ask 'Account to remove (enter number): ' do |q|
+            q.validate = /\d+/
+          end
+        end
+        
+        begin
+          account = account_id ? Account.find(account_id) : account
+          account_name = account.username
+          account.destroy
+          say "#{account_name} has been removed.\n"
+        rescue ActiveRecord::RecordNotFound
+          say "ERROR: Account could not be found. Try again. \n"
+        end
+        
+      end
+    end
+  end
+  
+  mode 'list' do
+    description 'Lists all the accounts that have been added and puts a * by the current one that is used for posting, etc.'
+    def run
+      do_work do
+        if Account.count == 0
+          say 'No accounts have been added.' 
+        else
+          say 'Account List'
+          Account.find(:all, :order => 'username').each do |a|
+            say a 
+          end
+        end
+      end
+    end
+  end
+  
+  mode 'change' do
+    description 'Changes the current account being used for posting etc. to the username provided. If no username is provided, a list is presented and you can choose the account from there.'
+    argument( 'username' ) { 
+      optional
+      description 'username of account you would like to switched to' 
+    }
+    
+    def run
+      do_work do
+        if params['username'].given?
+          new_current = Account.find_by_username(params['username'].value)
+        else
+          Account.find(:all, :order => 'username').each do |a|
+            say "#{a.id}. #{a}"
+          end
+          new_current = ask 'Change current account to (enter number): ' do |q|
+            q.validate = /\d+/
+          end
+        end
+        
+        begin
+          current = Account.set_current(new_current)
+          say "#{current} is now the current account.\n"
+        rescue ActiveRecord::RecordNotFound
+          say "ERROR: Account could not be found. Try again. \n"
+        end
+      end
+    end
+  end
+  
+  mode 'post' do
+    description "Posts a message to twitter using the current account. The following are all valid examples from the command line:
+    $ twitter post 'my update'
+    $ twitter post my update with quotes
+    $ echo 'my update from stdin' | twitter post"
+    def run
+      do_work do
+        post = ARGV.size > 1 ? ARGV.join(" ") : ARGV.shift
+        say "Sending twitter update"
+        finished, status = false, nil
+        progress_thread = Thread.new { until finished; print "."; $stdout.flush; sleep 0.5; end; }
+        post_thread = Thread.new(binding()) do |b|
+          status = base.post(post, :source => Twitter::SourceName)
+          finished = true
+        end
+        post_thread.join
+        progress_thread.join
+        say "Got it! New twitter created at: #{status.created_at}\n"
+      end
+    end
+  end
+
+  mode 'befriend' do
+    description "Allows you to add a user as a friend"
+    argument('username') {
+      required
+      description 'username or id of twitterrer to befriend'
+    }
+
+    def run
+      do_work do
+        username = params['username'].value
+        base.create_friendship(username)
+        say "#{username} has been added as a friend. follow notifications with 'twitter follow #{username}'"
+      end
+    end
+  end
+
+  mode 'defriend' do
+    description "Allows you to remove a user from being a friend"
+    argument('username') {
+      required
+      description 'username or id of twitterrer to defriend'
+    }
+
+    def run
+      do_work do
+        username = params['username'].value
+        base.destroy_friendship(username)
+        say "#{username} has been removed from your friends"
+      end
+    end
+  end
+  
+  mode 'follow' do
+    description "Allows you to turn on notifications for a user"
+    argument('username') {
+      required
+      description 'username or id of twitterrer to follow'
+    }
+    
+    def run
+      do_work do
+        username = params['username'].value
+        base.follow(username)
+        say "You are now following notifications from #{username}"
+      end
+    end
+  end
+  
+  mode 'leave' do
+    description "Allows you to turn off notifications for a user"
+    argument('username') {
+      required
+      description 'username or id of twitterrer to leave'
+    }
+    
+    def run
+      do_work do
+        username = params['username'].value
+        base.leave(username)
+        say "You are no longer following notifications from #{username}"
+      end
+    end
+  end
+  
+  mode 'd' do
+    description "Allows you to direct message a user. The following are all valid examples from the command line:
+    $ twitter d jnunemaker 'yo homeboy'
+    $ twitter d jnunemaker yo homeboy
+    $ echo 'yo homeboy' | twitter d jnunemaker"
+    argument('username') {
+      required
+      description 'username or id of twitterrer to direct message'
+    }
+    
+    def run
+      do_work do
+        username = params['username'].value
+        post = ARGV.size > 1 ? ARGV.join(" ") : ARGV.shift
+        base.d(username, post)
+        say "Direct message sent to #{username}"
+      end
+    end
+  end
+  
+  mode 'timeline' do
+    description "Allows you to view your timeline, your friends or the public one"
+    argument( 'timeline' ) {
+      description 'the timeline you wish to see (friends, public, me)'
+      default 'friends'
+    }
+    
+    def run
+      do_work do
+        timeline = params['timeline'].value == 'me' ? 'user' : params['timeline'].value
+        options, since_id = {}, Configuration["#{timeline}_last_id"]
+        options[:since_id] = since_id unless since_id.blank?
+        statuses = base.timeline(timeline.to_sym, options)
+        username_length = statuses.collect { |s| s.user.screen_name }.max { |a,b| a.length <=> b.length }.length rescue 0
+        if statuses.size > 0
+          statuses.each do |s|
+            Tweet.create_from_tweet(current_account, s) if timeline != :public
+            say "#{CGI::unescapeHTML(s.user.screen_name.ljust(username_length+1))}: #{CGI::unescapeHTML(s.text)} on #{Time.parse(s.created_at).strftime('%b %d at %l:%M%P')}"
+          end
+          Configuration["#{timeline}_last_id"] = statuses.first.id
+        else
+          say 'Nothing new since your last check'
+        end
+      end
+    end
+  end
+  
+  mode 'replies' do
+    description 'Allows you to view all @replies at you'
+    
+    def run
+      do_work do
+        options, since_id = {}, Configuration["replies_since_id"]
+        options[:since_id] = since_id if !since_id.blank?
+        replies = base.replies(options)
+        username_length = replies.collect { |s| s.user.screen_name }.max { |a,b| a.length <=> b.length }.length rescue 0
+        if replies.size > 0 
+          replies.each do |s|
+            say "#{CGI::unescapeHTML(s.user.screen_name.ljust(username_length+1))}: #{CGI::unescapeHTML(s.text)} on #{Time.parse(s.created_at).strftime('%b %d at %l:%M%P')}"
+          end
+          Configuration["replies_since_id"] = replies.first.id
+        else
+          say 'No new replies since your last check'
+        end
+      end
+    end
+  end
+}

data/lib/twitter/direct_message.rb

@@ -0,0 +1,22 @@
+module Twitter
+  class DirectMessage
+    include EasyClassMaker
+    
+    attributes :id, :text, :sender_id, :recipient_id, :created_at, :sender_screen_name, :recipient_screen_name
+    
+    class << self
+      # Creates a new status from a piece of xml
+      def new_from_xml(xml)
+        DirectMessage.new do |d|
+          d.id                    = (xml).at('id').innerHTML
+          d.text                  = (xml).get_elements_by_tag_name('text').innerHTML
+          d.sender_id             = (xml).at('sender_id').innerHTML
+          d.recipient_id          = (xml).at('recipient_id').innerHTML
+          d.created_at            = (xml).at('created_at').innerHTML
+          d.sender_screen_name    = (xml).at('sender_screen_name').innerHTML
+          d.recipient_screen_name = (xml).at('recipient_screen_name').innerHTML
+        end
+      end
+    end
+  end
+end

data/lib/twitter/easy_class_maker.rb

@@ -0,0 +1,43 @@
+# This is pretty much just a macro for creating a class that allows
+# using a block to initialize stuff and to define getters and setters
+# really quickly.
+module Twitter
+  module EasyClassMaker
+    
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+    
+    module ClassMethods
+      # creates the attributes class variable and creates each attribute's accessor methods
+      def attributes(*attrs)
+        @@attributes = attrs
+        @@attributes.each { |a| attr_accessor a }
+      end
+      
+      # read method for attributes class variable
+      def self.attributes; @@attributes end
+    end
+    
+    # allows for any class that includes this to use a block to initialize
+    # variables instead of assigning each one seperately
+    # 
+    # Example: 
+    # 
+    # instead of...
+    # 
+    # s = Status.new
+    # s.foo = 'thing'
+    # s.bar = 'another thing'
+    #
+    # you can ...
+    # 
+    # Status.new do |s|
+    #   s.foo = 'thing'
+    #   s.bar = 'another thing'
+    # end
+    def initialize
+      yield self if block_given?
+    end
+  end
+end

data/lib/twitter/rate_limit_status.rb

@@ -0,0 +1,19 @@
+module Twitter
+  class RateLimitStatus
+    include EasyClassMaker
+    
+    attributes :reset_time_in_seconds, :reset_time, :remaining_hits, :hourly_limit
+    
+    class << self
+      # Creates a new rate limi status from a piece of xml
+      def new_from_xml(xml)
+        RateLimitStatus.new do |s|
+          s.reset_time_in_seconds = xml.at('reset-time-in-seconds').inner_html.to_i
+          s.reset_time            = Time.parse xml.at('reset-time').inner_html
+          s.remaining_hits        = xml.at('remaining-hits').inner_html.to_i
+          s.hourly_limit          = xml.at('hourly-limit').inner_html.to_i
+        end
+      end
+    end
+  end
+end

data/lib/twitter/status.rb

@@ -0,0 +1,22 @@
+module Twitter
+  class Status
+    include EasyClassMaker
+    
+    attributes :created_at, :id, :text, :user, :source, :truncated, :in_reply_to_status_id, :in_reply_to_user_id, :favorited
+    
+    # Creates a new status from a piece of xml
+    def self.new_from_xml(xml)
+      s = new
+      s.id                    = (xml).at('id').innerHTML
+      s.created_at            = (xml).at('created_at').innerHTML
+      s.text                  = (xml).get_elements_by_tag_name('text').innerHTML
+      s.source                = (xml).at('source').innerHTML
+      s.truncated             = (xml).at('truncated').innerHTML == 'false' ? false : true
+      s.favorited             = (xml).at('favorited').innerHTML == 'false' ? false : true
+      s.in_reply_to_status_id = (xml).at('in_reply_to_status_id').innerHTML
+      s.in_reply_to_user_id   = (xml).at('in_reply_to_user_id').innerHTML
+      s.user                  = User.new_from_xml(xml.at('user')) if (xml).at('user')
+      s
+    end
+  end
+end

data/lib/twitter/user.rb

@@ -0,0 +1,37 @@
+module Twitter
+  class User    
+    include EasyClassMaker
+    
+    attributes  :id, :name, :screen_name, :status, :location, :description, :url,
+                :profile_image_url, :profile_background_color, :profile_text_color, :profile_link_color, 
+                :profile_sidebar_fill_color, :profile_sidebar_border_color, :friends_count, :followers_count,
+                :favourites_count, :statuses_count, :utc_offset , :protected
+    
+    # Creates a new user from a piece of xml
+    def self.new_from_xml(xml)
+      u = new
+      u.id                           = (xml).at('id').innerHTML
+      u.name                         = (xml).at('name').innerHTML
+      u.screen_name                  = (xml).at('screen_name').innerHTML
+      u.location                     = (xml).at('location').innerHTML
+      u.description                  = (xml).at('description').innerHTML
+      u.url                          = (xml).at('url').innerHTML
+      u.profile_image_url            = (xml).at('profile_image_url').innerHTML          
+      
+      # optional, not always present
+      u.profile_background_color     = (xml).at('profile_background_color').innerHTML if (xml).at('profile_background_color')
+      u.profile_text_color           = (xml).at('profile_text_color').innerHTML if (xml).at('profile_text_color')
+      u.profile_link_color           = (xml).at('profile_link_color').innerHTML if (xml).at('profile_link_color')
+      u.profile_sidebar_fill_color   = (xml).at('profile_sidebar_fill_color').innerHTML if (xml).at('profile_sidebar_fill_color')
+      u.profile_sidebar_border_color = (xml).at('profile_sidebar_border_color').innerHTML if (xml).at('profile_sidebar_border_color')
+      u.friends_count                = (xml).at('friends_count').innerHTML if (xml).at('friends_count')
+      u.followers_count              = (xml).at('followers_count').innerHTML if (xml).at('followers_count')
+      u.favourites_count             = (xml).at('favourites_count').innerHTML if (xml).at('favourites_count')
+      u.statuses_count               = (xml).at('statuses_count').innerHTML if (xml).at('statuses_count')
+      u.utc_offset                   = (xml).at('utc_offset').innerHTML if (xml).at('utc_offset')
+      u.protected                    = (xml).at('protected').innerHTML == 'false' ? false : true if (xml).at('protected')
+      u.status                       = Status.new_from_xml(xml.at('status')) if (xml).at('status')
+      u
+    end
+  end
+end

data/lib/twitter/version.rb

@@ -0,0 +1,9 @@
+module Twitter #:nodoc:
+  module VERSION #:nodoc:
+    MAJOR = 0
+    MINOR = 3
+    TINY  = 0
+
+    STRING = [MAJOR, MINOR, TINY].join('.')
+  end
+end

data/lib/twitter.rb

@@ -0,0 +1,20 @@
+%w(uri cgi net/http yaml rubygems hpricot active_support).each { |f| require f }
+
+$:.unshift(File.join(File.dirname(__FILE__)))
+require 'twitter/version'
+require 'twitter/easy_class_maker'
+require 'twitter/base'
+require 'twitter/user'
+require 'twitter/status'
+require 'twitter/direct_message'
+require 'twitter/rate_limit_status'
+
+module Twitter
+  class Unavailable < StandardError; end
+  class CantConnect < StandardError; end
+  class BadResponse < StandardError; end
+  class UnknownTimeline < ArgumentError; end
+  class RateExceeded < StandardError; end
+
+  SourceName = 'twittergem'
+end

data/script/destroy

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+APP_ROOT = File.join(File.dirname(__FILE__), '..')
+
+begin
+  require 'rubigen'
+rescue LoadError
+  require 'rubygems'
+  require 'rubigen'
+end
+require 'rubigen/scripts/destroy'
+
+ARGV.shift if ['--help', '-h'].include?(ARGV[0])
+RubiGen::Base.use_component_sources! [:rubygems, :newgem, :newgem_theme, :test_unit]
+RubiGen::Scripts::Destroy.new.run(ARGV)

data/script/generate

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+APP_ROOT = File.join(File.dirname(__FILE__), '..')
+
+begin
+  require 'rubigen'
+rescue LoadError
+  require 'rubygems'
+  require 'rubigen'
+end
+require 'rubigen/scripts/generate'
+
+ARGV.shift if ['--help', '-h'].include?(ARGV[0])
+RubiGen::Base.use_component_sources! [:rubygems, :newgem, :newgem_theme, :test_unit]
+RubiGen::Scripts::Generate.new.run(ARGV)

data/script/txt2html

@@ -0,0 +1,74 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+begin
+  require 'newgem'
+rescue LoadError
+  puts "\n\nGenerating the website requires the newgem RubyGem"
+  puts "Install: gem install newgem\n\n"
+  exit(1)
+end
+require 'redcloth'
+require 'syntax/convertors/html'
+require 'erb'
+require File.dirname(__FILE__) + '/../lib/twitter/version.rb'
+
+version  = Twitter.git::VERSION::STRING
+download = 'http://rubyforge.org/projects/twitter'
+
+class Fixnum
+  def ordinal
+    # teens
+    return 'th' if (10..19).include?(self % 100)
+    # others
+    case self % 10
+    when 1: return 'st'
+    when 2: return 'nd'
+    when 3: return 'rd'
+    else    return 'th'
+    end
+  end
+end
+
+class Time
+  def pretty
+    return "#{mday}#{mday.ordinal} #{strftime('%B')} #{year}"
+  end
+end
+
+def convert_syntax(syntax, source)
+  return Syntax::Convertors::HTML.for_syntax(syntax).convert(source).gsub(%r!^<pre>|</pre>$!,'')
+end
+
+if ARGV.length >= 1
+  src, template = ARGV
+  template ||= File.join(File.dirname(__FILE__), '/../website/template.rhtml')
+  
+else
+  puts("Usage: #{File.split($0).last} source.txt [template.rhtml] > output.html")
+  exit!
+end
+
+template = ERB.new(File.open(template).read)
+
+title = nil
+body = nil
+File.open(src) do |fsrc|
+  title_text = fsrc.readline
+  body_text = fsrc.read
+  syntax_items = []
+  body_text.gsub!(%r!<(pre|code)[^>]*?syntax=['"]([^'"]+)[^>]*>(.*?)</\1>!m){
+    ident = syntax_items.length
+    element, syntax, source = $1, $2, $3
+    syntax_items << "<#{element} class='syntax'>#{convert_syntax(syntax, source)}</#{element}>"
+    "syntax-temp-#{ident}"
+  }
+  title = RedCloth.new(title_text).to_html.gsub(%r!<.*?>!,'').strip
+  body = RedCloth.new(body_text).to_html
+  body.gsub!(%r!(?:<pre><code>)?syntax-temp-(\d+)(?:</code></pre>)?!){ syntax_items[$1.to_i] }
+end
+stat = File.stat(src)
+created = stat.ctime
+modified = stat.mtime
+
+$stdout << template.result(binding)