Merge pull request #27 from KintoXYZ/viewer-has-trait

Adds hasTrait to viewer
KintoXYZ · Jan 4, 2024 · 86fa905 · 86fa905
2 parents 6cc4ac7 + 0cb5b1b
commit 86fa905
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/src/interfaces/IKYCViewer.sol b/src/interfaces/IKYCViewer.sol
@@ -17,6 +17,8 @@ interface IKYCViewer {
 
     function isIndividual(address _account) external view returns (bool);
 
+    function hasTrait(address _account, uint8 _traitId) external view returns (bool);
+
     /* ============ Constants and attrs ============ */
 
     function kintoID() external view returns (IKintoID);

diff --git a/src/viewers/KYCViewer.sol b/src/viewers/KYCViewer.sol
@@ -72,6 +72,10 @@ contract KYCViewer is Initializable, UUPSUpgradeable, OwnableUpgradeable, IKYCVi
         return kintoID.isIndividual(_getFinalAddress(_account));
     }
 
+    function hasTrait(address _account, uint8 _traitId) external view returns (bool) {
+        return kintoID.hasTrait(_getFinalAddress(_account), _traitId);
+    }
+
     /* ============ Helpers ============ */
 
     function _getFinalAddress(address _address) private view returns (address) {

diff --git a/test/KYCViewer.t.sol b/test/KYCViewer.t.sol
@@ -104,5 +104,6 @@ contract KYCViewerTest is Create2Helper, UserOp, AATestScaffolding {
         assertEq(_kycViewer.isKYC(address(_kintoWalletv1)), _kycViewer.isKYC(_owner));
         assertEq(_kycViewer.isIndividual(address(_kintoWalletv1)), _kycViewer.isIndividual(_owner));
         assertEq(_kycViewer.isCompany(address(_kintoWalletv1)), false);
+        assertEq(_kycViewer.hasTrait(address(_kintoWalletv1), 6), false);
     }
 }