lib/arxiv/references/ArxivReferences.rb in arxiv-references-0.1.6.5 vs lib/arxiv/references/ArxivReferences.rb in arxiv-references-0.1.7.0

@@ -3,38 +3,32 @@
 require 'thor'
 require 'pathname'
 lib = Pathname.new(__FILE__).dirname.join().expand_path
 $:.unshift lib.to_s
 require 'myUtil'
+require 'json'
 
 
 module ArxivReferences
   class CLI < Thor
     include ArxivUtil
+    class_option 'work_dir', type: :string, aliases: 'Working dir', desc: 'Set working dir(default: /tmp)', default: '/tmp'
+    class_option 'dir', type: :boolean, aliases: 'Working in dir', desc: 'work to make dir or not(default: true)', default: true
+    class_option 'pdf', type: :boolean, aliases: 'Parse PDF', desc: 'fetch pdf information(defaut: true)', default: true
+
+
     desc 'url', 'Extract references from arxiv URL'
-    option 'work_dir', type: :string, aliases: '-work', desc: 'Set working dir(default: /tmp)'
-    option 'use_dir', type: :boolean, aliases: '-use_dir', desc: 'work to make dir or not(default: true)'
     def url(urlName)
-      work_dir = options['work_dir'].nil? ? '/tmp' : options['work_dir']
-      use_dir = options['use_dir'].nil? ? true : options['use_dir']
-      puts ArxivUtil.fetchFromUrl(urlName, work_dir, use_dir)
+      puts ArxivUtil.fetchFromUrl(urlName, options[:work_dir], options[:use_dir], options[:no_pdf]).to_json
     end
 
     desc 'id', 'Extract references from Arxiv id'
-    option 'work_dir', type: :string, aliases: '-work', desc: 'Set working dir(default: /tmp)'
-    option 'use_dir', type: :boolean, aliases: '-use_dir', desc: 'work to make dir or not(default: true)'
     def arxivid(idName)
-      work_dir = options['work_dir'].nil? ? '/tmp' : options['work_dir']
-      use_dir = options['use_dir'].nil? ? true : options['use_dir']
-      puts ArxivUtil.fetchFromArxivId(idName, work_dir, use_dir)
+      puts ArxivUtil.fetchFromArxivId(idName, options[:work_dir], options[:use_dir], options[:no_pdf]).to_json
     end
 
     desc 'pdfurl', 'Extract references from pdf URL'
-    option 'work_dir', type: :string, aliases: '-work', desc: 'Set working dir(default: /tmp)'
-    option 'use_dir', type: :boolean, aliases: '-use_dir', desc: 'work to make dir or not(default: true)'
     def pdfurl(pdfUrlName)
-      work_dir = options['work_dir'].nil? ? '/tmp' : options['work_dir']
-      use_dir = options['use_dir'].nil? ? true : options['use_dir']
-      puts ArxivUtil.fetchFromPdfUrl(pdfUrlName, work_dir, use_dir)
+      puts ArxivUtil.fetchFromPdfUrl(pdfUrlName, options[:work_dir], options[:use_dir]).to_json
     end
   end
 end