refactor: new project struct (#8199)

packages/frontend/web -> packages/frontend/apps/web packages/frontend/mobile -> packages/frontend/apps/mobile packages/frontend/electron -> packages/frontend/apps/electron
2026-07-25 14:28:51 +08:00 · 2024-09-12 07:42:57 +00:00
parent 7c4eab6cd3
commit cc5a6e6d40
291 changed files with 139 additions and 134 deletions
@@ -0,0 +1,101 @@
+import path from 'node:path';
+import { setTimeout } from 'node:timers/promises';
+
+import { removeWithRetry } from '@affine-test/kit/utils/utils';
+import { v4 } from 'uuid';
+import { afterAll, afterEach, beforeEach, expect, test, vi } from 'vitest';
+
+const tmpDir = path.join(__dirname, 'tmp');
+const appDataPath = path.join(tmpDir, 'app-data');
+
+vi.doMock('@affine/electron/helper/main-rpc', () => ({
+  mainRPC: {
+    getPath: async () => appDataPath,
+  },
+}));
+
+const constructorStub = vi.fn();
+const destroyStub = vi.fn();
+destroyStub.mockReturnValue(Promise.resolve());
+
+function existProcess() {
+  process.emit('beforeExit', 0);
+}
+
+vi.doMock('@affine/electron/helper/db/secondary-db', () => {
+  return {
+    SecondaryWorkspaceSQLiteDB: class {
+      constructor(...args: any[]) {
+        constructorStub(...args);
+      }
+
+      connectIfNeeded = () => Promise.resolve();
+
+      pull = () => Promise.resolve();
+
+      destroy = destroyStub;
+    },
+  };
+});
+
+beforeEach(() => {
+  vi.useFakeTimers({ shouldAdvanceTime: true });
+});
+
+afterEach(async () => {
+  existProcess();
+  await removeWithRetry(tmpDir);
+  vi.useRealTimers();
+});
+
+afterAll(() => {
+  vi.doUnmock('@affine/electron/helper/main-rpc');
+});
+
+test('can get a valid WorkspaceSQLiteDB', async () => {
+  const { ensureSQLiteDB } = await import(
+    '@affine/electron/helper/db/ensure-db'
+  );
+  const workspaceId = v4();
+  const db0 = await ensureSQLiteDB('workspace', workspaceId);
+  expect(db0).toBeDefined();
+  expect(db0.workspaceId).toBe(workspaceId);
+
+  const db1 = await ensureSQLiteDB('workspace', v4());
+  expect(db1).not.toBe(db0);
+  expect(db1.workspaceId).not.toBe(db0.workspaceId);
+
+  // ensure that the db is cached
+  expect(await ensureSQLiteDB('workspace', workspaceId)).toBe(db0);
+});
+
+test('db should be destroyed when app quits', async () => {
+  const { ensureSQLiteDB } = await import(
+    '@affine/electron/helper/db/ensure-db'
+  );
+  const workspaceId = v4();
+  const db0 = await ensureSQLiteDB('workspace', workspaceId);
+  const db1 = await ensureSQLiteDB('workspace', v4());
+
+  expect(db0.adapter).not.toBeNull();
+  expect(db1.adapter).not.toBeNull();
+
+  existProcess();
+
+  // wait the async `db.destroy()` to be called
+  await setTimeout(100);
+
+  expect(db0.adapter.db).toBeNull();
+  expect(db1.adapter.db).toBeNull();
+});
+
+test('db should be removed in db$Map after destroyed', async () => {
+  const { ensureSQLiteDB, db$Map } = await import(
+    '@affine/electron/helper/db/ensure-db'
+  );
+  const workspaceId = v4();
+  const db = await ensureSQLiteDB('workspace', workspaceId);
+  await db.destroy();
+  await setTimeout(100);
+  expect(db$Map.has(`workspace:${workspaceId}`)).toBe(false);
+});
@@ -0,0 +1,56 @@
+import path from 'node:path';
+
+import { removeWithRetry } from '@affine-test/kit/utils/utils';
+import fs from 'fs-extra';
+import { v4 } from 'uuid';
+import { afterAll, afterEach, beforeAll, expect, test, vi } from 'vitest';
+
+const tmpDir = path.join(__dirname, 'tmp');
+const appDataPath = path.join(tmpDir, 'app-data');
+
+beforeAll(() => {
+  vi.doMock('@affine/electron/helper/main-rpc', () => ({
+    mainRPC: {
+      getPath: async () => appDataPath,
+    },
+  }));
+});
+
+afterEach(async () => {
+  await removeWithRetry(tmpDir);
+});
+
+afterAll(() => {
+  vi.doUnmock('@affine/electron/helper/main-rpc');
+});
+
+test('can create new db file if not exists', async () => {
+  const { openWorkspaceDatabase } = await import(
+    '@affine/electron/helper/db/workspace-db-adapter'
+  );
+  const workspaceId = v4();
+  const db = await openWorkspaceDatabase('workspace', workspaceId);
+  const dbPath = path.join(
+    appDataPath,
+    `workspaces/${workspaceId}`,
+    `storage.db`
+  );
+  expect(await fs.exists(dbPath)).toBe(true);
+  await db.destroy();
+});
+
+test('on destroy, check if resources have been released', async () => {
+  const { openWorkspaceDatabase } = await import(
+    '@affine/electron/helper/db/workspace-db-adapter'
+  );
+  const workspaceId = v4();
+  const db = await openWorkspaceDatabase('workspace', workspaceId);
+  const updateSub = {
+    complete: vi.fn(),
+    next: vi.fn(),
+  };
+  db.update$ = updateSub as any;
+  await db.destroy();
+  expect(db.adapter.db).toBe(null);
+  expect(updateSub.complete).toHaveBeenCalled();
+});