lib/arxiv.rb in arxiv-0.0.3 vs lib/arxiv.rb in arxiv-0.0.4

@@ -18,16 +18,32 @@
   end
 
   ID_FORMAT = /^\d{4}\.\d{4}(?:v\d+)?$/
 
   def self.get(id)
+
+    id = parse_arxiv_id(id)
+
     raise Arxiv::Error::MalformedId, "Manuscript ID format is invalid" unless id =~ ID_FORMAT
 
     url = ::URI.parse("http://export.arxiv.org/api/query?id_list=#{id}")
     response = ::Nokogiri::XML(open(url)).remove_namespaces!
     manuscript = Arxiv::Manuscript.parse(response.to_s, single: id)
 
     raise Arxiv::Error::ManuscriptNotFound, "Manuscript #{id} doesn't exist on arXiv" if manuscript.title.nil?
     manuscript
+  end
+
+  private
+
+  def self.parse_arxiv_id(id)
+    if id =~ ID_FORMAT
+      id
+    elsif id =~ /arxiv.org/
+      match = id.match(/[^\/]+$/)
+      match[0] if match
+    else
+      id
+    end
   end
 
 end