lib/lionel/export.rb in lionel_richie-0.1.4 vs lib/lionel/export.rb in lionel_richie-0.1.5

@@ -1,11 +1,17 @@
 module Lionel
   class Export
     include Configurable
 
+    attr_reader :options
+
     config_accessor :google_doc_id, :trello_board_id
 
+    def initialize(options = {})
+      @options = options
+    end
+
     def data
       {
         trello_board_id: trello_board_id,
         google_doc_id: google_doc_id
       }
@@ -14,33 +20,44 @@
     def board
       @board ||= Trello::Board.find(trello_board_id)
     end
 
     def cards
-      cards ||= [].tap do |c|
-        # iterate over active lists rather
-        # than retrieving all historical cards;
-        # trello api returns association proxy
-        # that does not respond to "flatten"
-        board.lists.each do |list|
-          list.cards.each do |card|
-            c << card
-          end
-        end
-      end.map { |c| Lionel::ProxyCard.new(c) }
+      # iterate over active lists rather
+      # than retrieving all historical cards;
+      # trello api returns association proxy
+      # that does not respond to "flatten"
+      @cards ||= begin
+        case options.fetch('filter', 'open-lists')
+        when 'open-cards'
+          retrieve_open_cards
+        when 'open-lists'
+          retrieve_open_cards_in_open_lists
+        end.map { |c| Lionel::ProxyCard.new(c) }
+      end
     end
 
     def spreadsheet
       @spreadsheet ||= google_session.spreadsheet_by_key(google_doc_id)
     end
 
     def worksheet
-      @worksheet ||= Lionel::ProxyWorksheet.new(spreadsheet.worksheets[0])
+      @worksheet ||= get_worksheet
     end
 
+    def process
+      download
+
+      if options['print']
+        rows.each { |row| Lionel.logger.info row.inspect }
+      else
+        upload
+      end
+    end
+
     def download
-      puts "Exporting trello board '#{board.name}' (#{trello_board_id}) to " + "google doc '#{spreadsheet.title}' (#{google_doc_id})"
+      Lionel.logger.info "Exporting trello board '#{board.name}' (#{trello_board_id}) to " + "google doc '#{spreadsheet.title}' (#{google_doc_id})"
 
       start_row = 2
       rows = worksheet.size
 
       card_rows = {}
@@ -60,29 +77,15 @@
         row = rows + i + 1
         card_rows[row] = card
       end
 
       card_rows.each do |row, card|
-        Timeout.timeout(5) { sync_row(row, card) }
-      end
-    end
-
-    class CardMap
-      include Enumerable
-
-      attr_reader :cards, :worksheet
-
-      def initialize(cards, worksheet)
-        @cards, @worksheet = cards, worksheet
-      end
-
-      def each(&block)
-        card_rows.each(&block)
-      end
-
-      def card_rows
-        @card_rows ||= {}.tap do |map|
+        begin
+          Timeout.timeout(5) { sync_row(row, card) }
+        rescue Timeout::Error, Trello::Error => e
+          Lionel.logger.warn e.inspect
+          Lionel.logger.warn card.inspect
         end
       end
     end
 
     def upload
@@ -92,11 +95,11 @@
     def rows
       worksheet.rows
     end
 
     def sync_row(row, card)
-      puts "row[#{row}] : #{card.name}"
+      Lionel.logger.info "row[#{row}] = #{card.name}"
 
       worksheet["B",row] = card.id
 
       # Card link
       worksheet["C",row] = card.link
@@ -131,28 +134,67 @@
       # Estimate
       worksheet["L",row] = card.estimate
 
       # Due Date
       worksheet["M",row] = card.due_date
-
-    rescue Trello::Error => e
-      puts e.inspect
-      puts card.inspect
     end
 
     def authenticate
       return if @authenticated
+      authenticate_trello
+      authenticate_google
+      @authenticated
+    end
+
+    def authenticate_trello
       trello_session.configure
+      @board = Trello::Board.find(trello_board_id)
+    end
+
+    def authenticate_google
       google_session
-      @authenticated
+      @worksheet = Lionel::ProxyWorksheet.new(spreadsheet.worksheets[0])
     end
 
     def google_session
       @google_session ||= GoogleDrive.login_with_oauth(configuration.google_token)
     end
 
     def trello_session
       @trello_session ||= TrelloAuthentication.new
+    end
+
+    def retrieve_open_cards
+      board.cards(filter: :open)
+    end
+
+    def retrieve_open_cards_in_open_lists
+      [].tap do |c|
+        board.lists(filter: :open).each do |list|
+          list.cards(filter: :open).each do |card|
+            c << card
+          end
+        end
+      end
+    end
+
+    class CardMap
+      include Enumerable
+
+      attr_reader :cards, :worksheet
+
+      def initialize(cards, worksheet)
+        @cards, @worksheet = cards, worksheet
+      end
+
+      def each(&block)
+        card_rows.each(&block)
+      end
+
+      def card_rows
+        @card_rows ||= {}.tap do |map|
+        end
+      end
     end
 
   end
 end