Fix bugs in streaming code

mislav · May 8, 2012 · 8a107df · 8a107df
1 parent 56171d5
commit 8a107df
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/lib/t/stream.rb b/lib/t/stream.rb
@@ -29,18 +29,20 @@ def all
         if options['csv']
           say STATUS_HEADINGS.to_csv
         elsif options['long'] && STDOUT.tty?
-          say STATUS_HEADINGS.size.times.map do |index|
+          headings = STATUS_HEADINGS.size.times.map do |index|
             STATUS_HEADINGS_FORMATTING[index] % STATUS_HEADINGS[index]
           end
+          print_table([headings])
         end
       end
       client.on_timeline_status do |status|
         if options['csv']
           print_csv_status(status)
         elsif options['long']
-          print_table build_long_status(status).each_with_index.map do |element, index|
+          array = build_long_status(status).each_with_index.map do |element, index|
             STATUS_HEADINGS_FORMATTING[index] % element
           end
+          print_table([array])
         else
           print_status(status)
         end
@@ -74,9 +76,10 @@ def search(keyword, *keywords)
         if options['csv']
           print_csv_status(status)
         elsif options['long']
-          print_table build_long_status(status).each_with_index.map do |element, index|
+          array = build_long_status(status).each_with_index.map do |element, index|
             STATUS_HEADINGS_FORMATTING[index] % element
           end
+          print_table([array])
         else
           print_status(status)
         end
@@ -100,9 +103,10 @@ def timeline
         if options['csv']
           print_csv_status(status)
         elsif options['long']
-          print_table build_long_status(status).each_with_index.map do |element, index|
+          array = build_long_status(status).each_with_index.map do |element, index|
             STATUS_HEADINGS_FORMATTING[index] % element
           end
+          print_table([array])
         else
           print_status(status)
         end
@@ -120,16 +124,20 @@ def users(screen_name, *screen_names)
         if options['csv']
           say STATUS_HEADINGS.to_csv
         elsif options['long'] && STDOUT.tty?
-          say STATUS_HEADINGS.size.times.map do |index|
+          headings = STATUS_HEADINGS.size.times.map do |index|
             STATUS_HEADINGS_FORMATTING[index] % STATUS_HEADINGS[index]
           end
+          print_table([headings])
         end
       end
       client.on_timeline_status do |status|
         if options['csv']
           print_csv_status(status)
         elsif options['long']
-          print_table(build_long_status(status))
+          array = build_long_status(status).each_with_index.map do |element, index|
+            STATUS_HEADINGS_FORMATTING[index] % element
+          end
+          print_table([array])
         else
           print_status(status)
         end