legionio 1.6.21 → 1.6.24

data/lib/legion/extensions/actors/once.rb

@@ -1,13 +1,17 @@
 # frozen_string_literal: true
 
 require_relative 'base'
+require_relative 'dsl'
 
 module Legion
   module Extensions
     module Actors
       class Once
+        extend Legion::Extensions::Actors::Dsl
         include Legion::Extensions::Actors::Base
 
+        define_dsl_accessor :delay, default: 1.0
+
         def initialize
           return unless enabled?
 

data/lib/legion/extensions/actors/poll.rb

@@ -2,15 +2,22 @@
 
 require_relative 'base'
 require_relative 'fingerprint'
+require_relative 'dsl'
 require 'time'
 
 module Legion
   module Extensions
     module Actors
       class Poll
+        extend Legion::Extensions::Actors::Dsl
         include Legion::Extensions::Actors::Base
         include Legion::Extensions::Actors::Fingerprint
 
+        define_dsl_accessor :time, default: 9
+        define_dsl_accessor :timeout, default: 5
+        define_dsl_accessor :run_now, default: true
+        define_dsl_accessor :int_percentage_normalize, default: 0.00
+
         def initialize
           log.debug "Starting timer for #{self.class} with #{{ execution_interval: time, run_now: run_now?,
 check_subtask: check_subtask? }}"
@@ -64,20 +71,8 @@ check_subtask: check_subtask? }}"
           0.00
         end
 
-        def time
-          9
-        end
-
         def run_now?
-          true
-        end
-
-        def check_subtask?
-          true
-        end
-
-        def timeout
-          5
+          run_now
         end
 
         def action(_payload = {})

data/lib/legion/extensions/actors/subscription.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require_relative 'base'
+require_relative 'dsl'
 require 'date'
 require 'securerandom'
 
@@ -8,10 +9,17 @@ module Legion
   module Extensions
     module Actors
       class Subscription
+        extend Legion::Extensions::Actors::Dsl
         include Concurrent::Async
         include Legion::Extensions::Actors::Base
         include Legion::Extensions::Helpers::Transport
 
+        define_dsl_accessor :consumers, default: 1
+        define_dsl_accessor :manual_ack, default: true
+        define_dsl_accessor :delay_start, default: 0
+        define_dsl_accessor :block, default: false
+        define_dsl_accessor :prefetch, default: 2
+
         def initialize(**_options)
           super()
           @queue = queue.new
@@ -25,8 +33,7 @@ module Legion
           exchange_object = default_exchange.new
           queue_object = Kernel.const_get(queue_string).new
 
-          queue_object.bind(exchange_object, routing_key: actor_name)
-          queue_object.bind(exchange_object, routing_key: "#{lex_name}.#{actor_name}.#")
+          queue_object.bind(exchange_object, routing_key: "#{amqp_prefix}.runners.#{runner_name}.#")
         end
 
         def queue
@@ -93,22 +100,6 @@ module Legion
           log.info "[Subscription] activated: #{lex_name}/#{runner_name} (consumer registered)"
         end
 
-        def block
-          false
-        end
-
-        def consumers
-          1
-        end
-
-        def manual_ack
-          true
-        end
-
-        def delay_start
-          0
-        end
-
         def include_metadata_in_message?
           true
         end

data/lib/legion/extensions/builders/hooks.rb

@@ -28,8 +28,8 @@ module Legion
             hook_class = Kernel.const_get(hook_class_name)
             next unless hook_class < Legion::Extensions::Hooks::Base
 
-            mount_suffix = hook_class.mount_path || ''
-            route_path = "#{extension_name}/#{hook_name}#{mount_suffix}"
+            route_path = "#{extension_name}/#{hook_name}"
+            runner = hook_class.respond_to?(:runner_class) ? hook_class.runner_class : nil
 
             @hooks[hook_name.to_sym] = {
               extension:      lex_class.to_s.downcase,
@@ -38,6 +38,23 @@ module Legion
               hook_class:     hook_class,
               route_path:     route_path
             }
+
+            next unless defined?(Legion::API) && Legion::API.respond_to?(:router)
+
+            # Register hook component in the router (explicit methods derived from hook class)
+            hook_methods = hook_class.public_instance_methods(false).reject { |m| m.to_s.start_with?('_') }
+            hook_methods = [:handle] if hook_methods.empty?
+            hook_methods.each do |method_name|
+              Legion::API.router.register_extension_route(
+                lex_name:       extension_name,
+                amqp_prefix:    respond_to?(:amqp_prefix) ? amqp_prefix : "lex.#{extension_name.to_s.tr('_', '.')}",
+                component_type: 'hooks',
+                component_name: hook_name,
+                method_name:    method_name.to_s,
+                runner_class:   runner || hook_class,
+                definition:     hook_class.respond_to?(:definition_for) ? hook_class.definition_for(method_name) : nil
+              )
+            end
           end
         end
 

data/lib/legion/extensions/builders/routes.rb

@@ -28,14 +28,29 @@ module Legion
 
             methods.each do |function|
               route_path = "#{extension_name}/#{runner_name}/#{function}"
+              defn = runner_module.respond_to?(:definition_for) ? runner_module.definition_for(function) : nil
               log.info "[Routes] auto-route registered: POST /api/lex/#{route_path}"
               @routes[route_path] = {
-                lex_name:     extension_name,
-                runner_name:  runner_name,
-                function:     function,
-                runner_class: runner_class,
-                route_path:   route_path
+                lex_name:       extension_name,
+                runner_name:    runner_name,
+                function:       function,
+                component_type: 'runners',
+                runner_class:   runner_class,
+                route_path:     route_path,
+                definition:     defn
               }
+
+              next unless defined?(Legion::API) && Legion::API.respond_to?(:router)
+
+              Legion::API.router.register_extension_route(
+                lex_name:       extension_name,
+                amqp_prefix:    respond_to?(:amqp_prefix) ? amqp_prefix : "lex.#{extension_name.to_s.tr('_', '.')}",
+                component_type: 'runners',
+                component_name: runner_name,
+                method_name:    function.to_s,
+                runner_class:   runner_class,
+                definition:     defn
+              )
             end
           end
         end

data/lib/legion/extensions/builders/runners.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require_relative 'base'
+require_relative '../definitions'
 
 module Legion
   module Extensions
@@ -22,6 +23,7 @@ module Legion
             runner_name = file.split('/').last.sub('.rb', '')
             runner_class =  "#{lex_class}::Runners::#{runner_name.split('_').collect(&:capitalize).join}"
             loaded_runner = Kernel.const_get(runner_class)
+            loaded_runner.extend(Legion::Extensions::Definitions) unless loaded_runner.respond_to?(:definition)
             Legion::Logging.debug "[Runners] registered: #{runner_class}" if defined?(Legion::Logging)
 
             @runners[runner_name.to_sym] = {

data/lib/legion/extensions/core.rb

@@ -76,6 +76,7 @@ module Legion
         end
         build_helpers
         build_runners
+        generate_messages_from_definitions
         build_absorbers
         build_actors
         build_hooks
@@ -113,6 +114,55 @@ module Legion
         true
       end
 
+      # Auto-generate AMQP message classes for each runner method that has a definition.
+      # Explicit Messages::* classes in the transport directory take precedence.
+      # Runs after build_runners so definitions are populated.
+      def generate_messages_from_definitions
+        ctx = message_generation_context
+        return unless ctx
+
+        @runners.each do |runner_name, attr|
+          generate_runner_messages(ctx, runner_name, attr)
+        end
+      rescue StandardError => e
+        log.warn "[Core] generate_messages_from_definitions failed: #{e.message}" if defined?(log)
+      end
+
+      def message_generation_context
+        return unless defined?(Legion::Transport::Message)
+        return unless lex_class.const_defined?('Transport', false)
+
+        transport_mod = lex_class::Transport
+        return unless transport_mod.const_defined?('Messages', false) && transport_mod.const_defined?('Exchanges', false)
+
+        default_exch = transport_mod.default_exchange
+        { messages_mod: transport_mod::Messages, default_exch: default_exch, prefix: amqp_prefix }
+      rescue StandardError
+        nil
+      end
+
+      def generate_runner_messages(ctx, runner_name, attr)
+        runner_module = attr[:runner_module]
+        return unless runner_module.respond_to?(:definitions)
+
+        runner_module.definitions.each_key do |method_name|
+          const_name = "#{camelize(runner_name)}#{camelize(method_name)}"
+          next if ctx[:messages_mod].const_defined?(const_name, false)
+
+          rk_value = "#{ctx[:prefix]}.runners.#{runner_name}.#{method_name}"
+          ctx[:messages_mod].const_set(const_name, Class.new(Legion::Transport::Message) do
+            define_method(:exchange) { ctx[:default_exch] }
+            define_method(:routing_key) { rk_value }
+          end)
+        end
+      rescue StandardError => e
+        log.warn "[Core] message generation error for #{runner_name}: #{e.message}" if defined?(log)
+      end
+
+      def camelize(name)
+        name.to_s.split('_').collect(&:capitalize).join
+      end
+
       def build_data
         Legion::Logging.debug "[Core] build_data: #{name}" if defined?(Legion::Logging)
         auto_generate_data

data/lib/legion/extensions/definitions.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Definitions
+      DEFAULTS = {
+        remote_invocable: true,
+        mcp_exposed:      true,
+        idempotent:       false,
+        risk_tier:        :standard,
+        tags:             [],
+        requires:         [],
+        inputs:           {},
+        outputs:          {}
+      }.freeze
+
+      def definition(method_name, **opts)
+        base = DEFAULTS.transform_values do |value|
+          case value
+          when Array, Hash
+            value.dup
+          else
+            value
+          end
+        end
+        own_definitions[method_name.to_sym] = base.merge(opts)
+      end
+
+      def definitions
+        if respond_to?(:superclass) && superclass.respond_to?(:definitions)
+          superclass.definitions.merge(own_definitions)
+        else
+          own_definitions.dup
+        end
+      end
+
+      def definition_for(method_name)
+        definitions[method_name.to_sym]
+      end
+
+      private
+
+      def own_definitions
+        @own_definitions ||= {}
+      end
+    end
+  end
+end

data/lib/legion/extensions/helpers/lex.rb

@@ -13,6 +13,7 @@ module Legion
         include Legion::Extensions::Helpers::Secret
 
         module ClassMethods
+          # @deprecated Use mcp_exposed: flag in definition DSL instead
           def expose_as_mcp_tool(value = :_unset)
             if value == :_unset
               return @expose_as_mcp_tool unless @expose_as_mcp_tool.nil?
@@ -27,6 +28,7 @@ module Legion
             end
           end
 
+          # @deprecated Use mcp_exposed: flag in definition DSL instead
           def mcp_tool_prefix(value = :_unset)
             if value == :_unset
               @mcp_tool_prefix
@@ -36,42 +38,52 @@ module Legion
           end
         end
 
+        # @deprecated Use definition DSL instead: definition :method, desc:, inputs:, outputs:
         def function_example(function, example)
           function_set(function, :example, example)
         end
 
+        # @deprecated Use definition DSL instead: definition :method, inputs: { ... }
         def function_options(function, options)
           function_set(function, :options, options)
         end
 
+        # @deprecated Use definition DSL instead: definition :method, desc: '...'
         def function_desc(function, desc)
           function_set(function, :desc, desc)
         end
 
+        # @deprecated Use definition DSL instead: definition :method, outputs: { ... }
         def function_outputs(function, outputs)
           function_set(function, :outputs, outputs)
         end
 
+        # @deprecated Use definition DSL instead: definition :method, category: '...'
         def function_category(function, category)
           function_set(function, :category, category)
         end
 
+        # @deprecated Use definition DSL instead: definition :method, tags: [...]
         def function_tags(function, tags)
           function_set(function, :tags, tags)
         end
 
+        # @deprecated Use definition DSL instead: definition :method, risk_tier: :standard
         def function_risk_tier(function, tier)
           function_set(function, :risk_tier, tier)
         end
 
+        # @deprecated Use definition DSL instead: definition :method, idempotent: true
         def function_idempotent(function, value)
           function_set(function, :idempotent, value)
         end
 
+        # @deprecated Use definition DSL instead: definition :method, requires: [...]
         def function_requires(function, deps)
           function_set(function, :requires, deps)
         end
 
+        # @deprecated Use definition DSL instead: definition :method, mcp_exposed: true
         def function_expose(function, value)
           function_set(function, :expose, value)
         end

data/lib/legion/extensions/helpers/segments.rb

@@ -27,7 +27,7 @@ module Legion
         end
 
         def segments_to_amqp_prefix(segments)
-          "legion.#{segments.join('.')}"
+          "lex.#{segments.join('.')}"
         end
 
         def segments_to_settings_path(segments)

data/lib/legion/extensions/hooks/base.rb

@@ -1,9 +1,12 @@
 # frozen_string_literal: true
 
+require_relative '../definitions'
+
 module Legion
   module Extensions
     module Hooks
       class Base
+        extend Legion::Extensions::Definitions
         include Legion::Extensions::Helpers::Lex
 
         class << self
@@ -44,12 +47,8 @@ module Legion
             @verify_config = { header: header.upcase.tr('-', '_'), secret: secret }
           end
 
-          def mount(path)
-            @mount_path = path
-          end
-
           attr_reader :route_type, :route_header_name, :route_field_name,
-                      :route_mapping, :verify_type, :verify_config, :mount_path
+                      :route_mapping, :verify_type, :verify_config
         end
 
         # Instance methods called by the API layer
@@ -63,7 +62,7 @@ module Legion
           when :field
             route_by_field(payload)
           else
-            :handle
+            :handle # deprecated fallback; prefer explicit route_header/route_field
           end
         end
 

data/lib/legion/extensions/transport.rb

@@ -146,7 +146,7 @@ module Legion
         return [] if @exchanges.count != 1
 
         @queues.map do |queue|
-          { from: @exchanges.first, to: queue, routing_key: queue }
+          { from: @exchanges.first, to: queue, routing_key: "#{amqp_prefix}.runners.#{queue}.#" }
         end
       end
 

data/lib/legion/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Legion
-  VERSION = '1.6.21'
+  VERSION = '1.6.24'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: legionio
 version: !ruby/object:Gem::Version
-  version: 1.6.21
+  version: 1.6.24
 platform: ruby
 authors:
 - Esity
@@ -470,6 +470,8 @@ files:
 - lib/legion/api/helpers.rb
 - lib/legion/api/hooks.rb
 - lib/legion/api/lex.rb
+- lib/legion/api/lex_dispatch.rb
+- lib/legion/api/library_routes.rb
 - lib/legion/api/llm.rb
 - lib/legion/api/marketplace.rb
 - lib/legion/api/metrics.rb
@@ -485,9 +487,11 @@ files:
 - lib/legion/api/prompts.rb
 - lib/legion/api/rbac.rb
 - lib/legion/api/relationships.rb
+- lib/legion/api/router.rb
 - lib/legion/api/schedules.rb
 - lib/legion/api/settings.rb
 - lib/legion/api/stats.rb
+- lib/legion/api/sync_dispatch.rb
 - lib/legion/api/tasks.rb
 - lib/legion/api/tenants.rb
 - lib/legion/api/token.rb
@@ -511,6 +515,7 @@ files:
 - lib/legion/cli.rb
 - lib/legion/cli/absorb_command.rb
 - lib/legion/cli/acp_command.rb
+- lib/legion/cli/admin/purge_topology.rb
 - lib/legion/cli/apollo_command.rb
 - lib/legion/cli/audit_command.rb
 - lib/legion/cli/auth_command.rb
@@ -524,6 +529,7 @@ files:
 - lib/legion/cli/chat/checkpoint.rb
 - lib/legion/cli/chat/context.rb
 - lib/legion/cli/chat/context_manager.rb
+- lib/legion/cli/chat/daemon_chat.rb
 - lib/legion/cli/chat/extension_tool.rb
 - lib/legion/cli/chat/extension_tool_loader.rb
 - lib/legion/cli/chat/markdown_renderer.rb
@@ -749,6 +755,7 @@ files:
 - lib/legion/extensions/actors/absorber_dispatch.rb
 - lib/legion/extensions/actors/base.rb
 - lib/legion/extensions/actors/defaults.rb
+- lib/legion/extensions/actors/dsl.rb
 - lib/legion/extensions/actors/every.rb
 - lib/legion/extensions/actors/fingerprint.rb
 - lib/legion/extensions/actors/loop.rb
@@ -771,6 +778,7 @@ files:
 - lib/legion/extensions/data.rb
 - lib/legion/extensions/data/migrator.rb
 - lib/legion/extensions/data/model.rb
+- lib/legion/extensions/definitions.rb
 - lib/legion/extensions/helpers/base.rb
 - lib/legion/extensions/helpers/cache.rb
 - lib/legion/extensions/helpers/core.rb