Types

import { actor } from "rivetkit";
import { db } from "rivetkit/db";

const counter = actor({
  db: db({
    onMigrate: async (db) => {
      await db.execute(
        "CREATE TABLE IF NOT EXISTS counter (id INTEGER PRIMARY KEY CHECK (id = 1), count INTEGER NOT NULL)",
      );
    },
  }),

  // CreateContext in onCreate hook
  onCreate: async (c, input: { initial: number }) => {
    await c.db.execute("INSERT INTO counter (id, count) VALUES (1, ?)", input.initial);
  },

  // ActionContext in actions
  actions: {
    increment: async (c) => {
      await c.db.execute("UPDATE counter SET count = count + 1 WHERE id = 1");
      const [row] = (await c.db.execute(
        "SELECT count FROM counter WHERE id = 1",
      )) as { count: number }[];
      return row.count;
    }
  }
});

import { actor, CreateContextOf, ActionContextOf } from "rivetkit";

const gameRoom = actor({
  onCreate: (c, input: { roomId: string }) => {
    initializeRoom(c, input.roomId);
  },

  createVars: () => ({ players: [] as string[] }),

  actions: {
    addPlayer: (c, playerId: string) => {
      validatePlayer(c, playerId);
      c.vars.players.push(playerId);
    }
  }
});

// Extract CreateContext type for onCreate hook
function initializeRoom(
  context: CreateContextOf<typeof gameRoom>,
  roomId: string
) {
  console.log(`Initializing room: ${roomId}`);
  // context.vars is not available here (not created yet)
}

// Extract ActionContext type for actions
function validatePlayer(
  context: ActionContextOf<typeof gameRoom>,
  playerId: string
) {
  // Full context available in actions
  if (context.vars.players.includes(playerId)) {
    throw new Error("Player already in room");
  }
}