import {
  Bool,
  CircuitValue,
  Field,
  prop,
  PublicKey,
  UInt64,
  Poseidon,
  MerkleWitness,
} from 'snarkyjs';

export class MyMerkleWitness extends MerkleWitness(3) {}
let w = {
  isLeft: false,
  sibling: Field(0),
};
let dummyWitness = Array.from(Array(MyMerkleWitness.height - 1).keys()).map(
  () => w
);

export class Member extends CircuitValue {
  @prop publicKey: PublicKey;
  @prop balance: UInt64;

  // will need this to keep track of votes for candidates
  @prop votes: Field;

  @prop witness: MyMerkleWitness;
  @prop votesWitness: MyMerkleWitness;

  constructor(publicKey: PublicKey, balance: UInt64) {
    super();
    this.publicKey = publicKey;
    this.balance = balance;
    this.votes = Field(0);

    this.witness = new MyMerkleWitness(dummyWitness);
    this.votesWitness = new MyMerkleWitness(dummyWitness);
  }

  getHash(): Field {
    return Poseidon.hash(
      this.publicKey
        .toFields()
        .concat(this.balance.toFields())
        .concat(this.votes.toFields())
    );
  }

  addVote(): Member {
    this.votes = this.votes.add(1);
    return this;
  }

  static empty() {
    return new Member(PublicKey.empty(), UInt64.zero);
  }

  static from(publicKey: PublicKey, balance: UInt64) {
    return new Member(publicKey, balance);
  }
}