From f5756f1c597393f72594a82c51fe6879a93490e1 Mon Sep 17 00:00:00 2001
From: Neil Alexander <neilalexander@users.noreply.github.com>
Date: Fri, 30 Jul 2021 12:22:33 +0100
Subject: [PATCH] Test sig upload

---
 keyserver/internal/cross_signing.go | 69 ++++++++++++++++++++++++++++-
 1 file changed, 68 insertions(+), 1 deletion(-)

diff --git a/keyserver/internal/cross_signing.go b/keyserver/internal/cross_signing.go
index a99a1f6f..523480a0 100644
--- a/keyserver/internal/cross_signing.go
+++ b/keyserver/internal/cross_signing.go
@@ -218,13 +218,32 @@ func (a *KeyInternalAPI) PerformUploadDeviceSignatures(ctx context.Context, req
 }
 
 func (a *KeyInternalAPI) processSelfSignatures(
-	ctx context.Context, userID string,
+	ctx context.Context, _ string,
 	signatures map[string]map[gomatrixserverlib.KeyID]gomatrixserverlib.CrossSigningForKeyOrDevice,
 ) error {
 	// Here we will process:
 	// * The user signing their own devices using their self-signing key
 	// * The user signing their master key using one of their devices
 
+	for targetUserID, forTargetUserID := range signatures {
+		for targetKeyID, signature := range forTargetUserID {
+			switch sig := signature.CrossSigningBody.(type) {
+			case *gomatrixserverlib.CrossSigningForKey:
+				for originUserID, forOriginUserID := range sig.Signatures {
+					for originKeyID, originSig := range forOriginUserID {
+						if err := a.DB.StoreCrossSigningSigsForTarget(
+							ctx, originUserID, originKeyID, targetUserID, targetKeyID, originSig,
+						); err != nil {
+							return fmt.Errorf("a.DB.StoreCrossSigningKeysForTarget: %w", err)
+						}
+					}
+				}
+
+			case *gomatrixserverlib.CrossSigningForDevice:
+			}
+		}
+	}
+
 	return nil
 }
 
@@ -238,6 +257,54 @@ func (a *KeyInternalAPI) processOtherSignatures(
 	return nil
 }
 
+/*
+func (a *KeyInternalAPI) crossSigningKeysForUser(
+	ctx context.Context, userID string,
+) (api.CrossSigningKeyMap, error) {
+	keymap, err := a.DB.CrossSigningKeysForUser(ctx, userID)
+	if err != nil && err != sql.ErrNoRows {
+		return nil, fmt.Errorf("a.DB.CrossSigningKeysForUser: %w", err)
+	}
+
+	if len(keymap) > 0 {
+		return keymap, nil
+	}
+
+	req := &api.QueryKeysRequest{
+		UserToDevices: map[string][]string{
+			userID: {},
+		},
+	}
+	res := &api.QueryKeysResponse{}
+	a.QueryKeys(ctx, req, res)
+	if res.Error != nil {
+		return nil, fmt.Errorf("a.QueryKeys: %s", res.Error.Error())
+	}
+
+	result := api.CrossSigningKeyMap{}
+	if masterKey, ok := res.MasterKeys[userID]; ok {
+		if err := sanityCheckKey(masterKey, userID, gomatrixserverlib.CrossSigningKeyPurposeMaster); err != nil {
+			return nil, fmt.Errorf("sanityCheckKey: %w", err)
+		}
+		for _, key := range masterKey.Keys {
+			result[gomatrixserverlib.CrossSigningKeyPurposeMaster] = key
+			break
+		}
+	}
+	if selfSigningKey, ok := res.SelfSigningKeys[userID]; ok {
+		if err := sanityCheckKey(selfSigningKey, userID, gomatrixserverlib.CrossSigningKeyPurposeSelfSigning); err != nil {
+			return nil, fmt.Errorf("sanityCheckKey: %w", err)
+		}
+		for _, key := range selfSigningKey.Keys {
+			result[gomatrixserverlib.CrossSigningKeyPurposeSelfSigning] = key
+			break
+		}
+	}
+
+	return result, nil
+}
+*/
+
 func (a *KeyInternalAPI) crossSigningKeysFromDatabase(
 	ctx context.Context, req *api.QueryKeysRequest, res *api.QueryKeysResponse,
 ) {