renting.ts

/*
# Renting Nodes Test Script.

This script is intended for testing the process of renting nodes. It includes functions to:
1. Get rentable nodes
2. Reserve a node
3. Get rent contract ID for a reserved node
4. Unreserve a node

** HINT: Modify it according to your needs and follow coding best practices.
** HINT: If you intend to unreserve a node you've rented, modify the 'unreserve' caller to have the rented node's ID hard-coded and comment out the rest of the script to prevent re-renting another node.
*/

// Import required modules
import { GridClient } from "../src/client";
import { getClient } from "./client_loader";
import { log } from "./utils";

// Get rent contract ID for a node
async function getRentContract(client: GridClient, nodeId: number) {
  const rentContractId = await client.nodes.getRentContractId({ nodeId });
  log("================= Rent Contract ID =================");
  log(rentContractId);
  log("================= Rent Contract ID =================");
  return rentContractId;
}

// Reserve a node
async function reserveNode(client: GridClient, nodeId: number) {
  const reserved = await client.nodes.reserve({ nodeId });
  log("================= Reserve Nodes =================");
  log(reserved);
  log("================= Reserve Nodes =================");
  return reserved;
}

// Get a list of rentable nodes
async function getRentableNodes(client: GridClient) {
  const rentable = await client.capacity.filterNodes({ rentable: true });
  if (rentable.length > 0) {
    log("================= Rentable Nodes =================");
    log(rentable);
    log("================= Rentable Nodes =================");
    return rentable;
  } else {
    console.error("No rentable nodes available.");
    return [];
  }
}

// Unreserve a node
// eslint-disable-next-line @typescript-eslint/no-unused-vars
async function unreserve(client: GridClient, nodeId: number) {
  const unreserved = await client.nodes.unreserve({ nodeId });
  log("================= Unreserved Node =================");
  log(unreserved);
  log("================= Unreserved Node =================");
}

// Main function
async function main() {
  try {
    // Initialize the client
    const client = await getClient();

    // Get a list of rentable nodes
    const rentableNodes = await getRentableNodes(client);

    // If there are rentable nodes available
    if (rentableNodes.length) {
      const reservedNodeId = rentableNodes[0].nodeId;

      // Reserve the first available node
      await reserveNode(client, reservedNodeId);

      // Get the rent contract for the reserved node
      await getRentContract(client, reservedNodeId);

      // Uncomment the line below if you intend to perform unreserve.
      // await unreserve(client, reservedNodeId);
    }

    // Disconnect the client
    await client.disconnect();
  } catch (error) {
    console.error("An error occurred:", error);
  }
}

// Call the main function
main();