Add getIndexSet (#7)

* add getIndexSet and tests

* add extra test case

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polymarket/sdk",
-  "version": "5.0.0",
+  "version": "5.0.1",
   "description": "SDK to simplify common interactions with the Polymarket proxy wallet",
   "author": "Tom French <[email protected]>",
   "repository": "https://github.com/TokenUnion/polymarket-sdk.git",

src/index.ts

@@ -2,6 +2,7 @@ export * from "./conditionalTokens";
 export * from "./matic";
 export * from "./markets";
 export * from "./debt";
-export { ethTransferTransaction, erc20TransferTransaction } from "./utils";
+
+export { ethTransferTransaction, erc20TransferTransaction, getIndexSet } from "./utils";
 export { getProxyWalletAddress } from "./proxyWallet";
-export { negRisk } from "./negRisk";
+export { negRiskOperations } from "./negRisk";

src/negRisk/index.ts

@@ -39,6 +39,6 @@ const redeemPositions = ({ negRiskAdapterAddress, conditionId, amounts }: Redeem
   value: "0",
 });
 
-const negRisk = { convertPositions, redeemPositions };
+const negRiskOperations = { convertPositions, redeemPositions };
 
-export { negRisk };
+export { negRiskOperations };

src/utils/getIndexSet.ts

@@ -0,0 +1,5 @@
+const getIndexSet = (indices: number[]) =>
+  // eslint-disable-next-line no-bitwise
+  [...new Set(indices)].reduce((acc, index) => acc + (1 << index), 0);
+
+export { getIndexSet };

src/utils/index.ts

@@ -2,3 +2,4 @@ export { erc20ApprovalTransaction } from "./approveErc20";
 export { erc1155ApprovalTransaction } from "./approveErc1155";
 export { erc20TransferTransaction } from "./transferErc20";
 export { ethTransferTransaction } from "./transferEth";
+export { getIndexSet } from "./getIndexSet";

test/getIndexSet.test.ts

@@ -0,0 +1,28 @@
+/* eslint-env jest */
+
+import { getIndexSet } from "../src";
+
+const testCases = [
+  [[0], 1],
+  [[1], 2],
+  [[0, 1], 3],
+  [[0, 0, 1], 3],
+  [[2], 4],
+  [[0, 2], 5],
+  [[1, 2], 6],
+  [[0, 1, 2], 7],
+  [[3], 8],
+  [[0, 3], 9],
+  [[1, 3], 10],
+  [[0, 1, 3], 11],
+  [[2, 3], 12],
+  [[0, 2, 3], 13],
+  [[1, 2, 3], 14],
+  [[0, 1, 2, 3], 15],
+] as [number[], number][];
+
+describe("getIndexSet", () => {
+  it.each(testCases)(`should compute the index set`, (indices, expectedIndexSet) => {
+    expect(getIndexSet(indices)).toEqual(expectedIndexSet);
+  });
+});