rjp-twittermoo 0.0.4 → 0.0.5

data/bin/twittermoo.rb

@@ -1,3 +1,5 @@
+#! /usr/bin/env ruby
+
 require 'rubygems'
 require 'twitter'
 require 'gdbm'
@@ -5,6 +7,13 @@ require 'sha1'
 require 'optparse'
 require 'socket'
 
+# TODO move this to something like log4r if they have it
+def log(x)
+    if $options[:verbose] then
+        puts *x
+    end
+end
+
 $options = {
     :host => 'localhost',
     :port => nil,
@@ -64,33 +73,45 @@ OptionParser.new do |opts|
 end.parse!
 
 if $options[:keyfile] then
-    puts "loading secret key from #{$options[:keyfile]}"
-    $options[:secret_key] = File.open($options[:keyfile]).read.chomp
+    log "loading secret key from #{$options[:keyfile]}"
+    begin
+        $options[:secret_key] = File.open($options[:keyfile]).read.chomp
+    rescue => e
+        $stderr.puts "Error getting secret key from #{$options[:keyfile]}: #{e}"
+        exit(1)
+    end
 end
 
 def send_message(x)
     if $options[:port].nil? then
         puts "! #{x}"
     else
-        begin
-            # irc_cat doesn't seem to like persistent connections
-            $socket = TCPSocket.new($options[:host], $options[:port])
-            if $options[:secret_key] then
-                x = "%/#{$options[:secret_key]}/% #{x}"
-            end
-            $socket.puts(x)
-            $socket.close
+        attempts = 5
+        loop do
+	        begin
+	            # irc_cat doesn't seem to like persistent connections
+	            $socket = TCPSocket.new($options[:host], $options[:port])
+	            if $options[:secret_key] then
+	                x = "%/#{$options[:secret_key]}/% #{x}"
+	            end
+	            $socket.puts(x)
+	            $socket.close
+	        rescue
+	            log "E $!"
+	            attempts = attempts - 1
+	            if attempts == 0 then
+	                log "too many failures, bailing for 120s"
+	                sleep 120
+	                attempts = 5
+	            else
+	                log "transient failure, sleeping for 30s"
+	                sleep 30
+	            end
+	        end
         end
     end
 end
 
-# TODO move this to something like log4r if they have it
-def log(x)
-    if $options[:verbose] then
-        puts *x
-    end
-end
-
 config = YAML::load(open($options[:config]))
 
 # allow settings of options from the config file

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: rjp-twittermoo
 version: !ruby/object:Gem::Version 
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors: 
 - Rob Partington
@@ -24,8 +24,8 @@ dependencies:
     version: 
 description: 
 email: zimpenfish@gmail.com
-executables: []
-
+executables: 
+- twittermoo.rb
 extensions: []
 
 extra_rdoc_files: []