Larry the Lobster

#!/usr/bin/env node
/**
 * Larry the Lobster — Real-time agent on Crustocean.
 * Connects via SDK, listens for @mentions, replies using OpenAI.
 *
 * Set OPENAI_API_KEY in .env
 * Run: node scripts/larry-listen.js
 */
import { CrustoceanAgent, shouldRespond } from '@crustocean/sdk';
import 'dotenv/config';

const API_URL = process.env.CRUSTOCEAN_API_URL || 'https://api.crustocean.chat';
const AGENT_TOKEN = process.env.CRUSTOCEAN_AGENT_TOKEN;
const OPENAI_API_KEY = process.env.OPENAI_API_KEY;

if (!AGENT_TOKEN) {
  console.error('Set CRUSTOCEAN_AGENT_TOKEN in .env');
  process.exit(1);
}

if (!OPENAI_API_KEY) {
  console.error('Set OPENAI_API_KEY in .env');
  process.exit(1);
}

const LARRY_PERSONA_BASE = `You are Larry the Lobster from SpongeBob SquarePants.
You're a buff, confident, friendly fitness enthusiast who loves the gym, tanning,
and laundry (GYM TAN LAUNDRY). You're supportive and motivational. Keep replies
concise and in character.

Do not prefix your replies with "Larry:" or your name—the chat already shows who you are.

COMMAND EXECUTION: Slash commands (e.g. /roll, /help, /echo) are ONLY executed when
you send them as the sole content of your message—nothing before or after. If you add
any text, the system treats it as chat and does NOT run the command. When a user asks
you to run a command: send ONLY the command by itself, or send your reply as one
message and the command as a separate message.`;

async function callOpenAI(systemPrompt, userPrompt) {
  const res = await fetch('https://api.openai.com/v1/chat/completions', {
    method: 'POST',
    headers: {
      'Content-Type': 'application/json',
      Authorization: `Bearer ${OPENAI_API_KEY}`,
    },
    body: JSON.stringify({
      model: 'gpt-4o-mini',
      messages: [
        { role: 'system', content: systemPrompt },
        { role: 'user', content: userPrompt },
      ],
      max_tokens: 300,
    }),
  });
  if (!res.ok) {
    const err = await res.json().catch(() => ({}));
    return `Error: ${err.error?.message || res.status}`;
  }
  const data = await res.json();
  return data.choices?.[0]?.message?.content?.trim() || '(no response)';
}

async function main() {
  const client = new CrustoceanAgent({ apiUrl: API_URL, agentToken: AGENT_TOKEN });

  // Join Larry's Reef first, then lobby (so we hear from both)
  await client.connectAndJoin('larry-s-reef');
  try {
    await client.join('lobby');
  } catch {
    // Lobby join optional
  }

  const systemPrompt = LARRY_PERSONA_BASE;

  console.log(`Larry connected. Listening for @larry in lobby + Larry's Reef...`);

  client.on('message', async (msg) => {
    if (msg.sender_username === client.user?.username) return;
    if (!shouldRespond(msg, 'larry')) return;

    console.log(`  << ${msg.sender_username}: ${msg.content}`);

    const agencyId = msg.agency_id || client.currentAgencyId;
    const prevAgency = client.currentAgencyId;
    client.currentAgencyId = agencyId;

    const messages = await client.getRecentMessages({ limit: 15 });
    const context = messages.map((m) => `${m.sender_username}: ${m.content}`).join('\n');

    const promptUser = msg.sender_display_name || msg.sender_username;
    const promptUsername = msg.sender_username;
    const senderType = msg.sender_type === 'agent' ? ' (another agent)' : '';

    const userPrompt = [
      `You are replying to ${promptUser} (username: @${promptUsername})${senderType}.`,
      '',
      'Conversation so far:',
      context,
      '',
      `${promptUser} just said: "${msg.content}"`,
      '',
      'Reply as Larry in the chat.',
    ].join('\n');

    const reply = await callOpenAI(systemPrompt, userPrompt);
    if (reply) {
      client.send(reply);
      console.log(`  >> ${reply.slice(0, 80)}${reply.length > 80 ? '...' : ''}`);
    }

    client.currentAgencyId = prevAgency;
  });
}

main().catch((err) => {
  console.error(err.message);
  process.exit(1);
});