fixup! add queries to support rotating dbcrypt keys

johnstcn · johnstcn · commit 0c01b36523d1 · 2023-08-28T19:56:53.000Z
diff --git a/enterprise/dbcrypt/dbcrypt.go b/enterprise/dbcrypt/dbcrypt.go
@@ -27,6 +27,7 @@ import (
 	"database/sql"
 	"encoding/base64"
 	"errors"
+	"github.com/google/uuid"
 	"strings"
 	"sync/atomic"
 
@@ -142,6 +143,19 @@ func (db *dbCrypt) GetUserLinkByLinkedID(ctx context.Context, linkedID string) (
 	return link, db.decryptFields(&link.OAuthAccessToken, &link.OAuthRefreshToken)
 }
 
+func (db *dbCrypt) GetUserLinksByUserID(ctx context.Context, userID uuid.UUID) ([]database.UserLink, error) {
+	links, err := db.Store.GetUserLinksByUserID(ctx, userID)
+	if err != nil {
+		return nil, err
+	}
+	for _, link := range links {
+		if err := db.decryptFields(&link.OAuthAccessToken, &link.OAuthRefreshToken); err != nil {
+			return nil, err
+		}
+	}
+	return links, nil
+}
+
 func (db *dbCrypt) GetUserLinkByUserIDLoginType(ctx context.Context, params database.GetUserLinkByUserIDLoginTypeParams) (database.UserLink, error) {
 	link, err := db.Store.GetUserLinkByUserIDLoginType(ctx, params)
 	if err != nil {
@@ -182,6 +196,19 @@ func (db *dbCrypt) GetGitAuthLink(ctx context.Context, params database.GetGitAut
 	return link, db.decryptFields(&link.OAuthAccessToken, &link.OAuthRefreshToken)
 }
 
+func (db *dbCrypt) GetGitAuthLinksByUserID(ctx context.Context, userID uuid.UUID) ([]database.GitAuthLink, error) {
+	links, err := db.Store.GetGitAuthLinksByUserID(ctx, userID)
+	if err != nil {
+		return nil, err
+	}
+	for _, link := range links {
+		if err := db.decryptFields(&link.OAuthAccessToken, &link.OAuthRefreshToken); err != nil {
+			return nil, err
+		}
+	}
+	return links, nil
+}
+
 func (db *dbCrypt) UpdateGitAuthLink(ctx context.Context, params database.UpdateGitAuthLinkParams) (database.GitAuthLink, error) {
 	err := db.encryptFields(&params.OAuthAccessToken, &params.OAuthRefreshToken)
 	if err != nil {
