Merge pull request #20 from BooDoo/feature/jsonify-csv

jsonify: Support tweets.csv

bin/ebooks

@@ -2,6 +2,7 @@
 # encoding: utf-8
 
 require 'twitter_ebooks'
+require 'csv'
 
 $debug = true
 
@@ -154,6 +155,12 @@ STR
 
       tweets = []
       id = nil
+      if path.split('.')[-1] == "csv" #from twitter archive
+        csv_archive = CSV.read(path, :headers=>:first_row)
+        tweets = csv_archive.map do |tweet|
+          { text: tweet['text'], id: tweet['tweet_id'] }
+        end
+      else
         File.read(path).split("\n").each do |l|
           if l.start_with?('# ')
             id = l.split('# ')[-1]
@@ -166,6 +173,7 @@ STR
             tweets << tweet
           end
         end
+      end
 
       File.open(new_path, 'w') do |f|
         log "Writing #{tweets.length} tweets to #{new_path}"