feat: add ipAddressHbarSpendingPlan

Signed-off-by: Ivo Yankov <[email protected]>

packages/relay/src/lib/db/entities/hbarLimiter/ipAddressHbarSpendingPlan.ts

@@ -0,0 +1,31 @@
+/*
+ *
+ * Hedera JSON RPC Relay
+ *
+ * Copyright (C) 2022-2024 Hedera Hashgraph, LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+import { IIPAddressHbarSpendingPlan } from '../../types/hbarLimiter/ipAddressHbarSpendingPlan';
+
+export class IPAddressHbarSpendingPlan implements IIPAddressHbarSpendingPlan {
+  ipAddress: string;
+  planId: string;
+
+  constructor(data: IIPAddressHbarSpendingPlan) {
+    this.ipAddress = data.ipAddress;
+    this.planId = data.planId;
+  }
+}

packages/relay/src/lib/db/repositories/hbarLimiter/ipAddressHbarSpendingPlanRepository.ts

@@ -0,0 +1,97 @@
+/*
+ *
+ * Hedera JSON RPC Relay
+ *
+ * Copyright (C) 2022-2024 Hedera Hashgraph, LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+import { CacheService } from '../../../services/cacheService/cacheService';
+import { Logger } from 'pino';
+import { IIPAddressHbarSpendingPlan } from '../../types/hbarLimiter/ipAddressHbarSpendingPlan';
+import { IPAddressHbarSpendingPlanNotFoundError } from '../../types/hbarLimiter/errors';
+import { IPAddressHbarSpendingPlan } from '../../entities/hbarLimiter/ipAddressHbarSpendingPlan';
+
+export class IPAddressHbarSpendingPlanRepository {
+  private readonly collectionKey = 'ipAddressHbarSpendingPlan';
+  private readonly threeMonthsInMillis = 90 * 24 * 60 * 60 * 1000;
+
+  /**
+   * The cache service used for storing data.
+   * @private
+   */
+  private readonly cache: CacheService;
+
+  /**
+   * The logger used for logging all output from this class.
+   * @private
+   */
+  private readonly logger: Logger;
+
+  constructor(cache: CacheService, logger: Logger) {
+    this.cache = cache;
+    this.logger = logger;
+  }
+
+  /**
+   * Finds an {@link IPAddressHbarSpendingPlan} for an IP address.
+   *
+   * @param {string} ipAddress - The IP address to search for.
+   * @returns {Promise<IPAddressHbarSpendingPlan>} - The associated plan for the IP address.
+   */
+  async findByAddress(ipAddress: string): Promise<IPAddressHbarSpendingPlan> {
+    const key = this.getKey(ipAddress);
+    const addressPlan = await this.cache.getAsync<IIPAddressHbarSpendingPlan>(key, 'findByAddress');
+    if (!addressPlan) {
+      throw new IPAddressHbarSpendingPlanNotFoundError(ipAddress);
+    }
+    this.logger.trace(`Retrieved IPAddressHbarSpendingPlan with address ${ipAddress}`);
+    return new IPAddressHbarSpendingPlan(addressPlan);
+  }
+
+  /**
+   * Saves an {@link IPAddressHbarSpendingPlan} to the cache, linking the plan to the IP address.
+   *
+   * @param {IIPAddressHbarSpendingPlan} addressPlan - The plan to save.
+   * @returns {Promise<void>} - A promise that resolves when the IP address is linked to the plan.
+   */
+  async save(addressPlan: IIPAddressHbarSpendingPlan): Promise<void> {
+    const key = this.getKey(addressPlan.ipAddress);
+    await this.cache.set(key, addressPlan, 'save', this.threeMonthsInMillis);
+    this.logger.trace(`Saved IPAddressHbarSpendingPlan with address ${addressPlan.ipAddress}`);
+  }
+
+  /**
+   * Deletes an {@link IPAddressHbarSpendingPlan} from the cache, unlinking the plan from the IP address.
+   *
+   * @param {string} ipAddress - The IP address to unlink the plan from.
+   * @returns {Promise<void>} - A promise that resolves when the IP address is unlinked from the plan.
+   */
+  async delete(ipAddress: string): Promise<void> {
+    const key = this.getKey(ipAddress);
+    await this.cache.delete(key, 'delete');
+    this.logger.trace(`Deleted IPAddressHbarSpendingPlan with address ${ipAddress}`);
+  }
+
+  /**
+   * Gets the cache key for an {@link IPAddressHbarSpendingPlan}.
+   *
+   * @param {string} ipAddress - The IP address to get the key for.
+   * @private
+   */
+  private getKey(ipAddress: string): string {
+    return `${this.collectionKey}:${ipAddress}`;
+  }
+}

packages/relay/src/lib/db/types/hbarLimiter/errors.ts

@@ -38,3 +38,10 @@ export class EthAddressHbarSpendingPlanNotFoundError extends Error {
     this.name = 'EthAddressHbarSpendingPlanNotFoundError';
   }
 }
+
+export class IPAddressHbarSpendingPlanNotFoundError extends Error {
+  constructor(ipAddress: string) {
+    super(`IPAddressHbarSpendingPlan with address ${ipAddress} not found`);
+    this.name = 'IPAddressHbarSpendingPlanNotFoundError';
+  }
+}

packages/relay/src/lib/db/types/hbarLimiter/ipAddressHbarSpendingPlan.ts

@@ -0,0 +1,24 @@
+/*
+ *
+ * Hedera JSON RPC Relay
+ *
+ * Copyright (C) 2022-2024 Hedera Hashgraph, LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+export interface IIPAddressHbarSpendingPlan {
+  ipAddress: string;
+  planId: string;
+}

packages/relay/src/lib/services/hbarLimitService/index.ts

@@ -21,6 +21,7 @@
 import { IHbarLimitService } from './IHbarLimitService';
 import { HbarSpendingPlanRepository } from '../../db/repositories/hbarLimiter/hbarSpendingPlanRepository';
 import { EthAddressHbarSpendingPlanRepository } from '../../db/repositories/hbarLimiter/ethAddressHbarSpendingPlanRepository';
+import { IPAddressHbarSpendingPlanRepository } from '../../db/repositories/hbarLimiter/ipAddressHbarSpendingPlanRepository';
 import { IDetailedHbarSpendingPlan } from '../../db/types/hbarLimiter/hbarSpendingPlan';
 import { SubscriptionType } from '../../db/types/hbarLimiter/subscriptionType';
 import { Logger } from 'pino';
@@ -37,6 +38,7 @@ export class HbarLimitService implements IHbarLimitService {
   constructor(
     private readonly hbarSpendingPlanRepository: HbarSpendingPlanRepository,
     private readonly ethAddressHbarSpendingPlanRepository: EthAddressHbarSpendingPlanRepository,
+    private readonly ipAddressHbarSpendingPlanRepository: IPAddressHbarSpendingPlanRepository,
     private readonly logger: Logger,
   ) {}
 
@@ -95,7 +97,11 @@ export class HbarLimitService implements IHbarLimitService {
       }
     }
     if (ipAddress) {
-      // TODO: Implement this with https://github.com/hashgraph/hedera-json-rpc-relay/issues/2888
+      try {
+        return await this.getSpendingPlanByIPAddress(ipAddress);
+      } catch (error) {
+        this.logger.warn(error, `Failed to get spending plan for IP address '${ipAddress}'`);
+      }
     }
     return null;
   }
@@ -111,6 +117,17 @@ export class HbarLimitService implements IHbarLimitService {
     return this.hbarSpendingPlanRepository.findByIdWithDetails(ethAddressHbarSpendingPlan.planId);
   }
 
+  /**
+   * Gets the spending plan for the given IP address.
+   * @param {string} ipAddress - The IP address to get the spending plan for.
+   * @returns {Promise<IDetailedHbarSpendingPlan>} - A promise that resolves with the spending plan.
+   * @private
+   */
+  private async getSpendingPlanByIPAddress(ipAddress: string): Promise<IDetailedHbarSpendingPlan> {
+    const ipAddressHbarSpendingPlan = await this.ipAddressHbarSpendingPlanRepository.findByAddress(ipAddress);
+    return this.hbarSpendingPlanRepository.findByIdWithDetails(ipAddressHbarSpendingPlan.planId);
+  }
+
   /**
    * Creates a basic spending plan for the given eth address.
    * @param {string} ethAddress - The eth address to create the spending plan for.
@@ -128,8 +145,7 @@ export class HbarLimitService implements IHbarLimitService {
       await this.ethAddressHbarSpendingPlanRepository.save({ ethAddress, planId: spendingPlan.id });
     } else if (ipAddress) {
       this.logger.trace(`Linking spending plan with ID ${spendingPlan.id} to ip address ${ipAddress}`);
-      // TODO: Implement this with https://github.com/hashgraph/hedera-json-rpc-relay/issues/2888
-      // await this.ipAddressHbarSpendingPlanRepository.save({ ipAddress, planId: spendingPlan.id });
+      await this.ipAddressHbarSpendingPlanRepository.save({ ipAddress, planId: spendingPlan.id });
     }
     return spendingPlan;
   }