aftermath-ts-sdk
Version:
Aftermath TypeScript SDK
126 lines (125 loc) • 6.42 kB
JavaScript
;
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());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.ReferralVaultApi = void 0;
const transactions_1 = require("@mysten/sui/transactions");
const bcs_1 = require("@mysten/sui/bcs");
const utils_1 = require("../../../general/utils");
class ReferralVaultApi {
// =========================================================================
// Constructor
// =========================================================================
constructor(Provider) {
this.Provider = Provider;
// =========================================================================
// Public Methods
// =========================================================================
// =========================================================================
// Transaction Commands
// =========================================================================
this.updateReferrerTx = (inputs) => {
// TODO: handle this case better
// in try catch in case referrer is invalid address and throws
try {
const { tx, referrer } = inputs;
if (tx.blockData.sender &&
utils_1.Helpers.addLeadingZeroesToType(tx.blockData.sender) ===
utils_1.Helpers.addLeadingZeroesToType(referrer))
return;
return tx.moveCall({
target: utils_1.Helpers.transactions.createTxTarget(this.addresses.packages.referralVault, ReferralVaultApi.constants.moduleNames.referralVault, "update_referrer_address"),
typeArguments: [],
arguments: [
tx.object(this.addresses.objects.referralVault),
tx.pure.address(referrer),
],
});
}
catch (e) { }
};
this.withdrawRebateTx = (inputs) => {
const { tx } = inputs;
return tx.moveCall({
target: utils_1.Helpers.transactions.createTxTarget(this.addresses.packages.referralVault, ReferralVaultApi.constants.moduleNames.referralVault, inputs.withTransfer
? "withdraw_and_transfer"
: "withdraw_rebate"),
typeArguments: [inputs.coinType],
arguments: [tx.object(this.addresses.objects.referralVault)],
});
};
this.balanceOfRebateTx = (inputs) => {
const { tx } = inputs;
return tx.moveCall({
target: utils_1.Helpers.transactions.createTxTarget(this.addresses.packages.referralVault, ReferralVaultApi.constants.moduleNames.referralVault, "balance_of"),
typeArguments: [inputs.coinType],
arguments: [
tx.object(this.addresses.objects.referralVault),
tx.pure.address(inputs.referrer),
],
});
};
this.referrerForTx = (inputs) => {
const { tx } = inputs;
return tx.moveCall({
target: utils_1.Helpers.transactions.createTxTarget(this.addresses.packages.referralVault, ReferralVaultApi.constants.moduleNames.referralVault, "referrer_for"),
typeArguments: [],
arguments: [
tx.object(this.addresses.objects.referralVault),
tx.pure.address(inputs.referee),
],
});
};
this.hasReffererTx = (inputs) => {
const { tx } = inputs;
return tx.moveCall({
target: utils_1.Helpers.transactions.createTxTarget(this.addresses.packages.referralVault, ReferralVaultApi.constants.moduleNames.referralVault, "has_referrer"),
typeArguments: [],
arguments: [
tx.object(this.addresses.objects.referralVault),
tx.pure.address(inputs.referee),
],
});
};
// =========================================================================
// Inspections
// =========================================================================
this.fetchBalanceOfRebate = (inputs) => __awaiter(this, void 0, void 0, function* () {
const tx = new transactions_1.Transaction();
this.balanceOfRebateTx(Object.assign(Object.assign({}, inputs), { tx }));
const bytes = yield this.Provider.Inspections().fetchFirstBytesFromTxOutput({
tx,
});
return utils_1.Casting.bigIntFromBytes(bytes);
});
this.fetchReferrer = (inputs) => __awaiter(this, void 0, void 0, function* () {
const tx = new transactions_1.Transaction();
this.referrerForTx(Object.assign(Object.assign({}, inputs), { tx }));
const bytes = yield this.Provider.Inspections().fetchFirstBytesFromTxOutput({
tx,
});
const unwrapped = bcs_1.bcs.option(bcs_1.bcs.Address).parse(new Uint8Array(bytes));
return unwrapped !== null && unwrapped !== void 0 ? unwrapped : undefined;
});
const addresses = this.Provider.addresses.referralVault;
if (!addresses)
throw new Error("not all required addresses have been set in provider");
this.addresses = addresses;
}
}
exports.ReferralVaultApi = ReferralVaultApi;
// =========================================================================
// Constants
// =========================================================================
ReferralVaultApi.constants = {
moduleNames: {
referralVault: "referral_vault",
},
};