start-os/sdk/base/lib/interfaces/setupInterfaces.ts

import * as T from "../types"
import { AddressReceipt } from "./AddressReceipt"

declare const UpdateServiceInterfacesProof: unique symbol
export type UpdateServiceInterfacesReceipt = {
  [UpdateServiceInterfacesProof]: never
}

export type ServiceInterfacesReceipt = Array<T.AddressInfo[] & AddressReceipt>
export type SetServiceInterfaces<Output extends ServiceInterfacesReceipt> =
  (opts: { effects: T.Effects }) => Promise<Output>
export type UpdateServiceInterfaces<Output extends ServiceInterfacesReceipt> =
  (opts: {
    effects: T.Effects
  }) => Promise<Output & UpdateServiceInterfacesReceipt>
export type SetupServiceInterfaces = <Output extends ServiceInterfacesReceipt>(
  fn: SetServiceInterfaces<Output>,
) => UpdateServiceInterfaces<Output>
export const NO_INTERFACE_CHANGES = {} as UpdateServiceInterfacesReceipt
export const setupServiceInterfaces: SetupServiceInterfaces = <
  Output extends ServiceInterfacesReceipt,
>(
  fn: SetServiceInterfaces<Output>,
) =>
  ((options: { effects: T.Effects }) => {
    const updater = async (options: { effects: T.Effects }) => {
      const bindings: T.BindId[] = []
      const interfaces: T.ServiceInterfaceId[] = []
      const res = await fn({
        effects: {
          ...options.effects,
          bind: (params: T.BindParams) => {
            bindings.push({ id: params.id, internalPort: params.internalPort })
            return options.effects.bind(params)
          },
          exportServiceInterface: (params: T.ExportServiceInterfaceParams) => {
            interfaces.push(params.id)
            return options.effects.exportServiceInterface(params)
          },
        },
      })
      await options.effects.clearBindings({ except: bindings })
      await options.effects.clearServiceInterfaces({ except: interfaces })
      return res
    }
    const updaterCtx = { options }
    updaterCtx.options = {
      effects: {
        ...options.effects,
        constRetry: () => updater(updaterCtx.options),
      },
    }
    return updater(updaterCtx.options)
  }) as UpdateServiceInterfaces<Output>