lib/vedeu/repository/command_repository.rb in vedeu-0.0.21 vs lib/vedeu/repository/command_repository.rb in vedeu-0.0.22

@@ -1,19 +1,36 @@
 module Vedeu
-  class CommandRepository
+  module CommandRepository
     extend Repository
+    extend self
 
-    class << self
-      def by_keypress(input)
-        query(entity, :keypress, input)
-      end
+    def by_keypress(input)
+      query(entity, :keypress, input)
+    end
 
-      def by_keyword(input)
-        query(entity, :keyword, input)
-      end
+    def by_keyword(input)
+      query(entity, :keyword, input)
+    end
 
-      def entity
-        Command
-      end
+    def create(attributes)
+      super(Command.new(attributes))
     end
+
+    def entity
+      Command
+    end
   end
+
+  # :nocov:
+  module ClassMethods
+    def command(name, entity, options = {})
+      command_name = name.is_a?(Symbol) ? name.to_s : name
+
+      CommandRepository.create({
+        name:    command_name,
+        entity:  entity,
+        options: options
+      })
+    end
+  end
+  # :nocov:
 end