1
0
mirror of https://github.com/bitwarden/server.git synced 2024-12-05 14:23:30 +01:00
bitwarden-server/util/Migrator/DbScripts/2024-09-05_00_SyncDuoVersionFourMetaDataToVersionTwo.sql
Ike 02fee8c1e9
[PM-8108] Add Duo SDK v4 metadata to Duo Two Factor Provider (#4774)
* Migrate Duo Two Factor Configuration to support both v2 and v4

* Postgres Migrations

* SQLite migrations

* comment updates for SQLite; Query changes for consistency;

* comment clean up; formatting
2024-09-23 15:51:04 -07:00

82 lines
2.4 KiB
Transact-SQL

-- Create temporary table to store User IDs
CREATE TABLE #TempUserIDs (
RowNum INT IDENTITY(1,1) PRIMARY KEY,
ID UNIQUEIDENTIFIER
);
-- Populate temporary table with User IDs
INSERT INTO #TempUserIDs (ID)
SELECT Id
FROM [dbo].[User]
WHERE TwoFactorProviders LIKE '%"2":%'
AND ISJSON(TwoFactorProviders) = 1;
DECLARE @UserBatchSize INT = 1000;
DECLARE @TotalUserRows INT = (SELECT COUNT(*) FROM #TempUserIDs);
DECLARE @UserBatchNum INT = 0;
WHILE @UserBatchNum * @UserBatchSize < @TotalUserRows
BEGIN
-- Update Users
UPDATE U
SET TwoFactorProviders = JSON_MODIFY(
JSON_MODIFY(
U.TwoFactorProviders,
'$."2".MetaData.ClientSecret',
JSON_VALUE(U.TwoFactorProviders, '$."2".MetaData.SKey')
),
'$."2".MetaData.ClientId',
JSON_VALUE(U.TwoFactorProviders, '$."2".MetaData.IKey')
)
FROM [dbo].[User] U
INNER JOIN #TempUserIDs T ON U.Id = T.ID
WHERE T.RowNum > @UserBatchNum * @UserBatchSize
AND T.RowNum <= (@UserBatchNum + 1) * @UserBatchSize;
SET @UserBatchNum = @UserBatchNum + 1;
END
-- Clean up
DROP TABLE #TempUserIDs;
-- Create temporary table to store Organization IDs
CREATE TABLE #TempOrganizationIDs (
RowNum INT IDENTITY(1,1) PRIMARY KEY,
ID UNIQUEIDENTIFIER
);
-- Populate temporary table with Organization IDs
INSERT INTO #TempOrganizationIDs (ID)
SELECT Id
FROM [dbo].[Organization]
WHERE TwoFactorProviders LIKE '%"6":%'
AND ISJSON(TwoFactorProviders) = 1;
DECLARE @OrganizationBatchSize INT = 1000;
DECLARE @TotalOrganizationRows INT = (SELECT COUNT(*) FROM #TempOrganizationIDs);
DECLARE @OrganizationBatchNum INT = 0;
WHILE @OrganizationBatchNum * @OrganizationBatchSize < @TotalOrganizationRows
BEGIN
-- Update Organizations
UPDATE Org
SET TwoFactorProviders = JSON_MODIFY(
JSON_MODIFY(
Org.TwoFactorProviders,
'$."6".MetaData.ClientSecret',
JSON_VALUE(Org.TwoFactorProviders, '$."6".MetaData.SKey')
),
'$."6".MetaData.ClientId',
JSON_VALUE(Org.TwoFactorProviders, '$."6".MetaData.IKey')
)
FROM [dbo].[Organization] Org
INNER JOIN #TempOrganizationIDs T ON Org.Id = T.ID
WHERE T.RowNum > @OrganizationBatchNum * @OrganizationBatchSize
AND T.RowNum <= (@OrganizationBatchNum + 1) * @OrganizationBatchSize;
SET @OrganizationBatchNum = @OrganizationBatchNum + 1;
END
-- Clean up
DROP TABLE #TempOrganizationIDs;