Runtime exports

export function getTools() {
  return [
    {
      type: 'function',
      function: {
        name: 'send_snapshot',
        description: 'Send the user a photo matching the description.',
        parameters: {
          type: 'object',
          properties: {
            description: { type: 'string', description: 'What the photo shows.' },
            mood: { type: 'string', enum: ['soft', 'teasing', 'stern'] },
          },
          required: ['description'],
        },
      },
      // Optional. Only the chat UI sees these; they are stripped before the model does.
      displayLabels: {
        pending: 'Sending photo…',
        success: 'Sent you a photo',
        error: "Couldn't send the photo",
      },
    },
  ];
}

export async function executeTool(name, args, ctx) {
  if (name !== 'send_snapshot') throw new Error(`Unknown tool: ${name}`);

  const record = await ctx.runtime.emitImage({
    url: await renderPhoto(args.description, args.mood),
    prompt: args.description,
    sourceExtensionKey: ctx.extensionKey,
    sourceToolName: name,
    sourceKind: 'tool',
    displayMode: 'gallery',
  });

  // Whatever you return is serialised and sent back to the model as the tool result.
  return { ok: true, artifactId: record.id };
}

const state = { connected: false, bpm: null };

export function useExtension() {
  const React = globalThis.React;
  const [, rerender] = React.useState(0);

  React.useEffect(() => {
    const id = setInterval(() => rerender((n) => n + 1), 1000);
    return () => clearInterval(id);
  }, []);

  return {
    isInstalled: true,
    isAvailable: true,
    isEnabled: true,        // marketplace installs replace this with the persona toggle
    isActive: state.connected,
    statusIndicator: state.connected
      ? { color: 'green', label: 'Connected' }
      : { color: 'yellow', label: 'Ready' },
    globalSettings: {},     // replaced by the host
    personaSettings: {},    // replaced by the host
    resolvedSettings: {},   // replaced by the host
    promptState: state.bpm == null ? undefined : {
      playerInfo: [
        { kind: 'player_info', id: 'hr', label: 'Current Heart Rate', visibility: 'visible', value: state.bpm, unit: 'bpm' },
      ],
    },
    enable: async () => {},
    disable: async () => {},
    updateGlobalSettings: async () => {},
    updatePersonaSettings: async () => {},
    updateSettings: async () => {},
  };
}

export async function onBeforeUserMessageSend(ctx) {
  if (/\bpause\b/i.test(ctx.userMessage || '')) {
    return { blocked: true, message: 'Paused. Send anything else to continue.' };
  }
}

export async function onBeforeLLMRequest(ctx) {
  // Hide our own tool for this round without touching anyone else's.
  const mine = ctx.resolveOwnToolName('send_snapshot');
  return {
    tools: ctx.tools.filter((tool) => tool.function.name !== mine),
    promptContributions: [
      { placement: 'system_post', text: 'Photos are unavailable this round.' },
    ],
  };
}

export async function onAfterAssistantMessage(ctx) {
  const praised = /good girl/i.test(ctx.assistantText || '');
  if (!praised) return;
  const current = await ctx.runtime.variables.get('praise_count', { scope: 'persona', defaultValue: 0 });
  await ctx.runtime.variables.set('praise_count', Number(current.value) + 1, { scope: 'persona' });
}