bridge/crypto: add ShareOneTimeKeys function to CryptoHelper

Signed-off-by: Sumner Evans <sumner@beeper.com>

bridge/bridge.go

@@ -7,6 +7,7 @@
 package bridge
 
 import (
+	"context"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -219,6 +220,7 @@ type Crypto interface {
 	Stop()
 	Reset(startAfterReset bool)
 	Client() *mautrix.Client
+	ShareOneTimeKeys(context.Context) error
 }
 
 func (br *Bridge) GenerateRegistration() {

bridge/crypto.go

@@ -436,6 +436,11 @@ func (helper *CryptoHelper) HandleMemberEvent(evt *event.Event) {
 	helper.mach.HandleMemberEvent(0, evt)
 }
 
+// ShareOneTimeKeys uploads the given number of one-time-keys to the server.
+func (helper *CryptoHelper) ShareOneTimeKeys(ctx context.Context) error {
+	return helper.mach.ShareKeys(ctx, -1)
+}
+
 type cryptoSyncer struct {
 	*crypto.OlmMachine
 }

crypto/machine.go

@@ -622,8 +622,8 @@ func (mach *OlmMachine) ShareKeys(ctx context.Context, currentOTKCount int) erro
 	start := time.Now()
 	mach.otkUploadLock.Lock()
 	defer mach.otkUploadLock.Unlock()
-	if mach.lastOTKUpload.Add(1 * time.Minute).After(start) {
-		log.Debug().Msg("Checking OTK count from server due to suspiciously close share keys requests")
+	if mach.lastOTKUpload.Add(1*time.Minute).After(start) || currentOTKCount < 0 {
+		log.Debug().Msg("Checking OTK count from server due to suspiciously close share keys requests or negative OTK count")
 		resp, err := mach.Client.UploadKeys(&mautrix.ReqUploadKeys{})
 		if err != nil {
 			return fmt.Errorf("failed to check current OTK counts: %w", err)