chore: simplify pinocchio counter example

pinocchio-counter/Cargo.toml

@@ -3,14 +3,36 @@ name = "pinocchio-counter"
 version = "0.1.0"
 edition = "2021"
 
+[features]
+logging = []
+
 [dependencies]
-pinocchio = { version = "0.10", features = ["cpi", "alloc", "copy"] }
+pinocchio = { version = "0.10.1", features = ["cpi", "copy"] }
 pinocchio-log = { version = "0.5" }
-pinocchio-pubkey = { version = "0.3", features = ["const"] }
 pinocchio-system = { version = "0.5" }
-pinocchio-token = { version = "0.5" }
-ephemeral-rollups-pinocchio = { version = "0.8.0", features = ["address-find-program-address"] }
+ephemeral-rollups-pinocchio = { version = "0.8.1" }
+
+[dev-dependencies]
+solana-instruction = "3.0.0"
+solana-program = "3.0.0"
+solana-program-test = "=3.0.2"
+solana-pubkey = "4.0.0"
+solana-signer = "3.0.0"
+solana-transaction = "3.0.1"
+solana-system-interface = { version = "2.0.0", features = ["bincode"] }
 
 [lib]
 name = "pinocchio_counter"
-crate-type = ["cdylib", "lib"]
+crate-type = ["cdylib"]
+
+[[test]]
+name = "delegate_counter"
+path = "tests-rs/delegate_counter.rs"
+
+[[test]]
+name = "increase_counter"
+path = "tests-rs/increase_counter.rs"
+
+[[test]]
+name = "initialize_counter"
+path = "tests-rs/initialize_counter.rs"

pinocchio-counter/package.json

@@ -33,5 +33,6 @@
     "@solana/web3.js": "^1.93.0",
     "borsh": "^0.7.0",
     "dotenv": "^16.0.0"
-  }
+  },
+  "packageManager": "yarn@1.22.22+sha512.a6b2f7906b721bba3d67d4aff083df04dad64c399707841b7acf00f6b133b7ac24255f2652fa22ae3534329dc6180534e98d17432037ff6fd140556e2bb3137e"
 }

pinocchio-counter/src/entrypoint.rs

@@ -1,15 +1,98 @@
-use crate::processor;
+use crate::processor::{
+    process_commit, process_commit_and_undelegate, process_delegate, process_increase_counter,
+    process_increment_commit, process_increment_undelegate, process_initialize_counter,
+    process_undelegation_callback,
+};
+use core::{mem::MaybeUninit, slice::from_raw_parts};
 use pinocchio::{
-    account::AccountView, entrypoint, Address, ProgramResult,
+    entrypoint::deserialize, error::ProgramError, no_allocator, nostd_panic_handler, AccountView,
+    Address, ProgramResult, MAX_TX_ACCOUNTS, SUCCESS,
 };
 
-entrypoint!(process_instruction);
+// Do not allocate memory.
+no_allocator!();
+// Use the no_std panic handler.
+nostd_panic_handler!();
+
+#[no_mangle]
+#[allow(clippy::arithmetic_side_effects)]
+pub unsafe extern "C" fn entrypoint(input: *mut u8) -> u64 {
+    const UNINIT: MaybeUninit<AccountView> = MaybeUninit::<AccountView>::uninit();
+    let mut accounts = [UNINIT; { MAX_TX_ACCOUNTS }];
+
+    let (program_id, count, instruction_data) = deserialize::<MAX_TX_ACCOUNTS>(input, &mut accounts);
+
+    match process_instruction(
+        &program_id,
+        from_raw_parts(accounts.as_ptr() as _, count),
+        instruction_data,
+    ) {
+        Ok(()) => SUCCESS,
+        Err(error) => error.into(),
+    }
+}
 
+/// Log an error.
+#[cold]
+fn log_error(_error: &ProgramError) {
+    #[cfg(feature = "logging")]
+    pinocchio_log::log!("Program error");
+}
+
+/// Process an instruction.
+#[inline(always)]
 pub fn process_instruction(
     program_id: &Address,
     accounts: &[AccountView],
     instruction_data: &[u8],
 ) -> ProgramResult {
-    processor::process_instruction(program_id, accounts, instruction_data)?;
-    Ok(())
+    let result = inner_process_instruction(program_id, accounts, instruction_data);
+    result.inspect_err(log_error)
+}
+
+/// Process an instruction.
+#[inline(always)]
+pub(crate) fn inner_process_instruction(
+    program_id: &Address,
+    accounts: &[AccountView],
+    instruction_data: &[u8],
+) -> ProgramResult {
+    if instruction_data.len() < 8 {
+        return Err(ProgramError::InvalidInstructionData);
+    }
+
+    let discriminator = &instruction_data[..8];
+    let payload = &instruction_data[8..];
+
+    match discriminator {
+        [0, 0, 0, 0, 0, 0, 0, 0] => process_initialize_counter(program_id, accounts),
+        [1, 0, 0, 0, 0, 0, 0, 0] => {
+            let increase_by = read_u64(payload)?;
+            process_increase_counter(program_id, accounts, increase_by)
+        }
+        [2, 0, 0, 0, 0, 0, 0, 0] => process_delegate(program_id, accounts),
+        [3, 0, 0, 0, 0, 0, 0, 0] => process_commit_and_undelegate(program_id, accounts),
+        [4, 0, 0, 0, 0, 0, 0, 0] => process_commit(program_id, accounts),
+        [5, 0, 0, 0, 0, 0, 0, 0] => {
+            let increase_by = read_u64(payload)?;
+            process_increment_commit(program_id, accounts, increase_by)
+        }
+        [6, 0, 0, 0, 0, 0, 0, 0] => {
+            let increase_by = read_u64(payload)?;
+            process_increment_undelegate(program_id, accounts, increase_by)
+        }
+        [196, 28, 41, 206, 48, 37, 51, 167] => {
+            process_undelegation_callback(program_id, accounts, payload)
+        }
+        _ => Err(ProgramError::InvalidInstructionData),
+    }
+}
+
+fn read_u64(input: &[u8]) -> Result<u64, ProgramError> {
+    if input.len() < 8 {
+        return Err(ProgramError::InvalidInstructionData);
+    }
+    let mut bytes = [0u8; 8];
+    bytes.copy_from_slice(&input[..8]);
+    Ok(u64::from_le_bytes(bytes))
 }

pinocchio-counter/src/lib.rs

@@ -1,4 +1,8 @@
-pub mod entrypoint;
-pub mod instruction;
-pub mod processor;
-pub mod state;
+#![no_std]
+#![allow(unexpected_cfgs)]
+
+mod entrypoint;
+mod processor;
+mod state;
+
+pub use crate::entrypoint::process_instruction;