belt 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 330331693d70f3a18ad04c47b4f0b07c69b6a75538291be891946e12c7863dfc
-  data.tar.gz: 643ed0cad19c1124581d86b7f816ce199d372be0415c03b53c8f9d5f990bc3a9
+  metadata.gz: b09432f7426352033b67c6cc696bc90cddcae4b1facae08ce7b6fc08763b7eef
+  data.tar.gz: bb4617d6514bef6636b5b2efcfd32eaee6ebbc66b09a7e7f83581340c0fe5714
 SHA512:
-  metadata.gz: ca9f5d87f6b8ab597106789e5a381aa3ba6f33b85cc2620552b8c9380ad5508d7f27d5ed34cea52b59858db91ab02653ee4a487044562f7c3c68739c3fa40779
-  data.tar.gz: 36cbc894c45482981d1c3ddd34f0b59dd7e5259dd5de975175af756c54488693e6b0396c44b81b396442149a70ff6067d8d40046cef749d7119d248f49d57990
+  metadata.gz: 2c67896f36d6b09e394e19a73a92a6c8e31e9ec43071bf92c905f7d78ba260766d5b46aa2ee9d6c4d665c3313679557df527b866089bad240693fcdd1ce89e91
+  data.tar.gz: 8cf8956ea9a6227237be1dccc0cfe15b96ac9e75c3ddae543873cb067fcc5b8648fa23a1a258ccd3470afb9097ba54a7bc2dedfadcc701294b6ff8d9449ac755

data/lib/belt/assets/welcome.css

@@ -23,7 +23,7 @@ body {
   left: 50%;
   transform: translate(-50%, -50%);
   padding: 2rem 3rem;
-  background: rgba(255, 255, 255, 0.88);
+  background: rgba(255, 255, 255, 0.50);
   backdrop-filter: blur(6px);
   -webkit-backdrop-filter: blur(6px);
   border-radius: 12px;

data/lib/belt/cli/app_detection.rb

@@ -3,7 +3,9 @@
 module Belt
   module CLI
     module AppDetection
-      def detect_app_name
+      # Detects the primary namespace from the route definitions.
+      # Used by generators to determine controller directory and route file naming.
+      def detect_namespace
         routes_file = 'infrastructure/routes.tf.rb'
         if File.exist?(routes_file)
           match = File.read(routes_file).match(/namespace :(\w+)/)
@@ -12,6 +14,20 @@ module Belt
         File.basename(Dir.pwd)
       end
 
+      # Detects the application name.
+      # Prefers terraform.tfvars app_name, then falls back to directory name.
+      def detect_app_name
+        # Check any environment tfvars for the app_name
+        Dir.glob('infrastructure/*/terraform.tfvars').each do |f|
+          content = File.read(f)
+          match = content.match(/app_name\s*=\s*"([^"]+)"/)
+          return match[1] if match
+        end
+
+        # Fall back to directory name
+        File.basename(Dir.pwd)
+      end
+
       # S3 bucket names follow DNS rules — underscores are not allowed.
       # App names often use underscores (Ruby namespaces); convert for S3.
       def s3_safe_name(name)

data/lib/belt/cli/deploy_command.rb

@@ -1,5 +1,9 @@
 # frozen_string_literal: true
 
+require 'fileutils'
+require 'json'
+require 'open3'
+require 'tmpdir'
 require_relative 'env_resolver'
 require_relative 'terraform_command'
 
@@ -15,12 +19,15 @@ module Belt
         end
 
         auto_approve = false
+        rebuild = false
         filtered_args = []
 
         args.each do |arg|
           case arg
           when '--auto', '--yes', '-y'
             auto_approve = true
+          when '--rebuild'
+            rebuild = true
           when '-h', '--help'
             puts help_text
             exit 0
@@ -41,18 +48,24 @@ module Belt
             puts "\nDefaults to BELT_ENV or the first available environment."
             puts "\nOptions:"
             puts '  --auto, --yes, -y    Skip confirmation prompt (auto-approve)'
+            puts '  --rebuild            Rebuild and push Lambda code directly (bypasses Terraform)'
             puts '  -h, --help           Show this help'
             puts "\nExamples:"
             puts '  belt deploy                # Deploy dev (or BELT_ENV)'
             puts '  belt deploy prod           # Deploy to prod'
             puts '  belt deploy dev --auto     # Deploy without confirmation'
+            puts '  belt deploy --rebuild      # Fast code push (no infra changes)'
             puts '  belt deploy frontend dev   # Deploy frontend only'
             puts "\nNo environments found. Run `belt generate environment dev` first."
             exit 1
           end
         end
 
-        new(env, auto_approve: auto_approve, extra_args: filtered_args).run
+        if rebuild
+          new(env, auto_approve: auto_approve, extra_args: filtered_args).run_rebuild
+        else
+          new(env, auto_approve: auto_approve, extra_args: filtered_args).run
+        end
       end
 
       def self.help_text
@@ -63,22 +76,28 @@ module Belt
                  belt deploy frontend <environment>
 
           This runs the full deployment lifecycle:
-            1. terraform init    (initialize providers/modules)
-            2. terraform plan    (preview changes)
-            3. Prompt for confirmation (unless --auto)
-            4. terraform apply   (deploy changes)
+            1. Ensure Gemfile.lock is consistent (fix stale PATH refs)
+            2. terraform init    (initialize providers/modules)
+            3. terraform plan    (preview changes)
+            4. Prompt for confirmation (unless --auto)
+            5. terraform apply   (deploy changes)
 
           Options:
             --auto, --yes, -y    Skip confirmation prompt (auto-approve)
+            --rebuild            Rebuild and push Lambda code directly (bypasses Terraform).
+                                 Much faster for code-only changes — packages gems via Docker,
+                                 zips, and pushes with `aws lambda update-function-code`.
             -h, --help           Show this help
 
           Environment:
-            Defaults to BELT_ENV if set, otherwise requires an argument.
+            Defaults to BELT_ENV if set, otherwise the first available environment.
 
           Examples:
             belt deploy                # Deploy dev (or BELT_ENV)
             belt deploy prod           # Deploy to prod
             belt deploy dev --auto     # Deploy without confirmation (CI mode)
+            belt deploy --rebuild      # Fast code push to dev (no infra changes)
+            belt deploy prod --rebuild # Fast code push to prod
             belt deploy frontend dev   # Deploy frontend assets only
         HELP
       end
@@ -88,6 +107,7 @@ module Belt
         @auto_approve = auto_approve
         @extra_args = extra_args
         @infra_dir = TerraformCommand.find_infrastructure_dir
+        @project_root = find_project_root
       end
 
       def run
@@ -96,6 +116,8 @@ module Belt
 
         puts "belt → deploying #{@env} (in #{env_dir}/)\n\n"
 
+        ensure_lockfile_consistent!
+
         Dir.chdir(env_dir) do
           run_init
           run_plan
@@ -108,8 +130,43 @@ module Belt
         puts "\n   Run `belt server` to view your app locally (auto-connects to the deployed API)."
       end
 
+      def run_rebuild
+        validate!
+        validate_aws!
+
+        puts "belt → rebuilding Lambda for #{@env}\n\n"
+
+        lambda_dir = find_lambda_dir
+        abort 'Error: Cannot find lambda/ directory.' unless lambda_dir
+
+        function_name = find_lambda_function_name
+        abort 'Error: Cannot determine Lambda function name from Terraform state.' unless function_name
+
+        ensure_lockfile_consistent!
+        generate_routes(lambda_dir)
+        build_and_deploy(lambda_dir, function_name)
+      end
+
       private
 
+      def find_project_root
+        # Walk up from infra dir to find project root (where Gemfile lives)
+        dir = @infra_dir ? File.dirname(@infra_dir) : Dir.pwd
+        while dir != '/'
+          return dir if File.exist?(File.join(dir, 'Gemfile'))
+          dir = File.dirname(dir)
+        end
+        Dir.pwd
+      end
+
+      def find_lambda_dir
+        candidates = [
+          File.join(@project_root, 'lambda'),
+          File.join(Dir.pwd, 'lambda')
+        ]
+        candidates.find { |d| Dir.exist?(d) }
+      end
+
       def validate!
         unless @infra_dir
           abort "Error: No infrastructure/ directory found.\n" \
@@ -124,6 +181,262 @@ module Belt
               "Create it with: belt generate environment #{@env}"
       end
 
+      def validate_aws!
+        stdout, status = Open3.capture2('aws', 'sts', 'get-caller-identity', '--output', 'json')
+        unless status.success?
+          abort "Error: AWS credentials not available.\n" \
+                "Run `aws sso login` or configure AWS_PROFILE."
+        end
+        @aws_account = JSON.parse(stdout)['Account'] rescue nil
+      end
+
+      # Ensure Gemfile.lock doesn't have stale PATH references that will break
+      # the Docker-based gem build. If the Gemfile no longer uses `path:` but the
+      # lockfile still references a PATH source, refresh the lock.
+      def ensure_lockfile_consistent!
+        gemfile = File.join(@project_root, 'Gemfile')
+        lockfile = File.join(@project_root, 'Gemfile.lock')
+        return unless File.exist?(gemfile) && File.exist?(lockfile)
+
+        gemfile_content = File.read(gemfile)
+        lockfile_content = File.read(lockfile)
+
+        # Find gems that are PATH-sourced in the lockfile but NOT path-sourced in the Gemfile
+        stale_path_gems = detect_stale_path_gems(gemfile_content, lockfile_content)
+        return if stale_path_gems.empty?
+
+        puts "  🔧 Fixing stale Gemfile.lock (#{stale_path_gems.join(', ')} referenced as PATH but Gemfile uses RubyGems)"
+        Dir.chdir(@project_root) do
+          success = system('bundle', 'lock', '--update', *stale_path_gems)
+          unless success
+            puts "  ⚠ `bundle lock --update #{stale_path_gems.join(' ')}` failed — trying full bundle update"
+            system('bundle', 'update', *stale_path_gems)
+          end
+        end
+        puts ''
+      end
+
+      def detect_stale_path_gems(gemfile_content, lockfile_content)
+        stale = []
+
+        # Parse PATH sections from lockfile
+        # Format:
+        #   PATH
+        #     remote: ../something
+        #     specs:
+        #       gem_name (version)
+        lockfile_content.scan(/^PATH\n\s+remote:.*?\n\s+specs:\n((?:\s{4}\S.*\n?)+)/m).each do |match|
+          match[0].scan(/^\s{4}(\S+)\s/).each do |gem_match|
+            gem_name = gem_match[0]
+            # Check if the Gemfile still uses path: for this gem
+            # Match patterns like: gem 'name', path: ... or gem "name", path: ...
+            has_path_in_gemfile = gemfile_content.match?(/gem\s+['"]#{Regexp.escape(gem_name)}['"].*path:/)
+            stale << gem_name unless has_path_in_gemfile
+          end
+        end
+
+        stale
+      end
+
+      def find_lambda_function_name
+        env_dir = File.join(@infra_dir, @env)
+        return nil unless Dir.exist?(File.join(env_dir, '.terraform'))
+
+        # Get function name from terraform state
+        Dir.chdir(env_dir) do
+          output, status = Open3.capture2('terraform', 'output', '-json')
+          return nil unless status.success?
+
+          data = JSON.parse(output) rescue {}
+
+          # Try lambda_functions output first
+          if data['lambda_functions']
+            funcs = data['lambda_functions']['value']
+            if funcs.is_a?(Hash) && funcs.any?
+              return funcs.values.first # ARN — we need just the function name
+            end
+          end
+
+          # Fallback: parse from terraform state directly
+          state_output, state_status = Open3.capture2('terraform', 'state', 'show', 'conveyor_belt.main')
+          return nil unless state_status.success?
+
+          # Look for lambda function names in state
+          state_output.match(/function_name\s*=\s*"([^"]+)"/)&.captures&.first
+        end
+      end
+
+      def generate_routes(lambda_dir)
+        routes_dir = File.join(lambda_dir, 'lib', 'routes')
+        return unless Dir.exist?(routes_dir)
+
+        puts '  🗺️  Regenerating route manifests...'
+        success = system('belt', 'routes', '--namespace', 'all', '--output-dir', routes_dir)
+        if success
+          puts '  ✅ Routes updated'
+        else
+          puts '  ⚠ Route generation failed — using existing manifests'
+        end
+        puts ''
+      end
+
+      def build_and_deploy(lambda_dir, function_identifier)
+        # Extract function name from ARN if needed
+        function_name = if function_identifier.include?(':')
+                          function_identifier.split(':').last
+                        else
+                          function_identifier
+                        end
+
+        build_dir = Dir.mktmpdir('belt-rebuild-')
+
+        begin
+          puts '  📦 Building Lambda package...'
+          assemble_package(lambda_dir, build_dir)
+          build_gems(build_dir)
+          zip_file = create_zip(build_dir)
+
+          puts '  🚀 Deploying to AWS...'
+          deploy_to_lambda(function_name, zip_file)
+
+          puts "\n✅ Lambda #{function_name} rebuilt and deployed!"
+          puts ''
+          puts '  📋 View logs:'
+          puts "    aws logs tail /aws/lambda/#{function_name} --follow"
+        ensure
+          FileUtils.rm_rf(build_dir)
+        end
+      end
+
+      def assemble_package(lambda_dir, build_dir)
+        # Copy handler file(s) — .rb files at root of lambda/
+        Dir.glob(File.join(lambda_dir, '*.rb')).each do |f|
+          FileUtils.cp(f, build_dir)
+        end
+
+        # Copy shared directories
+        %w[controllers models lib helpers templates serializers jobs config].each do |dir|
+          src = File.join(lambda_dir, dir)
+          FileUtils.cp_r(src, File.join(build_dir, dir)) if Dir.exist?(src)
+        end
+
+        # Copy Gemfile from project root
+        gemfile = File.join(@project_root, 'Gemfile')
+        lockfile = File.join(@project_root, 'Gemfile.lock')
+        FileUtils.cp(gemfile, build_dir) if File.exist?(gemfile)
+        FileUtils.cp(lockfile, build_dir) if File.exist?(lockfile)
+
+        puts '    📁 Copied handlers, controllers, models, lib, config'
+      end
+
+      def build_gems(build_dir)
+        puts '    🐳 Building gems in Docker (Lambda-compatible)...'
+
+        docker_cmd = [
+          'docker', 'run', '--rm',
+          '--platform', 'linux/amd64',
+          '-v', "#{build_dir}:/var/task",
+          '-w', '/var/task',
+          'public.ecr.aws/sam/build-ruby3.4:latest-x86_64',
+          '/bin/bash', '-c',
+          "bundle config set --local path 'vendor/bundle' && " \
+          "bundle config set --local without 'development test' && " \
+          'bundle config set silence_root_warning 1 && ' \
+          'bundle install --jobs 4 && ' \
+          'bundle clean --force'
+        ]
+
+        output, status = Open3.capture2e(*docker_cmd)
+        unless status.success?
+          puts output
+          abort "\n✗ Gem build failed. Check Docker is running and the Gemfile is valid."
+        end
+
+        # Strip fat from vendor bundle
+        strip_vendor_fat(build_dir)
+        puts '    ✅ Gems installed'
+      end
+
+      def strip_vendor_fat(build_dir)
+        vendor_dir = File.join(build_dir, 'vendor')
+        return unless Dir.exist?(vendor_dir)
+
+        # Remove docs, tests, and non-runtime files from gems
+        Dir.glob(File.join(vendor_dir, 'bundle/ruby/*/gems/*/')).each do |gem_dir|
+          %w[spec test tests doc docs examples benchmarks].each do |fat_dir|
+            FileUtils.rm_rf(File.join(gem_dir, fat_dir))
+          end
+        end
+
+        # Remove non-essential files
+        exts = %w[*.md *.rdoc *.txt *.c *.h *.o Makefile *.log CHANGELOG* HISTORY* LICENSE* README* .gitignore .travis.yml .rubocop.yml Rakefile]
+        exts.each do |pattern|
+          Dir.glob(File.join(vendor_dir, "**", pattern)).each do |f|
+            File.delete(f) if File.file?(f)
+          end
+        end
+
+        # Strip debug symbols from native extensions
+        Dir.glob(File.join(vendor_dir, '**/*.so')).each do |so_file|
+          system('strip', '--strip-debug', so_file, err: File::NULL, out: File::NULL)
+        end
+      end
+
+      def create_zip(build_dir)
+        zip_file = File.join(Dir.tmpdir, "belt-lambda-#{@env}-#{Time.now.to_i}.zip")
+
+        Dir.chdir(build_dir) do
+          output, status = Open3.capture2e('zip', '-qr', zip_file, '.')
+          unless status.success?
+            puts output
+            abort "\n✗ Failed to create ZIP"
+          end
+        end
+
+        size_mb = (File.size(zip_file) / 1024.0 / 1024.0).round(1)
+        puts "    🗜️  Package: #{size_mb}MB"
+        zip_file
+      end
+
+      def deploy_to_lambda(function_name, zip_file)
+        _output, status = Open3.capture2(
+          'aws', 'lambda', 'update-function-code',
+          '--function-name', function_name,
+          '--zip-file', "fileb://#{zip_file}",
+          '--output', 'json'
+        )
+
+        File.delete(zip_file) if File.exist?(zip_file)
+
+        unless status.success?
+          abort "\n✗ Failed to update Lambda function code.\n" \
+                "  Function: #{function_name}\n" \
+                "  Check that the function exists and you have permissions."
+        end
+
+        # Wait for update to complete
+        print '    Waiting for Lambda to be ready...'
+        30.times do
+          sleep 1
+          state_output, = Open3.capture2(
+            'aws', 'lambda', 'get-function',
+            '--function-name', function_name,
+            '--query', 'Configuration.LastUpdateStatus',
+            '--output', 'text'
+          )
+          case state_output.strip
+          when 'Successful'
+            puts ' ✅'
+            return
+          when 'Failed'
+            puts ' ✗'
+            abort '    Lambda update failed. Check CloudWatch logs.'
+          end
+          print '.'
+        end
+        puts ' (timed out waiting, but code was pushed)'
+      end
+
       def run_init
         puts '━━━ terraform init ━━━'
         success = system('terraform', 'init')

data/lib/belt/cli/generate_command.rb

@@ -210,7 +210,7 @@ module Belt
         @name = name.downcase.gsub(/[^a-z0-9_]/, '_')
         @fields = fields
         @skip_views = skip_views
-        @app_name = detect_app_name
+        @app_name = detect_namespace
         @module_name = @app_name.split(/[-_]/).map(&:capitalize).join
         @resource_name = @name.end_with?('s') ? @name : "#{@name}s"
         @singular_name = @name.end_with?('s') ? @name.chomp('s') : @name

data/lib/belt/cli/new_command.rb

@@ -107,7 +107,7 @@ module Belt
 
       def directories
         %W[
-          #{@app_name}/lambda/controllers/#{@app_name}
+          #{@app_name}/lambda/controllers/api
           #{@app_name}/lambda/models
           #{@app_name}/lambda/models/concerns
           #{@app_name}/lambda/lib/routes
@@ -121,13 +121,13 @@ module Belt
         {
           'Gemfile.erb' => "#{@app_name}/Gemfile",
           'Rakefile.erb' => "#{@app_name}/Rakefile",
-          'lambda/api.rb.erb' => "#{@app_name}/lambda/#{@app_name}.rb",
+          'lambda/api.rb.erb' => "#{@app_name}/lambda/api.rb",
           'lambda/config/environment.rb.erb' => "#{@app_name}/lambda/config/environment.rb",
           'lambda/models/application_record.rb.erb' => "#{@app_name}/lambda/models/application_record.rb",
           'lambda/models/concerns/timestampable.rb.erb' => "#{@app_name}/lambda/models/concerns/timestampable.rb",
           'lambda/controllers/application_controller.rb.erb' =>
-            "#{@app_name}/lambda/controllers/#{@app_name}/application_controller.rb",
-          'lambda/lib/routes/routes.rb.erb' => "#{@app_name}/lambda/lib/routes/#{@app_name}_routes.rb",
+            "#{@app_name}/lambda/controllers/api/application_controller.rb",
+          'lambda/lib/routes/routes.rb.erb' => "#{@app_name}/lambda/lib/routes/api_routes.rb",
           'infrastructure/routes.tf.rb.erb' => "#{@app_name}/infrastructure/routes.tf.rb",
           'infrastructure/schema.tf.rb.erb' => "#{@app_name}/infrastructure/schema.tf.rb",
           'README.md.erb' => "#{@app_name}/README.md",

data/lib/belt/cli/server_command.rb

@@ -144,7 +144,7 @@ module Belt
         subtitle = ENV.fetch('WELCOME_SUBTITLE', 'Your app is ready. Deploy to see it live on AWS.')
 
         css = load_asset('welcome.css') || ''
-        image_b64 = load_image_base64('belt-default.png') || ''
+        image_b64 = load_image_base64('belt-default.jpg') || ''
 
         if @api_url
           deployed_html(css: css, image_b64: image_b64)
@@ -175,7 +175,7 @@ module Belt
           </head>
           <body>
             <div class="hero">
-              <img src="data:image/png;base64,#{image_b64}" alt="Belt" class="hero-bg" />
+              <img src="data:image/jpeg;base64,#{image_b64}" alt="Belt" class="hero-bg" />
               <div class="overlay">
                 <h1>#{@app_name}</h1>
                 <p class="subtitle">Deployed and running on AWS</p>
@@ -208,7 +208,7 @@ module Belt
           </head>
           <body>
             <div class="hero">
-              <img src="data:image/png;base64,#{image_b64}" alt="Belt" class="hero-bg" />
+              <img src="data:image/jpeg;base64,#{image_b64}" alt="Belt" class="hero-bg" />
               <div class="overlay">
                 <h1>#{title}</h1>
                 <p class="subtitle">#{subtitle}</p>

data/lib/belt/controllers/welcome_controller.rb

@@ -12,69 +12,17 @@ module Belt
     ASSETS_DIR = File.expand_path('../assets', __dir__)
 
     def show
-      title = ENV.fetch('WELCOME_TITLE', 'Welcome to Belt')
-      subtitle = ENV.fetch('WELCOME_SUBTITLE', 'API Gateway → Lambda → DynamoDB — all connected.')
-      app_name = ENV.fetch('APP_NAME', 'my-app')
+      @title = ENV.fetch('WELCOME_TITLE', 'Welcome to Belt')
+      @subtitle = ENV.fetch('WELCOME_SUBTITLE', 'API Gateway → Lambda → DynamoDB — all connected.')
+      @css = welcome_css
+      @background_image = background_image_base64
+      @dynamodb_connected = dynamodb_connected?
 
-      html_response(render_welcome_page(title: title, subtitle: subtitle, app_name: app_name))
+      render
     end
 
     private
 
-    def render_welcome_page(title:, subtitle:, app_name:)
-      <<~HTML
-        <!DOCTYPE html>
-        <html lang="en">
-        <head>
-          <meta charset="UTF-8">
-          <meta name="viewport" content="width=device-width, initial-scale=1.0">
-          <title>#{escape_html(title)}</title>
-          <style>#{welcome_css}</style>
-        </head>
-        <body>
-          <div class="hero">
-            <img src="data:image/png;base64,#{background_image_base64}" alt="Belt" class="hero-bg" />
-            <div class="overlay">
-              <h1>#{escape_html(title)}</h1>
-              <p class="subtitle">#{escape_html(subtitle)}</p>
-            </div>
-          </div>
-          <div class="container">
-            <div class="stack-check">
-              <div class="check-item check-pass">
-                <span class="icon">✓</span>
-                <span>API Gateway</span>
-              </div>
-              <div class="arrow">→</div>
-              <div class="check-item check-pass">
-                <span class="icon">✓</span>
-                <span>Lambda</span>
-              </div>
-              <div class="arrow">→</div>
-              #{dynamodb_check_html}
-            </div>
-            <div class="next-steps">
-              <h2>Next Steps</h2>
-              <ol>
-                <li>Generate a resource: <code>belt g resource post title:string body:text</code></li>
-                <li>Deploy: <code>belt deploy</code></li>
-                <li>This page will be replaced once you define your own root route.</li>
-              </ol>
-            </div>
-          </div>
-        </body>
-        </html>
-      HTML
-    end
-
-    def dynamodb_check_html
-      if dynamodb_connected?
-        '<div class="check-item check-pass"><span class="icon">✓</span><span>DynamoDB</span></div>'
-      else
-        '<div class="check-item check-warn"><span class="icon">⚠</span><span>DynamoDB</span></div>'
-      end
-    end
-
     def dynamodb_connected?
       return false unless defined?(Aws::DynamoDB::Client)
 
@@ -85,17 +33,13 @@ module Belt
       false
     end
 
-    def escape_html(text)
-      text.to_s.gsub('&', '&amp;').gsub('<', '&lt;').gsub('>', '&gt;').gsub('"', '&quot;')
-    end
-
     def welcome_css
-      @welcome_css ||= File.read(File.join(ASSETS_DIR, 'welcome.css'))
+      @welcome_css_content ||= File.read(File.join(ASSETS_DIR, 'welcome.css'))
     end
 
     def background_image_base64
-      @background_image_base64 ||= Base64.strict_encode64(
-        File.binread(File.join(ASSETS_DIR, 'belt-default.png'))
+      @background_image_content ||= Base64.strict_encode64(
+        File.binread(File.join(ASSETS_DIR, 'belt-default.jpg'))
       )
     end
   end

data/lib/belt/rendering.rb

@@ -0,0 +1,136 @@
+# frozen_string_literal: true
+
+require 'erb'
+require 'cgi'
+
+module Belt
+  # Provides Rails-like ERB rendering for BeltController.
+  #
+  # Convention: templates live at `views/<controller_name>/<action>.html.erb`
+  # relative to the controller's gem or app directory.
+  #
+  # Usage:
+  #   class WelcomeController < BeltController::Base
+  #     def show
+  #       @title = "Hello"
+  #       render  # auto-resolves to views/welcome/show.html.erb
+  #     end
+  #   end
+  #
+  # Or explicit:
+  #   render template: "welcome/show"
+  #   render inline: "<h1><%= @title %></h1>"
+  #
+  module Rendering
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+
+    module ClassMethods
+      # Override in subclasses to set a custom view path.
+      # Defaults to `views/` relative to the file that defines the controller.
+      def view_paths
+        @view_paths ||= []
+      end
+
+      def prepend_view_path(path)
+        view_paths.unshift(path)
+      end
+
+      def append_view_path(path)
+        view_paths << path
+      end
+    end
+
+    private
+
+    # Render an ERB template and return an html_response.
+    #
+    # Options:
+    #   render                         — auto-resolve from controller/action
+    #   render template: "ctrl/action" — explicit template path (no extension)
+    #   render inline: "<%= @x %>"     — render a string as ERB
+    #   render status: 201             — custom status code
+    #   render layout: false           — skip layout (default: no layout)
+    #
+    def render(options = {})
+      status = options.fetch(:status, 200)
+
+      html = if options.key?(:inline)
+               render_erb_string(options[:inline])
+             else
+               template_path = resolve_template_path(options[:template])
+               render_erb_file(template_path)
+             end
+
+      html_response(html, status)
+    end
+
+    def render_erb_string(source)
+      erb = ERB.new(source, trim_mode: '-')
+      erb.result(binding)
+    end
+
+    def render_erb_file(path)
+      raise Belt::TemplateNotFound, "Template not found: #{path}" unless File.exist?(path)
+
+      source = File.read(path)
+      erb = ERB.new(source, trim_mode: '-')
+      erb.filename = path
+      erb.result(binding)
+    end
+
+    # HTML-escape helper, available in templates as `h(value)`
+    def h(text)
+      CGI.escapeHTML(text.to_s)
+    end
+
+    def resolve_template_path(explicit_template = nil)
+      template_name = explicit_template || default_template_name
+      search_paths = build_view_search_paths
+
+      search_paths.each do |base|
+        full_path = File.join(base, "#{template_name}.html.erb")
+        return full_path if File.exist?(full_path)
+      end
+
+      # If nothing found, return the first candidate so the error message is useful
+      File.join(search_paths.first || '.', "#{template_name}.html.erb")
+    end
+
+    def default_template_name
+      # PostsController → posts, Belt::WelcomeController → welcome
+      controller = self.class.name.split('::').last
+                       .sub(/Controller$/, '')
+                       .gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
+                       .gsub(/([a-z\d])([A-Z])/, '\1_\2')
+                       .downcase
+      "#{controller}/#{action_name}"
+    end
+
+    def build_view_search_paths
+      paths = []
+
+      # 1. Class-level view paths (set by the controller or gem)
+      paths.concat(self.class.view_paths) if self.class.respond_to?(:view_paths)
+
+      # 2. Walk up the class hierarchy for inherited view paths
+      klass = self.class.superclass
+      while klass.respond_to?(:view_paths)
+        paths.concat(klass.view_paths)
+        klass = klass.superclass
+      end
+
+      # 3. App-level views (convention: lambda/views/ from Belt.root)
+      if defined?(Belt.root) && Belt.root
+        paths << File.join(Belt.root, 'lambda', 'views')
+        paths << File.join(Belt.root, 'views')
+      end
+
+      # 4. Gem-level fallback (views/ relative to this file — belt gem's own views)
+      paths << File.expand_path('views', __dir__)
+
+      paths.uniq
+    end
+  end
+end

data/lib/belt/version.rb

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

data/lib/belt/views/welcome/show.html.erb

@@ -0,0 +1,51 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <title><%= h(@title) %></title>
+  <style><%= @css %></style>
+</head>
+<body>
+  <div class="hero">
+    <img src="data:image/jpeg;base64,<%= @background_image %>" alt="Belt" class="hero-bg" />
+    <div class="overlay">
+      <h1><%= h(@title) %></h1>
+      <p class="subtitle"><%= h(@subtitle) %></p>
+    </div>
+  </div>
+  <div class="container">
+    <div class="stack-check">
+      <div class="check-item check-pass">
+        <span class="icon">✓</span>
+        <span>API Gateway</span>
+      </div>
+      <div class="arrow">→</div>
+      <div class="check-item check-pass">
+        <span class="icon">✓</span>
+        <span>Lambda</span>
+      </div>
+      <div class="arrow">→</div>
+      <%- if @dynamodb_connected -%>
+      <div class="check-item check-pass">
+        <span class="icon">✓</span>
+        <span>DynamoDB</span>
+      </div>
+      <%- else -%>
+      <div class="check-item check-warn">
+        <span class="icon">⚠</span>
+        <span>DynamoDB</span>
+      </div>
+      <%- end -%>
+    </div>
+    <div class="next-steps">
+      <h2>Next Steps</h2>
+      <ol>
+        <li>Generate a resource: <code>belt g resource post title:string body:text</code></li>
+        <li>Deploy: <code>belt deploy</code></li>
+        <li>This page will be replaced once you define your own root route.</li>
+      </ol>
+    </div>
+  </div>
+</body>
+</html>

data/lib/belt.rb

@@ -11,6 +11,7 @@ module Belt
   class AuthenticationError < StandardError; end
   class RecordNotFound < StandardError; end
   class ActionNotFound < StandardError; end
+  class TemplateNotFound < StandardError; end
 
   @controller_paths = []
 

data/lib/belt_controller/base.rb

@@ -6,10 +6,12 @@ require_relative '../belt/parameters'
 require_relative '../belt/helpers/response'
 require_relative '../belt/helpers/error_logging'
 require_relative '../belt/helpers/cors_origin'
+require_relative '../belt/rendering'
 
 module BeltController
   class Base
     include Belt::Helpers::Response
+    include Belt::Rendering
 
     attr_reader :event, :body
 

data/lib/templates/environment/main.tf.erb

@@ -20,7 +20,7 @@ provider "aws" {
 
   default_tags {
     tags = {
-      Project     = "<%= @app_name.capitalize %>"
+      Project     = var.app_name
       Environment = var.environment
       ManagedBy   = "Terraform"
     }
@@ -37,7 +37,20 @@ resource "conveyor_belt" "main" {
   source            = "${path.module}/../routes.tf.rb"
   app_name          = var.app_name
   lambda_source_dir = "${path.module}/../../lambda"
-  lambda_shared_dirs = ["controllers", "helpers", "lib", "models", "templates"]
+  lambda_shared_dirs = ["controllers", "helpers", "lib", "models", "views"]
   frontend_urls     = ["http://localhost:3000"]
-  friendly_errors   = true
+  friendly_errors   = var.environment != "prod"
+
+  # Per-lambda configuration: env vars, timeouts, memory
+  lambda_config = {
+    api = {
+      # timeout     = 30
+      # memory_size = 256
+
+      env_vars = {
+        WELCOME_TITLE    = "Welcome to ${var.app_name}"
+        WELCOME_SUBTITLE = "API Gateway → Lambda → DynamoDB — all connected."
+      }
+    }
+  }
 }

data/lib/templates/new_app/infrastructure/routes.tf.rb.erb

@@ -1,5 +1,5 @@
 Belt.application.routes.draw do
-  namespace :<%= @app_name %> do
+  namespace :api do
     get "/", action: :show, controller: :welcome
     # resources :posts
   end

data/lib/templates/new_app/lambda/api.rb.erb

@@ -10,12 +10,12 @@ ActiveItem.configure do |config|
   config.environment = ENV['ENVIRONMENT']
 end
 
-require_relative 'lib/routes/<%= @app_name %>_routes'
+require_relative 'lib/routes/api_routes'
 <% @resources&.each do |r| -%>
 require_relative 'controllers/<%= @app_name %>/<%= r %>_controller'
 <% end -%>
 
-ROUTER = Belt::ActionRouter.new(routes: Routes::<%= @module_name.upcase %>, namespace: '<%= @app_name %>')
+ROUTER = Belt::ActionRouter.new(routes: Routes::API, namespace: 'api')
 
 def execute(path:, body:, event:)
   ROUTER.route(event: event, body: body)

data/lib/templates/new_app/lambda/controllers/application_controller.rb.erb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-module <%= @module_name %>Controllers
+module ApiControllers
   class ApplicationController < BeltController::Base
   end
 end

data/lib/templates/new_app/lambda/lib/routes/routes.rb.erb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Routes
-  <%= @module_name.upcase %> = [
+  API = [
     { verb: 'GET', path: '/', controller: 'welcome', action: 'show' },
     # { verb: 'GET', path: '/posts', controller: 'posts', action: 'index' },
     # { verb: 'POST', path: '/posts', controller: 'posts', action: 'create' },

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: belt
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - Stowzilla
@@ -52,7 +52,7 @@ files:
 - exe/belt
 - lib/belt.rb
 - lib/belt/action_router.rb
-- lib/belt/assets/belt-default.png
+- lib/belt/assets/belt-default.jpg
 - lib/belt/assets/welcome.css
 - lib/belt/cli.rb
 - lib/belt/cli/app_detection.rb
@@ -82,10 +82,12 @@ files:
 - lib/belt/lambda_handler.rb
 - lib/belt/observability.rb
 - lib/belt/parameters.rb
+- lib/belt/rendering.rb
 - lib/belt/root.rb
 - lib/belt/route_dsl.rb
 - lib/belt/table_inference.rb
 - lib/belt/version.rb
+- lib/belt/views/welcome/show.html.erb
 - lib/belt_controller/base.rb
 - lib/templates/environment/backend.tf.erb
 - lib/templates/environment/main.tf.erb