ElMassimo/vite_ruby

# frozen_string_literal: true

require 'vite_rails'

module ViteRails::CLI
end

module ViteRails::CLI::Build
  def call(**options)
    ensure_rails_init
    super
  end

  # Internal: Attempts to initialize the Rails application.
  def ensure_rails_init
    require File.expand_path('config/environment', Dir.pwd)
  rescue StandardError, LoadError => error
    $stderr << "Unable to initialize Rails application before Vite build:\n\n\t#{ error.message }\n\n"
  end
end

# Internal: Extends the base installation script from Vite Ruby to work for a
# typical Rails app.
module ViteRails::CLI::Install
  RAILS_TEMPLATES = Pathname.new(File.expand_path('../../templates', __dir__))

  # Override: Setup a typical apps/web Hanami app to use Vite.
  def setup_app_files
    cp RAILS_TEMPLATES.join('config/rails-vite.json'), config.config_path
    if dir = %w[app/javascript app/packs].find { |path| root.join(path).exist? }
      replace_first_line config.config_path, 'app/frontend', %(    "sourceCodeDir": "#{ dir }",)
    end
    setup_content_security_policy root.join('config/initializers/content_security_policy.rb')
    append root.join('Procfile.dev'), 'web: bin/rails s'
  end

  # Internal: Configure CSP rules that allow to load @vite/client correctly.
  def setup_content_security_policy(csp_file)
    return unless csp_file.exist?

    inject_line_after csp_file, 'policy.script_src', <<~CSP
          # You may need to enable this in production as well depending on your setup.
      #    policy.script_src *policy.script_src, :blob if Rails.env.test?
    CSP
    inject_line_after csp_file, 'policy.connect_src', <<~CSP
          # Allow @vite/client to hot reload changes in development
      #    policy.connect_src *policy.connect_src, "ws://\#{ ViteRuby.config.host_with_port }" if Rails.env.development?
    CSP
    inject_line_after csp_file, 'policy.script_src', <<~CSP
          # Allow @vite/client to hot reload javascript changes in development
      #    policy.script_src *policy.script_src, :unsafe_eval, "http://\#{ ViteRuby.config.host_with_port }" if Rails.env.development?
    CSP
    inject_line_after csp_file, 'policy.style_src', <<~CSP
          # Allow @vite/client to hot reload style changes in development
      #    policy.style_src *policy.style_src, :unsafe_inline if Rails.env.development?
    CSP
  end

  # Override: Create a sample JS file and attempt to inject it in an HTML template.
  def install_sample_files
    unless config.resolved_entrypoints_dir.join('application.js').exist?
      cp RAILS_TEMPLATES.join('entrypoints/application.js'), config.resolved_entrypoints_dir.join('application.js')
    end

    if (layout_file = root.join('app/views/layouts/application.html.erb')).exist?
      inject_line_before layout_file, '</head>', <<-HTML
    <%= vite_client_tag %>
    <%= vite_javascript_tag 'application' %>
    <!--
      If using a TypeScript entrypoint file:
        vite_typescript_tag 'application'

      If using a .jsx or .tsx entrypoint, add the extension:
        vite_javascript_tag 'application.jsx'

      Visit the guide for more information: https://vite-ruby.netlify.app/guide/rails
    -->
      HTML
    end
  end
end

ViteRuby::CLI::Build.prepend(ViteRails::CLI::Build)
ViteRuby::CLI::Install.prepend(ViteRails::CLI::Install)