feat: register GPU & get cluster info

src/Scheduler.sol

@@ -0,0 +1,54 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.13;
+
+contract Scheduler {
+    struct Cluster {
+        address provider;
+        uint256 gpuId;
+        uint256 clusterSize;
+        bool available;
+    }
+
+    struct Task {
+        address provider;
+        address client;
+        string dataImage;
+        string trainImage;
+        uint256 status;
+    }
+
+    uint256 private randomNumber;
+    Cluster[] public clusters;
+    Task[] public tasks;
+
+    function getRandomNumber() internal returns (uint256) {
+        randomNumber = uint (keccak256(abi.encodePacked (msg.sender, block.timestamp, randomNumber)));
+        return randomNumber;
+    }
+
+    function registerCluster(uint256 gpuId, uint256 clusterSize) public {
+        address provider = msg.sender;
+        clusters.push(Cluster(provider, gpuId, clusterSize, false));
+    }
+
+    function registerTaskWithSpecificCluster(string memory dataImage, string memory trainImage, uint256 clusterIndex) public {
+        require(clusters[clusterIndex].available, "This GPU Cluster has been chosen");
+        clusters[clusterIndex].available = false;
+
+        address provider = clusters[clusterIndex].provider;
+        address client = msg.sender;
+        tasks.push(Task(provider, client, dataImage, trainImage, 1));
+    }
+
+    function getNumberOfClusters() public view returns (uint256) {
+        return clusters.length;
+    }
+
+    function getClusterInfo(uint256 clusterIndex) public view returns(Cluster memory) {
+        return clusters[clusterIndex];
+    }
+    
+    function getClusters() public view returns (Cluster[] memory) {
+        return clusters;
+    }
+}