[Bug]: Kiro compaction calls an extension the running agent did not advertise #35832

import { mkdtempSync, mkdirSync, writeFileSync, rmSync, readFileSync } from 'node:fs';
import { tmpdir } from 'node:os';
import { resolve, join } from 'node:path';
import { pathToFileURL } from 'node:url';
import { createRequire } from 'node:module';
import { execFileSync } from 'node:child_process';

const repo = resolve(process.argv[2]);
const scratch = mkdtempSync(join(tmpdir(), 'kiro-extension-repro-'));
let code = 2;
try {
  for (const dir of ['home', 'tmp', 'state', 'cwd']) mkdirSync(join(scratch, dir));
  Object.assign(process.env, { HOME: join(scratch, 'home'), TMPDIR: join(scratch, 'tmp'), T3CODE_HOME: join(scratch, 'state') });
  process.chdir(join(scratch, 'cwd'));
  const require = createRequire(join(repo, 'apps/server/package.json'));
  const load = name => {
    if (!name.startsWith('@t3tools/')) return import(pathToFileURL(require.resolve(name)).href);
    const [, pkg, ...rest] = name.split('/');
    const base = join(repo, 'packages', pkg);
    const manifest = JSON.parse(readFileSync(join(base, 'package.json')));
    return import(pathToFileURL(join(base, manifest.exports['./' + rest.join('/')].import)).href);
  };
  const [Effect, Exit, Layer, Stream, DateTime, Schema, NodeServices, NodeCrypto, TestProviderHost, IdAllocator, McpProviderSessions, ProviderAdapter, Kiro, KiroSettings, Runtime] = await Promise.all([
    'effect/Effect', 'effect/Exit', 'effect/Layer', 'effect/Stream', 'effect/DateTime', 'effect/Schema',
    '@effect/platform-node/NodeServices', '@effect/platform-node/NodeCrypto',
    '@t3tools/provider-testing/TestProviderHost', '@t3tools/provider-core/server/IdAllocator',
    '@t3tools/provider-core/server/McpProviderSessions', '@t3tools/provider-core/server/ProviderAdapter',
    '@t3tools/provider-kiro/testing', '@t3tools/provider-kiro/settings', '@t3tools/provider-acp/server/AcpSessionRuntime',
  ].map(load));
  console.log('Node ' + process.version);
  console.log(execFileSync('git', ['--version'], { encoding: 'utf8' }).trim());
  console.log('Revision ' + execFileSync('git', ['-C', repo, 'rev-parse', 'HEAD'], { encoding: 'utf8' }).trim());
  console.log('Effect ' + JSON.parse(readFileSync(require.resolve('effect/package.json'))).version);
  const peer = join(scratch, 'peer.mjs');
  writeFileSync(peer, `import { createInterface } from 'node:readline';
const supports=process.argv[2]==='supported';
for await (const line of createInterface({input:process.stdin})) {
 const r=JSON.parse(line); if(r.id===undefined)continue;
 let result={},error;
 if(r.method==='initialize')result={protocolVersion:1,agentCapabilities:{loadSession:true,_meta:{kiro:{extensionMethods:supports?['_kiro/session/compact']:[]}}},authMethods:[]};
 else if(r.method==='session/new')result={sessionId:'fixture-session',configOptions:[]};
 else if(r.method==='_kiro/session/compact') { if(supports)result={success:true}; else error={code:-32601,message:'Compaction extension not advertised'}; }
 else if(r.method==='session/prompt')result={stopReason:'end_turn'};
 else error={code:-32601,message:'Unsupported fixture method'};
 process.stdout.write(JSON.stringify({jsonrpc:'2.0',id:r.id,...(error?{error}:{result})})+'\\n');
}
`);
  const layer = Layer.mergeAll(NodeServices.layer, NodeCrypto.layer, IdAllocator.layer, McpProviderSessions.layer,
    TestProviderHost.layer().pipe(Layer.provide(Layer.mergeAll(NodeServices.layer, NodeCrypto.layer))));
  const run = async supports => {
    const methods=[];
    const exit = await Effect.runPromiseExit(Effect.gen(function* () {
      const adapter=yield* Kiro.makeKiroAdapterV2({
        instanceId:'kiro-fixture', settings:Schema.decodeSync(KiroSettings.KiroSettings)({enabled:true}), environment:{},
        selfInvocation:{command:process.execPath,args:[]},
        makeRuntime:input=>Runtime.make({...input,
          spawn:{command:process.execPath,args:[peer,supports?'supported':'unsupported'],extendEnv:false,env:{HOME:process.env.HOME,TMPDIR:process.env.TMPDIR,T3CODE_HOME:process.env.T3CODE_HOME}},
          requestLogger:event=>Effect.sync(()=>{if(event.status==='started')methods.push(event.method);}),
        }),
      });
      const modelSelection={instanceId:'kiro-fixture',model:'default'};
      const runtimePolicy=ProviderAdapter.ProviderAdapterV2RuntimePolicy.make({runtimeMode:'approval-required',interactionMode:'default',cwd:join(scratch,'cwd')});
      const session=yield* adapter.openSession({threadId:'fixture-thread',providerSessionId:'fixture-provider-session',modelSelection,runtimePolicy});
      const providerThread=yield* session.ensureThread({threadId:'fixture-thread',modelSelection,runtimePolicy});
      const now=yield* DateTime.now;
      yield* session.compactThread({
        appThread:{createdBy:'user',creationSource:'web',id:'fixture-thread',projectId:'fixture-project',title:'Fixture',providerInstanceId:'kiro-fixture',modelSelection,runtimeMode:'approval-required',interactionMode:'default',branch:null,worktreePath:null,activeProviderThreadId:providerThread.id,lineage:{parentThreadId:null,relationshipToParent:null,rootThreadId:'fixture-thread'},forkedFrom:null,createdAt:now,updatedAt:now,archivedAt:null,settledOverride:null,settledAt:null,lastVisitedAt:null,deletedAt:null},
        threadId:'fixture-thread',runId:'fixture-run',runOrdinal:1,providerTurnOrdinal:1,attemptId:'fixture-attempt',rootNodeId:'fixture-node',providerThread,
        message:{createdBy:'user',creationSource:'web',messageId:'fixture-message',text:'/compact',attachments:[]},modelSelection,runtimePolicy,
      });
      const events=yield* session.events.pipe(Stream.takeUntil(e=>e.type==='turn.terminal'),Stream.runCollect);
      return [...events].find(e=>e.type==='turn.terminal')?.status;
    }).pipe(Effect.scoped,Effect.provide(layer)));
    console.log(JSON.stringify({advertisedCompact:supports,methods,outcome:exit._tag,status:Exit.isSuccess(exit)?exit.value:undefined}));
    return {exit,methods};
  };
  const control=await run(true);
  if(!Exit.isSuccess(control.exit)||control.exit.value!=='completed'||!control.methods.includes('_kiro/session/compact'))throw new Error('Control failed: '+String(control.exit));
  const failing=await run(false);
  if(!failing.methods.includes('_kiro/session/compact'))code=0;
  else if(Exit.isSuccess(failing.exit)&&failing.exit.value==='failed')code=1;
} catch(error) { console.log('Setup error: '+error.message); }
finally { rmSync(scratch,{recursive:true,force:true}); }
console.log(code===0?'RESULT: expected behavior':code===1?'RESULT: defect reproduced':'RESULT: inconclusive');
process.exitCode=code;