Interactions.s.sol

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.19;


import {Script,console} from  "forge-std/Script.sol";
import {HelperConfig} from "./HelperConfig.s.sol";
import {VRFCoordinatorV2Mock} from "@chainlink/contracts/src/v0.8/mocks/VRFCoordinatorV2Mock.sol";
import {LinkToken} from "../test/mocks/LinkToken.sol";
import {DevOpsTools} from "lib/foundry-devops/src/DevOpsTools.sol";

contract CreateSubscription is Script {
    function createSubscriptionUsingConfig() public returns (uint64){
        HelperConfig helperConfig = new HelperConfig();
        (, , address vrfCoordinator, , , , ) = helperConfig.activeNetworkConfig();
        return createSubscription(vrfCoordinator);
    }
    function createSubscription(address vrfCoordinator)
    public returns (uint64) {
        console.log("Creating subscription on chainId :",block.chainid);
        vm.startBroadcast();
        uint64 subId = VRFCoordinatorV2Mock(vrfCoordinator).createSubscription();
        vm.stopBroadcast();
        console.log("Your subscription Id ", subId);
        console.log("Please update subscriptionId in HelperConfig.s.sol");
        return subId;
    }
    function run() external returns (uint64) {
        return createSubscriptionUsingConfig();
    }
}

contract FundSubscription is Script {
    uint96 public constant FUND_AMOUNT = 3 ether;
    function funcdSubscriptionUsingConfig() public  {
        HelperConfig helperConfig = new HelperConfig();
        (, ,address vrfCoordinator, ,uint64 subId, ,address link ) = helperConfig.activeNetworkConfig();
        fundSubscription(vrfCoordinator,subId,link);
    }
    function fundSubscription(address vrfCoordinator,uint64 subId,address link) public{
        console.log("Funding subscription :",subId);
        console.log("using VrfCorrdinator :", vrfCoordinator);
        console.log("ON-cahinid:", block.chainid);
        if(block.chainid == 31337){
            vm.startBroadcast();
            VRFCoordinatorV2Mock(vrfCoordinator).fundSubscription(subId,FUND_AMOUNT);
            vm.stopBroadcast();
        }
        else{
            vm.startBroadcast();
            LinkToken(link).transferAndCall(vrfCoordinator,FUND_AMOUNT,abi.encode(subId));
            vm.stopBroadcast();
            
                    }
    }

    function run() external{
        funcdSubscriptionUsingConfig();
    }
}

contract AddConsumer is Script {
    function addConsumer(
        address lottery,
        address vrfCoordinator,
        uint64 subId
    ) public {
        console.log("Adding consumer contract: ", lottery);
        console.log("Using vrfCoordinator: ", vrfCoordinator);
        console.log("On ChainID: ", block.chainid);
        vm.startBroadcast();
        VRFCoordinatorV2Mock(vrfCoordinator).addConsumer(
            subId,
            lottery
        );
        vm.stopBroadcast();
    }

    function addConsumerUsingConfig(address lottery) public {
        HelperConfig helperConfig = new HelperConfig();
        (
            ,
            ,
            address vrfCoordinator
            ,
            ,
            uint64 subId
            ,
            ,

        ) = helperConfig.activeNetworkConfig();
        addConsumer(lottery, vrfCoordinator, subId);
    }

    function run() external {
        address lottery = DevOpsTools.get_most_recent_deployment(
            "Lottery",
            block.chainid
        );
        addConsumerUsingConfig(lottery);
    }
}