Add public devnet faucet for staking

Author: cwalinapj

Anchor.toml

@@ -3,6 +3,9 @@ members = [
   "programs/dacit_program"
 ]
 
+[programs.devnet]
+dacit_program = "BHd8TSJu2GF5MntXPSsYrtderZYc4UGXzrVLj9GPoeh1"
+
 [provider]
 cluster = "devnet"
 wallet = "~/.config/solana/id.json"

frontend/components/StakePanel.tsx

@@ -11,11 +11,14 @@ import {
 } from "@solana/spl-token";
 import {
   getConfigPDA,
+  getFaucetStatePDA,
   getMintPDA,
   getProgram,
   getStakeStatePDA,
   getTokenProgramId,
   getVaultPDA,
+  PUBLIC_FAUCET_MAX_BASE_UNITS,
+  TOKEN_BASE_UNITS,
 } from "../utils/anchor";
 
 interface StakePanelProps {
@@ -26,9 +29,9 @@ interface StakePanelProps {
  * StakePanel component allows users to stake DACIT tokens.
  * Connects to the Anchor program and submits stake transactions.
  * 
- * NOTE: This panel is wired to the generated Anchor IDL and the active devnet
- * program address. The staking transaction still assumes the program has been
- * initialized and the connected wallet already holds DACIT tokens.
+ * NOTE: This panel is wired to the generated Anchor IDL and active devnet
+ * program address. It can bootstrap first-time wallets by initializing the
+ * program, creating the ATA, and using the public faucet when needed.
  */
 export default function StakePanel({ className = "" }: StakePanelProps) {
   const { connection } = useConnection();
@@ -52,13 +55,15 @@ export default function StakePanel({ className = "" }: StakePanelProps) {
         throw new Error("Connected wallet does not support Anchor transactions.");
       }
 
-      const stakeAmount = BigInt(Math.trunc(amount));
+      const normalizedAmount = Number.isFinite(amount) ? amount : 0;
+      const stakeAmount = BigInt(Math.round(normalizedAmount * TOKEN_BASE_UNITS));
       if (stakeAmount <= 0n) {
-        throw new Error("Enter a whole-number DACIT amount greater than zero.");
+        throw new Error("Enter a DACIT amount greater than zero.");
       }
 
       const program = getProgram(wallet as AnchorWallet);
       const [configPda] = getConfigPDA();
+      const [faucetStatePda] = getFaucetStatePDA(wallet.publicKey);
       const [stakeStatePda] = getStakeStatePDA(wallet.publicKey);
       const [mintPda] = getMintPDA();
       const [vaultPda] = getVaultPDA(mintPda);
@@ -107,28 +112,46 @@ export default function StakePanel({ className = "" }: StakePanelProps) {
       const userAccount = await getAccount(connection, userAta);
       const currentBalance = userAccount.amount;
       if (currentBalance < stakeAmount) {
-        if (!configAuthority || !configAuthority.equals(wallet.publicKey)) {
-          throw new Error(
-            `Wallet balance is ${currentBalance.toString()} DACIT. Connect the authority wallet to mint test tokens or lower the stake amount.`
-          );
-        }
-
         const topUpAmount = stakeAmount - currentBalance;
-        setStatus(`Minting ${topUpAmount.toString()} DACIT from the devnet faucet...`);
-        await program.methods
-          .mintToUser(new anchor.BN(topUpAmount.toString()))
-          .accounts({
-            config: configPda,
-            mint: mintPda,
-            userAta,
-            user: wallet.publicKey,
-            authority: wallet.publicKey,
-            tokenProgram,
-            systemProgram: anchor.web3.SystemProgram.programId,
-            associatedTokenProgram: anchor.utils.token.ASSOCIATED_PROGRAM_ID,
-            rent: anchor.web3.SYSVAR_RENT_PUBKEY,
-          })
-          .rpc();
+        if (configAuthority && configAuthority.equals(wallet.publicKey)) {
+          setStatus(`Minting ${(Number(topUpAmount) / TOKEN_BASE_UNITS).toFixed(9)} DACIT from the authority faucet...`);
+          await program.methods
+            .mintToUser(new anchor.BN(topUpAmount.toString()))
+            .accounts({
+              config: configPda,
+              mint: mintPda,
+              userAta,
+              user: wallet.publicKey,
+              authority: wallet.publicKey,
+              tokenProgram,
+              systemProgram: anchor.web3.SystemProgram.programId,
+              associatedTokenProgram: anchor.utils.token.ASSOCIATED_PROGRAM_ID,
+              rent: anchor.web3.SYSVAR_RENT_PUBKEY,
+            })
+            .rpc();
+        } else {
+          if (topUpAmount > BigInt(PUBLIC_FAUCET_MAX_BASE_UNITS)) {
+            throw new Error(
+              `Need ${(Number(topUpAmount) / TOKEN_BASE_UNITS).toFixed(9)} DACIT, but the public faucet caps each claim at ${(PUBLIC_FAUCET_MAX_BASE_UNITS / TOKEN_BASE_UNITS).toFixed(9)} DACIT. Lower the stake amount or wait for another faucet claim.`
+            );
+          }
+
+          setStatus(`Claiming ${(Number(topUpAmount) / TOKEN_BASE_UNITS).toFixed(9)} DACIT from the public faucet...`);
+          await program.methods
+            .claimFaucet(new anchor.BN(topUpAmount.toString()))
+            .accounts({
+              config: configPda,
+              mint: mintPda,
+              faucetState: faucetStatePda,
+              user: wallet.publicKey,
+              userAta,
+              tokenProgram,
+              systemProgram: anchor.web3.SystemProgram.programId,
+              associatedTokenProgram: anchor.utils.token.ASSOCIATED_PROGRAM_ID,
+              rent: anchor.web3.SYSVAR_RENT_PUBKEY,
+            })
+            .rpc();
+        }
       }
 
       setStatus("Submitting stake transaction...");
@@ -148,7 +171,7 @@ export default function StakePanel({ className = "" }: StakePanelProps) {
         })
         .rpc();
 
-      alert(`Successfully staked ${amount} DACIT tokens!`);
+      alert(`Successfully staked ${normalizedAmount} DACIT tokens!`);
       setAmount(0);
       setStatus(null);
     } catch (err) {
@@ -188,6 +211,7 @@ export default function StakePanel({ className = "" }: StakePanelProps) {
             type="number"
             value={amount}
             min={0}
+            step="0.000000001"
             onChange={(e) => setAmount(Number(e.target.value))}
             className="w-full p-3 border border-gray-300 rounded-lg focus:ring-2 focus:ring-blue-500 focus:border-transparent"
             placeholder="Enter amount"
@@ -215,8 +239,8 @@ export default function StakePanel({ className = "" }: StakePanelProps) {
 
         {wallet.publicKey && (
           <p className="text-xs text-gray-500 text-center">
-            First run will initialize the program, create your ATA, and mint DACIT if this wallet
-            is the configured authority.
+            First run will initialize the program, create your ATA, and claim up to 100 DACIT
+            from the public faucet when needed.
           </p>
         )}
       </div>

frontend/utils/anchor.ts

@@ -7,11 +7,14 @@ import { AnchorWallet } from "@solana/wallet-adapter-react";
 import idl from "./idl/dacit_program.json";
 
 const PROGRAM_ID = new anchor.web3.PublicKey(
-  "7rrFC6iDci7eDwq4Dvzhw1X5ejsL6EkpW79g5s25NbwQ"
+  "BHd8TSJu2GF5MntXPSsYrtderZYc4UGXzrVLj9GPoeh1"
 );
 
 // Network endpoint - change to mainnet-beta for production
 const DEVNET_ENDPOINT = "https://api.devnet.solana.com";
+export const TOKEN_DECIMALS = 9;
+export const TOKEN_BASE_UNITS = 10 ** TOKEN_DECIMALS;
+export const PUBLIC_FAUCET_MAX_BASE_UNITS = 100 * TOKEN_BASE_UNITS;
 
 /**
  * Creates an Anchor Provider instance for interacting with the Solana network.
@@ -76,6 +79,14 @@ export const getMintPDA = (): [anchor.web3.PublicKey, number] =>
     PROGRAM_ID
   );
 
+export const getFaucetStatePDA = (
+  userPubkey: anchor.web3.PublicKey
+): [anchor.web3.PublicKey, number] =>
+  anchor.web3.PublicKey.findProgramAddressSync(
+    [Buffer.from("faucet-state"), userPubkey.toBuffer()],
+    PROGRAM_ID
+  );
+
 /**
  * Derives a PDA for the emission schedule account.
  *