dougsko-apod 0.0.3 → 0.1.0

data/VERSION

@@ -1 +1 @@
-0.0.3
+0.1.0

data/apod.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{apod}
-  s.version = "0.0.3"
+  s.version = "0.1.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["dougsko"]

data/bin/apod-download

@@ -29,4 +29,4 @@ opts.parse(ARGV)
 save_path = options["o"]
 
 apod = Apod.new
-apod.download(0, save_path)
+apod.pictures.first.download(save_path)

data/lib/apod.rb

@@ -5,6 +5,7 @@
 
 require 'rubygems'
 require 'open-uri'
+require 'httpclient'
 require 'hpricot'
 require 'lib/pic'
 
@@ -23,16 +24,21 @@ class Apod
         @pictures.collect{ |pic| pic.title }
     end
 
-    def download(index, save_path)
-        doc = Hpricot(open(@pictures[index].website))
-        link = "http://apod.nasa.gov/" + doc.at("//center/p[2]/a")["href"]
-
-        url = URI.parse(link)
-
-        url.open do |pic|
-            File.open(save_path, "w") do |file|
-                file << pic.read
+    def search(query)
+        if query.size > 60
+            return "ERROR: Query must be less than 60 characters"
+        end
+        @pictures = []
+        clnt = HTTPClient.new("http://antwrp.gsfc.nasa.gov/cgi-bin/apod/apod_search")
+        results = clnt.post('http://antwrp.gsfc.nasa.gov/cgi-bin/apod/apod_search', {'tquery' => query}).content
+        doc = Hpricot(results)
+        doc.search("//p").each do |para|
+            title = para.at("a[2]")
+            link = para.at("a")
+            if title != nil and link != nil
+                @pictures << Pic.new(title.innerHTML, link["href"])
             end
         end
+        @pictures.pop
     end
 end

data/lib/pic.rb

@@ -10,4 +10,16 @@ class Pic
         @title = title
         @website = website
     end
+
+    def download(save_path)
+        doc = Hpricot(open(self.website))
+        link = "http://apod.nasa.gov/" + doc.at("//center/p[2]/a")["href"]
+        url = URI.parse(link)
+
+        url.open do |pic|
+            File.open(save_path, "w") do |file|
+                file << pic.read
+            end
+        end
+    end
 end

data/spec/apod_spec.rb

@@ -13,10 +13,16 @@ describe "Apod" do
         @apod.titles.class.to_s.should == "Array"
     end
 
-    it "should download today's pic" do
-        @apod.download(0, "/tmp/apod.jpg")
+    it "search and download the first picture" do
+        @apod.search("earth")
+        @apod.pictures.first.download("/tmp/apod.jpg")
         `file /tmp/apod.jpg`.should match(/image data/)
         `rm /tmp/apod.jpg`
     end
 
+    it "tries to search with a query that is too big" do
+        @apod.search("A"*61).should match(/ERROR/)
+    end
+
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: dougsko-apod
 version: !ruby/object:Gem::Version 
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors: 
 - dougsko
@@ -57,6 +57,7 @@ files:
 - spec/spec_helper.rb
 has_rdoc: false
 homepage: http://github.com/dougsko/apod
+licenses: 
 post_install_message: 
 rdoc_options: 
 - --charset=UTF-8
@@ -77,7 +78,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.2.0
+rubygems_version: 1.3.5
 signing_key: 
 specification_version: 3
 summary: A library to work with NASA's astronomy pictures of the day.