solana/web3.js/examples/budget-two-approvers.js

/*
  Example of using the Budget program to perform a payment authorized by two parties
*/

//eslint-disable-next-line import/no-commonjs
const solanaWeb3 = require('..');
//const solanaWeb3 = require('@solana/web3.js');

const account1 = new solanaWeb3.Account();
const account2 = new solanaWeb3.Account();
const contractFunds = new solanaWeb3.Account();
const contractState = new solanaWeb3.Account();

const approver1 = new solanaWeb3.Account();
const approver2 = new solanaWeb3.Account();

let url;
url = 'http://localhost:8899';
//url = 'http://testnet.solana.com:8899';
const connection = new solanaWeb3.Connection(url);

function showBalance() {
  console.log(`\n== Account State`);
  return Promise.all([
    connection.getBalance(account1.publicKey),
    connection.getBalance(account2.publicKey),
    connection.getBalance(contractFunds.publicKey),
    connection.getBalance(contractState.publicKey),
  ]).then(([fromBalance, toBalance, contractFundsBalance, contractStateBalance]) => {
    console.log(`Account1:       ${account1.publicKey} has a balance of ${fromBalance}`);
    console.log(`Account2:       ${account2.publicKey} has a balance of ${toBalance}`);
    console.log(`Contract Funds: ${contractFunds.publicKey} has a balance of ${contractFundsBalance}`);
    console.log(`Contract State: ${contractState.publicKey} has a balance of ${contractStateBalance}`);
  });
}

function confirmTransaction(signature) {
  console.log('Confirming transaction:', signature);
  return connection.getSignatureStatus(signature)
  .then((confirmation) => {
    if (confirmation !== 'Confirmed') {
      throw new Error(`Transaction was not confirmed (${confirmation})`);
    }
    console.log('Transaction confirmed');
  });
}

function airDrop() {
  console.log(`\n== Requesting airdrop of 100 to ${account1.publicKey}`);
  return connection.requestAirdrop(account1.publicKey, 100)
  .then(confirmTransaction);
}

showBalance()
.then(airDrop)
.then(() => {
  console.log(`\n== Move 1 token to approver1`);
  const transaction = solanaWeb3.SystemProgram.move(
    account1.publicKey,
    approver1.publicKey,
    1,
  );
  return connection.sendTransaction(account1, transaction);
})
.then(confirmTransaction)
.then(() => {
  console.log(`\n== Move 1 token to approver2`);
  const transaction = solanaWeb3.SystemProgram.move(
    account1.publicKey,
    approver2.publicKey,
    1,
  );
  return connection.sendTransaction(account1, transaction);
})
.then(confirmTransaction)
.then(showBalance)
.then(() => {
  console.log(`\n== Creating account for the contract funds`);
  const transaction = solanaWeb3.SystemProgram.createAccount(
    account1.publicKey,
    contractFunds.publicKey,
    50, // number of tokens to transfer
    0,
    solanaWeb3.BudgetProgram.programId,
  );
  return connection.sendTransaction(account1, transaction);
})
.then(confirmTransaction)
.then(showBalance)
.then(() => {
  console.log(`\n== Creating account for the contract state`);
  const transaction = solanaWeb3.SystemProgram.createAccount(
    account1.publicKey,
    contractState.publicKey,
    1, // account1 pays 1 token to hold the contract state
    solanaWeb3.BudgetProgram.space,
    solanaWeb3.BudgetProgram.programId,
  );
  return connection.sendTransaction(account1, transaction);
})
.then(confirmTransaction)
.then(showBalance)
.then(() => {
  console.log(`\n== Initializing contract`);
  const transaction = solanaWeb3.BudgetProgram.payOnBoth(
    contractFunds.publicKey,
    contractState.publicKey,
    account2.publicKey,
    50,
    solanaWeb3.BudgetProgram.signatureCondition(approver1.publicKey),
    solanaWeb3.BudgetProgram.signatureCondition(approver2.publicKey),
  );
  return connection.sendTransaction(contractFunds, transaction);
})
.then(confirmTransaction)
.then(showBalance)
.then(() => {
  console.log(`\n== Apply approver 1`);
  const transaction = solanaWeb3.BudgetProgram.applySignature(
    approver1.publicKey,
    contractState.publicKey,
    account2.publicKey,
  );
  return connection.sendTransaction(approver1, transaction);
})
.then(confirmTransaction)
.then(showBalance)
.then(() => {
  console.log(`\n== Apply approver 2`);
  const transaction = solanaWeb3.BudgetProgram.applySignature(
    approver2.publicKey,
    contractState.publicKey,
    account2.publicKey,
  );
  return connection.sendTransaction(approver2, transaction);
})
.then(confirmTransaction)
.then(showBalance)

.then(() => {
  console.log('\nDone');
})

.catch((err) => {
  console.log(err);
});