Function arrayifyFields

src/code-gen/codegen-helpers.ts:5–20

(
  beforeStr: string,
  arr: string[] | string,
  afterStr: string
)

import { toNatspec } from "./comments";

const arrayifyFields = (
  beforeStr: string,
  arr: string[] | string,
  afterStr: string
) => {
  if (Array.isArray(arr) && arr.length === 1) {
    arr = arr[0];
  }
  if (Array.isArray(arr)) {
    const withCommas = arr.map(
      (o, i) => `${o}${i === arr.length - 1 ? "" : ","}`
    );
    return [beforeStr, withCommas, afterStr];
  }
  return [[beforeStr, arr, afterStr].join("")];
};

export const buildAssemblyBlock = (lines: ArrayJoinInput<string>) => [
  `assembly {`,