lib/arxiv/references/ArxivReferences.rb in arxiv-references-0.1.4 vs lib/arxiv/references/ArxivReferences.rb in arxiv-references-0.1.5

@@ -9,20 +9,25 @@
 
 module ArxivReferences
   class CLI < Thor
     include ArxivUtil
     desc 'url', 'Extract references from arxiv URL'
+    option 'work_dir', type: :string, aliases: '-work', desc: 'Set working dir(default: /tmp)'
     def url(urlName)
-      puts ArxivUtil.fetchFromUrl(urlName)
+      puts ArxivUtil.fetchFromUrl(urlName, work_dir)
     end
 
     desc 'id', 'Extract references from Arxiv id'
+    option 'work_dir', type: :string, aliases: '-work', desc: 'Set working dir(default: /tmp)'
     def arxivid(idName)
-      puts ArxivUtil.fetchFromArxivId(idName)
+      work_dir = options['work_dir'].nil? ? '/tmp' : options['work_dir']
+      puts ArxivUtil.fetchFromArxivId(idName, work_dir)
     end
 
     desc 'pdfurl', 'Extract references from pdf URL'
+    option 'work_dir', type: :string, aliases: '-work', desc: 'Set working dir(default: /tmp)'
     def pdfurl(pdfUrlName)
-      puts ArxivUtil.fetchFromPdfUrl(pdfUrlName)
+      work_dir = options['work_dir'].nil? ? '/tmp' : options['work_dir']
+      puts ArxivUtil.fetchFromPdfUrl(pdfUrlName, work_dir)
     end
   end
 end