wip

harb-lib/jest.config.js

@@ -1,4 +1,5 @@
 module.exports = {
   preset: 'ts-jest',
   testEnvironment: 'node',
+  maxWorkers: 1
 };

harb-lib/package.json

@@ -1,6 +1,6 @@
 {
   "name": "harb-lib",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "helper functions and snatch selection",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

harb-lib/src/helpers.ts

@@ -0,0 +1,37 @@
+import { bytesToUint256LittleEndian } from "./subgraph";
+import { Position, PositionStatus } from './__generated__/graphql';
+
+const STAKE_TOTOAL_SUPPLY: bigint = 1000000000000000000n * 10000000n;
+
+const sortAndFilterPositions = (positions) => {
+    return positions
+        .filter(position => position.status === PositionStatus.Active)
+        .sort((a, b) => parseFloat(a.taxRate) - parseFloat(b.taxRate));
+};
+
+export function getSnatchList(
+	positions: Position[], // all active positions
+	needed: bigint,		// amount of stake requested by user
+	taxRate: number
+	): bigint[] {
+
+	const sortedActivePositions = sortAndFilterPositions(positions);
+
+	const rv: bigint[] = [];
+	let i = 0;
+	while (needed > 0 && i < positions.length) {
+		const available = (STAKE_TOTOAL_SUPPLY * BigInt(Math.round(sortedActivePositions[i].share * 100000))) / BigInt(100000);
+		if (sortedActivePositions[i].taxRate < taxRate) {
+			rv.push(bytesToUint256LittleEndian(sortedActivePositions[i].id));
+			console.log(rv, available);
+			needed -= available;
+		}
+		i++;
+	    // code block to be executed
+	}
+	
+	if (needed > 0)
+		throw new Error('Not enough capacity');
+
+	return rv;
+}

harb-lib/src/index.ts

@@ -1 +1,3 @@
-export {bytesToUint256LittleEndian, uint256ToBytesLittleEndian} from './subgraph'
+export {bytesToUint256LittleEndian, uint256ToBytesLittleEndian} from './subgraph'
+
+export {getSnatchList} from './helpers'

harb-lib/src/tests/functions.test.ts

@@ -1,4 +1,4 @@
-import { bytesToUint256LittleEndian, uint256ToBytesLittleEndian } from "../subgraph"; // Adjust the import path to where your functions are defined
+import { bytesToUint256LittleEndian, uint256ToBytesLittleEndian } from "../subgraph";
 import { Position, PositionStatus } from '../__generated__/graphql';
 
 

harb-lib/src/tests/helpers.test.ts

@@ -0,0 +1,42 @@
+import { uint256ToBytesLittleEndian } from "../subgraph";
+import { getSnatchList } from "../helpers";
+import { Position, PositionStatus } from '../__generated__/graphql';
+
+
+describe('BigInt Conversion Functions', () => {
+  test('get a list of positions to snatch', async () => {
+
+    const mockPos: Position[] = [
+      {
+        __typename: 'Position',
+        id: uint256ToBytesLittleEndian(3n),
+        owner: '0x8db6b632d743aef641146dc943acb64957155388',
+        share: 0.000001,
+        creationTime: 1610000000,
+        taxRate: 0.05,
+        status: PositionStatus.Active, // Enum usage
+      },
+      {
+        __typename: 'Position',
+        id: uint256ToBytesLittleEndian(4n),
+        owner: '0x8db6b632d743aef641146dc943acb64957155388',
+        share: 0.000001,
+        creationTime: 1610000000,
+        taxRate: 0.01,
+        status: PositionStatus.Active, // Enum usage
+      },
+      {
+        __typename: 'Position',
+        id: uint256ToBytesLittleEndian(5n),
+        owner: '0x8db6b632d743aef641146dc943acb64957155388',
+        share: 0.000001,
+        creationTime: 1610000000,
+        taxRate: 0.02,
+        status: PositionStatus.Active, // Enum usage
+      }
+    ];
+
+    expect(getSnatchList(mockPos, 20000000000000000000n, 0.03)).toEqual([4n, 5n]);
+  });
+
+});