@accessprotocol/distributor
Version:
Access Protocol Distributor Library
179 lines (178 loc) • 7.64 kB
JavaScript
"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.MerkleDistributor = void 0;
const web3_js_1 = require("@solana/web3.js");
const bn_js_1 = __importDefault(require("bn.js")); // eslint-disable-line @typescript-eslint/no-unused-vars
const borsh = __importStar(require("@coral-xyz/borsh")); // eslint-disable-line @typescript-eslint/no-unused-vars
const programId_1 = require("../programId");
/** State for the account which distributes tokens. */
class MerkleDistributor {
constructor(fields) {
this.bump = fields.bump;
this.version = fields.version;
this.root = fields.root;
this.mint = fields.mint;
this.tokenVault = fields.tokenVault;
this.maxTotalClaim = fields.maxTotalClaim;
this.maxNumNodes = fields.maxNumNodes;
this.totalAmountClaimed = fields.totalAmountClaimed;
this.numNodesClaimed = fields.numNodesClaimed;
this.startTs = fields.startTs;
this.endTs = fields.endTs;
this.creator = fields.creator;
this.admin = fields.admin;
this.clawedBack = fields.clawedBack;
}
static fetch(c_1, address_1) {
return __awaiter(this, arguments, void 0, function* (c, address, programId = programId_1.PROGRAM_ID) {
const info = yield c.getAccountInfo(address);
if (info === null) {
return null;
}
if (!info.owner.equals(programId)) {
throw new Error("account doesn't belong to this program");
}
return this.decode(info.data);
});
}
static fetchMultiple(c_1, addresses_1) {
return __awaiter(this, arguments, void 0, function* (c, addresses, programId = programId_1.PROGRAM_ID) {
const infos = yield c.getMultipleAccountsInfo(addresses);
return infos.map((info) => {
if (info === null) {
return null;
}
if (!info.owner.equals(programId)) {
throw new Error("account doesn't belong to this program");
}
return this.decode(info.data);
});
});
}
static decode(data) {
if (!data.slice(0, 8).equals(MerkleDistributor.discriminator)) {
throw new Error("invalid account discriminator");
}
const dec = MerkleDistributor.layout.decode(data.slice(8));
return new MerkleDistributor({
bump: dec.bump,
version: dec.version,
root: dec.root,
mint: dec.mint,
tokenVault: dec.tokenVault,
maxTotalClaim: dec.maxTotalClaim,
maxNumNodes: dec.maxNumNodes,
totalAmountClaimed: dec.totalAmountClaimed,
numNodesClaimed: dec.numNodesClaimed,
startTs: dec.startTs,
endTs: dec.endTs,
creator: dec.creator,
admin: dec.admin,
clawedBack: dec.clawedBack,
});
}
toJSON() {
return {
bump: this.bump,
version: this.version.toString(),
root: this.root,
mint: this.mint.toString(),
tokenVault: this.tokenVault.toString(),
maxTotalClaim: this.maxTotalClaim.toString(),
maxNumNodes: this.maxNumNodes.toString(),
totalAmountClaimed: this.totalAmountClaimed.toString(),
numNodesClaimed: this.numNodesClaimed.toString(),
startTs: this.startTs.toString(),
endTs: this.endTs.toString(),
creator: this.creator.toString(),
admin: this.admin.toString(),
clawedBack: this.clawedBack,
};
}
static fromJSON(obj) {
return new MerkleDistributor({
bump: obj.bump,
version: new bn_js_1.default(obj.version),
root: obj.root,
mint: new web3_js_1.PublicKey(obj.mint),
tokenVault: new web3_js_1.PublicKey(obj.tokenVault),
maxTotalClaim: new bn_js_1.default(obj.maxTotalClaim),
maxNumNodes: new bn_js_1.default(obj.maxNumNodes),
totalAmountClaimed: new bn_js_1.default(obj.totalAmountClaimed),
numNodesClaimed: new bn_js_1.default(obj.numNodesClaimed),
startTs: new bn_js_1.default(obj.startTs),
endTs: new bn_js_1.default(obj.endTs),
creator: new web3_js_1.PublicKey(obj.creator),
admin: new web3_js_1.PublicKey(obj.admin),
clawedBack: obj.clawedBack,
});
}
}
exports.MerkleDistributor = MerkleDistributor;
MerkleDistributor.discriminator = Buffer.from([
77, 119, 139, 70, 84, 247, 12, 26,
]);
MerkleDistributor.layout = borsh.struct([
borsh.u8("bump"),
borsh.u64("version"),
borsh.array(borsh.u8(), 32, "root"),
borsh.publicKey("mint"),
borsh.publicKey("tokenVault"),
borsh.u64("maxTotalClaim"),
borsh.u64("maxNumNodes"),
borsh.u64("totalAmountClaimed"),
borsh.u64("numNodesClaimed"),
borsh.i64("startTs"),
borsh.i64("endTs"),
borsh.publicKey("creator"),
borsh.publicKey("admin"),
borsh.bool("clawedBack"),
]);
MerkleDistributor.getAddress = (creator, mint, version, programId = programId_1.PROGRAM_ID) => {
return web3_js_1.PublicKey.findProgramAddressSync([Buffer.from("MerkleDistributor"), mint.toBuffer(), creator.toBuffer(), version.toArrayLike(Buffer, "le", 8)], programId)[0];
};