feat: Witnet*.class(): string

witnet · Nov 23, 2023 · 14cbe74 · 14cbe74
1 parent 77bee1d
commit 14cbe74
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 0 deletions.
diff --git a/contracts/WitnetBytecodes.sol b/contracts/WitnetBytecodes.sol
@@ -9,5 +9,8 @@ abstract contract WitnetBytecodes
     is
         IWitnetBytecodes
 {
+    function class() virtual external view returns (string memory) {
+        return type(WitnetBytecodes).name;
+    }   
     function specs() virtual external view returns (bytes4);
 }
diff --git a/contracts/WitnetRequestBoard.sol b/contracts/WitnetRequestBoard.sol
@@ -12,6 +12,9 @@ abstract contract WitnetRequestBoard
     is
         IWitnetRequestBoard
 {
+    function class() virtual external view returns (string memory) {
+        return type(WitnetRequestBoard).name;
+    }
     function factory() virtual external view returns (WitnetRequestFactory);
     function registry() virtual external view returns (WitnetBytecodes);
     function specs() virtual external view returns (bytes4);

diff --git a/contracts/WitnetRequestFactory.sol b/contracts/WitnetRequestFactory.sol
@@ -11,6 +11,7 @@ abstract contract WitnetRequestFactory
     is
         IWitnetRequestFactory
 {
+    function class() virtual external view returns (string memory);
     function registry() virtual external view returns (WitnetBytecodes);
     function specs() virtual external view returns (bytes4);
     function witnet() virtual external view returns (WitnetRequestBoard);

diff --git a/contracts/WitnetRequestTemplate.sol b/contracts/WitnetRequestTemplate.sol
@@ -11,6 +11,7 @@ abstract contract WitnetRequestTemplate
 {
     event WitnetRequestBuilt(address indexed request, bytes32 indexed radHash, string[][] args);
 
+    function class() virtual external view returns (string memory);
     function factory() virtual external view returns (WitnetRequestFactory);
     function registry() virtual external view returns (WitnetBytecodes);
     function specs() virtual external view returns (bytes4);

diff --git a/contracts/core/defaults/WitnetRequestFactoryDefault.sol b/contracts/core/defaults/WitnetRequestFactoryDefault.sol
@@ -184,6 +184,23 @@ contract WitnetRequestFactoryDefault
         );
     }
 
+    function class()
+        virtual override(WitnetRequestFactory, WitnetRequestTemplate)
+        external view
+        returns (string memory)
+    {
+        if (
+            address(this) == _SELF
+                || address(this) == __proxy()
+        ) {
+            return type(WitnetRequestFactory).name;
+        } else if (__witnetRequest().radHash != bytes32(0)) {
+            return type(WitnetRequest).name;
+        } else {
+            return type(WitnetRequestTemplate).name;
+        }
+    }
+
     function specs() 
         virtual override(WitnetRequestFactory, WitnetRequestTemplate)
         external view