fluence-js/packages/core/aqua-to-js/src/generate/service.ts

/**
 * Copyright 2023 Fluence Labs Limited
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { ServiceDef } from "@fluencelabs/interfaces";

import { recursiveRenameLaquaProps } from "../utils.js";

import { TypeGenerator } from "./interfaces.js";

// Actual value of defaultServiceId which comes from aqua-api
export interface DefaultServiceId {
  s_Some__f_value?: string;
}

export function generateServices(
  typeGenerator: TypeGenerator,
  services: Record<string, ServiceDef>,
) {
  const generated = Object.entries(services)
    .map(([srvName, srvDef]) => {
      return generateService(typeGenerator, srvName, srvDef);
    })
    .join("\n\n");

  return generated + "\n";
}

function generateService(
  typeGenerator: TypeGenerator,
  srvName: string,
  srvDef: ServiceDef,
) {
  return [
    typeGenerator.serviceType(srvName, srvDef),
    generateRegisterServiceOverload(typeGenerator, srvName, srvDef),
  ].join("\n");
}

function generateRegisterServiceOverload(
  typeGenerator: TypeGenerator,
  srvName: string,
  srvDef: ServiceDef,
) {
  return [
    `export function register${srvName}(${typeGenerator.type(
      "...args",
      "any[]",
    )}) {`,
    "    registerService$$(",
    "        args,",
    `        ${serviceToJson(srvDef)}`,
    "    );",
    "}",
  ].join("\n");
}

function serviceToJson(service: ServiceDef): string {
  return JSON.stringify(
    {
      // This assertion is required because aqua-api gives bad types
      // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
      ...((service.defaultServiceId as DefaultServiceId).s_Some__f_value != null
        ? {
            defaultServiceId:
              // This assertion is required because aqua-api gives bad types
              // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
              (service.defaultServiceId as DefaultServiceId).s_Some__f_value,
          }
        : {}),
      functions: recursiveRenameLaquaProps(service.functions),
    },
    null,
    4,
  );
}