rubydb 0.1.4 → 0.1.6

data/lib/rubydb/catalog/view.rb

@@ -1,66 +1,66 @@
-# frozen_string_literal: true
-
-module RubyDB
-  module Catalog
-    # View - represents a database view (virtual table)
-    class View
-      attr_reader :name, :query, :columns
-      attr_accessor :materialized
-
-      def initialize(name, query, materialized: false)
-        @name = name
-        @query = query
-        @columns = []
-        @materialized = materialized
-        @created_at = Time.now
-        @modified_at = Time.now
-      end
-
-      def add_column(name, type)
-        @columns << { name: name, type: type }
-        @modified_at = Time.now
-      end
-
-      def column_names
-        @columns.map { |c| c[:name] }
-      end
-
-      def to_s
-        if @materialized
-          "MATERIALIZED VIEW #{@name}"
-        else
-          "VIEW #{@name}"
-        end
-      end
-
-      def inspect
-        "#<View name=#{@name} materialized=#{@materialized} columns=#{@columns.size}>"
-      end
-
-      # --- Serialization ---
-
-      def serialize
-        {
-          name: @name,
-          query: @query,
-          columns: @columns,
-          materialized: @materialized,
-          created_at: @created_at.iso8601,
-          modified_at: @modified_at.iso8601
-        }
-      end
-
-      def self.deserialize(data)
-        view = new(data[:name], data[:query], materialized: data[:materialized] || false)
-        view.send(:columns=, data[:columns]) if data[:columns]
-        view.send(:created_at=, Time.parse(data[:created_at])) if data[:created_at]
-        view.send(:modified_at=, Time.parse(data[:modified_at])) if data[:modified_at]
-        view
-      end
-
-      protected
-
-      attr_writer :created_at, :modified_at, :columns
-    end
-  end
-end
+# frozen_string_literal: true
+
+module RubyDB
+  module Catalog
+    # View - represents a database view (virtual table)
+    class View
+      attr_reader :name, :query, :columns
+      attr_accessor :materialized
+
+      def initialize(name, query, materialized: false)
+        @name = name
+        @query = query
+        @columns = []
+        @materialized = materialized
+        @created_at = Time.now
+        @modified_at = Time.now
+      end
+
+      def add_column(name, type)
+        @columns << {name: name, type: type}
+        @modified_at = Time.now
+      end
+
+      def column_names
+        @columns.map { |c| c[:name] }
+      end
+
+      def to_s
+        if @materialized
+          "MATERIALIZED VIEW #{@name}"
+        else
+          "VIEW #{@name}"
+        end
+      end
+
+      def inspect
+        "#<View name=#{@name} materialized=#{@materialized} columns=#{@columns.size}>"
+      end
+
+      # --- Serialization ---
+
+      def serialize
+        {
+          name: @name,
+          query: @query,
+          columns: @columns,
+          materialized: @materialized,
+          created_at: @created_at.iso8601,
+          modified_at: @modified_at.iso8601
+        }
+      end
+
+      def self.deserialize(data)
+        view = new(data[:name], data[:query], materialized: data[:materialized] || false)
+        view.send(:columns=, data[:columns]) if data[:columns]
+        view.send(:created_at=, Time.parse(data[:created_at])) if data[:created_at]
+        view.send(:modified_at=, Time.parse(data[:modified_at])) if data[:modified_at]
+        view
+      end
+
+      protected
+
+      attr_writer :created_at, :modified_at, :columns
+    end
+  end
+end

data/lib/rubydb/cli/application.rb

@@ -1,163 +1,168 @@
-# frozen_string_literal: true
-
-require "optparse"
-
-module RubyDB
-  module CLI
-    # Application - Main CLI application
-    class Application
-      def self.start(argv = ARGV)
-        new.run(argv)
-      end
-
-      def initialize
-        @output = Output.new
-        @formatter = Formatter.new(@output)
-        @commands = {}
-        @global_options = {
-          config: nil,
-          env: "development",
-          verbose: false,
-          quiet: false,
-          no_color: false
-        }
-
-        register_commands
-      end
-
-      def run(argv = ARGV)
-        parse_global_options(argv)
-
-        command_name = argv.shift
-
-        if command_name.nil? || command_name == "help"
-          print_help
-          return 0
-        end
-
-        command = @commands[command_name.to_sym]
-        if command.nil?
-          @output.error("Unknown command: #{command_name}")
-          print_help
-          return 1
-        end
-
-        begin
-          command.call(argv, @global_options)
-          0
-        rescue => e
-          @output.error(e.message)
-          @output.debug(e.backtrace.join("\n"))
-          1
-        end
-      end
-
-      private
-
-      def register_commands
-        @commands[:init] = Commands::Init.new(@output, @formatter).method(:execute)
-        @commands[:create] = Commands::Create.new(@output, @formatter).method(:execute)
-        @commands[:drop] = Commands::Drop.new(@output, @formatter).method(:execute)
-        @commands[:start] = Commands::Start.new(@output, @formatter).method(:execute)
-        @commands[:stop] = Commands::Stop.new(@output, @formatter).method(:execute)
-        @commands[:restart] = Commands::Restart.new(@output, @formatter).method(:execute)
-        @commands[:status] = Commands::Status.new(@output, @formatter).method(:execute)
-        @commands[:shell] = Commands::Shell.new(@output, @formatter).method(:execute)
-        @commands[:migrate] = Commands::Migrate.new(@output, @formatter).method(:execute)
-        @commands[:backup] = Commands::Backup.new(@output, @formatter).method(:execute)
-        @commands[:restore] = Commands::Restore.new(@output, @formatter).method(:execute)
-        @commands[:snapshot] = Commands::Snapshot.new(@output, @formatter).method(:execute)
-        @commands[:branch] = Commands::Branch.new(@output, @formatter).method(:execute)
-        @commands[:checkout] = Commands::Checkout.new(@output, @formatter).method(:execute)
-        @commands[:merge] = Commands::Merge.new(@output, @formatter).method(:execute)
-        @commands[:diff] = Commands::Diff.new(@output, @formatter).method(:execute)
-        @commands[:inspect] = Commands::Inspect.new(@output, @formatter).method(:execute)
-        @commands[:vacuum] = Commands::Vacuum.new(@output, @formatter).method(:execute)
-        @commands[:doctor] = Commands::Doctor.new(@output, @formatter).method(:execute)
-      end
-
-      def parse_global_options(argv)
-        parser = OptionParser.new do |opts|
-          opts.banner = "Usage: rubydb [global_options] command [args]"
-
-          opts.on("-c", "--config FILE", "Configuration file") do |file|
-            @global_options[:config] = file
-          end
-
-          opts.on("-e", "--env ENV", "Environment (development, test, production)") do |env|
-            @global_options[:env] = env
-          end
-
-          opts.on("-v", "--verbose", "Verbose output") do
-            @global_options[:verbose] = true
-          end
-
-          opts.on("-q", "--quiet", "Quiet output") do
-            @global_options[:quiet] = true
-          end
-
-          opts.on("--no-color", "Disable colored output") do
-            @global_options[:no_color] = true
-          end
-
-          opts.on("-h", "--help", "Show help") do
-            print_help
-            exit(0)
-          end
-
-          opts.on("--version", "Show version") do
-            puts "RubyDB v#{RubyDB::VERSION}"
-            exit(0)
-          end
-        end
-
-        begin
-          parser.parse!(argv)
-        rescue OptionParser::ParseError => e
-          @output.error(e.message)
-          print_help
-          exit(1)
-        end
-      end
-
-      def print_help
-        @output.puts "RubyDB - Developer-first relational database", :bold
-        @output.puts "Version: #{RubyDB::VERSION}"
-        @output.puts
-        @output.puts "Usage: rubydb [global_options] <command> [args]"
-        @output.puts
-        @output.puts "Global Options:"
-        @output.puts "  -c, --config FILE        Configuration file"
-        @output.puts "  -e, --env ENV            Environment (development, test, production)"
-        @output.puts "  -v, --verbose            Verbose output"
-        @output.puts "  -q, --quiet              Quiet output"
-        @output.puts "  --no-color               Disable colored output"
-        @output.puts "  -h, --help               Show this help"
-        @output.puts "  --version                Show version"
-        @output.puts
-        @output.puts "Commands:"
-        @output.puts "  init                     Initialize a new database"
-        @output.puts "  create                   Create a database"
-        @output.puts "  drop                     Drop a database"
-        @output.puts "  start                    Start the database server"
-        @output.puts "  stop                     Stop the database server"
-        @output.puts "  restart                  Restart the database server"
-        @output.puts "  status                   Show database status"
-        @output.puts "  shell                    Open an interactive SQL shell"
-        @output.puts "  migrate                  Run database migrations"
-        @output.puts "  backup                   Create a database backup"
-        @output.puts "  restore                  Restore a database backup"
-        @output.puts "  snapshot                 Create a database snapshot"
-        @output.puts "  branch                   Manage database branches"
-        @output.puts "  checkout                 Switch to a branch"
-        @output.puts "  merge                    Merge a branch"
-        @output.puts "  diff                     Show differences between branches"
-        @output.puts "  inspect                  Inspect database internals"
-        @output.puts "  vacuum                   Vacuum the database"
-        @output.puts "  doctor                   Run health checks"
-        @output.puts
-        @output.puts "Run 'rubydb <command> --help' for more information on a command."
-      end
-    end
-  end
-end
+# frozen_string_literal: true
+
+require "optparse"
+
+module RubyDB
+  module CLI
+    # Application - Main CLI application
+    class Application
+      def self.start(argv = ARGV)
+        new.run(argv)
+      end
+
+      def initialize
+        @output = Output.new
+        @formatter = Formatter.new(@output)
+        @commands = {}
+        @global_options = {
+          config: nil,
+          env: "development",
+          verbose: false,
+          quiet: false,
+          no_color: false
+        }
+
+        register_commands
+      end
+
+      def run(argv = ARGV)
+        parse_global_options(argv)
+
+        command_name = argv.shift
+
+        if command_name.nil? || command_name == "help"
+          print_help
+          return 0
+        end
+
+        command = @commands[command_name.to_sym]
+        if command.nil?
+          @output.error("Unknown command: #{command_name}")
+          print_help
+          return 1
+        end
+
+        begin
+          command.call(argv, @global_options)
+          0
+        rescue => e
+          @output.error(e.message)
+          @output.debug(e.backtrace.join("\n"))
+          1
+        end
+      end
+
+      private
+
+      def register_commands
+        @commands[:init] = Commands::Init.new(@output, @formatter).method(:execute)
+        @commands[:create] = Commands::Create.new(@output, @formatter).method(:execute)
+        @commands[:drop] = Commands::Drop.new(@output, @formatter).method(:execute)
+        @commands[:start] = Commands::Start.new(@output, @formatter).method(:execute)
+        @commands[:stop] = Commands::Stop.new(@output, @formatter).method(:execute)
+        @commands[:restart] = Commands::Restart.new(@output, @formatter).method(:execute)
+        @commands[:status] = Commands::Status.new(@output, @formatter).method(:execute)
+        @commands[:shell] = Commands::Shell.new(@output, @formatter).method(:execute)
+        @commands[:migrate] = Commands::Migrate.new(@output, @formatter).method(:execute)
+        @commands[:backup] = Commands::Backup.new(@output, @formatter).method(:execute)
+        @commands[:restore] = Commands::Restore.new(@output, @formatter).method(:execute)
+        @commands[:snapshot] = Commands::Snapshot.new(@output, @formatter).method(:execute)
+        @commands[:branch] = Commands::Branch.new(@output, @formatter).method(:execute)
+        @commands[:checkout] = Commands::Checkout.new(@output, @formatter).method(:execute)
+        @commands[:merge] = Commands::Merge.new(@output, @formatter).method(:execute)
+        @commands[:diff] = Commands::Diff.new(@output, @formatter).method(:execute)
+        @commands[:inspect] = Commands::Inspect.new(@output, @formatter).method(:execute)
+        @commands[:vacuum] = Commands::Vacuum.new(@output, @formatter).method(:execute)
+        @commands[:doctor] = Commands::Doctor.new(@output, @formatter).method(:execute)
+        @commands[:accelerator] = Commands::Accelerator.new(@output, @formatter).method(:execute)
+      end
+
+      def parse_global_options(argv)
+        parser = OptionParser.new do |opts|
+          opts.banner = "Usage: rubydb [global_options] command [args]"
+
+          opts.on("-c", "--config FILE", "Configuration file") do |file|
+            @global_options[:config] = file
+          end
+
+          opts.on("-e", "--env ENV", "Environment (development, test, production)") do |env|
+            @global_options[:env] = env
+          end
+
+          opts.on("-v", "--verbose", "Verbose output") do
+            @global_options[:verbose] = true
+          end
+
+          opts.on("-q", "--quiet", "Quiet output") do
+            @global_options[:quiet] = true
+          end
+
+          opts.on("--no-color", "Disable colored output") do
+            @global_options[:no_color] = true
+          end
+
+          opts.on("-h", "--help", "Show help") do
+            print_help
+            exit(0)
+          end
+
+          opts.on("--version", "Show version") do
+            puts "RubyDB v#{RubyDB::VERSION}"
+            exit(0)
+          end
+        end
+
+        begin
+          # Leave command-specific flags (for example `accelerator --ping`)
+          # for the selected command parser. Global flags remain accepted
+          # before the command name.
+          parser.order!(argv)
+        rescue OptionParser::ParseError => e
+          @output.error(e.message)
+          print_help
+          exit(1)
+        end
+      end
+
+      def print_help
+        @output.puts "RubyDB - Developer-first relational database", :bold
+        @output.puts "Version: #{RubyDB::VERSION}"
+        @output.puts
+        @output.puts "Usage: rubydb [global_options] <command> [args]"
+        @output.puts
+        @output.puts "Global Options:"
+        @output.puts "  -c, --config FILE        Configuration file"
+        @output.puts "  -e, --env ENV            Environment (development, test, production)"
+        @output.puts "  -v, --verbose            Verbose output"
+        @output.puts "  -q, --quiet              Quiet output"
+        @output.puts "  --no-color               Disable colored output"
+        @output.puts "  -h, --help               Show this help"
+        @output.puts "  --version                Show version"
+        @output.puts
+        @output.puts "Commands:"
+        @output.puts "  init                     Initialize a new database"
+        @output.puts "  create                   Create a database"
+        @output.puts "  drop                     Drop a database"
+        @output.puts "  start                    Start the database server"
+        @output.puts "  stop                     Stop the database server"
+        @output.puts "  restart                  Restart the database server"
+        @output.puts "  status                   Show database status"
+        @output.puts "  shell                    Open an interactive SQL shell"
+        @output.puts "  migrate                  Run database migrations"
+        @output.puts "  backup                   Create a database backup"
+        @output.puts "  restore                  Restore a database backup"
+        @output.puts "  snapshot                 Create a database snapshot"
+        @output.puts "  branch                   Manage database branches"
+        @output.puts "  checkout                 Switch to a branch"
+        @output.puts "  merge                    Merge a branch"
+        @output.puts "  diff                     Show differences between branches"
+        @output.puts "  inspect                  Inspect database internals"
+        @output.puts "  vacuum                   Vacuum the database"
+        @output.puts "  doctor                   Run health checks"
+        @output.puts "  accelerator              Inspect or verify the Go accelerator"
+        @output.puts
+        @output.puts "Run 'rubydb <command> --help' for more information on a command."
+      end
+    end
+  end
+end

data/lib/rubydb/cli/commands/accelerator.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+module RubyDB
+  module CLI
+    module Commands
+      # Accelerator - Inspect and verify the optional Go execution layer.
+      class Accelerator
+        def initialize(output, formatter)
+          @output = output
+          @formatter = formatter
+        end
+
+        def execute(args, options)
+          parser = OptionParser.new do |opts|
+            opts.banner = "Usage: rubydb accelerator [options]"
+            opts.on("--ping", "Start the accelerator and verify the protocol") do
+              options[:ping] = true
+            end
+            opts.on("--mode MODE", %w[auto off required], "Override mode for this check") do |mode|
+              options[:accelerator_mode] = mode
+            end
+            opts.on("--json", "Output JSON") do
+              options[:json] = true
+            end
+            opts.on("-h", "--help", "Show help") do
+              @output.puts opts
+              exit(0)
+            end
+          end
+          parser.parse!(args)
+
+          configured_accelerator = RubyDB::Configuration::Config.instance.to_hash[:accelerator]
+          config = if configured_accelerator.is_a?(Hash)
+            configured_accelerator.to_h.dup
+          else
+            # RUBYDB_ACCELERATOR=off|auto|required is a shorthand environment
+            # override. The environment parser may therefore replace the
+            # nested accelerator hash with a string; normalize it back into
+            # the client configuration instead of calling Hash#to_h on it.
+            {mode: configured_accelerator.to_s}
+          end
+          config[:mode] = options[:accelerator_mode] if options[:accelerator_mode]
+          client = RubyDB::Accelerator::Client.new(config)
+          details = client.stats
+          if options[:ping] && details[:available]
+            details[:ping] = client.ping
+          end
+          details[:healthy] = if details[:mode] == "off"
+            true
+          elsif details[:available]
+            !options[:ping] || details[:ping].is_a?(Hash)
+          else
+            details[:mode] != "required"
+          end
+
+          if options[:json]
+            @output.json(details)
+          else
+            @output.puts "RubyDB Go accelerator"
+            details.each { |key, value| @output.puts "  #{key}: #{value.inspect}" }
+          end
+          details[:healthy] ? 0 : 1
+        rescue RubyDB::Accelerator::Error => error
+          @output.error(error.message)
+          1
+        ensure
+          client&.close
+        end
+      end
+    end
+  end
+end