Optimizing api, fixed bug when no tasks are shown

lib/burden.rb

@@ -11,11 +11,7 @@ require 'burden/version'
 require 'burden/wrapper'
 
 module Burden
-  include Config::Helper
-
-  def wrap(name, &block)
-    Wrapper.new(name, block).execute
-  end
-
-  extend self
+  extend Config::Helper
+  extend Storage::Helper
+  extend Wrapper::Helper
 end

lib/burden/statistics.rb

@@ -16,7 +16,7 @@ module Burden
         ActiveRecord::Base.establish_connection(conf)
 
         begin
-          Storage.run.create(name: name, success: success, execution_time: execution_time, timestamp: timestamp)
+          Burden.runs.create(name: name, success: success, execution_time: execution_time, timestamp: timestamp)
         rescue
           log_to_stdout(:failed)
         end

lib/burden/storage.rb

@@ -1,21 +1,23 @@
 module Burden
   module Storage
-    def run
-      case Burden.config.storage
-      when :active_record, :activerecord
-        require 'burden/storage/active_record/run'
-        Burden::Storage::ActiveRecord::Run
-      when :mongoid
-        require 'burden/storage/mongoid/run'
-        Burden::Storage::Mongoid::Run
-      when :mongo_mapper, :mongomapper
-        require 'burden/storage/mongo_mapper/run'
-        Burden::Storage::MongoMapper::Run
-      else
-        raise Exception.new("Unknown storage: #{storage}")
+    module Helper
+      def runs
+        @storage_class ||= begin
+          case Burden.config.storage
+          when :active_record, :activerecord
+            require 'burden/storage/active_record/run'
+            Burden::Storage::ActiveRecord::Run
+          when :mongoid
+            require 'burden/storage/mongoid/run'
+            Burden::Storage::Mongoid::Run
+          when :mongo_mapper, :mongomapper
+            require 'burden/storage/mongo_mapper/run'
+            Burden::Storage::MongoMapper::Run
+          else
+            raise Exception.new("Unknown storage: #{storage}")
+          end
+        end
       end
     end
-
-    extend self
   end
 end

lib/burden/version.rb

@@ -1,3 +1,3 @@
 module Burden
-  VERSION = '0.1.1'
+  VERSION = '0.1.2'
 end

lib/burden/wrapper.rb

@@ -54,5 +54,11 @@ module Burden
     def save_statistics
       Statistics.new(name: name, success: success, execution_time: execution_time, timestamp: timestamp).save
     end
+
+    module Helper
+      def wrap(name, &block)
+        Wrapper.new(name, block).execute
+      end
+    end
   end
 end