discourse/lib/stylesheet/watcher.rb

# frozen_string_literal: true

require 'listen'

module Stylesheet
  class Watcher

    def self.watch(paths = nil)
      watcher = new(paths)
      watcher.start
      watcher
    end

    def initialize(paths)
      @paths = paths || Watcher.default_paths
      @queue = Queue.new
    end

    def self.default_paths
      return @default_paths if @default_paths

      @default_paths = ["app/assets/stylesheets"]
      Discourse.plugins.each do |plugin|
        if plugin.path.to_s.include?(Rails.root.to_s)
          @default_paths << File.dirname(plugin.path).sub(Rails.root.to_s, '').sub(/^\//, '')
        else
          # if plugin doesn’t seem to be in our app, consider it as outside of the app
          # and ignore it
          warn("[stylesheet watcher] Ignoring outside of rails root plugin: #{plugin.path.to_s}")
        end
      end
      @default_paths
    end

    def start
      Thread.new do
        begin
          while true
            worker_loop
          end
        rescue => e
          STDERR.puts "CSS change notifier crashed \n#{e}"
          start
        end
      end

      listener_opts = { ignore: /xxxx/, only: /\.(css|scss)$/ }
      listener_opts[:force_polling] = true if ENV['FORCE_POLLING']

      Thread.new do
        begin
          plugins_paths = Dir.glob("#{Rails.root}/plugins/*").map do |file|
            if File.symlink?(file)
              File.expand_path(File.readlink(file), "#{Rails.root}/plugins")
            else
              file
            end
          end.compact

          listener = Listen.to(*@paths, listener_opts) do |modified, added, _|
            paths = [modified, added].flatten
            paths.compact!
            paths.map! do |long|
              plugin_name = nil
              plugins_paths.each do |plugin_path|
                if long.include?("#{plugin_path}/")
                  plugin_name = File.basename(plugin_path)
                  break
                end
              end

              target = nil
              target_match = long.match(/admin|desktop|mobile|publish/)
              if target_match&.length
                target = target_match[0]
              end

              {
                basename: File.basename(long),
                target: target,
                plugin_name: plugin_name
              }
            end

            process_change(paths)
          end
        rescue => e
          STDERR.puts "Failed to listen for CSS changes: \n#{e}"
        end
        listener.start
        sleep
      end
    end

    def core_assets_refresh(target)
      targets = target ? [target] : ["desktop", "mobile", "admin"]
      Stylesheet::Manager.clear_core_cache!(targets)
      message = targets.map! do |name|
        Stylesheet::Manager.new.stylesheet_data(name.to_sym)
      end.flatten!
      MessageBus.publish '/file-change', message
    end

    def plugin_assets_refresh(plugin_name, target)
      Stylesheet::Manager.clear_plugin_cache!(plugin_name)
      targets = []
      if target.present?
        targets.push("#{plugin_name}_#{target.to_s}") if DiscoursePluginRegistry.stylesheets_exists?(plugin_name, target.to_sym)
      else
        targets.push(plugin_name)
      end
      message = targets.map! do |name|
        Stylesheet::Manager.new.stylesheet_data(name.to_sym)
      end.flatten!
      MessageBus.publish '/file-change', message
    end

    def worker_loop
      path = @queue.pop

      while @queue.length > 0
        @queue.pop
      end

      if path[:plugin_name]
        plugin_assets_refresh(path[:plugin_name], path[:target])
      else
        core_assets_refresh(path[:target])
      end
    end

    def process_change(paths)
      paths.each do |path|
        @queue.push path
      end
    end

  end
end