import { Address, BigInt, ipfs, store } from '@graphprotocol/graph-ts';
import { Avatar } from '../types/Controller/Avatar';
import { DAO } from '../types/schema';

export function getDAO(id: string): DAO {
  let dao = store.get('DAO', id) as DAO;
  if (dao == null) {
    dao = new DAO(id);
    dao.name = '';
    dao.nativeToken = '';
    dao.nativeReputation = '';
    dao.reputationHoldersCount = BigInt.fromI32(0);
    dao.register = 'na';
    dao.metadata = '';
    dao.metadataHash = '';
    dao.numberOfQueuedProposals = BigInt.fromI32(0);
    dao.numberOfPreBoostedProposals = BigInt.fromI32(0);
    dao.numberOfBoostedProposals = BigInt.fromI32(0);
    dao.numberOfExpiredInQueueProposals = BigInt.fromI32(0);
    dao.numberOfQueuedProposalsUnregistered = BigInt.fromI32(0);
    dao.numberOfPreBoostedProposalsUnregistered = BigInt.fromI32(0);
    dao.numberOfBoostedProposalsUnregistered = BigInt.fromI32(0);
    dao.numberOfExpiredInQueueProposalsUnregistered = BigInt.fromI32(0);
    dao.ethBalance = BigInt.fromI32(0);
  }

  return dao;
}

export function increaseDAOmembersCount(id: string): void {
  let dao = getDAO(id);
  dao.reputationHoldersCount = dao.reputationHoldersCount.plus(BigInt.fromI32(1));
  saveDAO(dao);
}

export function decreaseDAOmembersCount(id: string): void {
  let dao = getDAO(id);
  dao.reputationHoldersCount = dao.reputationHoldersCount.minus(BigInt.fromI32(1));
  saveDAO(dao);
}

export function saveDAO(dao: DAO): void {
  store.set('DAO', dao.id, dao);
}

export function insertNewDAO(
  avatarAddress: Address,
  nativeTokenAddress: Address,
  nativeReputationAddress: Address,
): DAO {
  let avatar = Avatar.bind(avatarAddress);
  let dao = getDAO(avatarAddress.toHex());
  dao.name = avatar.orgName().toString();
  dao.nativeToken = nativeTokenAddress.toHex();
  dao.nativeReputation = nativeReputationAddress.toHex();
  dao.vault = avatar.vault();
  dao.reputationHoldersCount = BigInt.fromI32(0);
  dao.register = 'na';
  dao.metadata = '';
  dao.metadataHash = '';
  saveDAO(dao);

  return dao;
}

export function register(
  avatar: Address,
  tag: string,
): void {
  let dao = DAO.load(avatar.toHex());
  if (dao != null) {
    dao.register = tag;
    dao.save();
  }
}

export function metadata(
  avatar: Address,
  metadataHash: string,
): void {
  let dao = DAO.load(avatar.toHex());
  if (dao != null) {
    dao.metadataHash = metadataHash;
    dao.metadata = (ipfs.cat('/ipfs/' + metadataHash)).toString();
    dao.save();
  }
}

export function exists(
  avatar: Address,
): boolean {
  return (DAO.load(avatar.toHex()) != null);
}