import { ponder } from "ponder:registry";
import schema from "ponder:schema";

ponder.on("ERC1155:TransferSingle", async ({ event, context }) => {
  // Create an Account for the sender, or update the balance if it already exists.
  await context.db
    .insert(schema.account)
    .values({ address: event.args.from })
    .onConflictDoNothing();

  await context.db
    .insert(schema.tokenBalance)
    .values({
      owner: event.args.from,
      tokenId: event.args.id,
      balance: -event.args.amount,
    })
    .onConflictDoUpdate((row) => ({
      balance: row.balance - event.args.amount,
    }));

  // Create an Account for the recipient, or update the balance if it already exists.
  await context.db
    .insert(schema.account)
    .values({ address: event.args.to })
    .onConflictDoNothing();

  await context.db
    .insert(schema.tokenBalance)
    .values({
      owner: event.args.to,
      tokenId: event.args.id,
      balance: event.args.amount,
    })
    .onConflictDoUpdate((row) => ({
      balance: row.balance + event.args.amount,
    }));

  // Create a TransferEvent.
  await context.db.insert(schema.transferEvent).values({
    id: event.id,
    from: event.args.from,
    to: event.args.to,
    token: event.args.id,
    timestamp: Number(event.block.timestamp),
  });
});

ponder.on("ERC1155:TransferBatch", async ({ event, context }) => {
  await context.db
    .insert(schema.account)
    .values({ address: event.args.from })
    .onConflictDoNothing();
  await context.db
    .insert(schema.account)
    .values({ address: event.args.to })
    .onConflictDoNothing();

  for (let i = 0; i < event.args.ids.length; i++) {
    const id = event.args.ids[i]!;
    const amount = event.args.amounts[i]!;

    await context.db
      .insert(schema.tokenBalance)
      .values({
        owner: event.args.from,
        tokenId: id,
        balance: -amount,
      })
      .onConflictDoUpdate((row) => ({
        balance: row.balance - amount,
      }));

    await context.db
      .insert(schema.tokenBalance)
      .values({
        owner: event.args.to,
        tokenId: id,
        balance: amount,
      })
      .onConflictDoUpdate((row) => ({
        balance: row.balance + amount,
      }));

    await context.db.insert(schema.transferEvent).values({
      id: `${event.id}-${i}`,
      from: event.args.from,
      to: event.args.to,
      token: id,
      timestamp: Number(event.block.timestamp),
    });
  }
});