Add timer

Author: metaproph3t

programs/launchpad/src/error.rs

@@ -8,4 +8,6 @@ pub enum LaunchpadError {
     SupplyNonZero,
     #[msg("Invalid launch state")]
     InvalidLaunchState,
+    #[msg("Launch period not over")]
+    LaunchPeriodNotOver,
 } 

programs/launchpad/src/instructions/complete_launch.rs

@@ -25,6 +25,17 @@ pub struct CompleteLaunch<'info> {
 
 impl CompleteLaunch<'_> {
     pub fn validate(&self) -> Result<()> {
+        let clock = Clock::get()?;
+        
+        // 7 days in slots (assuming 400ms per slot)
+        const SLOTS_PER_DAY: u64 = 216_000; // (24 * 60 * 60 * 1000) / 400
+        const REQUIRED_SLOTS: u64 = SLOTS_PER_DAY * 5;
+
+        require!(
+            clock.slot >= self.launch.slot_initialized.saturating_add(REQUIRED_SLOTS),
+            LaunchpadError::LaunchPeriodNotOver
+        );
+
         Ok(())
     }
 

programs/launchpad/src/instructions/initialize_launch.rs

@@ -85,6 +85,7 @@ impl InitializeLaunch<'_> {
         ctx: Context<Self>,
         args: InitializeLaunchArgs,
     ) -> Result<()> {
+        let clock = Clock::get()?;
         let (dao_treasury, _) = Pubkey::find_program_address(
             &[ctx.accounts.dao.key().as_ref()],
             &AUTOCRAT_PROGRAM_ID
@@ -102,6 +103,7 @@ impl InitializeLaunch<'_> {
             pda_bump: ctx.bumps.launch,
             seq_num: 0,
             state: LaunchState::Live,
+            slot_initialized: clock.slot,
         });
 
         let clock = Clock::get()?;

programs/launchpad/src/state/launch.rs

@@ -31,4 +31,5 @@ pub struct Launch {
     /// The sequence number of this launch. Useful for sorting events.
     pub seq_num: u64,
     pub state: LaunchState,
+    pub slot_initialized: u64,
 }

sdk/src/v0.4/types/launchpad.ts

@@ -230,6 +230,10 @@ export type Launchpad = {
             type: {
               defined: "LaunchState";
             };
+          },
+          {
+            name: "slotInitialized";
+            type: "u64";
           }
         ];
       };
@@ -346,6 +350,11 @@ export type Launchpad = {
       code: 6002;
       name: "InvalidLaunchState";
       msg: "Invalid launch state";
+    },
+    {
+      code: 6003;
+      name: "LaunchPeriodNotOver";
+      msg: "Launch period not over";
     }
   ];
 };
@@ -583,6 +592,10 @@ export const IDL: Launchpad = {
               defined: "LaunchState",
             },
           },
+          {
+            name: "slotInitialized",
+            type: "u64",
+          },
         ],
       },
     },
@@ -699,5 +712,10 @@ export const IDL: Launchpad = {
       name: "InvalidLaunchState",
       msg: "Invalid launch state",
     },
+    {
+      code: 6003,
+      name: "LaunchPeriodNotOver",
+      msg: "Launch period not over",
+    },
   ],
 };

tests/launchpad/unit/completeLaunch.test.ts

@@ -1,4 +1,4 @@
-import { PublicKey } from "@solana/web3.js";
+import { ComputeBudgetProgram, PublicKey } from "@solana/web3.js";
 import { assert } from "chai";
 import {
   AutocratClient,
@@ -21,6 +21,7 @@ export default function suite() {
   let treasuryUsdcAccount: PublicKey;
 
   const minRaise = new BN(1000_000000); // 1000 USDC
+  const SLOTS_PER_DAY = 216_000n; // (24 * 60 * 60 * 1000) / 400
 
   before(async function () {
     autocratClient = this.autocratClient;
@@ -61,11 +62,10 @@ export default function suite() {
     ]).rpc();
   });
 
-  it("completes launch successfully when minimum raise is met", async function () {
+  it("completes launch successfully when minimum raise is met and time has passed", async function () {
     // Fund the launch with exactly minimum raise
     const userUsdcAccount = await this.createTokenAccount(USDC, this.payer.publicKey);
     const userTokenAccount = await this.createTokenAccount(META, this.payer.publicKey);
-    // await this.mintTo(USDC, userUsdcAccount, this.payer, minRaise.toNumber());
     await this.mintTo(USDC, this.payer.publicKey, this.payer, minRaise.toNumber());
 
     await launchpadClient.fundIx(
@@ -75,6 +75,9 @@ export default function suite() {
       META
     ).rpc();
 
+    // Advance clock past 7 days
+    await this.advanceBySlots(SLOTS_PER_DAY * 7n);
+
     // Complete the launch
     await launchpadClient.completeLaunchIx(launch, USDC, daoTreasury).rpc();
 
@@ -85,7 +88,39 @@ export default function suite() {
     assert.equal(treasuryBalance.toString(), minRaise.toString());
   });
 
-  it("moves to refunding state when minimum raise is not met", async function () {
+  it("fails when launch period has not passed", async function () {
+    // Fund the launch with exactly minimum raise
+    const userUsdcAccount = await this.createTokenAccount(USDC, this.payer.publicKey);
+    const userTokenAccount = await this.createTokenAccount(META, this.payer.publicKey);
+    await this.mintTo(USDC, this.payer.publicKey, this.payer, minRaise.toNumber());
+
+    await launchpadClient.fundIx(
+      launch,
+      minRaise,
+      USDC,
+      META
+    ).rpc();
+
+    // Try to complete immediately (should fail)
+    try {
+      await launchpadClient.completeLaunchIx(launch, USDC, daoTreasury).rpc();
+      assert.fail("Should have thrown error");
+    } catch (e) {
+      assert.include(e.message, "LaunchPeriodNotOver");
+    }
+
+    // Advance by 6 days (still not enough)
+    await this.advanceBySlots(SLOTS_PER_DAY * 3n);
+
+    try {
+      await launchpadClient.completeLaunchIx(launch, USDC, daoTreasury).preInstructions([ComputeBudgetProgram.setComputeUnitPrice({ microLamports: 1 })]).rpc();
+      assert.fail("Should have thrown error");
+    } catch (e) {
+      assert.include(e.message, "LaunchPeriodNotOver");
+    }
+  });
+
+  it("moves to refunding state when minimum raise is not met after period", async function () {
     // Fund the launch with less than minimum raise
     const userUsdcAccount = await this.createTokenAccount(USDC, this.payer.publicKey);
     const userTokenAccount = await this.createTokenAccount(META, this.payer.publicKey);
@@ -99,6 +134,9 @@ export default function suite() {
       META
     ).rpc();
 
+    // Advance clock past 7 days
+    await this.advanceBySlots(SLOTS_PER_DAY * 7n);
+
     // Complete the launch
     await launchpadClient.completeLaunchIx(launch, USDC, daoTreasury).rpc();
 
@@ -110,6 +148,9 @@ export default function suite() {
   });
 
   it("fails when launch is not in live state", async function () {
+    // Advance clock past 7 days
+    await this.advanceBySlots(SLOTS_PER_DAY * 7n);
+
     // Complete launch first time
     await launchpadClient.completeLaunchIx(launch, USDC, daoTreasury).rpc();
 

tests/main.test.ts

@@ -3,7 +3,7 @@ import amm from "./amm/main.test.js";
 import autocrat from "./autocrat/autocrat.js";
 import launchpad from "./launchpad/main.test.js";
 
-import { startAnchor } from "solana-bankrun";
+import { Clock, startAnchor } from "solana-bankrun";
 import { BankrunProvider } from "anchor-bankrun";
 import * as anchor from "@coral-xyz/anchor";
 import {
@@ -143,6 +143,22 @@ before(async function () {
       amount
     );
   };
+
+  this.advanceBySlots = async (
+    slots: bigint
+  ) => {
+    const currentClock = await this.context.banksClient.getClock();
+    this.context.setClock(
+      new Clock(
+        currentClock.slot + slots,
+        currentClock.epochStartTimestamp,
+        currentClock.epoch,
+        currentClock.leaderScheduleEpoch,
+        50n
+      )
+    );
+  };
+
 });
 
 describe("conditional_vault", conditionalVault);