belt 0.1.12 → 0.1.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 892f04b0175a10793ba63f3cf65182b3d3e380ab75446984004283b8b8cf7377
-  data.tar.gz: 701534fd954f5d444783d3adb085d0582c4f9d1b92bf4518249e677add64c1b7
+  metadata.gz: b3834960aad8c883c22d15acec6a320d860bd8b90acb3c7c74751c59ce49b959
+  data.tar.gz: 29f96098a7ac79c0f3eb62edd9950a9d1e95e2f56b219b3fcd024b19f3b4daac
 SHA512:
-  metadata.gz: '0682c1d964357b0109baca143e7c43fb8688a375aa84af0ba3ba2c46fbf0d56932a011b8eef6a0027dcf6b289f3eb6bc4617bcecd8717812246133b107563233'
-  data.tar.gz: 68d1e8b3a4311ca6bd3be6dbfeb48594efeb4afaf42500d27b7076e0068a8b21c8cd73a1e41a70c4ee6338e910dab96aaa8fdba09683c922027574141049a899
+  metadata.gz: 597c85007b7e8838f77605ec31de2b20bac85bf29ec20de371b85f03dd4a9eb20abba6be12f5b8f701e2c36ee02377b3488efa00292060f36b8bc956ef04d8ed
+  data.tar.gz: 4c24a2f2c554244010ec246bc337194f6616ed04394ff8aeea4d5aabe91c97adeac7aaad76fbfa9a812fa35397fc46770c347ee590a5d307e2073ee2e811ff13

data/CHANGELOG.md

@@ -1,5 +1,15 @@
 # Changelog
 
+## 0.1.13
+
+### Generator Extension API
+
+- Gems can now register generators by placing a file at `lib/belt/generators/<name>_generator.rb`.
+- Discovered automatically via `Gem.loaded_specs` — no manual registration needed.
+- Available as `belt generate <name>` / `belt destroy <name>`.
+- Generator contract: `.run(args)` (required), `.destroy(args)` (optional), `.description` (optional).
+- Help output includes gem-provided generators under "Gem Generators" section.
+
 ## Unreleased
 
 ## 0.1.11

data/lib/belt/cli/destroy_command.rb

@@ -2,6 +2,7 @@
 
 require 'fileutils'
 require_relative 'app_detection'
+require_relative 'generator_registry'
 require_relative 'tables_command'
 require_relative '../inflector'
 
@@ -20,13 +21,22 @@ module Belt
           exit 0
         end
 
-        unless GENERATORS.include?(generator)
+        unless GENERATORS.include?(generator) || GeneratorRegistry.generator_names.include?(generator)
           puts "Unknown generator: '#{generator}'"
-          puts "Available: #{GENERATORS.uniq.join(', ')}"
+          puts "Available: #{(GENERATORS + GeneratorRegistry.generator_names).uniq.join(', ')}"
           puts "\nRun 'belt destroy --help' for usage information."
           exit 1
         end
 
+        # Delegate to gem-provided generator if not a built-in
+        unless GENERATORS.include?(generator)
+          klass = GeneratorRegistry.find(generator)
+          return klass.destroy(args) if klass.respond_to?(:destroy)
+
+          puts "Generator '#{generator}' does not support destroy."
+          exit 1
+        end
+
         # Normalize: resource is an alias for scaffold
         generator = 'scaffold' if generator == 'resource'
 

data/lib/belt/cli/frontend_env_map.rb

@@ -181,7 +181,7 @@ module Belt
         return str if str.empty?
         return str unless str.match?(/[\s\#"'\\$`]/)
 
-        %("#{str.gsub('\\', '\\\\').gsub('"', '\\"')}")
+        %("#{str.gsub('\\') { '\\\\' }.gsub('"') { '\\"' }}")
       end
     end
   end

data/lib/belt/cli/generate_command.rb

@@ -7,6 +7,7 @@ require_relative 'environment_command'
 require_relative 'frontend_command'
 require_relative 'tables_command'
 require_relative 'views_command'
+require_relative 'generator_registry'
 require_relative '../inflector'
 
 module Belt
@@ -84,13 +85,19 @@ module Belt
           exit 0
         end
 
-        unless GENERATORS.include?(generator)
+        unless GENERATORS.include?(generator) || GeneratorRegistry.generator_names.include?(generator)
           puts "Unknown generator: '#{generator}'"
-          puts "Available generators: #{(GENERATORS - ['resource']).join(', ')}"
+          puts "Available generators: #{all_generator_names.join(', ')}"
           puts "\nRun 'belt generate --help' for usage information."
           exit 1
         end
 
+        # Delegate to gem-provided generator if not a built-in
+        unless GENERATORS.include?(generator)
+          klass = GeneratorRegistry.find(generator)
+          return klass.run(args)
+        end
+
         # Normalize: resource is an alias for scaffold
         generator = 'scaffold' if generator == 'resource'
 
@@ -131,6 +138,10 @@ module Belt
         belt setup environment frontend views routes
       ].freeze
 
+      def self.all_generator_names
+        ((GENERATORS - ['resource']) + GeneratorRegistry.generator_names).uniq
+      end
+
       def self.validate_resource_name!(name, generator)
         errors = []
 
@@ -152,6 +163,8 @@ module Belt
       end
 
       def self.print_generate_help
+        gem_generators = GeneratorRegistry.discovered_generators
+
         puts <<~HELP
           Usage: belt generate <generator> <name> [field:type ...] [options]
                  belt g <generator> <name> [field:type ...] [options]
@@ -167,6 +180,18 @@ module Belt
           Aliases:
             resource      Same as scaffold
 
+        HELP
+
+        if gem_generators.any?
+          puts '  Gem Generators:'
+          gem_generators.each do |name, klass|
+            desc = klass.respond_to?(:description) ? klass.description : "Generate #{name} (from gem)"
+            puts format('    %<name>-14s %<desc>s', name: name, desc: desc)
+          end
+          puts
+        end
+
+        puts <<~HELP
           Field Types:
             #{FIELD_TYPES.join(', ')}
             (defaults to string if omitted)

data/lib/belt/cli/generator_registry.rb

@@ -0,0 +1,92 @@
+# frozen_string_literal: true
+
+module Belt
+  module CLI
+    # Discovers and manages generators provided by external gems.
+    #
+    # Gems register generators by placing a file at:
+    #   lib/belt/generators/<name>_generator.rb
+    #
+    # Each generator class must:
+    #   - Be named <Name>Generator (e.g., MessagingGenerator)
+    #   - Live in the Belt::Generators module
+    #   - Implement .run(args) for generation
+    #   - Implement .destroy(args) for teardown (optional)
+    #   - Implement .description for help text (optional)
+    #
+    # Example gem structure:
+    #   belt-messaging/
+    #     lib/
+    #       belt/
+    #         generators/
+    #           messaging_generator.rb
+    #
+    # The generator is then available as:
+    #   belt generate messaging
+    #   belt destroy messaging
+    #
+    module GeneratorRegistry
+      GENERATOR_PATH = 'lib/belt/generators'
+
+      class << self
+        # Returns a hash of { "name" => generator_class } for all discovered generators.
+        def discovered_generators
+          @discovered_generators ||= discover_all
+        end
+
+        # Returns just the names of discovered generators.
+        def generator_names
+          discovered_generators.keys
+        end
+
+        # Loads and returns the generator class for a given name.
+        # Returns nil if not found.
+        def find(name)
+          discovered_generators[name]
+        end
+
+        # Reset cache (useful in tests)
+        def reset!
+          @discovered_generators = nil
+        end
+
+        private
+
+        def discover_all
+          generators = {}
+
+          Gem.loaded_specs.each_value do |spec|
+            generators_dir = File.join(spec.gem_dir, GENERATOR_PATH)
+            next unless File.directory?(generators_dir)
+
+            Dir.glob(File.join(generators_dir, '*_generator.rb')).each do |file|
+              name = File.basename(file, '.rb').sub(/_generator\z/, '')
+              klass = load_generator_class(file, name)
+              generators[name] = klass if klass
+            end
+          end
+
+          generators
+        end
+
+        def load_generator_class(file, name)
+          require file
+
+          class_name = "#{classify(name)}Generator"
+          Belt::Generators.const_get(class_name)
+        rescue LoadError, NameError => e
+          warn "[Belt] Warning: Failed to load generator '#{name}' from #{file}: #{e.message}"
+          nil
+        end
+
+        def classify(name)
+          name.split('_').map(&:capitalize).join
+        end
+      end
+    end
+  end
+
+  # Namespace for generator classes provided by external gems
+  module Generators
+  end
+end

data/lib/belt/cli.rb

@@ -56,7 +56,7 @@ module Belt
           # belt destroy --help → DestroyCommand help
           # belt destroy (no args) → terraform destroy (needs env)
           return DestroyCommand.run(args) if args.include?('--help') || args.include?('-h')
-        elsif DestroyCommand::GENERATORS.include?(args.first)
+        elsif DestroyCommand::GENERATORS.include?(args.first) || GeneratorRegistry.generator_names.include?(args.first)
           return DestroyCommand.run(args)
         end
       end

data/lib/belt/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Belt
-  VERSION = '0.1.12'
+  VERSION = '0.1.13'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: belt
 version: !ruby/object:Gem::Version
-  version: 0.1.12
+  version: 0.1.13
 platform: ruby
 authors:
 - Stowzilla
@@ -82,6 +82,7 @@ files:
 - lib/belt/cli/frontend_env_map.rb
 - lib/belt/cli/frontend_setup_command.rb
 - lib/belt/cli/generate_command.rb
+- lib/belt/cli/generator_registry.rb
 - lib/belt/cli/new_command.rb
 - lib/belt/cli/routes_command.rb
 - lib/belt/cli/routes_command/route_inference.rb