use filterworkerpoolorders

Author: SeddikBellamine

packages/sdk/src/lib/dataProtectorCore/processProtectedData.ts

@@ -34,7 +34,10 @@ import {
 import { IExecConsumer } from '../types/internalTypes.js';
 import { getWhitelistContract } from './smartContract/getWhitelistContract.js';
 import { isAddressInWhitelist } from './smartContract/whitelistContract.read.js';
-import { checkUserVoucher } from '../../utils/processProtectedData.models.js';
+import {
+  checkUserVoucher,
+  filterWorkerpoolOrders,
+} from '../../utils/processProtectedData.models.js';
 
 export type ProcessProtectedData = typeof processProtectedData;
 
@@ -182,11 +185,18 @@ export const processProtectedData = async ({
       title: 'FETCH_WORKERPOOL_ORDERBOOK',
       isDone: true,
     });
+    const desiredPriceWorkerpoolOrder = filterWorkerpoolOrders({
+      workerpoolOrders: [...workerpoolOrderbook.orders],
+      useVoucher: vUseVoucher,
+      userVoucher,
+    });
+    if (!desiredPriceWorkerpoolOrder) {
+      throw new Error('No Workerpool order found.');
+    }
 
     const underMaxPriceOrders = fetchOrdersUnderMaxPrice(
       datasetOrderbook,
       appOrderbook,
-      workerpoolOrderbook,
       vMaxPrice
     );
 
@@ -206,13 +216,13 @@ export const processProtectedData = async ({
     });
     const requestorderToSign = await iexec.order.createRequestorder({
       app: vApp,
-      category: underMaxPriceOrders.workerpoolorder.category,
+      category: desiredPriceWorkerpoolOrder.category,
       dataset: vProtectedData,
       appmaxprice: underMaxPriceOrders.apporder.appprice,
       datasetmaxprice: underMaxPriceOrders.datasetorder.datasetprice,
-      workerpoolmaxprice: underMaxPriceOrders.workerpoolorder.workerpoolprice,
+      workerpoolmaxprice: desiredPriceWorkerpoolOrder.workerpoolprice,
       tag: SCONE_TAG,
-      workerpool: underMaxPriceOrders.workerpoolorder.workerpool,
+      workerpool: desiredPriceWorkerpoolOrder.workerpool,
       params: {
         iexec_input_files: vInputFiles,
         iexec_secrets: secretsId,
@@ -227,6 +237,7 @@ export const processProtectedData = async ({
     const { dealid, txHash } = await iexec.order.matchOrders(
       {
         requestorder,
+        workerpoolorder: desiredPriceWorkerpoolOrder,
         ...underMaxPriceOrders,
       },
       matchOptions

packages/sdk/src/utils/processProtectedData.models.ts

@@ -1,4 +1,5 @@
 import { Address, BN } from 'iexec';
+import { PublishedWorkerpoolorder } from 'iexec/IExecOrderbookModule';
 
 type VoucherInfo = {
   owner: Address;
@@ -34,3 +35,51 @@ export function checkUserVoucher({
     );
   }
 }
+
+export function filterWorkerpoolOrders({
+  workerpoolOrders,
+  useVoucher,
+  userVoucher,
+}: {
+  workerpoolOrders: PublishedWorkerpoolorder[];
+  useVoucher: boolean;
+  userVoucher?: VoucherInfo;
+}) {
+  if (workerpoolOrders.length === 0) {
+    return null;
+  }
+
+  let eligibleWorkerpoolOrders = [...workerpoolOrders];
+  let maxVoucherSponsoredAmount = 0; // may be safer to use bigint
+
+  if (useVoucher) {
+    if (!userVoucher) {
+      throw new Error(
+        'useVoucher === true but userVoucher is undefined? Hum...'
+      );
+    }
+    // only voucher sponsored workerpoolorders
+    eligibleWorkerpoolOrders = eligibleWorkerpoolOrders.filter(({ order }) =>
+      userVoucher.sponsoredWorkerpools.includes(order.workerpool)
+    );
+    if (eligibleWorkerpoolOrders.length === 0) {
+      throw new Error(
+        'Found some workerpool orders but none can be sponsored by your voucher.'
+      );
+    }
+    maxVoucherSponsoredAmount = bnToNumber(userVoucher.balance);
+  }
+
+  const [cheapestOrder] = eligibleWorkerpoolOrders.sort(
+    (order1, order2) =>
+      order1.order.workerpoolprice - order2.order.workerpoolprice
+  );
+
+  if (
+    !cheapestOrder ||
+    cheapestOrder.order.workerpoolprice > maxVoucherSponsoredAmount
+  ) {
+    return null;
+  }
+  return cheapestOrder.order;
+}