chore(integration): snapshot local main worktree
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
import path from 'node:path';
|
||||
import { atomicWriteText } from '../../../coding-projects/atomic-json';
|
||||
|
||||
export const MAKELORE_PI_EXTENSION_VERSION = 3;
|
||||
export const MAKELORE_PI_EXTENSION_VERSION = 4;
|
||||
export const MAKELORE_PI_EXTENSION_FILENAME = `makelore-runtime-v${MAKELORE_PI_EXTENSION_VERSION}.mjs`;
|
||||
|
||||
const BUNDLE_SOURCE = String.raw`
|
||||
@@ -15,118 +15,156 @@ const MUTATION_TOOLS = new Set([
|
||||
'game_asset_browser',
|
||||
'game_asset_review',
|
||||
]);
|
||||
const WORKER_ROLE = process.env.MAKELORE_PI_WORKER_ROLE || 'parent';
|
||||
const leases = new Map();
|
||||
const touchedPaths = new Map();
|
||||
|
||||
async function runtimeContext() {
|
||||
const value = JSON.parse(await readFile(process.env.MAKELORE_PI_CONTEXT_FILE, 'utf8'));
|
||||
if (!value.runId) throw new Error('Makelore run context is unavailable');
|
||||
return value;
|
||||
}
|
||||
|
||||
async function bridge(action, body, signal) {
|
||||
const context = await runtimeContext();
|
||||
const response = await fetch(process.env.MAKELORE_PI_BRIDGE_URL, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
authorization: 'Bearer ' + process.env.MAKELORE_PI_WORKER_TOKEN,
|
||||
'content-type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({ ...context, action, ...body }),
|
||||
signal,
|
||||
});
|
||||
const result = await response.json().catch(() => ({}));
|
||||
if (!response.ok) throw new Error(result.error || 'Makelore runtime bridge rejected the request');
|
||||
return result;
|
||||
}
|
||||
|
||||
async function bridgeStream(action, body, signal, onUpdate) {
|
||||
const context = await runtimeContext();
|
||||
const response = await fetch(process.env.MAKELORE_PI_BRIDGE_URL, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
authorization: 'Bearer ' + process.env.MAKELORE_PI_WORKER_TOKEN,
|
||||
'content-type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({ ...context, action, ...body }),
|
||||
signal,
|
||||
});
|
||||
if (!response.ok) {
|
||||
const result = await response.json().catch(() => ({}));
|
||||
throw new Error(result.error || 'Makelore runtime bridge rejected the request');
|
||||
}
|
||||
if (!response.body) throw new Error('Makelore runtime bridge returned no stream');
|
||||
const reader = response.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buffered = '';
|
||||
let details;
|
||||
while (true) {
|
||||
const { value, done } = await reader.read();
|
||||
buffered += decoder.decode(value || new Uint8Array(), { stream: !done });
|
||||
let newline = buffered.indexOf('\n');
|
||||
while (newline >= 0) {
|
||||
const line = buffered.slice(0, newline);
|
||||
buffered = buffered.slice(newline + 1);
|
||||
if (line) {
|
||||
const item = JSON.parse(line);
|
||||
if (item.details?.schema === 'subagent.v1') {
|
||||
details = item.details;
|
||||
onUpdate?.({ content: [], details });
|
||||
}
|
||||
}
|
||||
newline = buffered.indexOf('\n');
|
||||
}
|
||||
if (done) break;
|
||||
}
|
||||
if (!details) throw new Error('Makelore subagent stream returned no details');
|
||||
return details;
|
||||
}
|
||||
|
||||
async function releaseLease(toolCallId) {
|
||||
const leaseId = leases.get(toolCallId);
|
||||
if (!leaseId) return;
|
||||
leases.delete(toolCallId);
|
||||
await bridge('lease.release', { leaseId, resourceId: toolCallId }).catch(() => undefined);
|
||||
}
|
||||
|
||||
async function releaseAll() {
|
||||
await Promise.all([...leases.keys()].map(releaseLease));
|
||||
touchedPaths.clear();
|
||||
}
|
||||
|
||||
function projectRelativePath(value) {
|
||||
if (typeof value !== 'string' || !value.trim()) return undefined;
|
||||
const projectPath = path.resolve(process.cwd());
|
||||
const targetPath = path.resolve(projectPath, value);
|
||||
const relative = path.relative(projectPath, targetPath);
|
||||
if (!relative || relative.startsWith('..') || path.isAbsolute(relative)) return undefined;
|
||||
return relative.split(path.sep).join('/');
|
||||
}
|
||||
|
||||
async function invokeProduct(toolCallId, toolName, input, signal) {
|
||||
const response = await bridge('product.invoke', {
|
||||
resourceId: toolCallId,
|
||||
toolName,
|
||||
input,
|
||||
}, signal);
|
||||
return response.result;
|
||||
}
|
||||
|
||||
function registerProductTool(pi, name, label, description, parameters) {
|
||||
pi.registerTool({
|
||||
name,
|
||||
label,
|
||||
description,
|
||||
parameters,
|
||||
async execute(toolCallId, params, signal) {
|
||||
return await invokeProduct(toolCallId, name, params, signal);
|
||||
},
|
||||
});
|
||||
}
|
||||
const RUNTIME_FLAGS = {
|
||||
bridgeUrl: 'makelore-bridge-url',
|
||||
workerToken: 'makelore-worker-token',
|
||||
contextFile: 'makelore-context-file',
|
||||
workerRole: 'makelore-worker-role',
|
||||
projectPath: 'makelore-project-path',
|
||||
};
|
||||
|
||||
export default function makeloreRuntime(pi) {
|
||||
if (WORKER_ROLE === 'parent') pi.registerTool({
|
||||
const fallbackFlagValues = new Map();
|
||||
const registerRuntimeFlag = (name, defaultValue) => {
|
||||
fallbackFlagValues.set(name, defaultValue);
|
||||
if (typeof pi.registerFlag === 'function') {
|
||||
pi.registerFlag(name, { type: 'string', default: defaultValue });
|
||||
}
|
||||
};
|
||||
registerRuntimeFlag(RUNTIME_FLAGS.bridgeUrl, process.env.MAKELORE_PI_BRIDGE_URL || '');
|
||||
registerRuntimeFlag(RUNTIME_FLAGS.workerToken, process.env.MAKELORE_PI_WORKER_TOKEN || '');
|
||||
registerRuntimeFlag(RUNTIME_FLAGS.contextFile, process.env.MAKELORE_PI_CONTEXT_FILE || '');
|
||||
registerRuntimeFlag(RUNTIME_FLAGS.workerRole, process.env.MAKELORE_PI_WORKER_ROLE || 'parent');
|
||||
registerRuntimeFlag(
|
||||
RUNTIME_FLAGS.projectPath,
|
||||
process.env.MAKELORE_PI_PROJECT_PATH || process.cwd(),
|
||||
);
|
||||
|
||||
const runtimeValue = (name) => {
|
||||
const value = typeof pi.getFlag === 'function'
|
||||
? pi.getFlag(name)
|
||||
: fallbackFlagValues.get(name);
|
||||
return typeof value === 'string' ? value : '';
|
||||
};
|
||||
const workerRole = runtimeValue(RUNTIME_FLAGS.workerRole) || 'parent';
|
||||
const leases = new Map();
|
||||
const touchedPaths = new Map();
|
||||
|
||||
async function runtimeContext() {
|
||||
const contextFile = runtimeValue(RUNTIME_FLAGS.contextFile);
|
||||
if (!contextFile) throw new Error('Makelore runtime context file is unavailable');
|
||||
const value = JSON.parse(await readFile(contextFile, 'utf8'));
|
||||
if (!value.runId) throw new Error('Makelore run context is unavailable');
|
||||
return value;
|
||||
}
|
||||
|
||||
async function bridge(action, body, signal) {
|
||||
const context = await runtimeContext();
|
||||
const bridgeUrl = runtimeValue(RUNTIME_FLAGS.bridgeUrl);
|
||||
const workerToken = runtimeValue(RUNTIME_FLAGS.workerToken);
|
||||
if (!bridgeUrl || !workerToken) throw new Error('Makelore runtime bridge is unavailable');
|
||||
const response = await fetch(bridgeUrl, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
authorization: 'Bearer ' + workerToken,
|
||||
'content-type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({ ...context, action, ...body }),
|
||||
signal,
|
||||
});
|
||||
const result = await response.json().catch(() => ({}));
|
||||
if (!response.ok) throw new Error(result.error || 'Makelore runtime bridge rejected the request');
|
||||
return result;
|
||||
}
|
||||
|
||||
async function bridgeStream(action, body, signal, onUpdate) {
|
||||
const context = await runtimeContext();
|
||||
const bridgeUrl = runtimeValue(RUNTIME_FLAGS.bridgeUrl);
|
||||
const workerToken = runtimeValue(RUNTIME_FLAGS.workerToken);
|
||||
if (!bridgeUrl || !workerToken) throw new Error('Makelore runtime bridge is unavailable');
|
||||
const response = await fetch(bridgeUrl, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
authorization: 'Bearer ' + workerToken,
|
||||
'content-type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({ ...context, action, ...body }),
|
||||
signal,
|
||||
});
|
||||
if (!response.ok) {
|
||||
const result = await response.json().catch(() => ({}));
|
||||
throw new Error(result.error || 'Makelore runtime bridge rejected the request');
|
||||
}
|
||||
if (!response.body) throw new Error('Makelore subagent stream returned no body');
|
||||
const reader = response.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buffered = '';
|
||||
let details;
|
||||
while (true) {
|
||||
const { value, done } = await reader.read();
|
||||
buffered += decoder.decode(value || new Uint8Array(), { stream: !done });
|
||||
let newline = buffered.indexOf('\n');
|
||||
while (newline >= 0) {
|
||||
const line = buffered.slice(0, newline);
|
||||
buffered = buffered.slice(newline + 1);
|
||||
if (line) {
|
||||
const item = JSON.parse(line);
|
||||
if (item.details?.schema === 'subagent.v1') {
|
||||
details = item.details;
|
||||
onUpdate?.({ content: [], details });
|
||||
}
|
||||
}
|
||||
newline = buffered.indexOf('\n');
|
||||
}
|
||||
if (done) break;
|
||||
}
|
||||
if (!details) throw new Error('Makelore subagent stream returned no details');
|
||||
return details;
|
||||
}
|
||||
|
||||
async function releaseLease(toolCallId) {
|
||||
const leaseId = leases.get(toolCallId);
|
||||
if (!leaseId) return;
|
||||
leases.delete(toolCallId);
|
||||
await bridge('lease.release', { leaseId, resourceId: toolCallId }).catch(() => undefined);
|
||||
}
|
||||
|
||||
async function releaseAll() {
|
||||
await Promise.all([...leases.keys()].map(releaseLease));
|
||||
touchedPaths.clear();
|
||||
}
|
||||
|
||||
function projectRelativePath(value) {
|
||||
if (typeof value !== 'string' || !value.trim()) return undefined;
|
||||
const projectPath = path.resolve(runtimeValue(RUNTIME_FLAGS.projectPath) || process.cwd());
|
||||
const targetPath = path.resolve(projectPath, value);
|
||||
const relative = path.relative(projectPath, targetPath);
|
||||
if (!relative || relative.startsWith('..') || path.isAbsolute(relative)) return undefined;
|
||||
return relative.split(path.sep).join('/');
|
||||
}
|
||||
|
||||
async function invokeProduct(toolCallId, toolName, input, signal) {
|
||||
const response = await bridge('product.invoke', {
|
||||
resourceId: toolCallId,
|
||||
toolName,
|
||||
input,
|
||||
}, signal);
|
||||
return response.result;
|
||||
}
|
||||
|
||||
function registerProductTool(name, label, description, parameters) {
|
||||
pi.registerTool({
|
||||
name,
|
||||
label,
|
||||
description,
|
||||
parameters,
|
||||
async execute(toolCallId, params, signal) {
|
||||
return await invokeProduct(toolCallId, name, params, signal);
|
||||
},
|
||||
});
|
||||
}
|
||||
|
||||
if (workerRole === 'parent') pi.registerTool({
|
||||
name: 'ask_user',
|
||||
label: 'Ask user',
|
||||
description: 'Ask the user for a selection, confirmation, short input, or editor text.',
|
||||
@@ -159,7 +197,7 @@ export default function makeloreRuntime(pi) {
|
||||
},
|
||||
});
|
||||
|
||||
if (WORKER_ROLE === 'parent') pi.registerTool({
|
||||
if (workerRole === 'parent') pi.registerTool({
|
||||
name: 'subagent',
|
||||
label: 'Subagent',
|
||||
description: 'Dispatch one or more managed project Agents in single, parallel, or chain mode.',
|
||||
@@ -205,8 +243,7 @@ export default function makeloreRuntime(pi) {
|
||||
},
|
||||
});
|
||||
|
||||
if (WORKER_ROLE === 'parent') registerProductTool(
|
||||
pi,
|
||||
if (workerRole === 'parent') registerProductTool(
|
||||
'agent_browser',
|
||||
'Agent browser',
|
||||
'Open, navigate, inspect, or close the Main-owned Makelore development browser.',
|
||||
@@ -222,15 +259,13 @@ export default function makeloreRuntime(pi) {
|
||||
},
|
||||
},
|
||||
);
|
||||
if (WORKER_ROLE === 'parent') registerProductTool(
|
||||
pi,
|
||||
if (workerRole === 'parent') registerProductTool(
|
||||
'game_asset_browser',
|
||||
'Game assets',
|
||||
'Load product-owned game asset candidates and their current review state.',
|
||||
{ type: 'object', additionalProperties: false, properties: { invocationId: { type: 'string' } } },
|
||||
);
|
||||
if (WORKER_ROLE === 'parent') registerProductTool(
|
||||
pi,
|
||||
if (workerRole === 'parent') registerProductTool(
|
||||
'game_asset_review',
|
||||
'Game asset review',
|
||||
'Load one versioned game asset review interaction without encoding decisions in message text.',
|
||||
@@ -242,8 +277,7 @@ export default function makeloreRuntime(pi) {
|
||||
},
|
||||
},
|
||||
);
|
||||
if (WORKER_ROLE === 'parent') registerProductTool(
|
||||
pi,
|
||||
if (workerRole === 'parent') registerProductTool(
|
||||
'task_state',
|
||||
'Task state',
|
||||
'Publish versioned task steps and progress to the Makelore product timeline.',
|
||||
@@ -263,8 +297,7 @@ export default function makeloreRuntime(pi) {
|
||||
},
|
||||
},
|
||||
);
|
||||
if (WORKER_ROLE === 'parent') registerProductTool(
|
||||
pi,
|
||||
if (workerRole === 'parent') registerProductTool(
|
||||
'changed_file',
|
||||
'Changed file',
|
||||
'Report project-relative paths touched by managed tools and refresh Conversation changes.',
|
||||
@@ -276,8 +309,7 @@ export default function makeloreRuntime(pi) {
|
||||
},
|
||||
},
|
||||
);
|
||||
if (WORKER_ROLE === 'parent') registerProductTool(
|
||||
pi,
|
||||
if (workerRole === 'parent') registerProductTool(
|
||||
'runtime_context',
|
||||
'Runtime context',
|
||||
'Read the safe selected-skill and command catalog for this managed worker.',
|
||||
|
||||
Reference in New Issue
Block a user