Enhancement/show available

bin/cheatsheet

@@ -4,4 +4,4 @@ require "cheatsheet"
 
 args = ARGV.clone
 
-Cheatsheet::Client.fetch args
+Cheatsheet::Client.start args

lib/cheatsheet.rb

@@ -1,6 +1,7 @@
 require "cheatsheet/version"
 require "cheatsheet/client"
 require "net/http"
+require "json"
 
 module Cheatsheet
 

lib/cheatsheet/client.rb

@@ -1,16 +1,55 @@
+
 module Cheatsheet
   class Client
 
     SOURCE = "https://raw.githubusercontent.com/rstacruz/cheatsheets/gh-pages/"
+    FILES_SOURCE = "https://api.github.com/repos/rstacruz/cheatsheets/contents/"
+
+    def self.start(*args)
+        begin
+          self.render(self.fetch(*args))
+        rescue CheatSheetClientException => e
+          self.render e
+        end
+    end
 
     def self.fetch(*args)
       key = args[0].first
-      uri = URI(SOURCE + key + ".md")
 
-      begin
-        puts self.fetch_raw(uri)
-      rescue CheatSheetClientException => e
-        puts e.message
+      # Show available cheatsheets
+      if (key === '-a')
+        files_uri = URI(FILES_SOURCE)
+        begin
+          files = JSON.parse(self.fetch_raw(files_uri))
+          filter = ''
+
+          if args[0].size > 1
+            filter = args[0].last
+          end
+
+          mds = files.select { |elem|
+            elem['name'].end_with?(".md") && elem['name'].include?(filter)
+          }.map { |elem|
+            File.basename(elem['name'],File.extname(elem['name']))
+          }
+
+          if mds.size === 0
+            raise CheatSheetClientException.new "We don't have any cheatsheet matching your search query"
+          end
+
+          return mds
+        rescue CheatSheetClientException => e
+          raise CheatSheetClientException.new e.message
+        rescue JSON::ParserError
+          raise CheatSheetClientException.new "Try again later"
+        end
+      else
+        uri = URI(SOURCE + key + ".md")
+        begin
+          return self.fetch_raw(uri)
+        rescue CheatSheetClientException => e
+          raise CheatSheetClientException.new e.message
+        end
       end
     end
 
@@ -27,6 +66,10 @@ def self.fetch_raw(uri)
       end
     end
 
+    def self.render(string)
+      puts string
+    end
+
   end
 end
 

test/cheatsheet_test.rb

@@ -7,11 +7,23 @@ def test_that_it_has_a_version_number
   end
 
   def test_client_success
-    assert_match /Jollibee/, Cheatsheet::Client.fetch("ph-food-delivery")
+    assert_match (/Jollibee/), Cheatsheet::Client.fetch(["ph-food-delivery"])
   end
 
   def test_client_failure
-    assert_raises(CheatSheetClientException) { Cheatsheet::Client.fetch("us-food-delivery") }
+    assert_raises(CheatSheetClientException) { Cheatsheet::Client.fetch(["us-food-delivery"]) }
+  end
+
+  def test_client_search_finds
+    assert_includes Cheatsheet::Client.fetch(["-a", "xpath"]), 'xpath'
+  end
+
+  def test_client_search_doesnt_find
+    assert_raises(CheatSheetClientException) { Cheatsheet::Client.fetch(["-a", "xmen"]) }
+  end
+
+  def test_client_search_finds_correctly
+    refute_includes Cheatsheet::Client.fetch(["-a", "xpath"]), 'react'
   end
 
   def test_client_invalid_result