Add convert command which export .enex to Boostnote

Author: asmsuechan

lib/ever2boost/cli.rb

@@ -1,16 +1,29 @@
 require 'thor'
 require 'ever2boost/evernote_authorizer'
+require 'ever2boost/enex_converter'
+require 'ever2boost/util'
 
 module Ever2boost
   class CLI < Thor
     DEFAULT_OUTPUT_DIR = "#{ENV['HOME']}/evernote_storage"
 
     desc "import", "import from evernote"
-    option :directory, aliases: :d, banner: 'DIRCTORY_PATH', desc: "make Boostnote storage in the directory\ndefault: ~/evernote_storage"
+    option :directory, aliases: :d, banner: 'DIRCTORY_PATH', desc: "make Boostnote storage in the directory default: ~/evernote_storage"
     def import
       output_dir = options[:directory] || DEFAULT_OUTPUT_DIR
+      Util.make_output_dir(output_dir)
       developer_token = ask('DEVELOPER_TOKEN:')
       EvernoteAuthorizer.new(developer_token).import(output_dir)
     end
+
+    desc "convert", "convert fron .enex"
+    option :directory, aliases: :d, banner: 'DIRCTORY_PATH', desc: "make Boostnote storage in the directory default: ~/evernote_storage"
+    def convert(path)
+      output_dir = options[:directory] || DEFAULT_OUTPUT_DIR
+      Util.make_output_dir(output_dir)
+      enex = File.read(path)
+      filename = File.basename(path, ".enex")
+      EnexConverter.convert(enex, output_dir, filename)
+    end
   end
 end

lib/ever2boost/cson_generator.rb

@@ -2,14 +2,13 @@ module Ever2boost
   class CsonGenerator
     class << self
       def build(folder_hash, note)
-        md_note_content = MdConverter.convert(note.content)
         cson = <<-EOS
 type: "MARKDOWN_NOTE"
 folder: "#{folder_hash}"
 title: "#{note.title}"
 content: '''
   # #{note.title}
-  #{md_note_content}
+  #{note.md_content}
 '''
 tags: []
 isStarred: false
@@ -23,7 +22,6 @@ def timestamp
       end
 
       def output(folder_hash, note, output_dir)
-        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(self.build(folder_hash, note))
         end

lib/ever2boost/enex_converter.rb

@@ -0,0 +1,22 @@
+require 'rexml/document'
+
+module Ever2boost
+  class EnexConverter
+    class << self
+      def convert(enex, output_dir, filename)
+        en_notes = []
+        REXML::Document.new(Ever2boost::MdConverter.convert(enex)).elements['en-export'].each { |el| en_notes.push(el.to_s) }
+        en_notes.delete("\n")
+        notes = en_notes.map { |note|
+          title = REXML::Document.new(note).elements['note/title'].text
+          Note.new(title: title, content: "<div>#{REXML::Document.new(note).elements['note/content/text()'].value.sub(/.+\n/, '').sub(/.+\n/, '')}</div>")
+        }
+        notebook_list = [NoteList.new(title: filename)]
+        JsonGenerator.output(notebook_list, output_dir)
+        notes.each do |note|
+          CsonGenerator.output(notebook_list.first.hash, note, output_dir)
+        end
+      end
+    end
+  end
+end

lib/ever2boost/md_converter.rb

@@ -1,5 +1,4 @@
 require 'rexml/document'
-require 'byebug'
 
 module Ever2boost
   class MdConverter
@@ -8,11 +7,15 @@ def convert(note_content)
         en_note = nil
         # TODO: convert evernote xml to md
         REXML::Document.new(note_content).elements.each('*') {|el| en_note = el.to_s || en_note + el.to_s }
-        en_note.gsub(/<en-note>/, '')
-               .gsub(/<\/en-note>/, '')
-               .gsub(/<div>/, '')
-               .gsub(/<\/div>/, '')
-               .gsub(/<br\/>/, '\n\r')
+        if en_note.nil?
+          note_content
+        else
+          en_note.gsub(/<en-note>/, '')
+                 .gsub(/<\/en-note>/, '')
+                 .gsub(/<div>/, '')
+                 .gsub(/<\/div>/, '')
+                 .gsub(/<br\/>/, '\n\r')
+        end
       end
     end
   end

lib/ever2boost/note.rb

@@ -10,5 +10,9 @@ def initialize(title: nil, content: nil, notebook_guid: nil)
       @notebook_guid = notebook_guid
       @file_name = SecureRandom.hex(DEFAULT_BYTES_NUMBER)
     end
+
+    def md_content
+      MdConverter.convert(self.content)
+    end
   end
 end

lib/ever2boost/util.rb

@@ -1,6 +1,9 @@
 module Ever2boost
   class Util
     class << self
+      def make_output_dir(output_dir)
+        FileUtils.mkdir_p("#{output_dir}/notes") unless FileTest.exist?("#{output_dir}/notes")
+      end
     end
   end
 end