@aerogel/plugin-local-first 0.0.0-next.4c77ee2164e8a4f72bdd920cb56ec059299ccc5f

package/src/lib/mirroring.ts

@@ -0,0 +1,327 @@
+import { bootModels } from 'soukai';
+import {
+    fail,
+    objectWithout,
+    requireUrlParentDirectory,
+    required,
+    tap,
+    toString,
+    urlResolveDirectory,
+    urlRoute,
+} from '@noeldemartin/utils';
+import { App } from '@aerogel/core';
+import { getTrackedModels, trackModelsCollection as trackSoukaiModelsCollection } from '@aerogel/plugin-soukai';
+import { Solid } from '@aerogel/plugin-solid';
+import { PropertyOperation, SolidContainer, SolidContainsRelation, Tombstone } from 'soukai-solid';
+import type { ObjectsMap } from '@noeldemartin/utils';
+import type { Attributes } from 'soukai';
+import type { SolidModel, SolidModelConstructor, SolidSchemaDefinition } from 'soukai-solid';
+
+import SyncQueue from '@aerogel/plugin-local-first/lib/SyncQueue';
+import { getContainerRelations, getRelatedAppModels } from '@aerogel/plugin-local-first/lib/inference';
+
+interface CloneOptions {
+    clean?: boolean;
+}
+
+const Cloud = required(() => App.service('$cloud'));
+const remoteClasses: WeakMap<SolidModelConstructor, SolidModelConstructor> = new WeakMap();
+const localClasses: WeakMap<SolidModelConstructor, SolidModelConstructor> = new WeakMap();
+
+function cleanRemoteModelClone(remoteModel: SolidModel): void {
+    const relatedModels = remoteModel.getRelatedModels().concat([remoteModel]);
+
+    for (const relatedModel of relatedModels) {
+        relatedModel.cleanDirty(true);
+    }
+}
+
+function fixRemoteModelClone(remoteModel: SolidModel): void {
+    const relatedModels = getRelatedAppModels(remoteModel).concat([remoteModel]);
+
+    for (const relatedModel of relatedModels) {
+        if (!relatedModel.operations || relatedModel.operations.length === 0) {
+            continue;
+        }
+
+        const operations = (relatedModel.operations ?? []).filter((operation) => {
+            return (
+                !(operation instanceof PropertyOperation) ||
+                !relatedModel.ignoreRdfPropertyHistory(operation.property, true)
+            );
+        });
+
+        if (!operations.some((operation) => !operation.isInception(relatedModel))) {
+            relatedModel.setRelationModels('operations', []);
+
+            continue;
+        }
+
+        relatedModel.setRelationModels('operations', operations);
+    }
+}
+
+export async function createRemoteModel(localModel: SolidModel): Promise<void> {
+    Cloud.setState({
+        localModelUpdates: {
+            ...Cloud.localModelUpdates,
+            [localModel.url]: 1,
+        },
+    });
+
+    if (isRegisteredModelClass(localModel.static())) {
+        await trackModelsCollection(localModel.static(), requireUrlParentDirectory(localModel.url));
+    }
+
+    if (Cloud.autoPush && Cloud.ready && !Cloud.syncing) {
+        SyncQueue.push(localModel);
+    }
+}
+
+export async function trackModelsCollection(modelClass: SolidModelConstructor, collection: string): Promise<void> {
+    if (Cloud.modelCollections[modelClass.modelName]?.includes(collection)) {
+        return;
+    }
+
+    Cloud.setState('modelCollections', {
+        ...Cloud.modelCollections,
+        [modelClass.modelName]: (Cloud.modelCollections[modelClass.modelName] ?? []).concat(collection),
+    });
+
+    await trackSoukaiModelsCollection(modelClass, collection);
+}
+
+export async function updateRemoteModel(localModel: SolidModel): Promise<void> {
+    if (ignoreModelUpdates(localModel)) {
+        return;
+    }
+
+    const modelUpdates = Cloud.localModelUpdates[localModel.url] ?? 0;
+
+    Cloud.setState({
+        localModelUpdates: localModel.isSoftDeleted()
+            ? objectWithout(Cloud.localModelUpdates, localModel.url)
+            : {
+                ...Cloud.localModelUpdates,
+                [localModel.url]: modelUpdates + 1,
+            },
+    });
+
+    if (Cloud.autoPush && Cloud.ready && !Cloud.syncing) {
+        SyncQueue.push(localModel);
+    }
+}
+
+export function clearLocalModelUpdates(models?: SolidModel[], malformedDocuments?: Set<string>): void {
+    const syncedModelUrls = models?.map((model) => toString(model.getDeletedPrimaryKey() ?? model.getPrimaryKey()));
+    const localModelUpdates = {} as Record<string, number>;
+
+    for (const [url, count] of Object.entries(Cloud.localModelUpdates)) {
+        const documentUrl = urlRoute(url);
+        const wasSynced =
+            !syncedModelUrls ||
+            syncedModelUrls.some(
+                (modelUrl) => url === modelUrl || (modelUrl.endsWith('/') && url.startsWith(modelUrl)),
+            );
+
+        if (wasSynced && !malformedDocuments?.has(documentUrl)) {
+            continue;
+        }
+
+        localModelUpdates[url] = count;
+    }
+
+    Cloud.setState({ localModelUpdates });
+}
+
+export function cloneLocalModel(localModel: SolidModel, options: CloneOptions = {}): SolidModel {
+    const localModelClasses = new Set<typeof SolidModel>();
+
+    localModelClasses.add(localModel.static());
+
+    if (localModel instanceof SolidContainer) {
+        for (const relation of getContainerRelations(localModel.static())) {
+            localModelClasses.add(localModel.requireRelation<SolidContainsRelation>(relation).relatedClass);
+        }
+    }
+
+    const constructors = [...localModelClasses.values()].map(
+        (localClass) => [localClass, getRemoteClass(localClass)] as [typeof SolidModel, typeof SolidModel],
+    );
+
+    return tap(localModel.clone({ constructors }), (model) => {
+        fixRemoteModelClone(model);
+
+        options.clean && cleanRemoteModelClone(model);
+    });
+}
+
+export function cloneRemoteModel(remoteModel: SolidModel): SolidModel {
+    const remoteModelClasses = new Set<typeof SolidModel>();
+
+    remoteModelClasses.add(remoteModel.static());
+
+    if (remoteModel instanceof SolidContainer) {
+        for (const relation of getContainerRelations(remoteModel.static())) {
+            remoteModelClasses.add(remoteModel.requireRelation<SolidContainsRelation>(relation).relatedClass);
+        }
+    }
+
+    const constructors = [...remoteModelClasses.values()].map(
+        (remoteClass) => [remoteClass, getLocalClass(remoteClass)] as [typeof SolidModel, typeof SolidModel],
+    );
+
+    return remoteModel.clone({ constructors });
+}
+
+export async function completeRemoteModels(
+    localModels: SolidModel[],
+    remoteModels: SolidModel[],
+    malformedDocuments: Set<string>,
+): Promise<SolidModel[]> {
+    const RemoteTombstone = getRemoteClass(Tombstone);
+    const remoteModelUrls = remoteModels.map((remoteModel) => remoteModel.url);
+    const missingModelDocumentUrls = localModels
+        .filter((localModel) => !remoteModelUrls.includes(localModel.url))
+        .map((localModel) => localModel.requireDocumentUrl())
+        .filter((documentUrl) => !malformedDocuments.has(documentUrl));
+    const tombstones = await RemoteTombstone.all({ $in: missingModelDocumentUrls });
+
+    return remoteModels.concat(tombstones);
+}
+
+export function ignoreModelUpdates(localModel: SolidModel): boolean {
+    const updatedAt = localModel.metadata.updatedAt?.getTime();
+
+    return !localModel.operations.some((operation) => {
+        if (operation.date.getTime() !== updatedAt) {
+            return false;
+        }
+
+        if (!(operation instanceof PropertyOperation)) {
+            return true;
+        }
+
+        return !localModel.ignoreRdfPropertyHistory(operation.property, true);
+    });
+}
+
+export function getLocalModel(remoteModel: SolidModel, localModels: ObjectsMap<SolidModel>): SolidModel {
+    return localModels.get(remoteModel.url) ?? cloneRemoteModel(remoteModel);
+}
+
+export function getLocalModels(): SolidModel[] {
+    return [...Cloud.registeredModels].map(({ modelClass }) => getTrackedModels(modelClass)).flat();
+}
+
+export function getSchemaMigrations(): Map<SolidModelConstructor, SolidModelConstructor | SolidSchemaDefinition> {
+    return Cloud.schemaMigrations;
+}
+
+export function clearSchemaMigrations(): void {
+    Cloud.schemaMigrations.clear();
+}
+
+export function getSchemaMigration(
+    modelClass: SolidModelConstructor,
+): SolidModelConstructor | SolidSchemaDefinition | undefined {
+    return Cloud.schemaMigrations.get(modelClass);
+}
+
+export function getRemoteClass<T extends SolidModelConstructor>(localClass: T): T {
+    return (remoteClasses.get(localClass) as T) ?? makeRemoteClass(localClass);
+}
+
+export function getRemoteContainersCollection(path?: string): string {
+    const rootStorage = Solid.requireUser().storageUrls[0];
+
+    return path
+        ? urlResolveDirectory((path.startsWith('/') ? rootStorage.slice(0, -1) : rootStorage) + path)
+        : rootStorage;
+}
+
+export function getRemoteModel(localModel: SolidModel, remoteModels: ObjectsMap<SolidModel>): SolidModel {
+    return remoteModels.get(localModel.url) ?? cloneLocalModel(localModel);
+}
+
+export function isRegisteredModel(model: SolidModel): boolean {
+    for (const { modelClass } of Cloud.registeredModels) {
+        const models = getTrackedModels(modelClass);
+
+        if (models.some((registeredModel) => registeredModel.url === model.url)) {
+            return true;
+        }
+    }
+
+    return false;
+}
+
+export function isRemoteModel(model: SolidModel): boolean {
+    return localClasses.has(model.static());
+}
+
+export function isLocalModel(model: SolidModel): boolean {
+    return !isRemoteModel(model);
+}
+
+export function isRegisteredModelClass(modelClass: SolidModelConstructor): boolean {
+    return Cloud.registeredModels.some(({ modelClass: registeredModelClass }) => registeredModelClass === modelClass);
+}
+
+export function makeRemoteClass<T extends SolidModelConstructor>(localClass: T): T {
+    const LocalClass = localClass as typeof SolidModel;
+    const withLocalHistoryConfig = (modelClass: typeof SolidModel, operation: () => unknown) => {
+        modelClass.timestamps = LocalClass.timestamps;
+        modelClass.history = LocalClass.history;
+
+        operation();
+
+        modelClass.timestamps = [];
+        modelClass.history = false;
+    };
+    const RemoteClass = class extends LocalClass {
+
+        public static override timestamps = false;
+        public static override history = false;
+
+        protected override initialize(attributes: Attributes, exists: boolean): void {
+            withLocalHistoryConfig(this._proxy.static(), () => super.initialize(attributes, exists));
+        }
+
+        protected override initializeRelations(): void {
+            super.initializeRelations();
+
+            for (const relation of Object.values(this._relations)) {
+                if (!(relation instanceof SolidContainsRelation)) {
+                    continue;
+                }
+
+                relation.relatedClass = getRemoteClass(relation.relatedClass);
+            }
+        }
+
+        public override fixMalformedAttributes(): void {
+            withLocalHistoryConfig(this.static(), () => super.fixMalformedAttributes());
+        }
+    
+    } as unknown as T;
+
+    remoteClasses.set(LocalClass, RemoteClass);
+    localClasses.set(RemoteClass, LocalClass);
+
+    LocalClass.on('schema-updated', async (schema) => {
+        await RemoteClass.updateSchema({
+            ...schema,
+            timestamps: false,
+            history: false,
+        });
+    });
+
+    bootModels({ [`Remote${localClass.modelName}`]: RemoteClass });
+
+    return RemoteClass;
+}
+
+export function getLocalClass<T extends SolidModelConstructor>(remoteClass: T): T {
+    return (localClasses.get(remoteClass) as T) ?? fail(`Couldn't find local class for ${remoteClass.modelName}`);
+}

package/src/lib/utils.ts

@@ -0,0 +1,21 @@
+export function mapFromArray<K, V>(items: [K, V][]): Map<K, V> {
+    const itemsMap = new Map();
+
+    for (const [key, value] of items) {
+        itemsMap.set(key, value);
+    }
+
+    return itemsMap;
+}
+
+export function lazy<T>(getter: () => Promise<T>): () => Promise<T> {
+    const result: { value?: T } = {};
+
+    return async () => {
+        if (!('value' in result)) {
+            result.value = await getter();
+        }
+
+        return result.value as T;
+    };
+}

package/src/services/Cloud.state.ts

@@ -0,0 +1,111 @@
+import { defineServiceState, replaceExisting, translateWithDefault } from '@aerogel/core';
+import { Solid } from '@aerogel/plugin-solid';
+import type { ErrorSource } from '@aerogel/core';
+import type { Nullable } from '@noeldemartin/utils';
+import type { SolidModelConstructor, SolidSchemaDefinition } from 'soukai-solid';
+
+import Migrate from '@aerogel/plugin-local-first/jobs/Migrate';
+import type Sync from '@aerogel/plugin-local-first/jobs/Sync';
+
+export const CloudStatus = {
+    Disconnected: 'disconnected',
+    Online: 'online',
+    Syncing: 'syncing',
+    Migrating: 'migrating',
+} as const;
+
+export interface ModelRegistration {
+    modelClass: SolidModelConstructor;
+    path?: string;
+}
+
+export type TCloudStatus = (typeof CloudStatus)[keyof typeof CloudStatus];
+
+export default defineServiceState({
+    name: 'cloud',
+    persist: [
+        'localModelUpdates',
+        'migrationDismissed',
+        'migrationJob',
+        'modelCollections',
+        'pollingEnabled',
+        'pollingMinutes',
+        'ready',
+        'startupSync',
+    ],
+    initialState: () => ({
+        autoPush: true,
+        localModelUpdates: {} as Record<string, number>,
+        migrationDismissed: false,
+        migrationJob: null as Nullable<Migrate>,
+        migrationPostponed: false,
+        modelCollections: {} as Record<string, string[]>,
+        pollingEnabled: true,
+        pollingMinutes: 10,
+        ready: false,
+        registeredModels: [] as ModelRegistration[],
+        schemaMigrations: new Map<SolidModelConstructor, SolidModelConstructor | SolidSchemaDefinition>(),
+        setupDismissed: false,
+        setupOngoing: false,
+        startupSync: true,
+        status: CloudStatus.Disconnected as TCloudStatus,
+        syncError: null as Nullable<ErrorSource>,
+        syncJob: null as Nullable<Sync>,
+    }),
+    serialize: (state) => replaceExisting(state, { migrationJob: state.migrationJob?.serialize() }),
+    restore: (state) =>
+        replaceExisting(state, {
+            migrationJob: state.migrationJob && Migrate.restore(state.migrationJob),
+        }),
+    watch: {
+        migrationJob(migrationJob) {
+            migrationJob?.listeners.add({ onUpdated: () => this.updatePersistedState('migrationJob') });
+        },
+    },
+    computed: {
+        disconnected: ({ status }) => status === CloudStatus.Disconnected,
+        online: ({ status }) => status === CloudStatus.Online,
+        syncing: ({ status }) => status === CloudStatus.Syncing,
+        migrating: ({ status }) => status === CloudStatus.Migrating,
+        localChanges({ localModelUpdates }) {
+            return Object.values(localModelUpdates).reduce((total, count) => total + count, 0);
+        },
+        dirty(): boolean {
+            return this.localChanges > 0;
+        },
+        migrationStarted({ migrationJob }): boolean {
+            return !!migrationJob;
+        },
+        migrationDisabledReason({ status, ready }): string | null {
+            if (!Solid.isLoggedIn()) {
+                return translateWithDefault(
+                    'cloud.migrationNotLoggedIn',
+                    'You need to reconnect before starting the migration',
+                );
+            }
+
+            if (!ready) {
+                return translateWithDefault(
+                    'cloud.migrationNotReady',
+                    'You need to backup your data before starting the migration',
+                );
+            }
+
+            if (status === CloudStatus.Syncing || status === CloudStatus.Migrating) {
+                return translateWithDefault(
+                    'cloud.migrationNotSyncing',
+                    'Wait for syncing to finish before starting the migration',
+                );
+            }
+
+            if (!this.online) {
+                return translateWithDefault(
+                    'cloud.migrationNotOnline',
+                    'You need to be online in order to start the migration',
+                );
+            }
+
+            return null;
+        },
+    },
+});