vite_rb 0.0.1.alpha1

data/lib/tasks/rails.rake

@@ -0,0 +1,31 @@
+namespace :vite do
+  task init: :environment do
+    Rake::Task["vite:init"].invoke
+  end
+
+  desc "Removes compiled assets"
+  task clobber: :environment do
+    build_dir = Rails.root.join(ViteRb.config.build_dir)
+    output_dir = File.join(build_dir, ViteRb.config.output_dir)
+    command = "rm -rf #{output_dir}"
+    logger = Logger.new(STDOUT)
+    logger.info(command)
+    system(command)
+  end
+
+  task build: :environment do
+    Rake::Task["vite:build"].invoke
+  end
+
+  task dev: :environment do
+    Rake::Task["vite:dev"].invoke
+  end
+end
+
+Rake::Task["assets:precompile"].enhance do
+  Rake::Task["vite:build"].invoke
+end
+
+Rake::Task["assets:clobber"].enhance do
+  Rake::Task["vite:clobber"].invoke
+end

data/lib/tasks/ruby.rake

@@ -0,0 +1,21 @@
+require "vite_rb/generator"
+
+namespace :vite do
+  desc "initializes vite"
+  task :init do
+    ViteRb::Generator.init
+  end
+
+  desc "Compiles assets using snowpack bundler"
+  task :build do
+    ViteRb::Runner.build
+  end
+
+  desc "Compiles assets using snowpack bundler"
+  task compile: :build
+
+  desc "Run a snowpack dev server"
+  task :dev do
+    ViteRb::Runner.dev
+  end
+end

data/lib/vite_rb.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require "vite_rb/configuration"
+require "vite_rb/env"
+require "vite_rb/generator"
+require "vite_rb/helpers"
+require "vite_rb/proxy"
+require "vite_rb/utils"
+require "vite_rb/manifest"
+
+# ViteRb is the top-level module for interacting with the Vite ESM bundler.
+module ViteRb
+  class << self
+    attr_accessor :config
+
+    def configure
+      self.config ||= Configuration.new
+      yield(config) if block_given?
+    end
+  end
+end
+
+require "vite_rb/version"
+require "vite_rb/runner"
+require "vite_rb/engine" if defined?(Rails)

data/lib/vite_rb/configuration.rb

@@ -0,0 +1,56 @@
+module ViteRb
+  class Configuration
+    attr_accessor :config_path
+    attr_accessor :config_file
+    attr_accessor :babel_config_file
+    attr_accessor :postcss_config_file
+    attr_accessor :build_dir
+    attr_accessor :mount_path
+    attr_accessor :manifest_file
+    attr_accessor :output_dir
+    attr_accessor :port, :hostname
+
+    def initialize
+      yield(self) if block_given?
+    end
+
+
+    # Allows dynamic definition of getters and setters
+    # The setter must be the value used first.
+    # @example
+    #   ViteRb.config.fake_attr # => Raises an error
+    #   ViteRb.config.fake_attr = "stuff"
+    #   ViteRb.config.fake_attr # => "stuff"
+    def method_missing(method_name, *args, &block)
+      # Check if the method missing an "attr=" method
+      raise unless method_name.to_s.end_with?("=")
+
+      setter = method_name
+      getter = method_name.to_s.slice(0...-1).to_sym
+      instance_var = "@#{getter}".to_sym
+
+      # attr_writer
+      define_singleton_method(setter) do |new_val|
+        instance_variable_set(instance_var, new_val)
+      end
+
+      # attr_reader
+      define_singleton_method(getter) { instance_variable_get(instance_var) }
+
+      # Ignores all arguments but the first one
+      value = args[0]
+
+      # Actually sets the value on the instance variable
+      send(setter, value)
+    rescue MethodMissing
+      # Raise error as normal, nothing to see here
+      super(method_name, *args, &block)
+    end
+
+    # rubocop:enable Style/MethodMissingSuper Metrics/MethodLength
+
+    def respond_to_missing?(method_name, include_private = false)
+      method_name.to_s.end_with?("=") || super
+    end
+  end
+end

data/lib/vite_rb/engine.rb

@@ -0,0 +1,5 @@
+# This is required for Rails to load our tasks.
+module ViteRb
+  class Engine < ::Rails::Engine
+  end
+end

data/lib/vite_rb/env.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module ViteR
+  # Sets the appropriately prefixed ENV variables with VITE_RB
+  class Env
+    ENV_PREFIX = 'VITE_RB'
+
+    class << self
+      def create_env_variables(config = ViteRb.config)
+        instance_variables.each do |var|
+          value = config.instance_variable_get(var)
+
+          # .slice removes the "@" from beginning of string
+          var = var.to_s.upcase.slice(1..-1)
+
+          set_env(var, value)
+        end
+      end
+
+      private
+
+      def set_env(env_var, value)
+        return if value.nil?
+
+        ENV["#{ENV_PREFIX}_#{env_var}"] ||= value.to_s
+      end
+    end
+  end
+end

data/lib/vite_rb/generator.rb

@@ -0,0 +1,81 @@
+require "thor"
+require "vite_rb/utils"
+
+module ViteRb
+  class Generator < Thor::Group
+    include Thor::Actions
+    extend Utils
+
+    TEMPLATES = File.join(File.expand_path(__dir__), "templates")
+    CONFIG_FILES = %w[
+      vite.config.js
+      postcss.config.js
+    ]
+
+    def self.source_root
+      TEMPLATES
+    end
+
+    def create_initializer_file
+      target = "vite.rb"
+      source = "#{target}.tt"
+
+      destination = File.join("config", "initializers", target)
+
+      if Utils.rails?
+        destination = Rails.root.join("config", "initializers", target)
+      end
+
+      # Creates a config/initializers/vite.rb file
+      say "\n\nCreating initializer file at #{destination}...\n\n", :magenta
+      template source, destination
+    end
+
+    def create_config_files
+      destination = File.join("config", "vite")
+
+      if Utils.rails?
+        destination = Rails.root.join("config", "vite")
+      end
+
+      Rake.mkdir_p destination
+
+      say "\n\nCreating config files @ #{destination}...\n\n", :magenta
+      CONFIG_FILES.each do |filename|
+        template filename, File.join(destination, filename)
+      end
+    end
+
+    def create_vite_files
+      destination = File.join("app", "vite")
+
+      if Utils.rails?
+        destination = Rails.root.join("app", "vite")
+      end
+      say "\n\nCreating vite files...\n\n", :magenta
+
+      directory "vite", destination
+    end
+
+    def init
+      create_initializer_file
+      create_config_files
+      create_vite_files
+      add_vite
+
+      say "Finished initializing vite", :green
+    end
+
+    def add_vite
+      if ENV["VITE_RB_TEST"] == "true"
+        return system("yarn add vite_rb file:../../")
+      end
+
+      system("yarn add vite_rb")
+    end
+
+    def self.init
+      new.init
+    end
+  end
+end

data/lib/vite_rb/helpers.rb

@@ -0,0 +1,51 @@
+require "vite_rb/utils"
+
+module ViteRb
+  module Helpers
+    # Injects an HMR tag during development via a websocket.
+    def vite_hmr_tag
+      return unless Rails.env == "development"
+
+      hostname = ViteRb
+      port = ViteRb
+
+      hmr = %(window.HMR_WEBSOCKET_URL = "ws:#{hostname}:#{port}")
+
+      return tag.script(hmr.html_safe) if Utils.rails?
+
+      hmr
+    end
+
+    def vite_asset_path(name, **options)
+      asset_path(File.join(vite_dir, name), options)
+    end
+
+    def javascript_vite_tag(name, **options)
+      options[:type] ||= "module"
+
+      if Utils.dev_server_running?
+        javascript_include_tag("/#{vite_dir}/#{entrypoints_dir}/#{name}", options)
+      end
+
+      ## TODO: Change to reading from manifest for production
+      javascript_include_tag("/#{vite_dir}/#{entrypoints_dir}/#{name}", options)
+    end
+
+    # Returns nothing when not in production. CSS only gets extracted
+    # during the final build.
+    def stylesheet_vite_tag(name, **options)
+      return unless Rails.env == "development"
+
+      options[:media] ||= "screen"
+      stylesheet_link_tag("/#{vite_dir}/#{name}", options)
+    end
+
+    def vite_dir
+      ViteRb
+    end
+
+    def entrypoints_dir
+      ViteRb
+    end
+  end
+end

data/lib/vite_rb/manifest.rb

@@ -0,0 +1,40 @@
+require "json"
+
+module ViteRb
+  module Manifest
+    VALID_TYPES = [:js, :"js.map", :css, :"css.map"]
+    class << self
+      attr_accessor :manifest_hash
+
+      def reload_manifest(manifest_file = ViteRb.config.manifest_file)
+        @manifest_hash = load_manifest(manifest_file)
+      end
+
+      def find_entrypoint(file_name, type)
+        @manifest_hash[:entrypoints][file_name.to_sym][type.to_sym]
+      end
+
+      def find_file(file_name)
+        @manifest_hash[file_name.to_sym]
+      end
+
+      def find_chunk(chunk_name)
+        @manifest_hash[:chunks][chunk_name.to_sym][:js]
+      end
+
+      private
+
+      # rubocop:disable Naming/MemoizedInstanceVariableName
+      def load_manifest(manifest_file = ViteRb.config.manifest_file)
+        @manifest_hash ||= JSON.parse(File.read(manifest_file), symbolize_names: true)
+      end
+      # rubocop:enable Naming/MemoizedInstanceVariableName
+
+      def valid_type?(type)
+        return true if VALID_TYPES.include?(type)
+
+        false
+      end
+    end
+  end
+end

data/lib/vite_rb/proxy.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require "rack/proxy"
+require "socket"
+require "vite_rb/utils"
+
+module ViteRb
+  # Proxy server for Vite
+  class Proxy < Rack::Proxy
+    def initialize(app = nil, opts = {})
+      opts[:streaming] = false if Rails.env.test? && !opts.key?(:streaming)
+      super
+    end
+
+    def perform_request(env)
+      output_dir = %r{/#{ViteRb.config.output_dir}/}
+
+      if env["PATH_INFO"].start_with?(output_dir) && Utils.dev_server_running?
+        env["HTTP_HOST"] = env["HTTP_X_FORWARDED_HOST"] = ViteRb
+        env["HTTP_X_FORWARDED_SERVER"] = Utils.host_with_port
+        env["HTTP_PORT"] = env["HTTP_X_FORWARDED_PORT"] = ViteRb
+        env["HTTP_X_FORWARDED_PROTO"] = env["HTTP_X_FORWARDED_SCHEME"] = "http"
+
+        unless Utils.https?
+          env["HTTPS"] = env["HTTP_X_FORWARDED_SSL"] = "off"
+        end
+
+        env["SCRIPT_NAME"] = ""
+        super(env)
+      else
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/vite_rb/runner.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require "socket"
+require "vite_rb/env"
+require "vite_rb/utils"
+require "thor"
+
+module ViteRb
+  class Runner < Thor::Group
+    include ::Thor::Actions
+
+    attr_reader :config_file
+
+    def initialize
+      Env.create_env_variables
+    rescue Errno::ENOENT, NoMethodError
+      $stdout.puts "Vite configuration not found in #{ViteRb.config.config_dir}"
+      $stdout.puts "Please run bundle exec rails generate vite_rb to install Vite"
+      exit!
+    end
+
+    class << self
+      # Build for production
+      def build
+        new
+        vite_rb_command(env: :production, cmd: :build)
+      end
+
+      # Serve for development
+      def dev
+        Utils.detect_port!
+        new
+        vite_rb_command(env: :development, cmd: :dev)
+      end
+
+      private
+
+      def vite_rb_command(env: "", cmd: "")
+        env = ENV["NODE_ENV"] || env
+        config_file = ViteRb
+        command = "NODE_ENV=#{env} yarn run vite #{cmd} --config #{config_file}"
+        Rake.sh(command)
+      end
+    end
+  end
+end

data/lib/vite_rb/templates/postcss.config.js

@@ -0,0 +1,12 @@
+module.exports = {
+  plugins: [
+    require('postcss-import'),
+    require('postcss-flexbugs-fixes'),
+    require('postcss-preset-env')({
+      autoprefixer: {
+        flexbox: 'no-2009'
+      },
+      stage: 3
+    })
+  ]
+}