allow skipping more modules

pk910 · Oct 30, 2023 · 87360d7 · 87360d7
1 parent 85539e3
commit 87360d7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/modules/faucet-balance/FaucetBalanceModule.ts b/src/modules/faucet-balance/FaucetBalanceModule.ts
@@ -32,6 +32,8 @@ export class FaucetBalanceModule extends BaseModule<IFaucetBalanceConfig> {
   }
 
   private async processSessionRewardFactor(session: FaucetSession, rewardFactors: ISessionRewardFactor[]): Promise<void> {
+    if(session.getSessionData<Array<string>>("skip.modules", []).indexOf(this.moduleName) !== -1)
+      return;
     await this.refreshBalanceRestriction();
     if(this.balanceRestriction !== 100) {
       rewardFactors.push({

diff --git a/src/modules/faucet-outflow/FaucetOutflowModule.ts b/src/modules/faucet-outflow/FaucetOutflowModule.ts
@@ -28,7 +28,7 @@ export class FaucetOutflowModule extends BaseModule<IFaucetOutflowConfig> {
     );
     this.moduleManager.addActionHook(
       this, ModuleHookAction.SessionRewarded, 5, "faucet outflow",
-      (session: FaucetSession, amount: bigint) => this.updateState(amount)
+      (session: FaucetSession, amount: bigint) => this.updateState(session, amount)
     );
     this.enforceBalanceBoundaries();
   }
@@ -44,6 +44,8 @@ export class FaucetOutflowModule extends BaseModule<IFaucetOutflowConfig> {
   }
 
   private async processSessionRewardFactor(session: FaucetSession, rewardFactors: ISessionRewardFactor[]): Promise<void> {
+    if(session.getSessionData<Array<string>>("skip.modules", []).indexOf(this.moduleName) !== -1)
+      return;
     let outflowRestriction = this.getOutflowRestriction();
     if(outflowRestriction < 100) {
       rewardFactors.push({
@@ -83,7 +85,9 @@ export class FaucetOutflowModule extends BaseModule<IFaucetOutflowConfig> {
     }
   }
 
-  public updateState(minedAmount: bigint) {
+  public updateState(session: FaucetSession, minedAmount: bigint) {
+    if(session.getSessionData<Array<string>>("skip.modules", []).indexOf(this.moduleName) !== -1)
+      return;
     if(minedAmount < 0)
       return;