add delete prelaunch

programs/drift/src/instructions/admin.rs

@@ -2468,6 +2468,21 @@ pub fn handle_update_prelaunch_oracle_params<'info>(
     Ok(())
 }
 
+pub fn handle_delete_prelaunch_oracle<'info>(
+    ctx: Context<DeletePrelaunchOracle<'info>>,
+    _perp_market_index: u16,
+) -> Result<()> {
+    let perp_market = ctx.accounts.perp_market.load()?;
+
+    validate!(
+        perp_market.amm.oracle != ctx.accounts.prelaunch_oracle.key(),
+        ErrorCode::DefaultError,
+        "prelaunch oracle currently in use"
+    )?;
+
+    Ok(())
+}
+
 #[derive(Accounts)]
 pub struct Initialize<'info> {
     #[account(mut)]
@@ -2942,3 +2957,25 @@ pub struct UpdatePrelaunchOracleParams<'info> {
     )]
     pub state: Box<Account<'info, State>>,
 }
+
+#[derive(Accounts)]
+#[instruction(perp_market_index: u16,)]
+pub struct DeletePrelaunchOracle<'info> {
+    #[account(mut)]
+    pub admin: Signer<'info>,
+    #[account(
+        mut,
+        seeds = [b"prelaunch_oracle".as_ref(), perp_market_index.to_le_bytes().as_ref()],
+        bump,
+        close = admin
+    )]
+    pub prelaunch_oracle: AccountLoader<'info, PrelaunchOracle>,
+    #[account(
+        constraint = perp_market.load()?.market_index == perp_market_index
+    )]
+    pub perp_market: AccountLoader<'info, PerpMarket>,
+    #[account(
+        has_one = admin
+    )]
+    pub state: Box<Account<'info, State>>,
+}

programs/drift/src/lib.rs

@@ -1171,6 +1171,13 @@ pub mod drift {
     ) -> Result<()> {
         handle_update_prelaunch_oracle_params(ctx, params)
     }
+
+    pub fn delete_prelaunch_oracle(
+        ctx: Context<DeletePrelaunchOracle>,
+        perp_market_index: u16,
+    ) -> Result<()> {
+        handle_delete_prelaunch_oracle(ctx, perp_market_index)
+    }
 }
 
 #[cfg(not(feature = "no-entrypoint"))]

sdk/src/adminClient.ts

@@ -1871,4 +1871,23 @@ export class AdminClient extends DriftClient {
 			},
 		});
 	}
+
+	public async deletePrelaunchOracle(
+		perpMarketIndex: number
+	): Promise<TransactionSignature> {
+		return await this.program.rpc.deletePrelaunchOracle(perpMarketIndex, {
+			accounts: {
+				admin: this.wallet.publicKey,
+				state: await this.getStatePublicKey(),
+				prelaunchOracle: await getPrelaunchOraclePublicKey(
+					this.program.programId,
+					perpMarketIndex
+				),
+				perpMarket: await getPerpMarketPublicKey(
+					this.program.programId,
+					perpMarketIndex
+				),
+			},
+		});
+	}
 }

sdk/src/idl/drift.json

@@ -5012,6 +5012,37 @@
           }
         }
       ]
+    },
+    {
+      "name": "deletePrelaunchOracle",
+      "accounts": [
+        {
+          "name": "admin",
+          "isMut": true,
+          "isSigner": true
+        },
+        {
+          "name": "prelaunchOracle",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "perpMarket",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "state",
+          "isMut": false,
+          "isSigner": false
+        }
+      ],
+      "args": [
+        {
+          "name": "perpMarketIndex",
+          "type": "u16"
+        }
+      ]
     }
   ],
   "accounts": [

tests/prelisting.ts

@@ -18,6 +18,7 @@ import {
 
 import {
 	initializeQuoteSpotMarket,
+	mockOracle,
 	mockUSDCMint,
 	mockUserUSDCAccount,
 } from './testHelpers';
@@ -209,7 +210,34 @@ describe('prelisting', () => {
 		const price = adminDriftClient.getOracleDataForPerpMarket(0);
 		assert(price.price.eq(new BN(40000000)));
 
-		const markTwap = adminDriftClient.getPerpMarketAccount(0).amm.lastMarkPriceTwap;
+		const markTwap =
+			adminDriftClient.getPerpMarketAccount(0).amm.lastMarkPriceTwap;
 		assert(markTwap.eq(new BN(40000000)));
 	});
+
+	it('delete', async () => {
+		try {
+			await adminDriftClient.deletePrelaunchOracle(0);
+			assert(false);
+		} catch (e) {
+			console.log('Delete successfully failed');
+		}
+
+		const oldOracleKey = adminDriftClient.getPerpMarketAccount(0).amm.oracle;
+
+		const newOracle = await mockOracle(40);
+		await adminDriftClient.updatePerpMarketOracle(
+			0,
+			newOracle,
+			OracleSource.PYTH
+		);
+
+		await adminDriftClient.deletePrelaunchOracle(0);
+
+		const result = await connection.getAccountInfoAndContext(
+			oldOracleKey,
+			'processed'
+		);
+		assert(result.value === null);
+	});
 });