Merge pull request #3 from orlando/master

[rbdr/lyricli.rb] / lib / lyricli.rb

diff --git a/lib/lyricli.rb b/lib/lyricli.rb
index 041959b505a486eb5f0d196a08ee51cdf4217fbc..73c5811a0e7a102d39f34ea2057407efb050a8c7 100755 (executable)
--- a/lib/lyricli.rb
+++ b/lib/lyricli.rb
@@ -1,4 +1,5 @@
 require 'uri'
+require 'cgi'
 require 'net/http'
 require 'multi_json'
 require 'nokogiri'
@@ -59,18 +60,37 @@ module Lyricli
   # Enables a source via the Source Manager
   def self.enable(source_name)
     source_manager = SourceManager.new
-    source_manager.enable(source_name)
+    begin
+      source_manager.enable(source_name)
+    rescue Exceptions::UnknownSourceError
+      "There is no such Source"
+    end
   end
 
   # Disables a source via the Source Manager
   def self.disable(source_name)
     source_manager = SourceManager.new
-    source_manager.disable(source_name)
+    begin
+      source_manager.disable(source_name)
+    rescue Exceptions::UnknownSourceError
+      "There is no such Source"
+    end
   end
 
   # Resets all configuration for a source via the Source Manager
   def self.reset(source_name)
     source_manager = SourceManager.new
-    source_manager.reset(source_name)
+    begin
+      source_manager.reset(source_name)
+    rescue Exceptions::UnknownSourceError
+      "There is no such Source"
+    end
+  end
+
+  # Returns the root of the Gem.
+  #
+  # @return [String] the root path for this gem
+  def self.root
+    File.expand_path('../..',__FILE__)
   end
 end