Modify to download notes in all of the notebooks

Author: asmsuechan

lib/ever2boost/cli.rb

@@ -3,10 +3,18 @@
 
 module Ever2boost
   class CLI < Thor
+    DEFAULT_OUTPUT_DIR = "#{ENV['HOME']}/ever2boost_store"
+
     desc "import", "import from evernote"
     def import
-      developer_token = ask('DEVELOPER_TOKEN: ')
-      EvernoteAuthorizer.new(developer_token).fetch_notes(100)
+      developer_token = ask('DEVELOPER_TOKEN:')
+      EvernoteAuthorizer.new(developer_token).fetch_notes(DEFAULT_OUTPUT_DIR)
+    end
+
+    class << self
+      def tell(message)
+        tell(message)
+      end
     end
   end
 end

lib/ever2boost/cson_generator.rb

@@ -1,27 +1,25 @@
 module Ever2boost
   class CsonGenerator
     class << self
-      DEFAULT_DIR_PATH = "#{ENV['HOME']}/ever2boost_store"
-
-      def generate(folder_hash, note)
+      def generate(folder_hash, note, output_dir)
         timestamp = self.timestamp
         md_note_content = MdConverter.convert(note.content)
         cson = <<-EOS
-          type: "MARKDOWN_NOTE"
-          folder: "#{folder_hash}"
-          title: "#{note.title}"
-          content:
-          '''
-#{md_note_content}
-          '''
-          tags: []
-          isStarred: false
-          createdAt: "#{timestamp}"
-          updatedAt: "#{timestamp}"
+type: "MARKDOWN_NOTE"
+folder: "#{folder_hash}"
+title: "#{note.title}"
+content: '''
+  # #{note.title}
+  #{md_note_content}
+'''
+tags: []
+isStarred: false
+createdAt: "#{timestamp}"
+updatedAt: "#{timestamp}"
         EOS
 
-        FileUtils.mkdir_p("#{DEFAULT_DIR_PATH}/notes") unless FileTest.exist?("#{DEFAULT_DIR_PATH}/notes")
-        File.open("#{DEFAULT_DIR_PATH}/notes/#{note.file_name}.cson", "w") do |f|
+        FileUtils.mkdir_p("#{output_dir}/notes") unless FileTest.exist?("#{output_dir}/notes")
+        File.open("#{output_dir}/notes/#{note.file_name}.cson", "w") do |f|
           f.write(cson)
         end
       end

lib/ever2boost/evernote_authorizer.rb

@@ -6,7 +6,6 @@
 module Ever2boost
   class EvernoteAuthorizer
     EVERNOTE_HOST = "www.evernote.com"
-    DEFAULT_DIR_PATH = "#{ENV['HOME']}/ever2boost_store"
 
     attr_accessor :developer_token, :note_store
 
@@ -23,27 +22,37 @@ def initialize(developer_token)
       @note_store = note_store
     end
 
-    def fetch_notes(number_of_notes)
-      FileUtils.mkdir_p(DEFAULT_DIR_PATH) unless FileTest.exist?(DEFAULT_DIR_PATH)
+    def fetch_notes(output_dir)
+      FileUtils.mkdir_p(output_dir) unless FileTest.exist?(output_dir)
 
-      start_index = 0
       note_store = self.note_store
       notebook_list = note_store.listNotebooks(self.developer_token)
       notebook_guids = notebook_list.map(&:guid)
       lists = notebook_list.map { |nl| Ever2boost::NoteList.new(title: nl.name, hash: hash, guid: nl.guid) }
-      Ever2boost::JsonGenerator.generate_boostnote_json(lists)
-      filter = Evernote::EDAM::NoteStore::NoteFilter.new
-      filter.notebookGuid = notebook_guids.first
-      note_list = note_store.findNotes(self.developer_token, filter, start_index, number_of_notes)
-      # TODO: note_listのtitleからboostnote.jsonを作成する。
-      note_guids = note_list.notes.map(&:guid)
-      # TODO: assign the booleans
-      en_notes = note_guids.map {|note_guid| note_store.getNote(self.developer_token, note_guid, true, true, false, false)}
-      notes = en_notes.map {|note| Note.new(title: note.title, content: note.content, notebook_guid: note.notebookGuid)}
-      notes.each do |note|
-        lists.each do |list|
-          # TODO: break if note found
-          CsonGenerator.generate(list.hash, note) if list.guid == note.notebook_guid
+      Ever2boost::JsonGenerator.generate_boostnote_json(lists, output_dir)
+
+      notebook_guids.each do |notebook_guid|
+        filter = Evernote::EDAM::NoteStore::NoteFilter.new
+        filter.notebookGuid = notebook_guid
+
+        spec = Evernote::EDAM::NoteStore::NotesMetadataResultSpec.new(includeTitle: true, includeNotebookGuid: true)
+
+        # get latest 250 notes
+          note_counts_hash = note_store.findNoteCounts(self.developer_token, filter, true).notebookCounts
+          number_of_notes = note_counts_hash.values.last || 0
+        # TODO: display message like "ignored first #{(number_of_notes - 250).to_s} notes due to EvernoteAPI access limitation" if number_of_notes > 250
+        start_index = number_of_notes > 250 ? number_of_notes - 250 : 0
+        note_list = note_store.findNotesMetadata(self.developer_token, filter, start_index, 15, spec)
+
+        note_guids = note_list.notes.map(&:guid)
+        # TODO: assign the booleans
+        en_notes = note_guids.map {|note_guid| note_store.getNote(self.developer_token, note_guid, true, true, false, false)}
+        notes = en_notes.map {|note| Note.new(title: note.title, content: note.content, notebook_guid: note.notebookGuid)}
+        notes.each do |note|
+          lists.each do |list|
+            # TODO: break if note found
+            CsonGenerator.generate(list.hash, note, output_dir) if list.guid == note.notebook_guid
+          end
         end
       end
     end

lib/ever2boost/json_generator.rb

@@ -11,7 +11,7 @@ class JsonGenerator
     ].freeze
 
     class << self
-      def generate_boostnote_json(note_list)
+      def generate_boostnote_json(note_list, output_dir)
         folders = <<-EOS
         EOS
 
@@ -36,7 +36,7 @@ def generate_boostnote_json(note_list)
           }
         EOS
 
-        File.open("#{ENV['HOME']}/ever2boost_store/boostnote.json","w") do |f|
+        File.open("#{output_dir}/boostnote.json","w") do |f|
           f.write(json)
         end
       end