liquid-funding/js/liquidPledging.js

/* eslint-disable no-await-in-loop */
const LiquidPledgingAbi = require('../build/LiquidPledging.sol').LiquidPledgingAbi;
const LiquidPledgingCode = require('../build/LiquidPledging.sol').LiquidPledgingByteCode;
const LiquidPledgingMockAbi = require('../build/LiquidPledgingMock.sol').LiquidPledgingMockAbi;
const LiquidPledgingMockCode = require('../build/LiquidPledgingMock.sol').LiquidPledgingMockByteCode;
const generateClass = require('./generateClass');

module.exports = (test) => {
  const $abi = (test) ? LiquidPledgingMockAbi : LiquidPledgingAbi;
  const $byteCode = (test) ? LiquidPledgingMockCode : LiquidPledgingCode;

  const LiquidPledging = generateClass($abi, $byteCode);

  async function $getNote(idNote) {
    const note = {
      delegates: [],
    };
    const res = await this.getNote(idNote);
    note.amount = this.$toNumber(res.amount);
    note.owner = res.owner;
    for (let i = 1; i <= this.$toDecimal(res.nDelegates); i += 1) {
      const delegate = {};
      const resd = await this.getNoteDelegate(idNote, i);
      delegate.id = this.$toDecimal(resd.idDelegate);
      delegate.addr = resd.addr;
      delegate.name = resd.name;
      note.delegates.push(delegate);
    }
    if (res.proposedProject) {
      note.proposedProject = this.$toDecimal(res.proposedProject);
      note.commmitTime = this.$toDecimal(res.commitTime);
    }
    if (res.oldNote) {
      note.oldProject = this.$toDecimal(res.oldNote);
    }
    if (res.paymentState === '0') {
      note.paymentState = 'NotPaid';
    } else if (res.paymentState === '1') {
      note.paymentState = 'Paying';
    } else if (res.paymentState === '2') {
      note.paymentState = 'Paid';
    } else {
      note.paymentState = 'Unknown';
    }
    return note;
  }

  LiquidPledging.prototype.$getNote = $getNote;

  async function $getManager(idManager) {
    const manager = {};
    const res = await this.getNoteManager(idManager);
    if (res.managerType === '0') {
      manager.type = 'Donor';
    } else if (res.managerType === '1') {
      manager.type = 'Delegate';
    } else if (res.managerType === '2') {
      manager.type = 'Project';
    } else {
      manager.type = 'Unknown';
    }
    manager.addr = res.addr;
    manager.name = res.name;
    manager.commitTime = this.$toDecimal(res.commitTime);
    if (manager.paymentState === 'Project') {
      manager.parentProject = res.parentProject;
      manager.canceled = res.canceled;
    }
    return manager;
  };

  LiquidPledging.prototype.$getManager = $getManager;

  async function getState() {
    const st = {
      notes: [null],
      managers: [null],
    };
    const nNotes = await this.numberOfNotes();
    for (let i = 1; i <= nNotes; i += 1) {
      const note = await this.$getNote(i);
      st.notes.push(note);
    }

    const nManagers = await this.numberOfNoteManagers();
    for (let i = 1; i <= nManagers; i += 1) {
      const manager = await this.$getManager(i);
      st.managers.push(manager);
    }
    return st;
  };

  LiquidPledging.prototype.getState = getState;

  LiquidPledging.prototype.generateDonorsState = function () {
    const donorsState = [];

    const getDonor = (idNote) => {
      let note = this.notes[idNote];
      while (note.oldNode) note = this.notes[idNote];
      return note.owner;
    };

      // Add a donor structure to the list
    const addDonor = (_list, idDonor) => {
      const list = _list;
      if (!list[idDonor]) {
        list[idDonor] = {
          idDonor,
          notAssigned: {
            notes: [],
            delegates: [],
          },
          precommitedProjects: [],
          commitedProjects: [],
        };
      }
    };

      // Add a delegate structure to the list
    const addDelegate = (_list, idDelegate) => {
      const list = _list;
      if (!list[idDelegate]) {
        list[idDelegate] = {
          idDelegate,
          name: this.managers[idDelegate].name,
          notes: [],
          delegtes: [],
        };
      }
    };

    const addProject = (_list, idProject) => {
      const list = _list;
      if (!list[idProject]) {
        list[idProject] = {
          idProject,
          notes: [],
          commitedProjects: [],
          name: this.managers[idProject].name,
          commitTime: this.managers[idProject].commitTime,
          owner: this.managers[idProject].owner,
          parentProject: this.managers[idProject].parentProject,
        };
      }
    };

    const addDelegateNote = (stDonor, idNote) => {
      const note = this.notes[idNote];
      stDonor.notAssigned.notes.push(idNote);
      let list = stDonor.notAssigned.delegates;
      for (let i = 0; i < note.delegationChain.length; i += 1) {
        const idDelegate = note.delegationChain[i];
        addDelegate(list, idDelegate);
        list = list[idDelegate].delegates;
      }
    };

    const addProjectNote = (stDonor, idNote) => {
      const note = this.notes[idNote];

      const projectList = [];
      let n = note;
      while (n.oldNode) {
        projectList.unshift(n.owner);
        n = this.notes[n.oldNode];
      }

      let list = stDonor.commitedProjects;
      for (let j = 0; j < projectList.length; j += 1) {
        addProject(list, projectList[j]);
        list[projectList[j]].notes.push(idNote);
        list = list[projectList[j]].commitedProjects;
      }
    };

    for (let i = 0; i < this.notes; i += 1) {
      const idNote = this.notes[i];
      const idDonor = getDonor(idNote);
      addDonor(donorsState, idDonor);
      const stDonor = donorsState[idDonor];
      const note = this.notes[idNote];
      if ((note.owner === idDonor) && (note.precommitedProject === 0)) {
        addDelegateNote(stDonor, idNote);
      } else if ((note.owner === idDonor) && (note.precommitedProject !== 0)) {
        addProject(stDonor.precommitedProjects, note.precommitedProject);
        stDonor.precommitedProjects[note.precommitedProject].notes.push(idNote);
      } else {
        addProjectNote(stDonor, idNote);
      }
    }

    this.donorsState = donorsState;
  };

  return LiquidPledging;
};