import {
  updateMembershipWitnessesPostBatchUpdates,
  updateMembershipWitnessPostAdd,
  updateMembershipWitnessPostRemove,
  updateMembershipWitnessUsingPublicInfoAfterBatchUpdate,
  updateMembershipWitnessUsingPublicInfoAfterMultipleBatchUpdates,
  updateNonMembershipWitnessesPostBatchUpdates,
  updateNonMembershipWitnessPostAdd,
  updateNonMembershipWitnessPostRemove,
  updateNonMembershipWitnessUsingPublicInfoAfterBatchUpdate,
  updateNonMembershipWitnessUsingPublicInfoAfterMultipleBatchUpdates
} from 'crypto-wasm-new';
import { getUint8ArraysFromObject } from '../util';
import { AccumulatorSecretKey } from './params-and-keys';
import { VBWitnessUpdateInfo, WitnessUpdateInfo } from './witness-update-info';

export abstract class AccumulatorWitness<V> {
  value: Uint8Array | object;

  constructor(value: Uint8Array | object) {
    this.value = value;
  }

  abstract updatePostAdd(addition: Uint8Array, element: Uint8Array, accumulatorValueBeforeAddition: V): void;
  abstract updatePostRemove(removal: Uint8Array, element: Uint8Array, accumulatorValueAfterRemoval: V): void;
  abstract updateUsingPublicInfoPostBatchUpdate(
    element: Uint8Array,
    additions: Uint8Array[],
    removals: Uint8Array[],
    publicInfo: WitnessUpdateInfo
  ): void;
  abstract updateUsingPublicInfoPostMultipleBatchUpdates(
    element: Uint8Array,
    additions: Uint8Array[][],
    removals: Uint8Array[][],
    publicInfo: WitnessUpdateInfo[]
  ): void;
}

export class VBMembershipWitness extends AccumulatorWitness<Uint8Array> {
  // @ts-ignore
  value: Uint8Array;

  /**
   * Update a membership witness after an element is added to the accumulator
   * @param addition - new addition to the accumulator
   * @param member - whose witness is being updated
   * @param accumulatorValueBeforeAddition - accumulator value before the addition was done.
   */
  updatePostAdd(addition: Uint8Array, member: Uint8Array, accumulatorValueBeforeAddition: Uint8Array) {
    this.value = updateMembershipWitnessPostAdd(this.value, member, addition, accumulatorValueBeforeAddition);
  }

  /**
   * Update a membership witness after an element is removed from the accumulator.
   * @param removal - removal from the accumulator
   * @param member - whose witness is being updated
   * @param accumulatorValueAfterRemoval - accumulator value after the removal was done.
   */
  updatePostRemove(removal: Uint8Array, member: Uint8Array, accumulatorValueAfterRemoval: Uint8Array) {
    this.value = updateMembershipWitnessPostRemove(this.value, member, removal, accumulatorValueAfterRemoval);
  }

  /**
   * Compute an update to the membership witness after adding and removing batches of elements from the accumulator.
   * @param member - the member whose witness is to be updated
   * @param additions - array of additions
   * @param removals - array of removals
   * @param publicInfo - witness update info published by the accumulator manager
   */
  updateUsingPublicInfoPostBatchUpdate(
    member: Uint8Array,
    additions: Uint8Array[],
    removals: Uint8Array[],
    publicInfo: VBWitnessUpdateInfo
  ) {
    this.value = updateMembershipWitnessUsingPublicInfoAfterBatchUpdate(
      this.value,
      member,
      additions,
      removals,
      publicInfo.value
    );
  }

  /**
   * Compute an update to the membership witness after adding and removing several batches of elements from the accumulator.
   * For each batch of updates, additions, removals and witness update info are provided.
   * @param member - the member whose witness is to be updated
   * @param additions - array of arrays of additions
   * @param removals - array of arrays of removals
   * @param publicInfo - array of witness update info
   */
  updateUsingPublicInfoPostMultipleBatchUpdates(
    member: Uint8Array,
    additions: Uint8Array[][],
    removals: Uint8Array[][],
    publicInfo: VBWitnessUpdateInfo[]
  ) {
    const info = publicInfo.map((i) => i.value);
    this.value = updateMembershipWitnessUsingPublicInfoAfterMultipleBatchUpdates(
      this.value,
      member,
      additions,
      removals,
      info
    );
  }

  /**
   * Compute an update to several membership witnesses after adding and removing batches of elements from the accumulator.
   * @param witnesses - membership witnesses to update
   * @param members - members corresponding to the witnesses
   * @param additions
   * @param removals
   * @param accumulatorValueBeforeUpdates - accumulator value before the updates
   * @param secretKey
   */
  static updateMultiplePostBatchUpdates(
    witnesses: VBMembershipWitness[],
    members: Uint8Array[],
    additions: Uint8Array[],
    removals: Uint8Array[],
    accumulatorValueBeforeUpdates: Uint8Array,
    secretKey: AccumulatorSecretKey
  ): VBMembershipWitness[] {
    const wits = witnesses.map((m) => m.value);
    return updateMembershipWitnessesPostBatchUpdates(
      wits,
      members,
      additions,
      removals,
      accumulatorValueBeforeUpdates,
      secretKey.value
    ).map((m) => new VBMembershipWitness(m));
  }

  toJSON(): string {
    return JSON.stringify({
      value: Array.from(this.value)
    });
  }

  static fromJSON(json: string): VBMembershipWitness {
    const obj = JSON.parse(json);
    const [value] = getUint8ArraysFromObject(obj, ['value']);
    return new VBMembershipWitness(value);
  }
}

export class VBNonMembershipWitness extends AccumulatorWitness<Uint8Array> {
  // @ts-ignore
  value: { d: Uint8Array; C: Uint8Array };

  /**
   * Update a non-membership witness after an element is added to the accumulator
   * @param addition - new addition to the accumulator
   * @param nonMember - whose witness is being updated
   * @param accumulatorValueBeforeAddition - accumulator value before the addition was done.
   */
  updatePostAdd(addition: Uint8Array, nonMember: Uint8Array, accumulatorValueBeforeAddition: Uint8Array) {
    this.value = updateNonMembershipWitnessPostAdd(this.value, nonMember, addition, accumulatorValueBeforeAddition);
  }

  /**
   * Update a non-membership witness after an element is removed from the accumulator.
   * @param removal - removal from the accumulator
   * @param nonMember - whose witness is being updated
   * @param accumulatorValueAfterRemoval - accumulator value after the removal was done.
   */
  updatePostRemove(removal: Uint8Array, nonMember: Uint8Array, accumulatorValueAfterRemoval: Uint8Array) {
    this.value = updateNonMembershipWitnessPostRemove(this.value, nonMember, removal, accumulatorValueAfterRemoval);
  }

  /**
   * Compute an update to the non-membership witness after adding and removing batches of elements from the accumulator.
   * @param nonMember - the non-member whose witness is to be updated
   * @param additions - array of additions
   * @param removals - array of removals
   * @param publicInfo - witness update info published by the accumulator manager
   */
  updateUsingPublicInfoPostBatchUpdate(
    nonMember: Uint8Array,
    additions: Uint8Array[],
    removals: Uint8Array[],
    publicInfo: VBWitnessUpdateInfo
  ) {
    this.value = updateNonMembershipWitnessUsingPublicInfoAfterBatchUpdate(
      this.value,
      nonMember,
      additions,
      removals,
      publicInfo.value
    );
  }

  /**
   * Compute an update to the non-membership witness after adding and removing several batches of elements from the accumulator.
   * For each batch of updates, additions, removals and witness update info are provided.
   * @param nonMember - the non-member whose witness is to be updated
   * @param additions - array of arrays of additions
   * @param removals - array of arrays of removals
   * @param publicInfo - array of witness update info
   */
  updateUsingPublicInfoPostMultipleBatchUpdates(
    nonMember: Uint8Array,
    additions: Uint8Array[][],
    removals: Uint8Array[][],
    publicInfo: VBWitnessUpdateInfo[]
  ) {
    const info = publicInfo.map((i) => i.value);
    this.value = updateNonMembershipWitnessUsingPublicInfoAfterMultipleBatchUpdates(
      this.value,
      nonMember,
      additions,
      removals,
      info
    );
  }

  /**
   * Compute an update to several non-membership witnesses after adding and removing batches of elements from the accumulator.
   * @param witnesses - non-membership witnesses to update
   * @param nonMembers - nonMembers corresponding to the witnesses
   * @param additions
   * @param removals
   * @param accumulatorValueBeforeUpdates - accumulator value before the updates
   * @param secretKey
   */
  static updateMultiplePostBatchUpdates(
    witnesses: VBNonMembershipWitness[],
    nonMembers: Uint8Array[],
    additions: Uint8Array[],
    removals: Uint8Array[],
    accumulatorValueBeforeUpdates: Uint8Array,
    secretKey: AccumulatorSecretKey
  ): VBNonMembershipWitness[] {
    const wits = witnesses.map((w) => w.value);
    return updateNonMembershipWitnessesPostBatchUpdates(
      wits,
      nonMembers,
      additions,
      removals,
      accumulatorValueBeforeUpdates,
      secretKey.value
    ).map((m) => new VBNonMembershipWitness(m));
  }

  toJSON(): string {
    return JSON.stringify({
      value: { d: Array.from(this.value.d), C: Array.from(this.value.C) }
    });
  }

  static fromJSON(json: string): VBNonMembershipWitness {
    const obj = JSON.parse(json);
    const [d, C] = getUint8ArraysFromObject(obj.value, ['d', 'C']);
    return new VBNonMembershipWitness({ d, C });
  }
}