defined typedDataTypes

iden3 · Sep 2, 2024 · f414b42 · f414b42
1 parent 62a3dd9
commit f414b42
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/pkg/services/did.go b/pkg/services/did.go
@@ -184,7 +184,7 @@ func (d *DidDocumentServices) GetDidDocument(ctx context.Context, did string, op
 	return didResolution, nil
 }
 
-func getTypedData(typeDataType TypedDataType, did w3c.DID, identityState IdentityState) (apitypes.TypedData, error) {
+func getTypedData(typedDataType TypedDataType, did w3c.DID, identityState IdentityState) (apitypes.TypedData, error) {
 	id, err := core.IDFromDID(did)
 	if err != nil {
 		return apitypes.TypedData{},
@@ -195,9 +195,10 @@ func getTypedData(typeDataType TypedDataType, did w3c.DID, identityState Identit
 
 	var apiTypes apitypes.Types
 	var message apitypes.TypedDataMessage
-	primaryType := ""
+	var primaryType string
 
-	if typeDataType == IdentityStateType {
+	switch typedDataType {
+	case IdentityStateType:
 		primaryType = "IdentityState"
 		apiTypes = apitypes.Types{
 			"IdentityState": []apitypes.Type{
@@ -223,7 +224,7 @@ func getTypedData(typeDataType TypedDataType, did w3c.DID, identityState Identit
 			"replacedAtTimestamp": replacedAtTimestamp,
 		}
 
-	} else {
+	case GlobalStateType:
 		primaryType = "GlobalState"
 		apiTypes = apitypes.Types{
 			"GlobalState": []apitypes.Type{
@@ -248,6 +249,8 @@ func getTypedData(typeDataType TypedDataType, did w3c.DID, identityState Identit
 			"root":                root,
 			"replacedAtTimestamp": replacedAtTimestamp,
 		}
+	default:
+		return apitypes.TypedData{}, fmt.Errorf("typedDataType %d not defined", typedDataType)
 	}
 
 	typedData := apitypes.TypedData{