@openzeppelin/hardhat-upgrades/dist/engine/force-import.js

[](https://docs.openzeppelin.com/upgrades-plugins/hardhat-upgrades) [](https://www.npmjs.org/package/@openzeppelin/h

import { Manifest, getImplementationAddressFromProxy, addProxyToManifest, isBeacon, getImplementationAddressFromBeacon, inferProxyKind, isBeaconProxy, hasCode, NoContractImportError, getAdminAddress, isEmptySlot, UpgradesError, } from '@openzeppelin/upgrades-core';
import { getDeployData, simulateDeployImpl } from './deploy-impl.js';
/**
 * Client-neutral orchestration of `forceImport`: records the implementation (and, for a proxy, the
 * proxy itself) in the manifest, and reports how the address was classified.
 */
export async function forceImport(binding, address, implInfo, opts) {
    const provider = binding.provider;
    const manifest = await Manifest.forNetwork(provider);
    const implAddress = await getImplementationAddressFromProxy(provider, address);
    if (implAddress !== undefined) {
        await importProxyToManifest(binding, provider, address, implAddress, implInfo, opts, manifest);
        return 'proxy';
    }
    else if (await isBeacon(provider, address)) {
        const beaconImplAddress = await getImplementationAddressFromBeacon(provider, address);
        await simulateDeployImpl(binding, implInfo, opts, beaconImplAddress);
        return 'beacon';
    }
    else {
        if (!(await hasCode(provider, address))) {
            throw new NoContractImportError(address);
        }
        await simulateDeployImpl(binding, implInfo, opts, address);
        return 'impl';
    }
}
async function importProxyToManifest(binding, provider, proxyAddress, implAddress, implInfo, opts, manifest) {
    await simulateDeployImpl(binding, implInfo, opts, implAddress);
    let importKind;
    if (opts.kind === undefined) {
        if (await isBeaconProxy(provider, proxyAddress)) {
            importKind = 'beacon';
        }
        else {
            const deployData = await getDeployData(binding, implInfo, opts);
            importKind = inferProxyKind(deployData.validations, deployData.version);
        }
    }
    else {
        importKind = opts.kind;
    }
    if (importKind === 'transparent') {
        await assertNonEmptyAdminSlot(provider, proxyAddress);
    }
    await addProxyToManifest(importKind, proxyAddress, manifest);
}
async function assertNonEmptyAdminSlot(provider, proxyAddress) {
    const adminAddress = await getAdminAddress(provider, proxyAddress);
    if (isEmptySlot(adminAddress)) {
        // Assert that the admin slot of a transparent proxy is not zero, otherwise the wrong kind may be imported.
        // Note: Transparent proxies should not have the zero address as the admin, according to TransparentUpgradeableProxy's _setAdmin function.
        throw new UpgradesError(`Proxy at ${proxyAddress} doesn't look like a transparent proxy`, () => `The proxy doesn't look like a transparent proxy because its admin address slot is empty. ` +
            `Set the \`kind\` option to the kind of proxy that was deployed at ${proxyAddress} (either 'uups' or 'beacon')`);
    }
}
//# sourceMappingURL=force-import.js.map