tweeter-bachue 0.0.1 → 0.0.2

data/README.md

@@ -44,13 +44,13 @@ Then please edit them:
 1. Edit `config/key.yml`
 2. Replace with your consumer key/secret, oauth token, oauth token secret (google if you need help)
 3. Then edit `config/proxies.yml`
-4. Add all your proxies info into this file. in this file, host, port and user are related to the ssh account, and gemset and path is related to `tweet` command
+4. Add all your proxies info into this file. in this file, host, port and user are related to the ssh account, and gemset and path is related to `tweet` command(To be security, password is totally forbidden)
 5. Don't forget to install and config `tweeter-bachue` in all your proxies in the same way
 6. You'd better delete `proxies.yml` if you haven't any proxy
 
 After all, try:
 
-	$ tweet "This is my 'hello world', sent from `tweet-bachue` gem"
+	$ tweet "This is my 'hello world', sent from 'tweet-bachue' gem"
 
 `tweeter-bachue` will try to tweet it by itself. if it failed for 3 times, will try the same command on each of your proxies by ssh one by one, until anyone of them puts "Succeed". when the `tweeter-bachue` in your proxies failed for 3 times, it will also do the same thing, try its proxies.
 
@@ -58,7 +58,7 @@ If all tries are failed, `tweeter-bachue` will write the tweet in a file which i
 
 ## Contributing
 
-1. Fork it from <https://gitcafe.com/bachue/tweeter.git>
+1. Fork it from `git@gitcafe.com:bachue/tweeter.git`
 2. Create your feature branch (`git checkout -b my-new-feature`)
 3. Commit your changes (`git commit -am 'Add some feature'`)
 4. Push to the branch (`git push origin my-new-feature`)

data/bin/tweet

@@ -4,44 +4,127 @@ require 'rubygems'
 require 'twitter'
 require 'yaml'
 require 'socket'
+require 'net/ssh'
 require 'active_support/all'
 
-def lost_file_name(path)
-  name = Time.now.to_f.to_s.sub('.', '') + '.tweet'
-  File.join(path, name)
-end
-
-if ARGV.first.downcase == 'proxy'
-  is_proxy = true 
-  ARGV.shift
-end
-
-tweet = ARGV.join(' ')
-
-exit if tweet.blank?
-
-root_dir = File.expand_path File.join(File.dirname(__FILE__), '..')
-config_dir = File.join(root_dir, 'config')
-lost_dir = File.join(root_dir, 'lost')
-client_info = YAML.load_file(File.join(config_dir, 'key.yml')).symbolize_keys
-proxies_info = YAML.load_file(File.join(config_dir, 'proxies.yml')).map(&:symbolize_keys) rescue []
-
-@client = Twitter::Client.new(client_info)
-retry_count = 0
-begin
-  retry_count += 1
-  @client.update(tweet)
-  puts 'Succeed'
-rescue
-  retry if retry_count < 3
-  puts "Failed to tweet by #{Socket.gethostname}, ask the proxies"
-  proxies_info = proxies_info.map { |proxy| [proxy[:host], proxy[:port], proxy[:user], proxy[:path], proxy[:gemset]] }
-  catch(:succ) {
-    proxies_info.each { |host, port, user, path, gemset|
-      puts input = "ssh #{"#{user}@" if user.present?}#{host} #{"-p #{port}" if port.present?} \"#{"rvm use #{gemset} &&" if gemset.present?} #{path.present? ? path : 'tweet' } proxy '#{tweet}'\""
-      puts output = system(input) rescue next
-      throw :succ if output =~ /Succeed/
+URL_REGEX = /((http|https):\/\/)?[a-z0-9]+([\-\.]{1}[a-z0-9]+)*\.[a-z]{2,5}(([0-9]{1,5})?\/.*)?/ix
+TWEET_MAX_LENGTH = 150 # The real value is 140, but set 150 here because the length limitation function is not exact
+SHORT_URL_LENGTH = 23
+
+def start!
+  exit_if_no_args!
+  set_proxy_if_it_is!
+  exit_if_invalid!(tweet)
+  twitter(tweet)
+end
+
+def twitter(tweet)
+  retry_count = 0
+  begin
+    retry_count += 1
+    client.update(tweet)
+    puts 'Succeed'
+  rescue
+    return if can_handle?
+    retry if retry_count < 3
+    puts "Failed to tweet by #{Socket.gethostname}, ask the proxies"
+    catch(:succ) {
+      proxies_info.each { |proxy|
+        remote_twitter proxy[:host], proxy[:port], proxy[:user], proxy[:path], proxy[:gemset] 
+      }
+      File.write(lost_file, tweet) unless $is_proxy # reach here means failed in all proxies
+      exit 1
     }
-    File.open(lost_file_name(lost_dir), 'w') { |f| f << tweet } unless is_proxy
+  end
+end
+
+def remote_twitter(host, port, user, path, gemset)
+  options = {:keys_only => true, :compression => true} # Password is forbidden here
+  options.merge! :port => port if port.present?
+  output = Net::SSH.start(host, user, :keys_only => true) { |ssh|
+    ssh.exec! <<-SHELL
+      #{"rvm use #{gemset} &&" if gemset.present?}
+      #{path.present? ? path : 'tweet'} proxy #{tweet.dump}
+    SHELL
   }
+  throw :succ if output =~ /Succeed/
+end
+
+def can_handle?
+  case $!.message
+  when /Status is a duplicate/
+    puts 'Succeed'
+    true
+  when /Could not authenticate you/
+    puts 'Failed to authenticate'
+    exit 4
+  when /Status is over \d+ characters/
+    puts 'Too long'
+    exit 3
+  else
+    false
+  end
+end
+
+def exit_if_invalid!(tweet)
+  unless valid?(tweet)
+    puts 'Too long'
+    exit 3
+  end
+end
+
+def valid?(tweet)
+  tweet.gsub(URL_REGEX, '*' * SHORT_URL_LENGTH).size <= TWEET_MAX_LENGTH
+end
+
+def exit_if_no_args!
+  exit 2 if ARGV.blank?
+end
+
+def set_proxy_if_it_is!
+  if ARGV.first.downcase == 'proxy'
+    $is_proxy = true 
+    ARGV.shift
+  end
+end
+
+def tweet
+  @tweet ||= ARGV.join(' ')
 end
+
+def lost_file
+  name = Time.now.to_f.to_s.sub('.', '') + '.tweet'
+  lost_path(name)
+end
+
+def root_path(filename = nil)
+  file = File.expand_path File.join(File.dirname(__FILE__), '..')
+  file = File.join(file, filename) if filename.present?
+  file
+end
+
+def config_path(filename = nil)
+  file = root_path('config')
+  file = File.join(file, filename) if filename.present?
+  file
+end
+
+def lost_path(filename = nil)
+  file = root_path('lost')
+  file = File.join(file, filename) if filename.present?
+  file
+end
+
+def client_info
+  @client_info ||= YAML.load_file(config_path('key.yml')).symbolize_keys
+end
+
+def proxies_info
+  @proxies_info ||= YAML.load_file(config_path('proxies.yml')).map(&:symbolize_keys) rescue []
+end
+
+def client
+  @client ||= Twitter::Client.new(client_info)
+end
+
+start!

data/lib/tweeter/version.rb

@@ -1,3 +1,3 @@
 module Tweeter
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/tweeter-bachue.gemspec

@@ -19,6 +19,7 @@ Gem::Specification.new do |gem|
   gem.add_dependency 'twitter'
   gem.add_dependency 'activesupport', '>= 3.0.0'
   gem.add_dependency 'rake'
+  gem.add_dependency 'net-ssh'
   gem.add_development_dependency 'pry'
   gem.add_development_dependency 'pry-doc'
   gem.add_development_dependency 'pry-nav'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tweeter-bachue
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-01-12 00:00:00.000000000 Z
+date: 2013-01-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: twitter
@@ -59,6 +59,22 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: pry
   requirement: !ruby/object:Gem::Requirement
@@ -157,7 +173,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -2088777374129425809
+      hash: 670411791857878890
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -166,7 +182,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -2088777374129425809
+      hash: 670411791857878890
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24