Add BuildContact and CRUD functionality for contacts

Author: fourjr

appstate/encode.go

@@ -32,6 +32,8 @@ type PatchInfo struct {
 	Timestamp time.Time
 	// Type is the app state type being mutated.
 	Type WAPatchName
+	// Operation is SET / REMOVE
+	Operation waServerSync.SyncdMutation_SyncdOperation
 	// Mutations contains the individual mutations to apply to the app state in this patch.
 	Mutations []MutationInfo
 }
@@ -46,7 +48,8 @@ func BuildMute(target types.JID, mute bool, muteDuration time.Duration) PatchInf
 	}
 
 	return PatchInfo{
-		Type: WAPatchRegularHigh,
+		Type:      WAPatchRegularHigh,
+		Operation: waServerSync.SyncdMutation_SET,
 		Mutations: []MutationInfo{{
 			Index:   []string{IndexMute, target.String()},
 			Version: 2,
@@ -60,6 +63,27 @@ func BuildMute(target types.JID, mute bool, muteDuration time.Duration) PatchInf
 	}
 }
 
+func BuildContact(target types.JID, fullName string, add bool) PatchInfo {
+	operation := waServerSync.SyncdMutation_SET
+	if !add {
+		operation = waServerSync.SyncdMutation_REMOVE
+	}
+	return PatchInfo{
+		Type:      WAPatchCriticalUnblockLow,
+		Operation: operation,
+		Mutations: []MutationInfo{{
+			Index:   []string{IndexContact, target.String()},
+			Version: 2,
+			Value: &waSyncAction.SyncActionValue{
+				ContactAction: &waSyncAction.ContactAction{
+					FullName:                 &fullName,
+					SaveOnPrimaryAddressbook: proto.Bool(add),
+				},
+			},
+		}},
+	}
+}
+
 func newPinMutationInfo(target types.JID, pin bool) MutationInfo {
 	return MutationInfo{
 		Index:   []string{IndexPin, target.String()},
@@ -75,7 +99,8 @@ func newPinMutationInfo(target types.JID, pin bool) MutationInfo {
 // BuildPin builds an app state patch for pinning or unpinning a chat.
 func BuildPin(target types.JID, pin bool) PatchInfo {
 	return PatchInfo{
-		Type: WAPatchRegularLow,
+		Type:      WAPatchRegularLow,
+		Operation: waServerSync.SyncdMutation_SET,
 		Mutations: []MutationInfo{
 			newPinMutationInfo(target, pin),
 		},
@@ -119,6 +144,7 @@ func BuildArchive(target types.JID, archive bool, lastMessageTimestamp time.Time
 
 	result := PatchInfo{
 		Type:      WAPatchRegularLow,
+		Operation: waServerSync.SyncdMutation_SET,
 		Mutations: mutations,
 	}
 
@@ -140,7 +166,8 @@ func newLabelChatMutation(target types.JID, labelID string, labeled bool) Mutati
 // BuildLabelChat builds an app state patch for labeling or un(labeling) a chat.
 func BuildLabelChat(target types.JID, labelID string, labeled bool) PatchInfo {
 	return PatchInfo{
-		Type: WAPatchRegular,
+		Type:      WAPatchRegular,
+		Operation: waServerSync.SyncdMutation_SET,
 		Mutations: []MutationInfo{
 			newLabelChatMutation(target, labelID, labeled),
 		},
@@ -162,7 +189,8 @@ func newLabelMessageMutation(target types.JID, labelID, messageID string, labele
 // BuildLabelMessage builds an app state patch for labeling or un(labeling) a message.
 func BuildLabelMessage(target types.JID, labelID, messageID string, labeled bool) PatchInfo {
 	return PatchInfo{
-		Type: WAPatchRegular,
+		Type:      WAPatchRegular,
+		Operation: waServerSync.SyncdMutation_SET,
 		Mutations: []MutationInfo{
 			newLabelMessageMutation(target, labelID, messageID, labeled),
 		},
@@ -186,7 +214,8 @@ func newLabelEditMutation(labelID string, labelName string, labelColor int32, de
 // BuildLabelEdit builds an app state patch for editing a label.
 func BuildLabelEdit(labelID string, labelName string, labelColor int32, deleted bool) PatchInfo {
 	return PatchInfo{
-		Type: WAPatchRegular,
+		Type:      WAPatchRegular,
+		Operation: waServerSync.SyncdMutation_SET,
 		Mutations: []MutationInfo{
 			newLabelEditMutation(labelID, labelName, labelColor, deleted),
 		},
@@ -208,7 +237,8 @@ func newSettingPushNameMutation(pushName string) MutationInfo {
 // BuildSettingPushName builds an app state patch for setting the push name.
 func BuildSettingPushName(pushName string) PatchInfo {
 	return PatchInfo{
-		Type: WAPatchCriticalBlock,
+		Type:      WAPatchCriticalBlock,
+		Operation: waServerSync.SyncdMutation_SET,
 		Mutations: []MutationInfo{
 			newSettingPushNameMutation(pushName),
 		},
@@ -238,7 +268,8 @@ func BuildStar(target, sender types.JID, messageID types.MessageID, fromMe, star
 		senderJID = "0"
 	}
 	return PatchInfo{
-		Type: WAPatchRegularHigh,
+		Type:      WAPatchRegularHigh,
+		Operation: waServerSync.SyncdMutation_SET,
 		Mutations: []MutationInfo{
 			newStarMutation(targetJID, senderJID, messageID, isFromMe, starred),
 		},
@@ -281,11 +312,11 @@ func (proc *Processor) EncodePatch(keyID []byte, state HashState, patchInfo Patc
 			return nil, fmt.Errorf("failed to encrypt mutation: %w", err)
 		}
 
-		valueMac := generateContentMAC(waServerSync.SyncdMutation_SET, encryptedContent, keyID, keys.ValueMAC)
+		valueMac := generateContentMAC(patchInfo.Operation, encryptedContent, keyID, keys.ValueMAC)
 		indexMac := concatAndHMAC(sha256.New, keys.Index, indexBytes)
 
 		mutations = append(mutations, &waServerSync.SyncdMutation{
-			Operation: waServerSync.SyncdMutation_SET.Enum(),
+			Operation: patchInfo.Operation.Enum(),
 			Record: &waServerSync.SyncdRecord{
 				Index: &waServerSync.SyncdIndex{Blob: indexMac},
 				Value: &waServerSync.SyncdValue{Blob: append(encryptedContent, valueMac...)},