Skip to content

chore: lint files and change to consts #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
bitcoinbrisbane wants to merge 1 commit into
base: main
Choose a base branch
from
+741 −654
Open

chore: lint files and change to consts #1

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
160 changes: 88 additions & 72 deletions functions/AppendWhitelist.js
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,95 +1,111 @@
import {MerkleTree} from "merkletreejs";
import { MerkleTree } from "merkletreejs";
import keccak256 from "keccak256";
import { deploy_nft_abi } from "../abi.js";
import { initializeApp } from "firebase/app";
import { ethers } from "ethers";
import { getFirestore, collection, query, getDocs, where, setDoc, doc } from 'firebase/firestore/lite';
import {
getFirestore,
collection,
query,
getDocs,
where,
setDoc,
doc,
} from "firebase/firestore/lite";
import { GetProvider } from "./GetProvider.js";

const firebaseConfig = {
apiKey: process.env.fb_key,
authDomain: process.env.authDomain,
projectId: process.env.projectId,
storageBucket: process.env.storageBucket,
messagingSenderId: process.env.messagingSenderId,
appId: process.env.appId,
measurementId: process.env.measurementId
apiKey: process.env.fb_key,
authDomain: process.env.authDomain,
projectId: process.env.projectId,
storageBucket: process.env.storageBucket,
messagingSenderId: process.env.messagingSenderId,
appId: process.env.appId,
measurementId: process.env.measurementId,
};

const fb = initializeApp(firebaseConfig);
const db = getFirestore(fb);

export const AppendWhitelist = async (req) => {
const contract = req.query.contract;
const addresses = req.query.wallets.split(',');
const network = req.query.network;
const sender = req.query.wallet;
const contract = req.query.contract;
const addresses = req.query.wallets.split(",");
const network = req.query.network;
const sender = req.query.wallet;

const provider = GetProvider(network);
const provider = GetProvider(network);

const code = network === 'goerli' ? await provider.getCode(contract) : network === 'mainnet' ? await provider.getCode(contract) : 'invalid';
const code =
network === "goerli"
? await provider.getCode(contract)
: network === "mainnet"
? await provider.getCode(contract)
: "invalid";

if(code === 'invalid') {
throw('Network was not specified');
}
if(code === '0x') {
throw('Address entered is not a contract address');
}
if (code === "invalid") {
throw "Network was not specified";
}
if (code === "0x") {
throw "Address entered is not a contract address";
}

const NFTContract = new ethers.Contract(contract, deploy_nft_abi, provider);
const isAdmin = await NFTContract.isAdmin(process.env.wallet_address);
const owner = await NFTContract.owner();
console.log(owner);
console.log(sender);
console.log(isAdmin);
const NFTContract = new ethers.Contract(contract, deploy_nft_abi, provider);
const isAdmin = await NFTContract.isAdmin(process.env.wallet_address);
const owner = await NFTContract.owner();
console.log(owner);
console.log(sender);
console.log(isAdmin);

if(!isAdmin) {
throw('We are not authorized to update this contract');
}
if(owner.toLowerCase() !== sender.toLowerCase()) {
throw('Sender is not the contract owner');
}
if (!isAdmin) {
throw "We are not authorized to update this contract";
}
if (owner.toLowerCase() !== sender.toLowerCase()) {
throw "Sender is not the contract owner";
}

const wlRef = collection(db, 'whitelists');
const q = query(collection(db, "whitelists"), where("contract", "==", contract));
const wlSnapshot = await getDocs(q);

let whitelist;
if(wlSnapshot.docs.length === 0) {
//wlRef = doc(db, 'whitelists', contract)
await setDoc(doc(wlRef, contract), {
addresses: addresses,
contract: contract
});

whitelist = addresses;
} else {
whitelist = wlSnapshot.docs[0].data().addresses;
for(let i = 0; i < addresses.length; i++) {
if(!whitelist.includes(addresses[i])){
whitelist.push(addresses[i]);
}
}

await setDoc(doc(wlRef, contract), {
addresses: whitelist,
contract: contract
});
}
const wlRef = collection(db, "whitelists");
const q = query(
collection(db, "whitelists"),
where("contract", "==", contract)
);
const wlSnapshot = await getDocs(q);

const leafNodes = whitelist.map(addr => keccak256(addr));
const merkleTree = new MerkleTree(leafNodes, keccak256, {sortPairs: true});
const root = merkleTree.getHexRoot();
let whitelist;
if (wlSnapshot.docs.length === 0) {
//wlRef = doc(db, 'whitelists', contract)
await setDoc(doc(wlRef, contract), {
addresses: addresses,
contract: contract,
});

const result = {
inputs: {contract: contract},
output: {data: root},
success: true
whitelist = addresses;
} else {
whitelist = wlSnapshot.docs[0].data().addresses;
for (let i = 0; i < addresses.length; i++) {
if (!whitelist.includes(addresses[i])) {
whitelist.push(addresses[i]);
}
}

const signer = new ethers.Wallet(process.env.PRIVATE_KEY, provider);

const tx = await NFTContract.connect(signer).setALRoot(result.output.data);
result.output.tx = tx.hash;
return result;
}
await setDoc(doc(wlRef, contract), {
addresses: whitelist,
contract: contract,
});
}

const leafNodes = whitelist.map((addr) => keccak256(addr));
const merkleTree = new MerkleTree(leafNodes, keccak256, { sortPairs: true });
const root = merkleTree.getHexRoot();

const result = {
inputs: { contract: contract },
output: { data: root },
success: true,
};

const signer = new ethers.Wallet(process.env.PRIVATE_KEY, provider);

const tx = await NFTContract.connect(signer).setALRoot(result.output.data);
result.output.tx = tx.hash;
return result;
};
24 changes: 12 additions & 12 deletions functions/ContractOwners.js
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,21 +1,21 @@
import { Network, Alchemy } from "alchemy-sdk";
const settings = {
apiKey: process.env.ALCHEMY_KEY,
network: Network.ETH_MAINNET
apiKey: process.env.ALCHEMY_KEY,
network: Network.ETH_MAINNET,
};

const alchemy = new Alchemy(settings);

export const ContractOwners = async (req) => {
const contract = req.query.contract;
const contract = req.query.contract;

const owners = await alchemy.nft.getOwnersForContract(contract);
const owners = await alchemy.nft.getOwnersForContract(contract);

const result = {
inputs: {contract: contract},
output: {data: owners},
success: true
}
const result = {
inputs: { contract: contract },
output: { data: owners },
success: true,
};

return result;
}
return result;
};
Loading