legionio 1.4.48 → 1.4.50

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d7a08e02740d6be6ecb76c91c3ee6f3a2ed8f594801994e8005c1ec3e3a3b47c
-  data.tar.gz: c45901fb93e4e97063e3044eee0d5646a50447acb818055ff75428a6c5f490a8
+  metadata.gz: 9f4884c77ed8ea44e74bdc3c9860dac014693cc90a3029fb06f1bc9e5a7edc27
+  data.tar.gz: 8da5570f871da743755e0c1c0fc1f81a8a5085061a6e27b5a8d9c2befe2400e0
 SHA512:
-  metadata.gz: 23eb01d19c9e38ab19d0bf28f5f2f9d32cb7bd64a47abd1fa0833b2e7d7ba13963c62eda860b43d5745cc54d77e4bacd94086e4749351e0b3af0489dd7ee5dc5
-  data.tar.gz: f436f935e81de29c1364022aad090c0e7c8bd7f1ee9419a1b33aaf1bb66800ffa117f699047ed02f314498161db79ad82193863893f5f538e9e08cf0147272b1
+  metadata.gz: be184813bae9bba912f9d190a06dbb9e3bc0a5b9c317019743fcaf1787de5c8aaad28fec85dcd14e1ffe27115cb80be3ea0970dd183cc002042aac1f2a12a7f1
+  data.tar.gz: e844ec3f09dd2f5bc6d8c908f884def2a6e4d71aa0ec5178d0e1ec1440c3a4cfe9d5d58996736acca92c67b7bced45ed8e8d2ffbc00a5f37a1ac4c6c4fb3ca37

data/CHANGELOG.md

@@ -1,5 +1,22 @@
 # Legion Changelog
 
+## [1.4.50] - 2026-03-17
+
+### Added
+- `Legion::Graph::Builder`: builds task relationship graph from relationships table with chain/worker filtering
+- `Legion::Graph::Exporter`: renders graphs to Mermaid and DOT (Graphviz) formats
+- `legion graph show`: CLI command with `--format mermaid|dot`, `--chain`, `--worker`, `--output`, `--limit` options
+
+## [1.4.49] - 2026-03-17
+
+### Added
+- `Legion::TenantContext`: thread-local tenant context propagation (set, clear, with block)
+- `Legion::Tenants`: tenant CRUD, suspension, and quota enforcement
+- `Middleware::Tenant`: extracts tenant_id from JWT/header, sets TenantContext per request
+- `GET/POST /api/tenants`: tenant listing and provisioning endpoints
+- `POST /api/tenants/:id/suspend`: tenant suspension
+- `GET /api/tenants/:id/quota/:resource`: quota check endpoint
+
 ## [1.4.48] - 2026-03-17
 
 ### Added

data/lib/legion/api/middleware/tenant.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module Legion
+  class API < Sinatra::Base
+    module Middleware
+      class Tenant
+        SKIP_PATHS = %w[/api/health /api/ready /api/openapi.json /metrics].freeze
+
+        def initialize(app, opts = {})
+          @app = app
+          @opts = opts
+        end
+
+        def call(env)
+          return @app.call(env) if skip_path?(env['PATH_INFO'])
+
+          tenant_id = extract_tenant(env)
+          Legion::TenantContext.set(tenant_id) if tenant_id
+          @app.call(env)
+        ensure
+          Legion::TenantContext.clear
+        end
+
+        private
+
+        def skip_path?(path)
+          SKIP_PATHS.any? { |sp| path.start_with?(sp) }
+        end
+
+        def extract_tenant(env)
+          env['legion.tenant_id'] || env['HTTP_X_TENANT_ID']
+        end
+      end
+    end
+  end
+end

data/lib/legion/api/tenants.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+require_relative '../tenants'
+
+module Legion
+  class API < Sinatra::Base
+    module Routes
+      module Tenants
+        def self.registered(app)
+          app.get '/api/tenants' do
+            tenants = Legion::Tenants.list
+            json_response(data: tenants)
+          end
+
+          app.post '/api/tenants' do
+            params = parsed_body
+            result = Legion::Tenants.create(
+              tenant_id:   params['tenant_id'],
+              name:        params['name'],
+              max_workers: params['max_workers'] || 10
+            )
+            status result[:error] ? 409 : 201
+            json_response(data: result)
+          end
+
+          app.get '/api/tenants/:tenant_id' do
+            tenant = Legion::Tenants.find(params[:tenant_id])
+            halt 404, json_response(error: 'not_found') unless tenant
+            json_response(data: tenant)
+          end
+
+          app.post '/api/tenants/:tenant_id/suspend' do
+            result = Legion::Tenants.suspend(tenant_id: params[:tenant_id])
+            json_response(data: result)
+          end
+
+          app.get '/api/tenants/:tenant_id/quota/:resource' do
+            result = Legion::Tenants.check_quota(
+              tenant_id: params[:tenant_id],
+              resource:  params[:resource].to_sym
+            )
+            json_response(data: result)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/cli/graph_command.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require 'thor'
+
+module Legion
+  module CLI
+    class GraphCommand < Thor
+      namespace 'graph'
+
+      desc 'show', 'Display task relationship graph'
+      option :chain, type: :string, desc: 'Filter by chain ID'
+      option :worker, type: :string, desc: 'Filter by worker ID'
+      option :format, type: :string, default: 'mermaid', enum: %w[mermaid dot]
+      option :output, type: :string, desc: 'Write to file'
+      option :limit, type: :numeric, default: 100
+      def show
+        require 'legion/graph/builder'
+        require 'legion/graph/exporter'
+
+        graph = Legion::Graph::Builder.build(
+          chain_id:  options[:chain],
+          worker_id: options[:worker],
+          limit:     options[:limit]
+        )
+
+        rendered = case options[:format]
+                   when 'dot' then Legion::Graph::Exporter.to_dot(graph)
+                   else Legion::Graph::Exporter.to_mermaid(graph)
+                   end
+
+        if options[:output]
+          File.write(options[:output], rendered)
+          say "Written to #{options[:output]}", :green
+        else
+          say rendered
+        end
+      end
+
+      default_task :show
+    end
+  end
+end

data/lib/legion/graph/builder.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+module Legion
+  module Graph
+    module Builder
+      class << self
+        def build(chain_id: nil, worker_id: nil, limit: 100) # rubocop:disable Lint/UnusedMethodArgument
+          return { nodes: {}, edges: [] } unless db_available?
+
+          ds = Legion::Data.connection[:relationships].limit(limit)
+          ds = ds.where(chain_id: chain_id) if chain_id
+
+          nodes = {}
+          edges = []
+
+          ds.each do |rel|
+            trigger = rel[:trigger] || "node_#{rel[:id]}_from"
+            action  = rel[:action] || "node_#{rel[:id]}_to"
+
+            nodes[trigger] ||= { label: trigger, type: 'trigger' }
+            nodes[action]  ||= { label: action, type: 'action' }
+            edges << {
+              from:     trigger,
+              to:       action,
+              label:    rel[:runner_function] || '',
+              chain_id: rel[:chain_id]
+            }
+          end
+
+          { nodes: nodes, edges: edges }
+        end
+
+        private
+
+        def db_available?
+          defined?(Legion::Data) && Legion::Data.connection&.table_exists?(:relationships)
+        rescue StandardError
+          false
+        end
+      end
+    end
+  end
+end

data/lib/legion/graph/exporter.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+module Legion
+  module Graph
+    module Exporter
+      class << self
+        def to_mermaid(graph)
+          lines = ['graph TD']
+          node_ids = {}
+          counter = 0
+
+          graph[:nodes].each do |key, node|
+            counter += 1
+            id = "N#{counter}"
+            node_ids[key] = id
+            lines << "  #{id}[#{node[:label]}]"
+          end
+
+          graph[:edges].each do |edge|
+            from = node_ids[edge[:from]]
+            to   = node_ids[edge[:to]]
+            next unless from && to
+
+            lines << if edge[:label] && !edge[:label].empty?
+                       "  #{from} -->|#{edge[:label]}| #{to}"
+                     else
+                       "  #{from} --> #{to}"
+                     end
+          end
+
+          lines.join("\n")
+        end
+
+        def to_dot(graph)
+          lines = ['digraph legion_tasks {', '  rankdir=LR;']
+
+          graph[:nodes].each do |key, node|
+            label = node[:label].gsub('"', '\\"')
+            shape = node[:type] == 'trigger' ? 'box' : 'ellipse'
+            lines << "  \"#{key}\" [label=\"#{label}\" shape=#{shape}];"
+          end
+
+          graph[:edges].each do |edge|
+            label = edge[:label] && !edge[:label].empty? ? " [label=\"#{edge[:label]}\"]" : ''
+            lines << "  \"#{edge[:from]}\" -> \"#{edge[:to]}\"#{label};"
+          end
+
+          lines << '}'
+          lines.join("\n")
+        end
+      end
+    end
+  end
+end

data/lib/legion/tenant_context.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Legion
+  module TenantContext
+    class << self
+      def current
+        Thread.current[:legion_tenant_id]
+      end
+
+      def set(tenant_id)
+        Thread.current[:legion_tenant_id] = tenant_id
+      end
+
+      def clear
+        Thread.current[:legion_tenant_id] = nil
+      end
+
+      def with(tenant_id)
+        prev = current
+        set(tenant_id)
+        yield
+      ensure
+        set(prev)
+      end
+    end
+  end
+end

data/lib/legion/tenants.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+module Legion
+  module Tenants
+    class << self
+      def create(tenant_id:, name: nil, max_workers: 10, max_queue_depth: 10_000, **)
+        return { error: 'tenant_exists' } if find(tenant_id)
+
+        Legion::Data.connection[:tenants].insert(
+          tenant_id:       tenant_id,
+          name:            name || tenant_id,
+          max_workers:     max_workers,
+          max_queue_depth: max_queue_depth,
+          status:          'active',
+          created_at:      Time.now.utc,
+          updated_at:      Time.now.utc
+        )
+        { created: true, tenant_id: tenant_id }
+      end
+
+      def find(tenant_id)
+        Legion::Data.connection[:tenants].where(tenant_id: tenant_id).first
+      rescue StandardError
+        nil
+      end
+
+      def suspend(tenant_id:, **)
+        Legion::Data.connection[:tenants]
+                    .where(tenant_id: tenant_id)
+                    .update(status: 'suspended', updated_at: Time.now.utc)
+        { suspended: true, tenant_id: tenant_id }
+      end
+
+      def list(**)
+        Legion::Data.connection[:tenants].all
+      end
+
+      def check_quota(tenant_id:, resource:, **)
+        tenant = find(tenant_id)
+        return { allowed: true } unless tenant
+
+        case resource
+        when :workers
+          count = Legion::Data.connection[:digital_workers].where(tenant_id: tenant_id).count
+          { allowed: count < tenant[:max_workers], current: count, limit: tenant[:max_workers] }
+        else
+          { allowed: true }
+        end
+      end
+    end
+  end
+end

data/lib/legion/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: legionio
 version: !ruby/object:Gem::Version
-  version: 1.4.48
+  version: 1.4.50
 platform: ruby
 authors:
 - Esity
@@ -333,6 +333,7 @@ files:
 - lib/legion/api/middleware/auth.rb
 - lib/legion/api/middleware/body_limit.rb
 - lib/legion/api/middleware/rate_limit.rb
+- lib/legion/api/middleware/tenant.rb
 - lib/legion/api/nodes.rb
 - lib/legion/api/oauth.rb
 - lib/legion/api/openapi.rb
@@ -341,6 +342,7 @@ files:
 - lib/legion/api/schedules.rb
 - lib/legion/api/settings.rb
 - lib/legion/api/tasks.rb
+- lib/legion/api/tenants.rb
 - lib/legion/api/token.rb
 - lib/legion/api/transport.rb
 - lib/legion/api/validators.rb
@@ -415,6 +417,7 @@ files:
 - lib/legion/cli/function.rb
 - lib/legion/cli/gaia_command.rb
 - lib/legion/cli/generate_command.rb
+- lib/legion/cli/graph_command.rb
 - lib/legion/cli/init/config_generator.rb
 - lib/legion/cli/init/environment_detector.rb
 - lib/legion/cli/init_command.rb
@@ -511,6 +514,8 @@ files:
 - lib/legion/extensions/helpers/transport.rb
 - lib/legion/extensions/hooks/base.rb
 - lib/legion/extensions/transport.rb
+- lib/legion/graph/builder.rb
+- lib/legion/graph/exporter.rb
 - lib/legion/guardrails.rb
 - lib/legion/ingress.rb
 - lib/legion/isolation.rb
@@ -566,6 +571,8 @@ files:
 - lib/legion/service.rb
 - lib/legion/supervision.rb
 - lib/legion/telemetry.rb
+- lib/legion/tenant_context.rb
+- lib/legion/tenants.rb
 - lib/legion/version.rb
 - lib/legion/webhooks.rb
 - scripts/rollout-ci-workflow.sh