modified clientSpec check to do JSON-singing only for BabyJubJubSigna…

…ture2023 signed document
hypersign-protocol · Nov 13, 2023 · d10e2ae · d10e2ae
1 parent 0c4abb1
commit d10e2ae
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/x/ssi/verification/client_spec.go b/x/ssi/verification/client_spec.go
@@ -56,7 +56,7 @@ func getPersonalSignSpecDocBytes(ssiMsg types.SsiMsg) ([]byte, error) {
 func getDocBytesByClientSpec(ssiMsg types.SsiMsg, extendedVm *types.ExtendedVerificationMethod) ([]byte, error) {
 	switch extendedVm.Proof.ClientSpecType {
 	case types.CLIENT_SPEC_TYPE_NONE:
-		if didDoc, ok := ssiMsg.(*types.DidDocument); ok && len(didDoc.Context) == 0 {
+		if extendedVm.Proof.Type == types.BabyJubJubSignature2023 {
 			return ssiMsg.GetSignBytes(), nil
 		}
 
@@ -67,7 +67,7 @@ func getDocBytesByClientSpec(ssiMsg types.SsiMsg, extendedVm *types.ExtendedVeri
 			return nil, err
 		}
 
-		if didDoc, ok := ssiMsg.(*types.DidDocument); ok && len(didDoc.Context) == 0 {
+		if extendedVm.Proof.Type == types.BabyJubJubSignature2023 {
 			return getCosmosADR036SignDocBytes(ssiMsg.GetSignBytes(), signerAddress)
 		}
 
@@ -79,7 +79,7 @@ func getDocBytesByClientSpec(ssiMsg types.SsiMsg, extendedVm *types.ExtendedVeri
 		return getCosmosADR036SignDocBytes(canonizedDidDocHash, signerAddress)
 
 	case types.CLIENT_SPEC_TYPE_ETH_PERSONAL_SIGN:
-		if didDoc, ok := ssiMsg.(*types.DidDocument); ok && len(didDoc.Context) == 0 {
+		if extendedVm.Proof.Type == types.BabyJubJubSignature2023 {
 			return getPersonalSignSpecDocBytes(ssiMsg)
 		}