Add next_available_task_ids, expose program

Author: madninja

tuktuk-sdk/src/lib.rs

@@ -7,6 +7,8 @@ pub mod pubsub_client;
 pub mod tuktuk;
 pub mod watcher;
 
+pub use tuktuk_program;
+
 pub mod prelude {
     pub use anchor_lang::prelude::*;
 

tuktuk-sdk/src/tuktuk.rs

@@ -388,7 +388,23 @@ pub mod task_queue {
 
     use self::tuktuk::types::InitializeTaskQueueArgsV0;
     use super::*;
-    use crate::client::GetAnchorAccount;
+    use crate::{
+        client::GetAnchorAccount,
+        compiled_transaction::next_available_task_ids_excluding_in_progress,
+    };
+
+    pub fn next_available_task_ids(
+        task_queue: &tuktuk::accounts::TaskQueueV0,
+        n: u8,
+    ) -> Result<Vec<u16>, Error> {
+        next_available_task_ids_excluding_in_progress(
+            task_queue.capacity,
+            &task_queue.task_bitmap,
+            n,
+            &Default::default(),
+            0,
+        )
+    }
 
     pub fn key(config_key: &Pubkey, next_task_queue_id: u32) -> Pubkey {
         Pubkey::find_program_address(
@@ -414,6 +430,10 @@ pub mod task_queue {
         .0
     }
 
+    pub fn queue_authority_key() -> Pubkey {
+        Pubkey::find_program_address(&[b"queue_authority"], &tuktuk::ID).0
+    }
+
     pub fn keys(config_key: &Pubkey, config: &TuktukConfigV0) -> Result<Vec<Pubkey>, Error> {
         let queue_ids = 0..config.next_task_queue_id;
         let queue_keys = queue_ids.map(|id| self::key(config_key, id)).collect_vec();