Contents

# frozen_string_literal: true

require 'rack/proxy'

# Public: Allows to relay asset requests to the Vite development server.
class ViteRails::DevServerProxy < Rack::Proxy
  VITE_DEPENDENCY_PREFIX = '/@'

  def initialize(app = nil, options = {})
    @vite_rails = options.delete(:vite_rails) || ViteRails.instance
    options[:streaming] = false if Rails.env.test? && !options.key?(:streaming)
    super
  end

  # Rack: Intercept asset requests and send them to the Vite server.
  def perform_request(env)
    if vite_should_handle?(env['REQUEST_URI']) && dev_server.running?
      env['REQUEST_URI'] = env['REQUEST_URI']
        .sub(vite_asset_url_prefix, '/')
        .sub('.ts.js', '.ts') # Patch: Rails helpers always append the extension.
      env['PATH_INFO'], env['QUERY_STRING'] = env['REQUEST_URI'].split('?')

      env['HTTP_HOST'] = env['HTTP_X_FORWARDED_HOST'] = config.host
      env['HTTP_X_FORWARDED_SERVER'] = config.host_with_port
      env['HTTP_PORT'] = env['HTTP_X_FORWARDED_PORT'] = config.port.to_s
      env['HTTP_X_FORWARDED_PROTO'] = env['HTTP_X_FORWARDED_SCHEME'] = config.protocol
      env['HTTPS'] = env['HTTP_X_FORWARDED_SSL'] = 'off' unless config.https
      env['SCRIPT_NAME'] = ''
      super(env)
    else
      @app.call(env)
    end
  end

private

  delegate :config, :dev_server, to: :@vite_rails

  def vite_should_handle?(url)
    url.start_with?(vite_asset_url_prefix) || url.start_with?(VITE_DEPENDENCY_PREFIX) ||
      url.include?('?t=') # Direct Hot Reload
  end

  def vite_asset_url_prefix
    @vite_asset_url_prefix ||= "/#{ config.public_output_dir }/"
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
vite_rails-1.0.4	lib/vite_rails/dev_server_proxy.rb
vite_rails-1.0.3	lib/vite_rails/dev_server_proxy.rb
vite_rails-1.0.2	lib/vite_rails/dev_server_proxy.rb
vite_rails-1.0.1	lib/vite_rails/dev_server_proxy.rb
vite_rails-1.0.0	lib/vite_rails/dev_server_proxy.rb