/*
 * Copyright (C)  2021-2022 Kristóf Marussy <kristof@marussy.com>
 *
 * This file is part of Sophie.
 *
 * Sophie is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, version 3.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 *
 * SPDX-License-Identifier: AGPL-3.0-only
 */

import {
  applySnapshot,
  applyPatch,
  getEnv as getAnyEnv,
  IAnyStateTreeNode,
  Instance,
  types
} from 'mobx-state-tree';
import { sharedStore, SophieRenderer } from '@sophie/shared';

export interface RootEnv {
  ipc: SophieRenderer;
}

/**
 * Gets a well-typed environment from `model`.
 *
 * Only useable inside state trees created by `createAndConnectRootStore`.
 *
 * @param model The state tree node.
 */
export function getEnv(model: IAnyStateTreeNode): RootEnv {
  return getAnyEnv<RootEnv>(model);
}

export const rootStore = types.model('RootStore', {
  shared: sharedStore,
}).actions((self) => ({
  buttonClick() {
    getEnv(self).ipc.buttonClick();
  },
}));

export interface RootStore extends Instance<typeof rootStore> {}

/**
 * Creates a new `RootStore` with a new environment and connects it to `ipc`.
 *
 * Changes to the `shared` store in the main process will be propagated to
 * the newly created store via `ipc`.
 *
 * @param ipc The `sophieRenderer` context bridge.
 */
export function createAndConnectRootStore(ipc: SophieRenderer): RootStore {
  const store = rootStore.create({
    shared: {},
  }, {
    ipc,
  });

  ipc.setSharedStoreListener({
    onSnapshot(snapshot) {
      applySnapshot(store.shared, snapshot);
    },
    onPatch(patch) {
      applyPatch(store.shared, patch);
    },
  });

  return store;
}