From 495644c8f0261926df336ff60899b9f973b30bef Mon Sep 17 00:00:00 2001
From: Alexander Onnikov <Alexander.Onnikov@xored.com>
Date: Wed, 4 Sep 2024 01:22:39 +0700
Subject: [PATCH] Sync files tool (#6478)

Signed-off-by: Alexander Onnikov <Alexander.Onnikov@xored.com>
---
 dev/tool/src/index.ts   | 40 +++++++++++++++++++++++++++++++++++++---
 dev/tool/src/storage.ts | 41 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 78 insertions(+), 3 deletions(-)

diff --git a/dev/tool/src/index.ts b/dev/tool/src/index.ts
index e8ec0c284f..3862426012 100644
--- a/dev/tool/src/index.ts
+++ b/dev/tool/src/index.ts
@@ -93,7 +93,7 @@ import { changeConfiguration } from './configuration'
 import { fixJsonMarkup, migrateMarkup } from './markup'
 import { fixMixinForeignAttributes, showMixinForeignAttributes } from './mixin'
 import { fixAccountEmails, renameAccount } from './renameAccount'
-import { moveFiles } from './storage'
+import { moveFiles, syncFiles } from './storage'
 
 const colorConstants = {
   colorRed: '\u001b[31m',
@@ -1020,9 +1020,43 @@ export function devTool (
                 continue
               }
 
-              console.log('start', workspace, index, '/', workspaces.length)
+              console.log('start', workspace.workspace, index, '/', workspaces.length)
               await moveFiles(toolCtx, getWorkspaceId(workspace.workspace), exAdapter, params)
-              console.log('done', workspace)
+              console.log('done', workspace.workspace)
+
+              index += 1
+            }
+          } catch (err: any) {
+            console.error(err)
+          }
+        })
+      })
+    })
+
+  program
+    .command('sync-files')
+    .option('-w, --workspace <workspace>', 'Selected workspace only', '')
+    .action(async (cmd: { workspace: string }) => {
+      const { mongodbUri } = prepareTools()
+      await withDatabase(mongodbUri, async (db) => {
+        await withStorage(mongodbUri, async (adapter) => {
+          try {
+            const exAdapter = adapter as StorageAdapterEx
+
+            console.log('syncing files from storage provider')
+
+            let index = 1
+            const workspaces = await listWorkspacesPure(db)
+            workspaces.sort((a, b) => b.lastVisit - a.lastVisit)
+
+            for (const workspace of workspaces) {
+              if (cmd.workspace !== '' && workspace.workspace !== cmd.workspace) {
+                continue
+              }
+
+              console.log('start', workspace.workspace, index, '/', workspaces.length)
+              await syncFiles(toolCtx, getWorkspaceId(workspace.workspace), exAdapter)
+              console.log('done', workspace.workspace)
 
               index += 1
             }
diff --git a/dev/tool/src/storage.ts b/dev/tool/src/storage.ts
index 8393d5feec..1dc0f55d96 100644
--- a/dev/tool/src/storage.ts
+++ b/dev/tool/src/storage.ts
@@ -23,6 +23,47 @@ export interface MoveFilesParams {
   move: boolean
 }
 
+export async function syncFiles (
+  ctx: MeasureContext,
+  workspaceId: WorkspaceId,
+  exAdapter: StorageAdapterEx
+): Promise<void> {
+  if (exAdapter.adapters === undefined) return
+
+  for (const [name, adapter] of exAdapter.adapters.entries()) {
+    await adapter.make(ctx, workspaceId)
+
+    await retryOnFailure(ctx, 5, async () => {
+      let time = Date.now()
+      let count = 0
+
+      const iterator = await adapter.listStream(ctx, workspaceId)
+      try {
+        while (true) {
+          const data = await iterator.next()
+          if (data === undefined) break
+
+          const blob = await exAdapter.stat(ctx, workspaceId, data._id)
+          if (blob !== undefined) continue
+
+          await exAdapter.syncBlobFromStorage(ctx, workspaceId, data._id, name)
+
+          count += 1
+          if (count % 100 === 0) {
+            const duration = Date.now() - time
+            time = Date.now()
+
+            console.log('...processed', count, Math.round(duration / 1000) + 's')
+          }
+        }
+        console.log('processed', count)
+      } finally {
+        await iterator.close()
+      }
+    })
+  }
+}
+
 export async function moveFiles (
   ctx: MeasureContext,
   workspaceId: WorkspaceId,