feat(core): desktop multiple server support (#8979)

packages/backend/server/src/core/auth/controller.ts

@@ -35,6 +35,7 @@ import { TokenService, TokenType } from './token';
 interface PreflightResponse {
   registered: boolean;
   hasPassword: boolean;
+  magicLink: boolean;
 }
 
 interface SignInCredential {
@@ -82,16 +83,20 @@ export class AuthController {
       params.email
     );
 
+    const magicLinkAvailable = !!this.config.mailer.host;
+
     if (!user) {
       return {
         registered: false,
         hasPassword: false,
+        magicLink: magicLinkAvailable,
       };
     }
 
     return {
       registered: user.registered,
       hasPassword: !!user.password,
+      magicLink: magicLinkAvailable,
     };
   }
 

packages/backend/server/src/index.ts

@@ -8,7 +8,7 @@ import { createApp } from './app';
 import { URLHelper } from './fundamentals';
 
 const app = await createApp();
-const listeningHost = AFFiNE.deploy ? '0.0.0.0' : 'localhost';
+const listeningHost = '0.0.0.0';
 await app.listen(AFFiNE.server.port, listeningHost);
 const url = app.get(URLHelper);
 

packages/common/infra/src/framework/core/event.ts

@@ -33,31 +33,33 @@ export class EventBus {
     });
 
     for (const handler of handlers.values()) {
-      this.on(handler.event.id, handler.handler);
+      this.on(handler.event, handler.handler);
     }
   }
 
   get root(): EventBus {
     return this.parent?.root ?? this;
   }
 
-  on<T>(id: string, listener: (event: FrameworkEvent<T>) => void) {
-    if (!this.listeners[id]) {
-      this.listeners[id] = [];
+  on<T>(event: FrameworkEvent<T>, listener: (event: T) => void) {
+    if (!this.listeners[event.id]) {
+      this.listeners[event.id] = [];
     }
-    this.listeners[id].push(listener);
-    const off = this.parent?.on(id, listener);
+    this.listeners[event.id].push(listener);
+    const off = this.parent?.on(event, listener);
     return () => {
-      this.off(id, listener);
+      this.off(event, listener);
       off?.();
     };
   }
 
-  off<T>(id: string, listener: (event: FrameworkEvent<T>) => void) {
-    if (!this.listeners[id]) {
+  off<T>(event: FrameworkEvent<T>, listener: (event: T) => void) {
+    if (!this.listeners[event.id]) {
       return;
     }
-    this.listeners[id] = this.listeners[id].filter(l => l !== listener);
+    this.listeners[event.id] = this.listeners[event.id].filter(
+      l => l !== listener
+    );
   }
 
   emit<T>(event: FrameworkEvent<T>, payload: T) {

packages/common/infra/src/modules/feature-flag/constant.ts

@@ -216,6 +216,15 @@ export const AFFINE_FLAGS = {
     configurable: false,
     defaultState: isMobile,
   },
+  enable_multiple_cloud_servers: {
+    category: 'affine',
+    displayName:
+      'com.affine.settings.workspace.experimental-features.enable-multiple-cloud-servers.name',
+    description:
+      'com.affine.settings.workspace.experimental-features.enable-multiple-cloud-servers.description',
+    configurable: isDesktopEnvironment,
+    defaultState: false,
+  },
 } satisfies { [key in string]: FlagInfo };
 
 export type AFFINE_FLAGS = typeof AFFINE_FLAGS;

packages/common/infra/src/modules/global-context/entities/global-context.ts

@@ -8,6 +8,9 @@ export class GlobalContext extends Entity {
   memento = new MemoryMemento();
 
   workspaceId = this.define<string>('workspaceId');
+  workspaceFlavour = this.define<string>('workspaceFlavour');
+
+  serverId = this.define<string>('serverId');
 
   /**
    * is in doc page

packages/common/infra/src/modules/workspace/__tests__/workspace.spec.ts

@@ -1,4 +1,3 @@
-import { WorkspaceFlavour } from '@affine/env/workspace';
 import { describe, expect, test } from 'vitest';
 
 import { Framework } from '../../../framework';
@@ -22,7 +21,7 @@ describe('Workspace System', () => {
     expect(workspaceService.list.workspaces$.value.length).toBe(0);
 
     const workspace = workspaceService.open({
-      metadata: await workspaceService.create(WorkspaceFlavour.LOCAL),
+      metadata: await workspaceService.create('local'),
     });
 
     expect(workspace.workspace).toBeInstanceOf(Workspace);

packages/common/infra/src/modules/workspace/entities/list.ts

@@ -1,34 +1,47 @@
+import { combineLatest, map, of, switchMap } from 'rxjs';
+
 import { Entity } from '../../../framework';
 import { LiveData } from '../../../livedata';
-import type { WorkspaceFlavourProvider } from '../providers/flavour';
+import type { WorkspaceMetadata } from '../metadata';
+import type { WorkspaceFlavoursService } from '../services/flavours';
 
 export class WorkspaceList extends Entity {
-  workspaces$ = new LiveData(this.providers.map(p => p.workspaces$))
-    .map(v => {
-      return v;
-    })
-    .flat()
-    .map(workspaces => {
-      return workspaces.flat();
-    });
-  isRevalidating$ = new LiveData(
-    this.providers.map(p => p.isRevalidating$ ?? new LiveData(false))
-  )
-    .flat()
-    .map(isLoadings => isLoadings.some(isLoading => isLoading));
+  workspaces$ = LiveData.from<WorkspaceMetadata[]>(
+    this.flavoursService.flavours$.pipe(
+      switchMap(flavours =>
+        combineLatest(flavours.map(flavour => flavour.workspaces$)).pipe(
+          map(workspaces => workspaces.flat())
+        )
+      )
+    ),
+    []
+  );
+
+  isRevalidating$ = LiveData.from<boolean>(
+    this.flavoursService.flavours$.pipe(
+      switchMap(flavours =>
+        combineLatest(
+          flavours.map(flavour => flavour.isRevalidating$ ?? of(false))
+        ).pipe(map(isLoadings => isLoadings.some(isLoading => isLoading)))
+      )
+    ),
+    false
+  );
 
   workspace$(id: string) {
     return this.workspaces$.map(workspaces =>
       workspaces.find(workspace => workspace.id === id)
     );
   }
 
-  constructor(private readonly providers: WorkspaceFlavourProvider[]) {
+  constructor(private readonly flavoursService: WorkspaceFlavoursService) {
     super();
   }
 
   revalidate() {
-    this.providers.forEach(provider => provider.revalidate?.());
+    this.flavoursService.flavours$.value.forEach(provider => {
+      provider.revalidate?.();
+    });
   }
 
   waitForRevalidation(signal?: AbortSignal) {

packages/common/infra/src/modules/workspace/entities/profile.ts

@@ -12,6 +12,7 @@ import {
 } from '../../../livedata';
 import type { WorkspaceMetadata } from '../metadata';
 import type { WorkspaceFlavourProvider } from '../providers/flavour';
+import type { WorkspaceFlavoursService } from '../services/flavours';
 import type { WorkspaceProfileCacheStore } from '../stores/profile-cache';
 import type { Workspace } from './workspace';
 
@@ -47,12 +48,14 @@ export class WorkspaceProfile extends Entity<{ metadata: WorkspaceMetadata }> {
 
   constructor(
     private readonly cache: WorkspaceProfileCacheStore,
-    providers: WorkspaceFlavourProvider[]
+    flavoursService: WorkspaceFlavoursService
   ) {
     super();
 
     this.provider =
-      providers.find(p => p.flavour === this.props.metadata.flavour) ?? null;
+      flavoursService.flavours$.value.find(
+        p => p.flavour === this.props.metadata.flavour
+      ) ?? null;
   }
 
   private setProfile(info: WorkspaceProfileInfo) {

packages/common/infra/src/modules/workspace/index.ts

@@ -5,7 +5,8 @@ export { getAFFiNEWorkspaceSchema } from './global-schema';
 export type { WorkspaceMetadata } from './metadata';
 export type { WorkspaceOpenOptions } from './open-options';
 export type { WorkspaceEngineProvider } from './providers/flavour';
-export { WorkspaceFlavourProvider } from './providers/flavour';
+export type { WorkspaceFlavourProvider } from './providers/flavour';
+export { WorkspaceFlavoursProvider } from './providers/flavour';
 export { WorkspaceLocalCache, WorkspaceLocalState } from './providers/storage';
 export { WorkspaceScope } from './scopes/workspace';
 export { WorkspaceService } from './services/workspace';
@@ -21,48 +22,50 @@ import {
   WorkspaceLocalCacheImpl,
   WorkspaceLocalStateImpl,
 } from './impls/storage';
-import { WorkspaceFlavourProvider } from './providers/flavour';
+import { WorkspaceFlavoursProvider } from './providers/flavour';
 import { WorkspaceLocalCache, WorkspaceLocalState } from './providers/storage';
 import { WorkspaceScope } from './scopes/workspace';
 import { WorkspaceDestroyService } from './services/destroy';
 import { WorkspaceEngineService } from './services/engine';
 import { WorkspaceFactoryService } from './services/factory';
+import { WorkspaceFlavoursService } from './services/flavours';
 import { WorkspaceListService } from './services/list';
 import { WorkspaceProfileService } from './services/profile';
 import { WorkspaceRepositoryService } from './services/repo';
 import { WorkspaceTransformService } from './services/transform';
 import { WorkspaceService } from './services/workspace';
 import { WorkspacesService } from './services/workspaces';
 import { WorkspaceProfileCacheStore } from './stores/profile-cache';
-import { TestingWorkspaceLocalProvider } from './testing/testing-provider';
+import { TestingWorkspaceFlavoursProvider } from './testing/testing-provider';
 
 export function configureWorkspaceModule(framework: Framework) {
   framework
     .service(WorkspacesService, [
-      [WorkspaceFlavourProvider],
+      WorkspaceFlavoursService,
       WorkspaceListService,
       WorkspaceProfileService,
       WorkspaceTransformService,
       WorkspaceRepositoryService,
       WorkspaceFactoryService,
       WorkspaceDestroyService,
     ])
-    .service(WorkspaceDestroyService, [[WorkspaceFlavourProvider]])
+    .service(WorkspaceFlavoursService, [[WorkspaceFlavoursProvider]])
+    .service(WorkspaceDestroyService, [WorkspaceFlavoursService])
     .service(WorkspaceListService)
-    .entity(WorkspaceList, [[WorkspaceFlavourProvider]])
+    .entity(WorkspaceList, [WorkspaceFlavoursService])
     .service(WorkspaceProfileService)
     .store(WorkspaceProfileCacheStore, [GlobalCache])
     .entity(WorkspaceProfile, [
       WorkspaceProfileCacheStore,
-      [WorkspaceFlavourProvider],
+      WorkspaceFlavoursService,
     ])
-    .service(WorkspaceFactoryService, [[WorkspaceFlavourProvider]])
+    .service(WorkspaceFactoryService, [WorkspaceFlavoursService])
     .service(WorkspaceTransformService, [
       WorkspaceFactoryService,
       WorkspaceDestroyService,
     ])
     .service(WorkspaceRepositoryService, [
-      [WorkspaceFlavourProvider],
+      WorkspaceFlavoursService,
       WorkspaceProfileService,
     ])
     .scope(WorkspaceScope)
@@ -82,8 +85,8 @@ export function configureWorkspaceModule(framework: Framework) {
 
 export function configureTestingWorkspaceProvider(framework: Framework) {
   framework.impl(
-    WorkspaceFlavourProvider('LOCAL'),
-    TestingWorkspaceLocalProvider,
+    WorkspaceFlavoursProvider('LOCAL'),
+    TestingWorkspaceFlavoursProvider,
     [GlobalState]
   );
 }

packages/common/infra/src/modules/workspace/metadata.ts

@@ -1,7 +1,5 @@
-import type { WorkspaceFlavour } from '@affine/env/workspace';
-
 export type WorkspaceMetadata = {
   id: string;
-  flavour: WorkspaceFlavour;
+  flavour: string;
   initialized?: boolean;
 };

packages/common/infra/src/modules/workspace/providers/flavour.ts

@@ -1,4 +1,3 @@
-import type { WorkspaceFlavour } from '@affine/env/workspace';
 import type { DocCollection } from '@blocksuite/affine/store';
 
 import { createIdentifier } from '../../../framework';
@@ -22,7 +21,7 @@ export interface WorkspaceEngineProvider {
 }
 
 export interface WorkspaceFlavourProvider {
-  flavour: WorkspaceFlavour;
+  flavour: string;
 
   deleteWorkspace(id: string): Promise<void>;
 
@@ -60,5 +59,9 @@ export interface WorkspaceFlavourProvider {
   onWorkspaceInitialized?(workspace: Workspace): void;
 }
 
-export const WorkspaceFlavourProvider =
-  createIdentifier<WorkspaceFlavourProvider>('WorkspaceFlavourProvider');
+export interface WorkspaceFlavoursProvider {
+  workspaceFlavours$: LiveData<WorkspaceFlavourProvider[]>;
+}
+
+export const WorkspaceFlavoursProvider =
+  createIdentifier<WorkspaceFlavoursProvider>('WorkspaceFlavoursProvider');

packages/common/infra/src/modules/workspace/services/destroy.ts

@@ -1,14 +1,16 @@
 import { Service } from '../../../framework';
 import type { WorkspaceMetadata } from '../metadata';
-import type { WorkspaceFlavourProvider } from '../providers/flavour';
+import type { WorkspaceFlavoursService } from './flavours';
 
 export class WorkspaceDestroyService extends Service {
-  constructor(private readonly providers: WorkspaceFlavourProvider[]) {
+  constructor(private readonly flavoursService: WorkspaceFlavoursService) {
     super();
   }
 
   deleteWorkspace = async (metadata: WorkspaceMetadata) => {
-    const provider = this.providers.find(p => p.flavour === metadata.flavour);
+    const provider = this.flavoursService.flavours$.value.find(
+      p => p.flavour === metadata.flavour
+    );
     if (!provider) {
       throw new Error(`Unknown workspace flavour: ${metadata.flavour}`);
     }

packages/common/infra/src/modules/workspace/services/factory.ts

@@ -1,12 +1,11 @@
-import type { WorkspaceFlavour } from '@affine/env/workspace';
 import type { DocCollection } from '@blocksuite/affine/store';
 
 import { Service } from '../../../framework';
 import type { BlobStorage, DocStorage } from '../../../sync';
-import type { WorkspaceFlavourProvider } from '../providers/flavour';
+import type { WorkspaceFlavoursService } from './flavours';
 
 export class WorkspaceFactoryService extends Service {
-  constructor(private readonly providers: WorkspaceFlavourProvider[]) {
+  constructor(private readonly flavoursService: WorkspaceFlavoursService) {
     super();
   }
 
@@ -17,14 +16,16 @@ export class WorkspaceFactoryService extends Service {
    * @returns workspace id
    */
   create = async (
-    flavour: WorkspaceFlavour,
+    flavour: string,
     initial: (
       docCollection: DocCollection,
       blobStorage: BlobStorage,
       docStorage: DocStorage
     ) => Promise<void> = () => Promise.resolve()
   ) => {
-    const provider = this.providers.find(x => x.flavour === flavour);
+    const provider = this.flavoursService.flavours$.value.find(
+      x => x.flavour === flavour
+    );
     if (!provider) {
       throw new Error(`Unknown workspace flavour: ${flavour}`);
     }