model_probe 0.0.2 → 0.0.3

data/lib/model_probe/color.rb

@@ -0,0 +1,21 @@
+module ModelProbe
+  module Color
+    extend self
+
+    colors = {
+      :red     => 31,
+      :green   => 32,
+      :yellow  => 33,
+      :blue    => 34,
+      :magenta => 35,
+      :cyan    => 36,
+      :white   => 37
+    }
+
+    colors.each do |name, code|
+      define_method name do |text|
+      "\e[#{code}m#{text}\e[0m"
+      end
+    end
+  end
+end

data/lib/model_probe/version.rb

@@ -1,3 +1,3 @@
 module ModelProbe
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/model_probe.rb

@@ -1,27 +1,14 @@
 require "model_probe/version"
+require "model_probe/color"
 
 module ModelProbe
-  colors = {
-    :red     => 31,
-    :green   => 32,
-    :yellow  => 33,
-    :blue    => 34,
-    :magenta => 35,
-    :cyan    => 36,
-    :white   => 37
-  }
-
-  colors.each do |name, code|
-    define_method name do |text|
-    "\e[#{code}m#{text}\e[0m"
-    end
-  end
+  include ModelProbe::Color
 
   # Pretty prints column meta data for an ActiveModel.
   def probe
-    name_pad = columns.map{|c| c.name.length}.max + 1
-    type_pad = columns.map{|c| c.type.length}.max + 2
-    sql_type_pad = columns.map{|c| c.sql_type.length}.max + 1
+    name_pad = columns.map{ |c| c.name.length }.max + 1
+    type_pad = columns.map{ |c| c.type.length }.max + 2
+    sql_type_pad = columns.map{ |c| c.sql_type.length }.max + 1
 
     columns.sort{ |a, b| a.name <=> b.name }.map do |column|
       name = column.name
@@ -30,7 +17,8 @@ module ModelProbe
       print " "
       print blue(column.type.to_s.ljust(type_pad, "."))
       print magenta(column.sql_type.to_s.ljust(sql_type_pad))
-      print red("NULL") if column.null
+      column.null ? print(red("NULL")) : print("    ")
+      print " [#{column.default}]" if column.default
       puts
     end
     nil
@@ -43,7 +31,7 @@ module ModelProbe
 
     columns.sort{ |a, b| a.name <=> b.name }.map do |column|
       next if column.name =~ /^(created_at|updated_at)$/
-      hash[name][column.name] = "value"
+      hash[name][column.name] = column.default || "value"
     end
 
     puts hash.to_yaml

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: model_probe
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -24,6 +24,7 @@ files:
 - README.md
 - Rakefile
 - lib/model_probe.rb
+- lib/model_probe/color.rb
 - lib/model_probe/version.rb
 - model_probe.gemspec
 homepage: ''