roda-debug_bar 0.1.1

data/lib/roda/plugins/debug_bar.rb

@@ -0,0 +1,299 @@
+require "tty-logger"
+require "pathname"
+require_relative "../debug_bar/current"
+require_relative "../debug_bar/instance"
+require_relative "../../sequel/extensions/debug_bar"
+require_relative "../../sequel/plugins/debug_bar"
+require 'rouge'
+
+require 'json'
+
+class Roda
+  module RodaPlugins
+    module DebugBar
+
+      DEFAULTS = {
+        db: nil,
+        log_time: false,
+        trace_missed: true,
+        ##### PUT TRACE ALL TO TRUE AGAIN
+        trace_all: false,
+        # trace_all: true,
+        filtered_params: %w[password password_confirmation _csrf],
+        handlers: [:console]
+      }.freeze
+
+      def self.load_dependencies(app, opts = {})
+        app.plugin :render
+
+        # just to display sessions
+        # app.plugin :sessions, secret: 'very-secret'*20
+
+        app.plugin :hooks
+        app.plugin :match_hook
+      end
+
+      def self.configure(app, opts = {})
+        # app.opts[:debug_bar] = opts
+
+        @@data_store = []
+
+        options = DEFAULTS.merge(opts)
+
+        logger = TTY::Logger.new { |config|
+          config.handlers = options[:handlers]
+          config.output = options.fetch(:output) { File.open('/dev/null', 'w') }
+          config.metadata.push(:time, :date) if options[:log_time]
+          config.filters.data = options[:filtered_params]
+          config.filters.mask = "<FILTERED>"
+        }
+
+        root = Pathname(app.opts[:root] || Dir.pwd)
+
+        db = options[:db] || (defined?(DB) && DB)
+        db&.extension :debug_bar
+
+        ::Sequel::Model.plugin :debug_bar
+
+        app.match_hook do
+          callee = caller_locations.find { |location|
+            location.path.start_with?(root.to_s)
+          }
+
+          @_debug_bar_instance.add_match(callee)
+        end
+
+        app.before do
+          @_debug_bar_instance = Roda::DebugBar::Instance.new(logger, env, object_id, root, options[:filter])
+
+          @halted = false
+
+          if request.path == '/debug_bar/ruby.png'
+            @halted = true
+            response.status = 200
+            response['content-type'] = 'image/png'
+            response.write(File.open(File.join(__dir__, '../../../public/ruby.png')).read)
+            request.halt
+          end
+
+          if request.path == '/debug_bar/out.css'
+            @halted = true
+            response.status = 200
+            response['content-type'] = 'text/css'
+            response.write(File.open(File.join(__dir__, '../../../public/out.css')).read)
+            request.halt
+          end
+
+          if request.path.start_with? '/debug_bar/last/'
+            @halted = true
+            id = request.path.split('/').last.to_i
+
+            if id > @@data_store.size
+              response.status = 404
+              response['content-type'] = 'text/plain'
+              response.write "Log ##{id} either doesn't exist yet, or wasn't stored. The current max store is 5."
+              request.halt
+            end
+
+            response.status = 200
+            response['content-type'] = 'application/json'
+            response.write(@@data_store[-id].to_json)
+            request.halt
+          end
+
+          # request.env['PATH_INFO'] is Rack's raw path
+          # request.path is a Roda abstraction over it, but can't be modified
+          @debug_catch = false
+          if request.env['PATH_INFO'].start_with? '/debug_bar/catch/'
+            @debug_catch = true
+            request.env['PATH_INFO'].sub!('/debug_bar/catch', '')
+          end
+          if request.env['PATH_INFO'].start_with? '/debug_bar/c/'
+            @debug_catch = true
+            request.env['PATH_INFO'].sub!('/debug_bar/c', '')
+          end
+          if request.env['PATH_INFO'].start_with? '/c/'
+            @debug_catch = true
+            request.env['PATH_INFO'].sub!('/c', '')
+          end
+
+        end
+
+        app.after do |res|
+
+          break if @halted
+
+          status, headers, content = res
+          @_debug_bar_instance.add(
+            status,
+            request,
+            (options[:trace_missed] && status == 404) || options[:trace_all]
+          )
+
+          # This @data is available to views
+          @data = @_debug_bar_instance.debug_data
+
+          if @debug_catch
+            response = @data.to_json
+            res[1]['content-type'] = 'application/json'
+            res[1]['Content-Length'] = response.bytesize.to_s
+            res[2] = [response]
+            @_debug_bar_instance.drain
+            @_debug_bar_instance.reset
+            break
+          end
+
+          if headers['content-type'] == 'text/html'
+            #####
+
+            # Should be a method, but I'm having trouble accessing the class variable from within an instance method
+            # def add_data(data)
+              if @@data_store.size < 5
+                @@data_store << @data
+              else
+                @@data_store.shift
+                @@data_store.push @data
+              end
+            # end
+
+            #####
+
+            # add_data(@data)
+
+            puts @@data_store.size
+          end
+
+          imports = <<-HTML
+  <script src="https://unpkg.com/alpinejs@3.14.8/dist/cdn.min.js" defer></script>
+  <!-- <script src="https://raw.githubusercontent.com/caldwell/renderjson/refs/heads/master/renderjson.js"></script>
+  <script src="https://cdn.jsdelivr.net/gh/caldwell/renderjson@master/renderjson.js"></script>
+  <script defer src="https://unpkg.com/pretty-json-custom-element/index.js"></script>
+  <script src="https://unpkg.com/@alenaksu/json-viewer@2.1.0/dist/json-viewer.bundle.js"></script> -->
+HTML
+
+          if headers['content-type'] == 'text/html'
+
+            res[2] = res[2].map do |body_part|
+              if body_part.include?("<head>")
+                body_part.sub("<head>", "<head>#{imports}")
+              else
+                "#{imports}#{body_part}"
+              end
+            end
+
+            debug_bar = relative_render('debug_bar')
+
+            res[2] = res[2].map do |body_part|
+              if body_part.include?("</body>")
+                body_part.sub("</body>", "#{debug_bar}</body>")
+              else
+                "#{body_part}#{debug_bar}"
+              end
+            end
+
+            res[1]["Content-Length"] = res[2].reduce(0) { |memo, chunk| memo + chunk.bytesize }.to_s
+
+          end
+
+          # if request.path == '/baz'
+          #   puts 'hi hi hi'
+          # end
+
+          @_debug_bar_instance.drain
+          @_debug_bar_instance.reset
+        end
+      end
+
+      #########
+
+      module InstanceMethods
+
+        # def add_data(data)
+        #   if @@data_store.size < 5
+        #     @@data_store << data
+        #   else
+        #     @@data_store.unshift
+        #     @@data_store.push data
+        #   end
+        # end
+
+        def render(path, opts = {})
+          unless opts[:ignore]
+            # puts "rendering #{path}"
+            Roda::DebugBar::Current.add_view(path)
+          end
+          super
+        end
+
+        def view(path, opts = {})
+          unless opts[:ignore]
+            # puts "view #{path}"
+            Roda::DebugBar::Current.add_view(path)
+          end
+          super
+        end
+
+        # def grid_component(opts)
+        # end
+
+        def relative_render view
+          render('', path: File.join(__dir__, "../debug_bar/views/#{view}.erb"), ignore: true)
+        end
+
+        def sql_highlight query
+          formatter = Rouge::Formatters::HTML.new
+          lexer = Rouge::Lexers::SQL.new
+          sql = formatter.format(lexer.lex(query))
+          sql.gsub('`', '&#96;')
+        end
+
+        def highlight_ruby_hash query
+          formatter = Rouge::Formatters::HTML.new
+          lexer = Rouge::Lexers::Ruby.new
+          formatter.format(lexer.lex(query))
+        end
+
+        def format_time time
+
+          # case time
+          # in 1e-6...1e-3
+          #   "#{time*1e6}μs"
+          # in 1e-3...
+          # end
+
+          if time < 1e-3
+            "#{(time*1e6).round}μs"
+          elsif time < 1
+            "#{(time*1e3).round(3)}ms"
+          else
+            "#{time}s"
+          end
+
+        end
+
+      #   # def _roda_run_main_route
+      #   def call
+      #     status, headers, content = super
+
+      #     debug_bar = relative_render('debug-bar')
+
+      #     content = content.map do |body_part|
+      #       if body_part.include?("</body>")
+      #         body_part.sub("</body>", "#{debug_bar}</body>")
+      #       else
+      #         "#{body_part}#{debug_bar}"
+      #       end
+      #     end
+
+      #     headers["Content-Length"] = content.reduce(0) { |memo, chunk| memo + chunk.bytesize }.to_s
+
+      #     [status, headers, content]
+      #   end
+
+      end
+
+    end
+
+  register_plugin(:debug_bar, RodaPlugins::DebugBar)
+  end
+end

data/lib/sequel/extensions/debug_bar.rb

@@ -0,0 +1,39 @@
+# frozen-string-literal: true
+
+require "sequel"
+
+module DebugBar
+  module Sequel
+
+    if ::Sequel::VERSION_NUMBER >= 50240
+      def skip_logging?
+        false
+      end
+    end
+
+    def self.extended(base)
+      if ::Sequel::VERSION_NUMBER < 50240
+        return if base.loggers.any?
+
+        require "logger"
+        base.loggers = [Logger.new("/dev/null")]
+      end
+
+      # ::Sequel::Model.plugin :after_initialize
+    end
+
+    # def self.configure(model, opts = {})
+    #   model.include InstanceMethods
+    # end
+
+    def log_duration(duration, message)
+      Roda::DebugBar::Current.increment_accrued_database_time(duration)
+      Roda::DebugBar::Current.increment_database_query_count
+      Roda::DebugBar::Current.add_database_query(message, duration.round(6))
+
+      super
+    end
+  end
+
+  ::Sequel::Database.register_extension :debug_bar, DebugBar::Sequel
+end

data/lib/sequel/main.rb

@@ -0,0 +1,36 @@
+require 'sequel'
+require_relative 'plugins/debug_bar'
+
+# require 'sequel/plugins/after_initialize'
+
+DB = Sequel.sqlite('/Users/avi/code/2025/jan/roda-30day-clone/database/database.sqlite')
+
+
+Sequel::Model.plugin :debug_bar
+
+class Employer < Sequel::Model
+end
+
+# puts Employer.first.inspect
+
+# puts Employer.plugins.include?(:after_initialize)
+
+
+# Sequel::Model.plugin :after_initialize
+# Sequel::Model.plugin :debug_bar
+
+# class Employer < Sequel::Model
+#   plugin :debug_bar
+# end
+
+# Employer.plugin :after_initialize
+# Employer.plugin :debug_bar
+
+# puts Employer.first.inspect
+
+# a = Employer.create(name: 'alksgj')
+
+# # puts Employer.plugins.include?(:after_initialization)
+# puts Employer.plugins.include?(:debug_bar)
+
+# # puts Sequel::Model.plugins.include?(:debug_bar)

data/lib/sequel/plugins/debug_bar.rb

@@ -0,0 +1,50 @@
+# frozen-string-literal: true
+
+require "sequel"
+
+module Sequel
+  module Plugins
+    module DebugBar
+
+      module ClassMethods
+
+        def call(_)
+          v = super
+          v.after_initialize
+          v
+        end
+
+      end
+
+      module InstanceMethods
+
+        def initialize(h={})
+          super
+          after_initialize
+        end
+
+        def after_initialize
+          Roda::DebugBar::Current.add_model(self.class, self.values)
+          # puts "created #{self.inspect}"
+        end
+      end
+
+      def self.extended(base)
+        # ::Sequel::Model.plugin :after_initialize
+      end
+
+      def self.apply(model, opts = {})
+        model.include InstanceMethods
+        model.include ClassMethods
+      end
+
+      # def self.configure(model, opts = {})
+      #   model.include InstanceMethods
+      #   model.include ClassMethods
+      # end
+
+    end
+  end
+
+  Model.plugin :debug_bar, Sequel::Plugins::DebugBar
+end

data/lib/version.rb

@@ -0,0 +1,3 @@
+module DebugBar
+  VERSION = "0.1.1"
+end