1 files changed, 102 insertions, 10 deletions

diff --git a/packages/main/src/stores/SharedStore.ts b/packages/main/src/stores/SharedStore.ts
index 76ce05c..861c8ce 100644
--- a/packages/main/src/stores/SharedStore.ts
+++ b/packages/main/src/stores/SharedStore.ts
@@ -19,19 +19,111 @@
  */
 
 import { sharedStore as originalSharedStore } from '@sophie/shared';
-import { Instance, types } from 'mobx-state-tree';
+import {
+  applySnapshot,
+  getSnapshot,
+  Instance,
+  resolveIdentifier,
+  types,
+} from 'mobx-state-tree';
 
-import { config } from './Config';
-import { runtimeService } from './RuntimeService';
+import SettingsWithId from '../utils/SettingsWithId';
+import { getLogger } from '../utils/log';
+import overrideProps from '../utils/overrideProps';
+
+import { globalSettings, GlobalSettingsSnapshotIn } from './GlobalSettings';
+import { addMissingProfileIds, profile, ProfileConfig } from './Profile';
+import {
+  addMissingServiceIdsAndProfiles,
+  service,
+  ServiceConfig,
+} from './Service';
+
+const log = getLogger('sharedStore');
+
+export interface Config extends GlobalSettingsSnapshotIn {
+  profiles?: ProfileConfig[] | undefined;
+
+  services?: ServiceConfig[] | undefined;
+}
+
+function getConfigs<T>(models: { config: T }[]): T[] | undefined {
+  return models.length === 0 ? undefined : models.map((model) => model.config);
+}
+
+function reconcileSettings<T>(
+  originalSnapshots: SettingsWithId<T>[],
+  settingsSnapshotsWithId: SettingsWithId<T>[],
+): SettingsWithId<T>[] {
+  const idToOriginalSnapshots = new Map(
+    originalSnapshots.map((originalSnapshot) => [
+      originalSnapshot.id,
+      originalSnapshot,
+    ]),
+  );
+  return settingsSnapshotsWithId.map(({ id, settings }) => ({
+    ...idToOriginalSnapshots.get(id),
+    id,
+    settings,
+  }));
+}
+
+export const sharedStore = overrideProps(originalSharedStore, {
+  settings: types.optional(globalSettings, {}),
+  profiles: types.array(profile),
+  services: types.array(service),
+  selectedService: types.safeReference(service),
+})
+  .views((self) => ({
+    get config(): Config {
+      const { settings, profiles, services } = self;
+      const globalSettingsConfig = getSnapshot(settings);
+      return {
+        ...globalSettingsConfig,
+        profiles: getConfigs(profiles),
+        services: getConfigs(services),
+      };
+    },
+  }))
+  .actions((self) => ({
+    loadConfig(config: Config): void {
+      const snapshot = getSnapshot(self);
+      const { profiles, services, ...settings } = config;
+      const profileSettingsSnapshots = addMissingProfileIds(profiles);
+      const serviceSettingsSnapshots = addMissingServiceIdsAndProfiles(
+        services,
+        profileSettingsSnapshots,
+      );
+      applySnapshot(self, {
+        ...snapshot,
+        settings,
+        profiles: reconcileSettings(
+          snapshot.profiles,
+          profileSettingsSnapshots,
+        ),
+        services: reconcileSettings(
+          snapshot.services,
+          serviceSettingsSnapshots,
+        ),
+      });
+    },
+    setShouldUseDarkColors(shouldUseDarkColors: boolean): void {
+      self.shouldUseDarkColors = shouldUseDarkColors;
+    },
+    setSelectedServiceId(serviceId: string): void {
+      const serviceInstance = resolveIdentifier(service, self, serviceId);
+      if (serviceInstance === undefined) {
+        log.warn('Trying to select unknown service', serviceId);
+        return;
+      }
+      self.selectedService = serviceInstance;
+      log.debug('Selected service', serviceId);
+    },
+  }));
+
+export interface SharedStore extends Instance<typeof sharedStore> {}
 
 export type {
   SharedStoreSnapshotIn,
   SharedStoreSnapshotOut,
 } from '@sophie/shared';
-
-export const sharedStore = originalSharedStore.props({
-  config: types.optional(config, {}),
-  runtimeServices: types.map(runtimeService),
-});
-
-export interface SharedStore extends Instance<typeof sharedStore> {}