import { Address, BigInt, store } from '@graphprotocol/graph-ts';
import { ProposalStake } from '../types/schema';
import { addStakeEvent } from './event';

export function getStake(id: string): ProposalStake {
  let stake = store.get('ProposalStake', id) as ProposalStake;
  if (stake == null) {
    stake = new ProposalStake(id);
  }
  return stake;
}

export function saveStake(stake: ProposalStake): void {
  store.set('ProposalStake', stake.id, stake);
}

export function insertStake(
  eventId: string,
  timestamp: BigInt,
  staker: Address,
  amount: BigInt,
  proposalId: string,
  daoId: string,
  outcome: string,
): void {
  let stake = getStake(eventId);
  stake.createdAt = timestamp;
  stake.staker = staker;
  stake.amount = amount;
  stake.proposal = proposalId;
  stake.dao = daoId;
  stake.outcome = outcome;
  saveStake(stake);

  addStakeEvent(eventId, proposalId, outcome, amount, staker, daoId, timestamp);
}