@maci-protocol/core 0.0.0-ci.185b643

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/README.md ADDED
@@ -0,0 +1,107 @@
1
+ # `maci-core`
2
+
3
+ [![NPM Package][core-npm-badge]][core-npm-link]
4
+ [![Actions Status][core-actions-badge]][core-actions-link]
5
+
6
+ This submodule assists with handling key business logic functions and
7
+ processes.
8
+
9
+ ## Overview
10
+
11
+ One may conceive of MACI as a state machine with 1) data and 2) functions which
12
+ transform said data. This makes it easier to reason about the system, write
13
+ tests, and implement functionality. It also allows us to implement the smart
14
+ contracts in discrete components which are easy to test.
15
+
16
+ To this end, this submodule exposes a `MaciState` class and a `Poll` class.
17
+ Developers should instantiate objects from these classes to test MACI. For
18
+ instance, [`MACI.test.ts`](https://github.com/privacy-scaling-explorations/maci/blob/dev/packages/contracts/tests/MACI.test.ts) creates a
19
+ `MaciState` object and every time it interacts with the MACI smart contract, it
20
+ mirrors said interaction on the `MaciState` and `Poll`. As such, the developer
21
+ can then use their helper functions like `maciState.signUp()`,
22
+ `poll.publishMessage`, `poll.processMessages()`, and `poll.tallyVotes()` to
23
+ step through the various stages of the MACI flow.
24
+
25
+ ## `MaciState`
26
+
27
+ ### Key functions
28
+
29
+ #### **`signUp`**
30
+
31
+ Accepts a user's public key and creates a new state leaf and ballot leaf.
32
+
33
+ In testing, whenever a test suite submits a `signUp()` transaction, it should
34
+ call `maciState.signUp()` as well, so that the off-chain representation of MACI
35
+ is kept up to date.
36
+
37
+ In production, `genMaciStateFromContract()` in
38
+ [`genMaciState.ts`](https://github.com/privacy-scaling-explorations/maci/blob/dev/contracts/ts/genMaciState.ts) uses this function when it
39
+ scans a MACI contract's event log for signups, so as to bring its `MaciState`
40
+ instance up to date.
41
+
42
+ #### **`deployPoll`**
43
+
44
+ Creates a new `Poll`. This should be done whenever the MACI contract's
45
+ `deployPoll()` function is called.
46
+
47
+ ### Helper functions
48
+
49
+ #### **`copy`**
50
+
51
+ A function that deep-copies an object.
52
+
53
+ ### Key data structures
54
+
55
+ #### **`stateTree`**
56
+
57
+ `stateTree` is a quinary Merkle tree, chosen over a binary tree due to the gas and circuit constraints associated with the Poseidon hash function ([details here][ethresearch-link]). Each leaf in this tree represents a participant's public key, their voice credit balance and the block timestamp at which they signed up. The tree features a configurable depth and an arity of five, with insertions starting at index 1. The zeroth leaf is reserved as a security measure against denial-of-service attacks.
58
+
59
+ ## **`Poll`**
60
+
61
+ A `Poll` is an off-chain representation of a Poll. In testing, `Poll` instances
62
+ should mirror their on-chain counterparts.
63
+
64
+ ### Key functions
65
+
66
+ #### **`publishMessage`**
67
+
68
+ Publishes a message by updating the message tree and message accumulation
69
+ queue.
70
+
71
+ #### **`processMessages`**
72
+
73
+ Processes a batch of messages and returns the inputs to the `processMessages`
74
+ circuit which can be used to prove correct execution.
75
+
76
+ #### **`tallyVotes`**
77
+
78
+ Tallies a batch of votes and returns the inputs to the `tallyVotes`
79
+ circuit which can be used to prove correct execution.
80
+
81
+ ### Helper functions
82
+
83
+ #### **`copy`**
84
+
85
+ Deep-copies and returns this object.
86
+
87
+ ### Key data structures
88
+
89
+ #### **`messageTree`**
90
+
91
+ `messageTree` is also a quinary Merkle tree. Each message in this tree represents an encrypted command, such as a user casting a vote in a poll or changing their public key.
92
+
93
+ ## Testing
94
+
95
+ For more details about testing please refer to the [tests documentation](https://maci.pse.dev/docs/testing).
96
+
97
+ [core-npm-badge]: https://img.shields.io/npm/v/maci-core.svg
98
+ [core-npm-link]: https://www.npmjs.com/package/maci-core
99
+ [core-actions-badge]: https://github.com/privacy-scaling-explorations/maci/actions/workflows/core-build.yml/badge.svg
100
+ [core-actions-link]: https://github.com/privacy-scaling-explorations/maci/actions?query=workflow%3Acore
101
+ [ethresearch-link]: https://ethresear.ch/t/gas-and-circuit-constraint-benchmarks-of-binary-and-quinary-incremental-merkle-trees-using-the-poseidon-hash-function/7446
102
+
103
+ ## Directory Structure
104
+
105
+ - `ts/`: The home for our core classes, `MaciState` and `Poll`
106
+ - `ts/utls/`: Contains supporting utilities
107
+ - `ts/__tests__/`: A dedicated test suite directory
@@ -0,0 +1,64 @@
1
+ import { IncrementalQuinTree } from "@maci-protocol/crypto";
2
+ import { PubKey, type Keypair } from "@maci-protocol/domainobjs";
3
+ import type { IJsonMaciState, IMaciState, TreeDepths } from "./utils/types";
4
+ import { Poll } from "./Poll";
5
+ /**
6
+ * A representation of the MACI contract.
7
+ */
8
+ export declare class MaciState implements IMaciState {
9
+ polls: Map<bigint, Poll>;
10
+ pubKeys: PubKey[];
11
+ stateTreeDepth: number;
12
+ stateTree?: IncrementalQuinTree;
13
+ numSignUps: number;
14
+ pollBeingProcessed?: boolean;
15
+ currentPollBeingProcessed?: bigint;
16
+ /**
17
+ * Constructs a new MaciState object.
18
+ * @param stateTreeDepth - The depth of the state tree.
19
+ */
20
+ constructor(stateTreeDepth: number);
21
+ /**
22
+ * Sign up a user with the given public key.
23
+ * @param pubKey - The public key of the user.
24
+ * @returns The index of the newly signed-up user in the state tree.
25
+ */
26
+ signUp(pubKey: PubKey): number;
27
+ /**
28
+ * Deploy a new poll with the given parameters.
29
+ * @param pollEndTimestamp - The Unix timestamp at which the poll ends.
30
+ * @param treeDepths - The depths of the tree.
31
+ * @param messageBatchSize - The batch size for processing messages.
32
+ * @param coordinatorKeypair - The keypair of the MACI round coordinator.
33
+ * @param voteOptions - The number of vote options for this poll.
34
+ * @returns The index of the newly deployed poll.
35
+ */
36
+ deployPoll(pollEndTimestamp: bigint, treeDepths: TreeDepths, messageBatchSize: number, coordinatorKeypair: Keypair, voteOptions: bigint): bigint;
37
+ /**
38
+ * Deploy a null poll.
39
+ */
40
+ deployNullPoll(): void;
41
+ /**
42
+ * Create a deep copy of the MaciState object.
43
+ * @returns A new instance of the MaciState object with the same properties.
44
+ */
45
+ copy: () => MaciState;
46
+ /**
47
+ * Check if the MaciState object is equal to another MaciState object.
48
+ * @param m - The MaciState object to compare.
49
+ * @returns True if the two MaciState objects are equal, false otherwise.
50
+ */
51
+ equals: (m: MaciState) => boolean;
52
+ /**
53
+ * Serialize the MaciState object to a JSON object.
54
+ * @returns A JSON object representing the MaciState object.
55
+ */
56
+ toJSON(): IJsonMaciState;
57
+ /**
58
+ * Create a new MaciState object from a JSON object.
59
+ * @param json - The JSON object representing the MaciState object.
60
+ * @returns A new instance of the MaciState object with the properties from the JSON object.
61
+ */
62
+ static fromJSON(json: IJsonMaciState): MaciState;
63
+ }
64
+ //# sourceMappingURL=MaciState.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"MaciState.d.ts","sourceRoot":"","sources":["../../ts/MaciState.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,KAAK,OAAO,EAAU,MAAM,2BAA2B,CAAC;AAEzE,OAAO,KAAK,EAAE,cAAc,EAAa,UAAU,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAEvF,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAG9B;;GAEG;AACH,qBAAa,SAAU,YAAW,UAAU;IAE1C,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAA2B;IAGnD,OAAO,EAAE,MAAM,EAAE,CAAM;IAGvB,cAAc,EAAE,MAAM,CAAC;IAGvB,SAAS,CAAC,EAAE,mBAAmB,CAAC;IAEhC,UAAU,SAAK;IAGf,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAE7B,yBAAyB,CAAC,EAAE,MAAM,CAAC;IAEnC;;;OAGG;gBACS,cAAc,EAAE,MAAM;IAUlC;;;;OAIG;IACH,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAO9B;;;;;;;;OAQG;IACH,UAAU,CACR,gBAAgB,EAAE,MAAM,EACxB,UAAU,EAAE,UAAU,EACtB,gBAAgB,EAAE,MAAM,EACxB,kBAAkB,EAAE,OAAO,EAC3B,WAAW,EAAE,MAAM,GAClB,MAAM;IAiBT;;OAEG;IACH,cAAc,IAAI,IAAI;IAItB;;;OAGG;IACH,IAAI,QAAO,SAAS,CAQlB;IAEF;;;;OAIG;IACH,MAAM,GAAI,GAAG,SAAS,KAAG,OAAO,CAsB9B;IAEF;;;OAGG;IACH,MAAM,IAAI,cAAc;IAWxB;;;;OAIG;IACH,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,cAAc,GAAG,SAAS;CAgBjD"}
@@ -0,0 +1,127 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.MaciState = void 0;
4
+ const domainobjs_1 = require("@maci-protocol/domainobjs");
5
+ const Poll_1 = require("./Poll");
6
+ const constants_1 = require("./utils/constants");
7
+ /**
8
+ * A representation of the MACI contract.
9
+ */
10
+ class MaciState {
11
+ /**
12
+ * Constructs a new MaciState object.
13
+ * @param stateTreeDepth - The depth of the state tree.
14
+ */
15
+ constructor(stateTreeDepth) {
16
+ // a MaciState can hold multiple polls
17
+ this.polls = new Map();
18
+ // the public keys of the users
19
+ this.pubKeys = [];
20
+ this.numSignUps = 0;
21
+ /**
22
+ * Create a deep copy of the MaciState object.
23
+ * @returns A new instance of the MaciState object with the same properties.
24
+ */
25
+ this.copy = () => {
26
+ const copied = new MaciState(this.stateTreeDepth);
27
+ copied.pubKeys = this.pubKeys.map((x) => x.copy());
28
+ copied.polls = new Map(Array.from(this.polls, ([key, value]) => [key, value.copy()]));
29
+ return copied;
30
+ };
31
+ /**
32
+ * Check if the MaciState object is equal to another MaciState object.
33
+ * @param m - The MaciState object to compare.
34
+ * @returns True if the two MaciState objects are equal, false otherwise.
35
+ */
36
+ this.equals = (m) => {
37
+ const result = this.stateTreeDepth === m.stateTreeDepth &&
38
+ this.polls.size === m.polls.size &&
39
+ this.pubKeys.length === m.pubKeys.length;
40
+ if (!result) {
41
+ return false;
42
+ }
43
+ for (let i = 0; i < this.polls.size; i += 1) {
44
+ if (!this.polls.get(BigInt(i))?.equals(m.polls.get(BigInt(i)))) {
45
+ return false;
46
+ }
47
+ }
48
+ for (let i = 0; i < this.pubKeys.length; i += 1) {
49
+ if (!this.pubKeys[i].equals(m.pubKeys[i])) {
50
+ return false;
51
+ }
52
+ }
53
+ return true;
54
+ };
55
+ this.stateTreeDepth = stateTreeDepth;
56
+ // we put a blank state leaf to prevent a DoS attack
57
+ this.pubKeys.push(domainobjs_1.padKey);
58
+ // we need to increase the number of signups by one given
59
+ // that we already added the blank leaf
60
+ this.numSignUps += 1;
61
+ }
62
+ /**
63
+ * Sign up a user with the given public key.
64
+ * @param pubKey - The public key of the user.
65
+ * @returns The index of the newly signed-up user in the state tree.
66
+ */
67
+ signUp(pubKey) {
68
+ this.numSignUps += 1;
69
+ this.stateTree?.insert(pubKey.hash());
70
+ return this.pubKeys.push(pubKey.copy()) - 1;
71
+ }
72
+ /**
73
+ * Deploy a new poll with the given parameters.
74
+ * @param pollEndTimestamp - The Unix timestamp at which the poll ends.
75
+ * @param treeDepths - The depths of the tree.
76
+ * @param messageBatchSize - The batch size for processing messages.
77
+ * @param coordinatorKeypair - The keypair of the MACI round coordinator.
78
+ * @param voteOptions - The number of vote options for this poll.
79
+ * @returns The index of the newly deployed poll.
80
+ */
81
+ deployPoll(pollEndTimestamp, treeDepths, messageBatchSize, coordinatorKeypair, voteOptions) {
82
+ const poll = new Poll_1.Poll(pollEndTimestamp, coordinatorKeypair, treeDepths, {
83
+ messageBatchSize,
84
+ tallyBatchSize: constants_1.STATE_TREE_ARITY ** treeDepths.intStateTreeDepth,
85
+ }, this, voteOptions);
86
+ this.polls.set(BigInt(this.polls.size), poll);
87
+ return BigInt(this.polls.size - 1);
88
+ }
89
+ /**
90
+ * Deploy a null poll.
91
+ */
92
+ deployNullPoll() {
93
+ this.polls.set(BigInt(this.polls.size), null);
94
+ }
95
+ /**
96
+ * Serialize the MaciState object to a JSON object.
97
+ * @returns A JSON object representing the MaciState object.
98
+ */
99
+ toJSON() {
100
+ return {
101
+ stateTreeDepth: this.stateTreeDepth,
102
+ polls: Array.from(this.polls.values()).map((poll) => poll.toJSON()),
103
+ pubKeys: this.pubKeys.map((pubKey) => pubKey.toJSON()),
104
+ pollBeingProcessed: Boolean(this.pollBeingProcessed),
105
+ currentPollBeingProcessed: this.currentPollBeingProcessed ? this.currentPollBeingProcessed.toString() : "",
106
+ numSignUps: this.numSignUps,
107
+ };
108
+ }
109
+ /**
110
+ * Create a new MaciState object from a JSON object.
111
+ * @param json - The JSON object representing the MaciState object.
112
+ * @returns A new instance of the MaciState object with the properties from the JSON object.
113
+ */
114
+ static fromJSON(json) {
115
+ const maciState = new MaciState(json.stateTreeDepth);
116
+ // assign the json values to the new instance
117
+ maciState.pubKeys = json.pubKeys.map((pubKey) => domainobjs_1.PubKey.fromJSON(pubKey));
118
+ maciState.pollBeingProcessed = json.pollBeingProcessed;
119
+ maciState.currentPollBeingProcessed = BigInt(json.currentPollBeingProcessed);
120
+ maciState.numSignUps = json.numSignUps;
121
+ // re-generate the polls and set the maci state reference
122
+ maciState.polls = new Map(json.polls.map((jsonPoll, index) => [BigInt(index), Poll_1.Poll.fromJSON(jsonPoll, maciState)]));
123
+ return maciState;
124
+ }
125
+ }
126
+ exports.MaciState = MaciState;
127
+ //# sourceMappingURL=MaciState.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"MaciState.js","sourceRoot":"","sources":["../../ts/MaciState.ts"],"names":[],"mappings":";;;AACA,0DAAyE;AAIzE,iCAA8B;AAC9B,iDAAqD;AAErD;;GAEG;AACH,MAAa,SAAS;IAoBpB;;;OAGG;IACH,YAAY,cAAsB;QAvBlC,sCAAsC;QACtC,UAAK,GAAsB,IAAI,GAAG,EAAgB,CAAC;QAEnD,+BAA+B;QAC/B,YAAO,GAAa,EAAE,CAAC;QAQvB,eAAU,GAAG,CAAC,CAAC;QAwEf;;;WAGG;QACH,SAAI,GAAG,GAAc,EAAE;YACrB,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAElD,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YAE3D,MAAM,CAAC,KAAK,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAEtF,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;QAEF;;;;WAIG;QACH,WAAM,GAAG,CAAC,CAAY,EAAW,EAAE;YACjC,MAAM,MAAM,GACV,IAAI,CAAC,cAAc,KAAK,CAAC,CAAC,cAAc;gBACxC,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,CAAC,KAAK,CAAC,IAAI;gBAChC,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;YAE3C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,KAAK,CAAC;YACf,CAAC;YAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAE,CAAC,EAAE,CAAC;oBAChE,OAAO,KAAK,CAAC;gBACf,CAAC;YACH,CAAC;YACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;gBAChD,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC1C,OAAO,KAAK,CAAC;gBACf,CAAC;YACH,CAAC;YAED,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;QArGA,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QAErC,oDAAoD;QACpD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,mBAAM,CAAC,CAAC;QAC1B,yDAAyD;QACzD,uCAAuC;QACvC,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;IACvB,CAAC;IAED;;;;OAIG;IACH,MAAM,CAAC,MAAc;QACnB,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;QAErB,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QACtC,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC;IAC9C,CAAC;IAED;;;;;;;;OAQG;IACH,UAAU,CACR,gBAAwB,EACxB,UAAsB,EACtB,gBAAwB,EACxB,kBAA2B,EAC3B,WAAmB;QAEnB,MAAM,IAAI,GAAS,IAAI,WAAI,CACzB,gBAAgB,EAChB,kBAAkB,EAClB,UAAU,EACV;YACE,gBAAgB;YAChB,cAAc,EAAE,4BAAgB,IAAI,UAAU,CAAC,iBAAiB;SACjE,EACD,IAAI,EACJ,WAAW,CACZ,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;QAC9C,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAuB,CAAC,CAAC;IACnE,CAAC;IA6CD;;;OAGG;IACH,MAAM;QACJ,OAAO;YACL,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACnE,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACtD,kBAAkB,EAAE,OAAO,CAAC,IAAI,CAAC,kBAAkB,CAAC;YACpD,yBAAyB,EAAE,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE;YAC1G,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,MAAM,CAAC,QAAQ,CAAC,IAAoB;QAClC,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAErD,6CAA6C;QAC7C,SAAS,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,mBAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;QAC1E,SAAS,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC;QACvD,SAAS,CAAC,yBAAyB,GAAG,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QAC7E,SAAS,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QAEvC,yDAAyD;QACzD,SAAS,CAAC,KAAK,GAAG,IAAI,GAAG,CACvB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,QAAmB,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,WAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,CACpG,CAAC;QAEF,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AApKD,8BAoKC"}
@@ -0,0 +1,219 @@
1
+ import { IncrementalQuinTree } from "@maci-protocol/crypto";
2
+ import { PCommand, Keypair, Ballot, PubKey, Message, StateLeaf } from "@maci-protocol/domainobjs";
3
+ import { LeanIMT } from "@zk-kit/lean-imt";
4
+ import type { TreeDepths, BatchSizes, IPoll, IJsonPoll, IProcessMessagesOutput, ITallyCircuitInputs, IProcessMessagesCircuitInputs, IPollJoiningCircuitInputs, IJoiningCircuitArgs } from "./index";
5
+ import type { MaciState } from "./MaciState";
6
+ import type { IJoinedCircuitArgs, IPollJoinedCircuitInputs } from "./utils/types";
7
+ /**
8
+ * A representation of the Poll contract.
9
+ */
10
+ export declare class Poll implements IPoll {
11
+ coordinatorKeypair: Keypair;
12
+ treeDepths: TreeDepths;
13
+ batchSizes: BatchSizes;
14
+ voteOptions: bigint;
15
+ maxVoteOptions: number;
16
+ actualStateTreeDepth: number;
17
+ pollEndTimestamp: bigint;
18
+ ballots: Ballot[];
19
+ ballotTree?: IncrementalQuinTree;
20
+ messages: Message[];
21
+ commands: PCommand[];
22
+ encPubKeys: PubKey[];
23
+ stateCopied: boolean;
24
+ pubKeys: PubKey[];
25
+ stateTree?: LeanIMT;
26
+ numBatchesProcessed: number;
27
+ currentMessageBatchIndex: number;
28
+ maciStateRef: MaciState;
29
+ pollId: bigint;
30
+ sbSalts: Record<number | string, bigint>;
31
+ resultRootSalts: Record<number | string, bigint>;
32
+ preVOSpentVoiceCreditsRootSalts: Record<number | string, bigint>;
33
+ spentVoiceCreditSubtotalSalts: Record<number | string, bigint>;
34
+ tallyResult: bigint[];
35
+ perVOSpentVoiceCredits: bigint[];
36
+ numBatchesTallied: number;
37
+ totalSpentVoiceCredits: bigint;
38
+ emptyBallot: Ballot;
39
+ emptyBallotHash?: bigint;
40
+ chainHash: bigint;
41
+ batchHashes: bigint[];
42
+ pollStateLeaves: StateLeaf[];
43
+ pollStateTree?: IncrementalQuinTree;
44
+ pollNullifiers: Map<bigint, boolean>;
45
+ private numSignups;
46
+ /**
47
+ * Constructs a new Poll object.
48
+ * @param pollEndTimestamp - The Unix timestamp at which the poll ends.
49
+ * @param coordinatorKeypair - The keypair of the coordinator.
50
+ * @param treeDepths - The depths of the trees used in the poll.
51
+ * @param batchSizes - The sizes of the batches used in the poll.
52
+ * @param maciStateRef - The reference to the MACI state.
53
+ * @param pollId - The poll id
54
+ */
55
+ constructor(pollEndTimestamp: bigint, coordinatorKeypair: Keypair, treeDepths: TreeDepths, batchSizes: BatchSizes, maciStateRef: MaciState, voteOptions: bigint);
56
+ /**
57
+ * Check if user has already joined the poll by checking if the nullifier is registered
58
+ */
59
+ hasJoined: (nullifier: bigint) => boolean;
60
+ /**
61
+ * Join the anonymous user to the Poll (to the tree)
62
+ * @param nullifier - Hashed private key used as nullifier
63
+ * @param pubKey - The poll public key.
64
+ * @param newVoiceCreditBalance - New voice credit balance of the user.
65
+ * @returns The index of added state leaf
66
+ */
67
+ joinPoll: (nullifier: bigint, pubKey: PubKey, newVoiceCreditBalance: bigint) => number;
68
+ /**
69
+ * Update a Poll with data from MaciState.
70
+ * This is the step where we copy the state from the MaciState instance,
71
+ * and set the number of signups we have so far.
72
+ * @note It should be called to generate the state for poll joining with numSignups set as
73
+ * the number of signups in the MaciState. For message processing, you should set numSignups as
74
+ * the number of users who joined the poll.
75
+ */
76
+ updatePoll: (numSignups: bigint) => void;
77
+ /**
78
+ * Process one message.
79
+ * @param message - The message to process.
80
+ * @param encPubKey - The public key associated with the encryption private key.
81
+ * @returns A number of variables which will be used in the zk-SNARK circuit.
82
+ */
83
+ processMessage: (message: Message, encPubKey: PubKey, qv?: boolean) => IProcessMessagesOutput;
84
+ /**
85
+ * Inserts a Message and the corresponding public key used to generate the
86
+ * ECDH shared key which was used to encrypt said message.
87
+ * @param message - The message to insert
88
+ * @param encPubKey - The public key used to encrypt the message
89
+ */
90
+ publishMessage: (message: Message, encPubKey: PubKey) => void;
91
+ /**
92
+ * Updates message chain hash
93
+ * @param messageHash hash of message with encPubKey
94
+ */
95
+ updateChainHash: (messageHash: bigint) => void;
96
+ /**
97
+ * Create circuit input for pollJoining
98
+ * @param args Poll joining circuit inputs
99
+ * @returns stringified circuit inputs
100
+ */
101
+ joiningCircuitInputs: ({ maciPrivKey, stateLeafIndex, pollPubKey, }: IJoiningCircuitArgs) => IPollJoiningCircuitInputs;
102
+ /**
103
+ * Create circuit input for pollJoined
104
+ * @param args Poll joined circuit inputs
105
+ * @returns stringified circuit inputs
106
+ */
107
+ joinedCircuitInputs: ({ maciPrivKey, stateLeafIndex, voiceCreditsBalance, }: IJoinedCircuitArgs) => IPollJoinedCircuitInputs;
108
+ /**
109
+ * Pad last unclosed batch
110
+ */
111
+ padLastBatch: () => void;
112
+ /**
113
+ * This method checks if there are any unprocessed messages in the Poll instance.
114
+ * @returns Returns true if the number of processed batches is
115
+ * less than the total number of batches, false otherwise.
116
+ */
117
+ hasUnprocessedMessages: () => boolean;
118
+ /**
119
+ * Process _batchSize messages starting from the saved index. This
120
+ * function will process messages even if the number of messages is not an
121
+ * exact multiple of _batchSize. e.g. if there are 10 messages, index is
122
+ * 8, and _batchSize is 4, this function will only process the last two
123
+ * messages in this.messages, and finally update the zeroth state leaf.
124
+ * Note that this function will only process as many state leaves as there
125
+ * are ballots to prevent accidental inclusion of a new user after this
126
+ * poll has concluded.
127
+ * @param pollId The ID of the poll associated with the messages to
128
+ * process
129
+ * @param quiet - Whether to log errors or not
130
+ * @returns stringified circuit inputs
131
+ */
132
+ processMessages: (pollId: bigint, qv?: boolean, quiet?: boolean) => IProcessMessagesCircuitInputs;
133
+ /**
134
+ * Generates partial circuit inputs for processing a batch of messages
135
+ * @param index - The index of the partial batch.
136
+ * @returns stringified partial circuit inputs
137
+ */
138
+ private genProcessMessagesCircuitInputsPartial;
139
+ /**
140
+ * Process all messages. This function does not update the ballots or state
141
+ * leaves; rather, it copies and then updates them. This makes it possible
142
+ * to test the result of multiple processMessage() invocations.
143
+ * @returns The state leaves and ballots of the poll
144
+ */
145
+ processAllMessages: () => {
146
+ stateLeaves: StateLeaf[];
147
+ ballots: Ballot[];
148
+ };
149
+ /**
150
+ * Checks whether there are any untallied ballots.
151
+ * @returns Whether there are any untallied ballots
152
+ */
153
+ hasUntalliedBallots: () => boolean;
154
+ /**
155
+ * This method tallies a ballots and updates the tally results.
156
+ * @returns the circuit inputs for the TallyVotes circuit.
157
+ */
158
+ tallyVotes: () => ITallyCircuitInputs;
159
+ tallyVotesNonQv: () => ITallyCircuitInputs;
160
+ /**
161
+ * This method generates a commitment to the total spent voice credits.
162
+ *
163
+ * This is the hash of the total spent voice credits and a salt, computed as Poseidon([totalCredits, _salt]).
164
+ * @param salt - The salt used in the hash function.
165
+ * @param numBallotsToCount - The number of ballots to count for the calculation.
166
+ * @param useQuadraticVoting - Whether to use quadratic voting or not. Default is true.
167
+ * @returns Returns the hash of the total spent voice credits and a salt, computed as Poseidon([totalCredits, _salt]).
168
+ */
169
+ private genSpentVoiceCreditSubtotalCommitment;
170
+ /**
171
+ * This method generates a commitment to the spent voice credits per vote option.
172
+ *
173
+ * This is the hash of the Merkle root of the spent voice credits per vote option and a salt, computed as Poseidon([root, _salt]).
174
+ * @param salt - The salt used in the hash function.
175
+ * @param numBallotsToCount - The number of ballots to count for the calculation.
176
+ * @param useQuadraticVoting - Whether to use quadratic voting or not. Default is true.
177
+ * @returns Returns the hash of the Merkle root of the spent voice credits per vote option and a salt, computed as Poseidon([root, _salt]).
178
+ */
179
+ private genPerVOSpentVoiceCreditsCommitment;
180
+ /**
181
+ * Create a deep copy of the Poll object.
182
+ * @returns A new instance of the Poll object with the same properties.
183
+ */
184
+ copy: () => Poll;
185
+ /**
186
+ * Check if the Poll object is equal to another Poll object.
187
+ * @param p - The Poll object to compare.
188
+ * @returns True if the two Poll objects are equal, false otherwise.
189
+ */
190
+ equals: (p: Poll) => boolean;
191
+ /**
192
+ * Serialize the Poll object to a JSON object
193
+ * @returns a JSON object
194
+ */
195
+ toJSON(): IJsonPoll;
196
+ /**
197
+ * Deserialize a json object into a Poll instance
198
+ * @param json the json object to deserialize
199
+ * @param maciState the reference to the MaciState Class
200
+ * @returns a new Poll instance
201
+ */
202
+ static fromJSON(json: IJsonPoll, maciState: MaciState): Poll;
203
+ /**
204
+ * Set the coordinator's keypair
205
+ * @param serializedPrivateKey - the serialized private key
206
+ */
207
+ setCoordinatorKeypair: (serializedPrivateKey: string) => void;
208
+ /**
209
+ * Set the number of signups to match the ones from the contract
210
+ * @param numSignups - the number of signups
211
+ */
212
+ setNumSignups: (numSignups: bigint) => void;
213
+ /**
214
+ * Get the number of signups
215
+ * @returns The number of signups
216
+ */
217
+ getNumSignups: () => bigint;
218
+ }
219
+ //# sourceMappingURL=Poll.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"Poll.d.ts","sourceRoot":"","sources":["../../ts/Poll.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,EAYpB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACL,QAAQ,EACR,OAAO,EACP,MAAM,EACN,MAAM,EAEN,OAAO,EACP,SAAS,EAMV,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,OAAO,EAAuB,MAAM,kBAAkB,CAAC;AAIhE,OAAO,KAAK,EAEV,UAAU,EACV,UAAU,EACV,KAAK,EACL,SAAS,EACT,sBAAsB,EACtB,mBAAmB,EACnB,6BAA6B,EAC7B,yBAAyB,EACzB,mBAAmB,EACpB,MAAM,SAAS,CAAC;AACjB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,KAAK,EAAE,kBAAkB,EAAE,wBAAwB,EAAE,MAAM,eAAe,CAAC;AAMlF;;GAEG;AACH,qBAAa,IAAK,YAAW,KAAK;IAGhC,kBAAkB,EAAE,OAAO,CAAC;IAE5B,UAAU,EAAE,UAAU,CAAC;IAEvB,UAAU,EAAE,UAAU,CAAC;IAEvB,WAAW,EAAE,MAAM,CAAC;IAEpB,cAAc,EAAE,MAAM,CAAC;IAGvB,oBAAoB,EAAE,MAAM,CAAC;IAE7B,gBAAgB,EAAE,MAAM,CAAC;IAEzB,OAAO,EAAE,MAAM,EAAE,CAAM;IAEvB,UAAU,CAAC,EAAE,mBAAmB,CAAC;IAEjC,QAAQ,EAAE,OAAO,EAAE,CAAM;IAEzB,QAAQ,EAAE,QAAQ,EAAE,CAAM;IAE1B,UAAU,EAAE,MAAM,EAAE,CAAM;IAE1B,WAAW,UAAS;IAEpB,OAAO,EAAE,MAAM,EAAE,CAAY;IAE7B,SAAS,CAAC,EAAE,OAAO,CAAC;IAGpB,mBAAmB,SAAK;IAExB,wBAAwB,EAAE,MAAM,CAAC;IAEjC,YAAY,EAAE,SAAS,CAAC;IAExB,MAAM,EAAE,MAAM,CAAC;IAEf,OAAO,EAAE,MAAM,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,CAAC,CAAM;IAE9C,eAAe,EAAE,MAAM,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,CAAC,CAAM;IAEtD,+BAA+B,EAAE,MAAM,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,CAAC,CAAM;IAEtE,6BAA6B,EAAE,MAAM,CAAC,MAAM,GAAG,MAAM,EAAE,MAAM,CAAC,CAAM;IAGpE,WAAW,EAAE,MAAM,EAAE,CAAM;IAE3B,sBAAsB,EAAE,MAAM,EAAE,CAAM;IAEtC,iBAAiB,SAAK;IAEtB,sBAAsB,SAAM;IAI5B,WAAW,EAAE,MAAM,CAAC;IAEpB,eAAe,CAAC,EAAE,MAAM,CAAC;IAGzB,SAAS,SAAwB;IAGjC,WAAW,WAA0B;IAGrC,eAAe,EAAE,SAAS,EAAE,CAAoB;IAGhD,aAAa,CAAC,EAAE,mBAAmB,CAAC;IAGpC,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAGrC,OAAO,CAAC,UAAU,CAAM;IAExB;;;;;;;;OAQG;gBAED,gBAAgB,EAAE,MAAM,EACxB,kBAAkB,EAAE,OAAO,EAC3B,UAAU,EAAE,UAAU,EACtB,UAAU,EAAE,UAAU,EACtB,YAAY,EAAE,SAAS,EACvB,WAAW,EAAE,MAAM;IA0BrB;;OAEG;IACH,SAAS,GAAI,WAAW,MAAM,KAAG,OAAO,CAA+C;IAEvF;;;;;;OAMG;IACH,QAAQ,GAAI,WAAW,MAAM,EAAE,QAAQ,MAAM,EAAE,uBAAuB,MAAM,KAAG,MAAM,CAYnF;IAEF;;;;;;;OAOG;IACH,UAAU,GAAI,YAAY,MAAM,KAAG,IAAI,CAoDrC;IAEF;;;;;OAKG;IACH,cAAc,GAAI,SAAS,OAAO,EAAE,WAAW,MAAM,EAAE,YAAS,KAAG,sBAAsB,CA+GvF;IAEF;;;;;OAKG;IACH,cAAc,GAAI,SAAS,OAAO,EAAE,WAAW,MAAM,KAAG,IAAI,CAiC1D;IAEF;;;OAGG;IACH,eAAe,GAAI,aAAa,MAAM,KAAG,IAAI,CAO3C;IAEF;;;;OAIG;IACH,oBAAoB,GAAI,8CAIrB,mBAAmB,KAAG,yBAAyB,CA2ChD;IAEF;;;;OAIG;IACH,mBAAmB,GAAI,uDAIpB,kBAAkB,KAAG,wBAAwB,CAuB9C;IAEF;;OAEG;IACH,YAAY,QAAO,IAAI,CAIrB;IAEF;;;;OAIG;IACH,sBAAsB,QAAO,OAAO,CAUlC;IAEF;;;;;;;;;;;;;OAaG;IACH,eAAe,GAAI,QAAQ,MAAM,EAAE,YAAS,EAAE,eAAY,KAAG,6BAA6B,CAsQxF;IAEF;;;;OAIG;IACH,OAAO,CAAC,sCAAsC,CAkF5C;IAEF;;;;;OAKG;IACH,kBAAkB,QAAO;QAAE,WAAW,EAAE,SAAS,EAAE,CAAC;QAAC,OAAO,EAAE,MAAM,EAAE,CAAA;KAAE,CAUtE;IAEF;;;OAGG;IACH,mBAAmB,QAAO,OAAO,CAAkF;IAEnH;;;OAGG;IACH,UAAU,QAAO,mBAAmB,CA0KlC;IAEF,eAAe,QAAO,mBAAmB,CAoIvC;IAEF;;;;;;;;OAQG;IACH,OAAO,CAAC,qCAAqC,CAiB3C;IAEF;;;;;;;;OAQG;IACH,OAAO,CAAC,mCAAmC,CAoBzC;IAEF;;;OAGG;IACH,IAAI,QAAO,IAAI,CA+Db;IAEF;;;;OAIG;IACH,MAAM,GAAI,GAAG,IAAI,KAAG,OAAO,CA2BzB;IAEF;;;OAGG;IACH,MAAM,IAAI,SAAS;IAwBnB;;;;;OAKG;IACH,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,GAAG,IAAI;IA6B5D;;;OAGG;IACH,qBAAqB,GAAI,sBAAsB,MAAM,KAAG,IAAI,CAE1D;IAEF;;;OAGG;IACH,aAAa,GAAI,YAAY,MAAM,KAAG,IAAI,CAExC;IAEF;;;OAGG;IACH,aAAa,QAAO,MAAM,CAAoB;CAC/C"}