lens/src/main/channel/message-to-channel.injectable.ts

/**
 * Copyright (c) OpenLens Authors. All rights reserved.
 * Licensed under MIT License. See LICENSE in root directory for more information.
 */
import { lensWindowInjectionToken } from "../start-main-application/lens-window/application-window/lens-window-injection-token";
import { pipeline } from "@ogre-tools/fp";
import { getInjectable } from "@ogre-tools/injectable";
import { filter } from "lodash/fp";
import { messageToChannelInjectionToken } from "../../common/channel/message-to-channel-injection-token";
import type { MessageChannel } from "../../common/channel/message-channel-injection-token";

const messageToChannelInjectable = getInjectable({
  id: "message-to-channel",

  instantiate: (di) => {
    const getAllLensWindows = () => di.injectMany(lensWindowInjectionToken);

    // TODO: Figure out way to improve typing in internals
    // Notice that this should be injected using "messageToChannelInjectionToken" which is typed correctly.
    return (channel: MessageChannel<unknown>, message?: unknown) => {
      const visibleWindows = pipeline(
        getAllLensWindows(),
        filter((lensWindow) => !!lensWindow.visible),
      );

      visibleWindows.forEach((lensWindow) =>
        lensWindow.send({ channel: channel.id, data: message ? [message] : [] }),
      );
    };
  },

  injectionToken: messageToChannelInjectionToken,
});

export default messageToChannelInjectable;