kolla 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aa7af9c82815bce009d2214337606318aaf9af44fa9874fbbd70793926ab9aca
-  data.tar.gz: 58d03e71d4ebb50a0ff1bdef9d33dfebb0cd9f3199d2a7f2e5212f90d3df4742
+  metadata.gz: 98522a2e33c960997f0f916d48c56aeb29fdade6f153e5b1e6098b769b4cf6a4
+  data.tar.gz: 15ac9b7c46afb68861a8243155a30ac825c5e6071bc4fe567722b42edf0fc963
 SHA512:
-  metadata.gz: eaf1ca3aaf552e86d5812f2a4f11be50a9766c5cc2e1fa862a77c74f3930381aa01e932cbf660d51bac053fe96b12d819da6c07f5ef0c03af599c547f2206745
-  data.tar.gz: 46903fc9c6446b1f51c8896ae88140e88ad9483e62336aa2dc1e9e3a95f3d994e61de5bfa09f15e8bce88a42d0faec91506cc20499f953733871c26edbd78ee5
+  metadata.gz: f51517d99c73fdb2748ce884c8ebe7959771f586dcc95e10e4d56434a530ec64b7a9a6504835e2097e5d36064efcdbb6a338119b57e475573e5c29141f5c11db
+  data.tar.gz: 1a632a31c1667b7fad4b37693f4241fd0ed0a77a44958f50ef1dc2f5e7066c10c94dcc4f40548d66ca459561275e30805eaed0ad6b5cbdd4e0d16457f0e60c2d

data/lib/kolla/parser.rb

@@ -5,6 +5,9 @@ module Kolla
     DEFAULT_FILE = File.join("db", "schema.rb").freeze
     IGNORED_PREFIXES = %w[action_ active_].freeze
 
+    Field = Struct.new(:name, :type, :constraints, keyword_init: true)
+    Index = Struct.new(:columns, :unique, keyword_init: true)
+
     def initialize(file_path: DEFAULT_FILE)
       @file_path = File.expand_path(file_path)
       @content = File.read(@file_path)
@@ -26,32 +29,61 @@ module Kolla
       block = match[1]
       fields = parse_fields(block)
       indexes = parse_indexes(block)
-      fields + indexes
+
+      format_output(name, fields, indexes)
     end
 
     private
 
     def parse_fields(block)
       field_regex = /t\.(\w+)\s+"(\w+)"(.*)/
-      block.scan(field_regex).map do |type, name, rest|
+      block.scan(field_regex).filter_map do |type, name, rest|
         next if type == "index"
 
-        parts = [type]
-        parts << "null: false" if rest.include?("null: false")
+        constraints = []
+        constraints << "null: false" if rest.include?("null: false")
         if (default_match = rest.match(/default:\s*("(?:[^"\\]|\\.)*"|[\w.]+)/))
-          parts << "default: #{default_match[1]}"
+          constraints << "default: #{default_match[1]}"
         end
-        "#{name} (#{parts.join(', ')})"
-      end.compact
+        Field.new(name: name, type: type, constraints: constraints)
+      end
     end
 
     def parse_indexes(block)
       index_regex = /t\.index\s+\[([^\]]+)\](.*)$/
       block.scan(index_regex).map do |columns, rest|
         cols = columns.scan(/"(\w+)"/).flatten
-        unique = rest.include?("unique: true") ? " (unique)" : ""
-        "index: [#{cols.join(', ')}]#{unique}"
+        Index.new(columns: cols, unique: rest.include?("unique: true"))
       end
     end
+
+    def format_output(table_name, fields, indexes)
+      return [] if fields.empty? && indexes.empty?
+
+      name_width = fields.map { |f| f.name.length }.max
+      type_width = fields.map { |f| f.type.length }.max
+
+      lines = []
+      lines << table_name
+      lines << "\u2500" * [name_width + type_width + 5, table_name.length].max
+
+      fields.each do |f|
+        line = "  #{f.name.ljust(name_width)}  #{f.type.ljust(type_width)}"
+        line << "  #{f.constraints.join(', ')}" unless f.constraints.empty?
+        lines << line
+      end
+
+      unless indexes.empty?
+        lines << ""
+        lines << "  Indexes:"
+        indexes.each do |idx|
+          line = "    [#{idx.columns.join(', ')}]"
+          line << " (unique)" if idx.unique
+          lines << line
+        end
+      end
+
+      lines
+    end
   end
 end

data/lib/kolla/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Kolla
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/kolla.rb

@@ -1,5 +1,9 @@
 # frozen_string_literal: true
 
+if RUBY_VERSION < "3.0"
+  abort "kolla requires Ruby 3.0 or later (you have #{RUBY_VERSION})"
+end
+
 require_relative "kolla/version"
 require_relative "kolla/errors"
 require_relative "kolla/parser"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: kolla
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Antony Sastre