Add BudgetProgram.payOnBoth

2018-09-26 09:49:59 -07:00
parent 63382b7e6b
commit e3703cec29
4 changed files with 208 additions and 4 deletions
--- a/web3.js/examples/budget-timestamp.js
+++ b/web3.js/examples/budget-timestamp.js
--- a/web3.js/examples/budget-two-approvers.js
+++ b/web3.js/examples/budget-two-approvers.js
@ -0,0 +1,146 @@
+/*
+  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.confirmTransaction(signature)
+  .then((confirmation) => {
+    if (!confirmation) {
+      throw new Error('Transaction was not confirmed');
+    }
+    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);
+});