Add error message on error when call EvernoteAPI

Author: asmsuechan

lib/ever2boost/cli.rb

@@ -10,11 +10,5 @@ def import
       developer_token = ask('DEVELOPER_TOKEN:')
       EvernoteAuthorizer.new(developer_token).import(DEFAULT_OUTPUT_DIR)
     end
-
-    class << self
-      def tell(message)
-        tell(message)
-      end
-    end
   end
 end

lib/ever2boost/evernote_authorizer.rb

@@ -19,6 +19,8 @@ def initialize(developer_token)
       note_store = Evernote::EDAM::NoteStore::NoteStore::Client.new(note_store_protocol)
       @developer_token = developer_token
       @note_store = note_store
+    rescue => e
+      abort_with_message e
     end
 
     def fetch_notebook_list
@@ -52,6 +54,7 @@ def fetch_notes(filter)
     # Download the all of notes fron Evernote and generate Boostnote storage from it
     # TODO: move this method to CLI
     def import(output_dir)
+      puts 'processing...'
       FileUtils.mkdir_p(output_dir) unless FileTest.exist?(output_dir)
       notebook_list = self.notebook_list
 
@@ -70,6 +73,19 @@ def import(output_dir)
           end
         end
       end
+      puts 'Successfully finished!'
+    rescue => e
+      abort_with_message e
+    end
+
+    def abort_with_message(exception)
+      if exception.class == Evernote::EDAM::Error::EDAMUserException
+        abort "\e[31mError! Confirm your developer token\e[0m"
+      elsif exception.class == Evernote::EDAM::Error::EDAMSystemException
+        abort "\e[31mError! You reached EvernoteAPI rate limitation\e[0m"
+      else
+        raise exception
+      end
     end
   end
 end