import { SourceUnit } from 'solidity-ast';
import { findAll } from 'solidity-ast/utils';

import { renameContract } from '../rename';
import { getNodeBounds } from '../solc/ast-utils';
import { Transformation } from './type';
import { TransformerTools } from '../transform';
import { matchBuffer } from '../utils/match';

export function* renameInheritdoc(
  sourceUnit: SourceUnit,
  { readOriginal, getData, resolver }: TransformerTools,
): Generator<Transformation> {
  for (const contract of findAll('ContractDefinition', sourceUnit)) {
    const baseContracts = contract.linearizedBaseContracts.map(i => resolver.resolveContract(i));

    for (const doc of findAll('StructuredDocumentation', contract)) {
      const bounds = getNodeBounds(doc);
      const re = /(@inheritdoc\s+)([a-zA-Z0-9$_]+)/;
      const match = matchBuffer(readOriginal(doc, 'buffer'), re);

      if (match) {
        yield {
          start: bounds.start + match.start + match.captureLengths[0],
          length: match.captureLengths[1],
          kind: 'rename-inheritdoc',
          transform: source => {
            const ref = baseContracts.find(c => c?.canonicalName == source);
            return ref !== undefined && getData(ref).importFromPeer !== undefined
              ? source
              : source.replace(/[a-zA-Z0-9$_]+$/, renameContract);
          },
        };
      }
    }
  }
}