kolla 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 132a825d42a4bea791e2e5e689d6d06ae3defd3d452ecf41220d9079e63ec475
-  data.tar.gz: 8db516431150f3b81246d8042f4a8d7fe3a7cf0ede830d2374589a48ba1224af
+  metadata.gz: e9f49f3b572118301b672c4352947ca6cda45b9c174e8b846ec6e3707d5d5320
+  data.tar.gz: 70332827226bda106ebc84029a6fc5bf8ad076a9786f281f238ba6506ed8b25d
 SHA512:
-  metadata.gz: 431e7bbeb1c5fcb61a13c3ef9007c19b7dafaede6228c5391ba102146f58adc00c867a7c26897887daee25cfe85e3568c2f8509f479ca9b936f0c360b5d9b189
-  data.tar.gz: d6159da48b3f8e3e2ae794666313ac3a0e20d2e1609b70591fee854cca244bd858e095343f18435a6fbffb2677abffa5788e510978cdfabe71ea1df382b1d90c
+  metadata.gz: 975308aca1e8ebfb2442296ef663c2e4cf4ad9a2c4ba6e8abc37e1bfb8d12ac75c63aa94ca99c66e9d1a5ca0d40e255e4451280628594543083a00388c1b702b
+  data.tar.gz: ba2c479eb4e04f3347e3e8e919a9c330e1dd36d1586047ca2a819974960ac9559c46430ce1d6939ac21adc0deb8911ef6bd04af366eb2cfc27bad6dd2c88a749

data/bin/kolla

@@ -1,4 +1,4 @@
 #!/usr/bin/env ruby
 
 require 'kolla'
-ARGV[0].nil? ? Kolla.run : Kolla.run(ARGV[0])
+ARGV.empty? ? Kolla.run : Kolla.run(ARGV)

data/lib/kolla/parser.rb

@@ -2,20 +2,40 @@
 
 module Kolla
   class Parser
-    def initialize(file_path)
-      @file_path = file_path
+    DEFAULT_FILE = File.join('db', 'schema.rb').freeze
+
+    def initialize(file_path: DEFAULT_FILE, options: {})
+      @file_path = File.expand_path(file_path)
+      @options = options
     end
 
-    def process
+    def run!
       content = File.read(file_path)
-      matches = content.scan(/create_table\s+"(\w+)"/)
-      matches.map(&:first).reject { |table| table.start_with?('action_', 'active_') }
+      if options[:model_name]
+        extract_fields(content, options[:model_name])
+      else
+        extract_tables(content)
+      end
     rescue Errno::ENOENT
-      raise "Schema file_path not found at #{file_path}"
+      Kolla.logger.warn "Schema file_path not found at #{file_path}"
     end
 
     private
 
-    attr_reader :file_path
+    def extract_tables(schema)
+      matches = schema.scan(/create_table\s+"(\w+)"/)
+      matches.map(&:first).reject { |table| table.start_with?('action_', 'active_') }
+    end
+
+    def extract_fields(schema, table_name)
+      table_regex = /create_table "#{table_name}".*?do \|t\|(.*?)end/m
+      match = schema.match(table_regex)
+      return [] unless match
+
+      field_regex = /t\.(\w+)\s+"(\w+)"/
+      match[1].scan(field_regex).map { |type, name| "#{name} (#{type})" }
+    end
+
+    attr_reader :file_path, :options
   end
 end

data/lib/kolla.rb

@@ -1,19 +1,22 @@
 # frozen_string_literal: true
 
+require 'logger'
+
 # Kolla parses a file and reports matching results.
 module Kolla
-  DEFAULT_FILE = 'db/schema.rb'
+  class << self
+    attr_accessor :logger
+  end
 
-  def self.run(file_path = DEFAULT_FILE)
-    absolute_path = File.expand_path(file_path)
+  self.logger = Logger.new($stdout)
 
-    unless File.exist?(absolute_path)
-      puts "File #{file_path} not found"
-      return
+  def self.run(args = [])
+    if !args[0].nil?
+      model_name = args[0]
+      puts Parser.new(options:{model_name:}).run!
+    else
+      puts Parser.new.run!
     end
-
-    parser = Parser.new(absolute_path)
-    puts parser.process
   end
 end
 

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: kolla
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Antony Sastre
 bindir: bin
 cert_chain: []
-date: 2025-02-03 00:00:00.000000000 Z
+date: 2025-02-15 00:00:00.000000000 Z
 dependencies: []
 email: antony.sastre@gmail.com
 executables: