lib/arql/commands/models.rb in arql-0.1.10 vs lib/arql/commands/models.rb in arql-0.1.11

@@ -2,26 +2,39 @@
 
 module Arql::Commands
   module Models
     class << self
       def models
-        Terminal::Table.new do |t|
-          t << ['Table Name', 'Model Class', 'Abbr']
-          t << :separator
-          Arql::Definition.models.each do |definition|
-            t << [definition[:table], definition[:model].name, definition[:abbr] || '']
-          end
+        t = []
+        t << ['Table Name', 'Model Class', 'Abbr']
+        t << nil
+        Arql::Definition.models.each do |definition|
+          t << [definition[:table], definition[:model].name, definition[:abbr] || '']
         end
+        t
       end
 
+      def models_table
+        Terminal::Table.new do |t|
+          models.each { |row| t << (row || :separator) }
+        end
+      end
     end
+  end
 
-    Pry.commands.block_command 'models' do
-      puts
-      puts Models::models
-    end
+  Pry.commands.block_command 'm' do
+    puts
+    puts Models::models_table
+  end
 
-    Pry.commands.alias_command 'm', 'models'
-    Pry.commands.alias_command 'l', 'models'
-    Pry.commands.alias_command 'tables', 'models'
+  Pry.commands.alias_command 'l', 'm'
+end
+
+module Kernel
+  def models
+    Arql::Commands::Models::models
+  end
+
+  def tables
+    models
   end
 end