start-os/sdk/base/lib/actions/setupActions.ts

import { InputSpec } from "./input/builder"
import { ExtractInputSpecType } from "./input/builder/inputSpec"
import * as T from "../types"
import { once } from "../util"

export type Run<
  A extends
    | Record<string, any>
    | InputSpec<Record<string, any>, any>
    | InputSpec<Record<string, never>, never>,
> = (options: {
  effects: T.Effects
  input: ExtractInputSpecType<A> & Record<string, any>
}) => Promise<T.ActionResult | null | void | undefined>
export type GetInput<
  A extends
    | Record<string, any>
    | InputSpec<Record<string, any>, any>
    | InputSpec<Record<string, any>, never>,
> = (options: {
  effects: T.Effects
}) => Promise<
  null | void | undefined | (ExtractInputSpecType<A> & Record<string, any>)
>

export type MaybeFn<T> = T | ((options: { effects: T.Effects }) => Promise<T>)
function callMaybeFn<T>(
  maybeFn: MaybeFn<T>,
  options: { effects: T.Effects },
): Promise<T> {
  if (maybeFn instanceof Function) {
    return maybeFn(options)
  } else {
    return Promise.resolve(maybeFn)
  }
}
function mapMaybeFn<T, U>(
  maybeFn: MaybeFn<T>,
  map: (value: T) => U,
): MaybeFn<U> {
  if (maybeFn instanceof Function) {
    return async (...args) => map(await maybeFn(...args))
  } else {
    return map(maybeFn)
  }
}

export class Action<
  Id extends T.ActionId,
  Store,
  InputSpecType extends
    | Record<string, any>
    | InputSpec<any, Store>
    | InputSpec<any, never>,
  Type extends
    ExtractInputSpecType<InputSpecType> = ExtractInputSpecType<InputSpecType>,
> {
  private constructor(
    readonly id: Id,
    private readonly metadataFn: MaybeFn<T.ActionMetadata>,
    private readonly inputSpec: InputSpecType,
    private readonly getInputFn: GetInput<Type>,
    private readonly runFn: Run<Type>,
  ) {}
  static withInput<
    Id extends T.ActionId,
    Store,
    InputSpecType extends
      | Record<string, any>
      | InputSpec<any, Store>
      | InputSpec<any, never>,
    Type extends
      ExtractInputSpecType<InputSpecType> = ExtractInputSpecType<InputSpecType>,
  >(
    id: Id,
    metadata: MaybeFn<Omit<T.ActionMetadata, "hasInput">>,
    inputSpec: InputSpecType,
    getInput: GetInput<Type>,
    run: Run<Type>,
  ): Action<Id, Store, InputSpecType, Type> {
    return new Action(
      id,
      mapMaybeFn(metadata, (m) => ({ ...m, hasInput: true })),
      inputSpec,
      getInput,
      run,
    )
  }
  static withoutInput<Id extends T.ActionId, Store>(
    id: Id,
    metadata: MaybeFn<Omit<T.ActionMetadata, "hasInput">>,
    run: Run<{}>,
  ): Action<Id, Store, {}, {}> {
    return new Action(
      id,
      mapMaybeFn(metadata, (m) => ({ ...m, hasInput: false })),
      {},
      async () => null,
      run,
    )
  }
  async exportMetadata(options: {
    effects: T.Effects
  }): Promise<T.ActionMetadata> {
    const metadata = await callMaybeFn(this.metadataFn, options)
    await options.effects.action.export({ id: this.id, metadata })
    return metadata
  }
  async getInput(options: { effects: T.Effects }): Promise<T.ActionInput> {
    return {
      spec: await this.inputSpec.build(options),
      value: (await this.getInputFn(options)) || null,
    }
  }
  async run(options: {
    effects: T.Effects
    input: Type
  }): Promise<T.ActionResult | null> {
    return (await this.runFn(options)) || null
  }
}

export class Actions<
  Store,
  AllActions extends Record<T.ActionId, Action<T.ActionId, Store, any, any>>,
> {
  private constructor(private readonly actions: AllActions) {}
  static of<Store>(): Actions<Store, {}> {
    return new Actions({})
  }
  addAction<A extends Action<T.ActionId, Store, any, any>>(
    action: A,
  ): Actions<Store, AllActions & { [id in A["id"]]: A }> {
    return new Actions({ ...this.actions, [action.id]: action })
  }
  async update(options: { effects: T.Effects }): Promise<null> {
    options.effects = {
      ...options.effects,
      constRetry: once(() => {
        this.update(options) // yes, this reuses the options object, but the const retry function will be overwritten each time, so the once-ness is not a problem
      }),
    }
    for (let action of Object.values(this.actions)) {
      await action.exportMetadata(options)
    }
    await options.effects.action.clear({ except: Object.keys(this.actions) })

    return null
  }
  get<Id extends T.ActionId>(actionId: Id): AllActions[Id] {
    return this.actions[actionId]
  }
}