wehub-resource-sync
86 lines
2.5 KiB
JavaScript
86 lines
2.5 KiB
JavaScript
const fs = require('fs');
|
|
const path = require('path');
|
|
const os = require('os');
|
|
const { getClaudeDir, getConfigDir } = require('./ponytail-config');
|
|
|
|
const STATE_FILE = '.ponytail-active';
|
|
const isCopilot = Boolean(process.env.COPILOT_PLUGIN_DATA);
|
|
const isCodex = !isCopilot && Boolean(process.env.PLUGIN_DATA);
|
|
const isQoder = !isCopilot && !isCodex && Boolean(process.env.QODER_SESSION_ID);
|
|
|
|
let stateDir = getClaudeDir();
|
|
if (isCodex) stateDir = process.env.PLUGIN_DATA;
|
|
if (isCopilot) stateDir = process.env.COPILOT_PLUGIN_DATA;
|
|
if (isQoder) stateDir = path.join(os.homedir(), '.qoder');
|
|
|
|
const statePath = path.join(stateDir, STATE_FILE);
|
|
|
|
function setMode(mode) {
|
|
fs.mkdirSync(path.dirname(statePath), { recursive: true });
|
|
fs.writeFileSync(statePath, mode);
|
|
}
|
|
|
|
function clearMode() {
|
|
try { fs.unlinkSync(statePath); } catch (e) {}
|
|
}
|
|
|
|
// Live mode written by activate/mode-tracker. Absent flag = ponytail off.
|
|
function readMode() {
|
|
try {
|
|
return fs.readFileSync(statePath, 'utf8').trim() || null;
|
|
} catch (e) {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
function writeHookOutput(event, mode, context = '') {
|
|
if (isCopilot) {
|
|
// Copilot reads additionalContext on SessionStart; ignores output elsewhere.
|
|
process.stdout.write(JSON.stringify(
|
|
event === 'SessionStart' && context ? { additionalContext: context } : {}));
|
|
return;
|
|
}
|
|
if (isCodex) {
|
|
const output = { systemMessage: `PONYTAIL:${mode.toUpperCase()}` };
|
|
if (context) {
|
|
output.hookSpecificOutput = {
|
|
hookEventName: event,
|
|
additionalContext: context,
|
|
};
|
|
}
|
|
process.stdout.write(JSON.stringify(output));
|
|
return;
|
|
}
|
|
if (isQoder) {
|
|
// Qoder: hookSpecificOutput JSON, same shape as Codex minus systemMessage.
|
|
// UserPromptSubmit additionalContext is injected into the Agent's conversation.
|
|
const output = {};
|
|
if (context) {
|
|
output.hookSpecificOutput = {
|
|
hookEventName: event,
|
|
additionalContext: context,
|
|
};
|
|
}
|
|
process.stdout.write(JSON.stringify(output));
|
|
return;
|
|
}
|
|
// Native Claude: SessionStart accepts raw stdout, but SubagentStart needs the
|
|
// hookSpecificOutput JSON form or the context is dropped.
|
|
if (event === 'SubagentStart') {
|
|
process.stdout.write(JSON.stringify(
|
|
{ hookSpecificOutput: { hookEventName: event, additionalContext: context } }));
|
|
return;
|
|
}
|
|
process.stdout.write(context);
|
|
}
|
|
|
|
module.exports = {
|
|
clearMode,
|
|
isCodex,
|
|
isCopilot,
|
|
isQoder,
|
|
readMode,
|
|
setMode,
|
|
writeHookOutput,
|
|
};
|