@solvprotocol/upgrade-safe-transpiler/src/generate-with-init.ts

Solidity preprocessor used to generate OpenZeppelin Contracts Upgrade Safe.

import path from 'path';
import { satisfies } from 'compare-versions';
import { Transform } from './transform';
import { formatLines, Line } from './transformations/utils/format-lines';
import { findAll } from 'solidity-ast/utils';
import { getConstructor } from './solc/ast-utils';
import { renameContract, renamePath } from './rename';
import { relativePath } from './utils/relative-path';
import { hasConstructorOverride } from './utils/upgrades-overrides';

export function generateWithInit(
  transform: Transform,
  destPath: string,
  solcVersion = '0.6.0',
): string {
  const pragmaVersion = satisfies(solcVersion, '>=0.7') ? '0.7' : '0.6';
  const res: Line[] = [
    `pragma solidity >=${pragmaVersion} <0.9;`,
    `pragma experimental ABIEncoderV2;`,
    ``,
  ];

  for (const sourceUnit of transform.asts()) {
    for (const contract of findAll('ContractDefinition', sourceUnit)) {
      if (contract.contractKind !== 'contract' || contract.abstract) {
        continue;
      }

      const constructorNode = getConstructor(contract);

      let argNames = '';
      if (constructorNode) {
        argNames = constructorNode.parameters.parameters.map(p => p.name).join(', ');
      }

      let argsList = '';

      if (constructorNode) {
        const source = transform.read(constructorNode);
        const argsMatch = source.match(/\((.*?)\)/s);
        if (argsMatch === null) {
          throw new Error(`Could not find constructor arguments for ${contract.name}`);
        }
        [, argsList] = argsMatch;
      }

      const renamedContract = renameContract(contract.name);

      let parents: string[] = [];
      let statements: string[] = [];
      if (hasConstructorOverride(contract)) {
        parents = [`${renameContract(contract.name)}(${argNames})`];
      } else {
        statements = [`__${contract.name}_init(${argNames});`];
      }

      res.push(
        `import "${relativePath(path.dirname(destPath), renamePath(sourceUnit.absolutePath))}";`,
        ``,
        `contract ${renamedContract}WithInit is ${renamedContract} {`,
        [
          [
            `constructor(${argsList})`,
            ...parents,
            ...(satisfies(pragmaVersion, '>=0.7') ? [] : [`public`]),
            `payable initializer {`,
          ].join(' '),
          statements,
          `}`,
        ],
        `}`,
      );
    }
  }

  return formatLines(0, res);
}