/*
 * 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 { ipcRenderer } from 'electron';
import type { IJsonPatch } from 'mobx-state-tree';
import {
  MainToRendererIpcMessage,
  RendererToMainIpcMessage,
  sharedStore,
  SharedStoreListener,
  SharedStoreSnapshotIn,
  SophieRenderer
} from '@sophie/shared';

export type MessageSender = (channel: RendererToMainIpcMessage, ...args: unknown[]) => void;

export class SophieRendererImpl implements SophieRenderer {
  readonly #send: MessageSender;

  #listener: SharedStoreListener | null = null;

  #snapshot: SharedStoreSnapshotIn | null = null;

  constructor(send: MessageSender) {
    this.#send = send;
  }

  sharedStoreSnapshotReceived(snapshot: unknown): void {
    if (sharedStore.is(snapshot)) {
      if (this.#listener === null) {
        this.#snapshot = snapshot;
      } else {
        this.#listener.onSnapshot(snapshot);
      }
    } else {
      console.error('Received invalid snapshot', snapshot);
      this.#snapshot = null;
    }
  }

  sharedStorePatchReceived(patch: unknown): void {
    if (this.#listener !== null) {
      // `mobx-state-tree` will validate the patch, so we can safely cast here.
      this.#listener.onPatch(patch as IJsonPatch);
    }
  }

  setSharedStoreListener(listener: SharedStoreListener): void {
    this.#listener = listener;
    if (this.#snapshot !== null) {
      listener.onSnapshot(this.#snapshot);
      this.#snapshot = null;
    }
    this.#send(RendererToMainIpcMessage.SharedStoreSnapshotRequest);
  }

  buttonClick(): void {
    this.#send(RendererToMainIpcMessage.ButtonClick);
  }
}

export function createSophieRenderer(): SophieRenderer {
  const impl = new SophieRendererImpl(ipcRenderer.send);

  ipcRenderer.on(MainToRendererIpcMessage.SharedStoreSnapshot, (_event, snapshot) => {
    impl.sharedStoreSnapshotReceived(snapshot);
  });

  ipcRenderer.on(MainToRendererIpcMessage.SharedStorePatch, (_event, patch) => {
    impl.sharedStorePatchReceived(patch);
  });

  return {
    setSharedStoreListener: impl.setSharedStoreListener.bind(impl),
    buttonClick: impl.buttonClick.bind(impl),
  };
}