pwn 0.5.749 → 0.5.751

data/spec/lib/pwn/migrate_spec.rb

@@ -80,7 +80,7 @@ describe PWN::Migrate do
       File.write(File.join(@tmp, '.schema'), JSON.generate(schema: 2))
       expect(described_class.needed?).to be(true)
       result = described_class.run(fix: false, backup: false, io: io)
-      expect(result[:applied_migrations]).to eq([3, 4, 5])
+      expect(result[:applied_migrations]).to eq([3, 4, 5, 6, 7])
       expected = Marshal.load(Marshal.dump(original))
       expected['custom']['model'] = nil
       expect(YAML.safe_load_file(path)).to eq(expected)
@@ -189,7 +189,7 @@ describe PWN::Migrate do
       File.write(File.join(@tmp, '.schema'), JSON.generate(schema: 3))
       expect(described_class.needed?).to be(true)
       result = described_class.run(fix: true, backup: false, io: io)
-      expect(result[:applied_migrations]).to eq([4, 5])
+      expect(result[:applied_migrations]).to eq([4, 5, 6, 7])
       creds = YAML.safe_load_file(dec, symbolize_names: true)
       user = PWN::Plugins::Vault.dump(file: yaml, key: creds[:key], iv: creds[:iv])
       expect(user[:ai][:active]).to eq('grok')
@@ -215,7 +215,7 @@ describe PWN::Migrate do
       PWN::Plugins::Vault.create(file: yaml, decryptor_file: dec)
       File.write(File.join(@tmp, '.schema'), JSON.generate(schema: 4))
       result = described_class.run(fix: true, backup: false, io: io)
-      expect(result[:applied_migrations]).to eq([5])
+      expect(result[:applied_migrations]).to eq([5, 6, 7])
       creds = YAML.safe_load_file(dec, symbolize_names: true)
       user = PWN::Plugins::Vault.dump(file: yaml, key: creds[:key], iv: creds[:iv])
       expect(user.dig(:ai, :agent, :skill_review)).to eq('off')
@@ -233,6 +233,43 @@ describe PWN::Migrate do
       expect(described_class.run(fix: true, backup: false, io: io)[:applied_migrations]).to eq([])
     end
 
+    it 'upgrades a schema-5 vault with ai.tui.theme without replacing a chosen color' do
+      yaml = File.join(@tmp, 'pwn.yaml')
+      dec = File.join(@tmp, 'pwn.yaml.decryptor')
+      stale = { ai: { active: 'grok', grok: { key: 'sk-KEEP' }, tui: { theme: { assistant: 'blue' } } } }
+      File.write(yaml, YAML.dump(stale).gsub(/^(\s*):/, '\1'))
+      PWN::Plugins::Vault.create(file: yaml, decryptor_file: dec)
+      File.write(File.join(@tmp, '.schema'), JSON.generate(schema: 5))
+      expect(described_class.run(fix: true, backup: false, io: io)[:applied_migrations]).to eq([6, 7])
+      creds = YAML.safe_load_file(dec, symbolize_names: true)
+      user = PWN::Plugins::Vault.dump(file: yaml, key: creds[:key], iv: creds[:iv])
+      theme = user.dig(:ai, :tui, :theme)
+      expect(theme[:assistant]).to eq('blue')
+      expect(theme[:border]).to eq('black')
+      expect(theme[:title]).to eq('red')
+      expect(theme[:request]).to eq('white')
+      expect(theme).to eq(PWN::Config.env_template.dig(:ai, :tui, :theme).merge(assistant: 'blue'))
+      expect(user.dig(:ai, :grok, :key)).to eq('sk-KEEP')
+      expect(described_class.run(fix: true, backup: false, io: io)[:applied_migrations]).to eq([])
+      expect(PWN::Plugins::Vault.dump(file: yaml, key: creds[:key], iv: creds[:iv]).dig(:ai, :tui, :theme)).to eq(theme)
+    end
+
+    it 'preserves every configured theme role when backfilling a current-schema vault' do
+      yaml = File.join(@tmp, 'pwn.yaml')
+      dec = File.join(@tmp, 'pwn.yaml.decryptor')
+      chosen = PWN::Config.env_template.dig(:ai, :tui, :theme).transform_values { 'blue' }
+      user = PWN::Config.env_template
+      user[:ai][:tui][:theme] = chosen
+      File.write(yaml, YAML.dump(user))
+      PWN::Plugins::Vault.create(file: yaml, decryptor_file: dec)
+      File.write(File.join(@tmp, '.schema'), JSON.generate(schema: described_class::SCHEMA_VERSION))
+      encrypted = File.binread(yaml)
+      expect(described_class.backfill_vault(io: io)[:added]).to eq(0)
+      expect(File.binread(yaml)).to eq(encrypted)
+      creds = YAML.safe_load_file(dec, symbolize_names: true)
+      expect(PWN::Plugins::Vault.dump(file: yaml, key: creds[:key], iv: creds[:iv]).dig(:ai, :tui, :theme)).to eq(chosen)
+    end
+
     it 'dry_run writes nothing' do
       r = PWN::Migrate.run(fix: true, backup: true, dry_run: true, io: io)
       expect(r[:dry_run]).to be true

data/spec/lib/pwn/plugins/repl/ai_console_commands_spec.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe PWN::Plugins::REPL::AIConsoleCommands do
+  it 'completes command names and parameter positions without a provider call' do
+    allow(PWN::Plugins::REPL).to receive(:pwn_ai_engines).and_return(%w[openai grok])
+    first = described_class.complete(line: '/mo', cursor: 3)
+    expect(first[:items].map { |item| item[:label] }).to include('/model')
+    expect(described_class.complete(line: '/', cursor: 1)[:items].map { |item| item[:label] }).to include('/system-role')
+    expect(described_class.complete(line: '/sys', cursor: 4)[:items].map { |item| item[:label] }).to eq(['/system-role'])
+    engines = described_class.complete(line: '/model ', cursor: 7)
+    expect(engines[:items].map { |item| item[:label] }).to include('openai', 'list')
+    expect(engines[:items].map { |item| item[:text] }).to include('/model openai')
+    verbose = described_class.complete(line: '/verbose ', cursor: 9)
+    expect(verbose[:items].map { |item| item[:label] }).to eq(%w[on off])
+    free = described_class.complete(line: '/steer keep ', cursor: 12)
+    expect(free[:items]).to be_empty
+    expect(free[:hint]).to include('free text')
+  end
+
+  it 'completes PWN constants and filesystem paths as they are typed' do
+    hits = described_class.complete(line: 'inspect PWN::Plug', cursor: 16)
+    expect(hits[:items].map { |item| item[:label] }).to include('PWN::Plugins::')
+    expect(hits[:hint]).to eq('PWN constant')
+    Dir.mktmpdir do |dir|
+      File.write(File.join(dir, 'sample.txt'), 'x')
+      path = described_class.complete(line: "#{dir}/sam", cursor: "#{dir}/sam".length)
+      expect(path[:items].map { |item| item[:label] }).to include(File.join(dir, 'sample.txt'))
+      expect(path[:hint]).to eq('path')
+    end
+  end
+
+  it 'does not autoload every constant candidate or fail on a scalar namespace' do
+    mod = Module.new
+    mod.autoload(:Unloaded, '/not/a/real/dependency.rb')
+    mod.const_set(:Scalar, 42)
+    stub_const('PWN::CompletionProbe', mod)
+    result = described_class.complete(line: 'PWN::CompletionProbe::', cursor: 22)
+    expect(result[:items].map { |item| item[:label] }).to include('PWN::CompletionProbe::Unloaded::')
+    expect(mod.autoload?(:Unloaded)).not_to be_nil
+    line = 'PWN::CompletionProbe::Scalar::'
+    expect(described_class.complete(line: line, cursor: line.length)[:items]).to be_empty
+  end
+
+  it 'recognizes absolute paths outside a hardcoded root list without swallowing slash commands' do
+    expect(PWN::Plugins::REPL).to receive(:pwn_ai_complete_path).with(target: '/srv/probe', line: '/srv/probe').and_return(['/srv/probe.txt'])
+    expect(described_class.complete(line: '/srv/probe', cursor: 10)[:items].first[:label]).to eq('/srv/probe.txt')
+    expect(described_class.complete(line: '/mo', cursor: 3)[:items].first[:label]).to eq('/model')
+  end
+
+  it 'preserves a Unicode suffix when replacing the token at the cursor' do
+    result = described_class.complete(line: '/verbose o café', cursor: 10)
+    chosen = result[:items].find { |item| item[:label] == 'on' }
+    expect(chosen[:text]).to eq('/verbose on café')
+    expect(chosen[:text].scan(/\X/)[chosen[:cursor]]).to eq(' ')
+  end
+
+  it 'keeps every command reachable and completes only the prefix before a mid-token cursor' do
+    labels = described_class.complete(line: '/', cursor: 1)[:items].map { |item| item[:label] }
+    expect(labels).to include('/swarm', '/verbose', '/steer', '/status')
+    result = described_class.complete(line: '/verbose ox café', cursor: 10)
+    expect(result[:items].map { |item| item[:text] }).to include('/verbose on café')
+  end
+end