cleaner invitation step messages

client/invitation/files/step2.go

@@ -12,8 +12,8 @@ import (
 )
 
 // Step2ReadAndAnswer reads an InvitationInitPayload from a .mwiv file, creates the
-// invitee's peer entry, and writes the encrypted ContactCard (PackedUserMessage) to a
-// .mwiv file for the initiator to pick up and process in step 3.
+// invitee's peer entry, and writes the serialized Invitation (step=2) to a .mwiv file
+// for the initiator to pick up and process in step 3.
 func Step2ReadAndAnswer(invitationFile string, nickname string, myNickname string, serverUids []string) error {
 	if _, err := os.Stat(invitationFile); os.IsNotExist(err) {
 		return err
@@ -29,35 +29,23 @@ func Step2ReadAndAnswer(invitationFile string, nickname string, myNickname strin
 	if err != nil {
 		return err
 	}
+	payloadBytes, err := proto.Marshal(payload)
+	if err != nil {
+		return err
+	}
 
 	mynick := myNickname
 	if mynick == "" {
 		mynick = client.GetConfig().GetIdentity().Nickname
 	}
 
-	packed, peer, err := messages.Step2InviteeCreatesInitiatorAndEncryptedContactCard(payload, nickname, mynick, serverUids)
-	if err != nil {
-		return err
-	}
-
-	// Wrap the PackedUserMessage in an Invitation so the initiator (step3) has the
-	// invitee's public key available for signature verification without an extra file.
-	packedBytes, err := proto.Marshal(packed)
-	if err != nil {
-		return err
-	}
-	invitation := &meowlib.Invitation{
-		Uuid:    peer.InvitationId,
-		Step:    2,
-		From:    peer.MyIdentity.Public,
-		Payload: packedBytes,
-	}
-	out, err := proto.Marshal(invitation)
+	// messages.Step2 returns a serialized Invitation ready to write directly to file.
+	invBytes, err := messages.Step2InviteeCreatesInitiatorAndEncryptedContactCard(payloadBytes, nickname, mynick, serverUids)
 	if err != nil {
 		return err
 	}
 
 	c := client.GetConfig()
 	filename := c.StoragePath + string(os.PathSeparator) + mynick + "-" + nickname + ".mwiv"
-	return os.WriteFile(filename, out, 0600)
+	return os.WriteFile(filename, invBytes, 0600)
 }