hayabusa-token 0.1.0

Hayabusa Token program interface
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

// Copyright (c) 2025, Arcane Labs <dev@arcane.fi>
// SPDX-License-Identifier: Apache-2.0

use crate::{write_bytes, UNINIT_BYTE};
use core::slice::from_raw_parts;
use hayabusa_cpi::{CpiCtx, CheckProgramId};
use hayabusa_errors::Result;
use pinocchio::{account_info::AccountInfo, cpi::{invoke, invoke_signed}, instruction::{AccountMeta, Instruction}, pubkey::Pubkey};

#[repr(u8)]
#[derive(Clone, Copy)]
pub enum AuthorityType {
    MintTokens = 0,
    FreezeAccount = 1,
    AccountOwner = 2,
    CloseAccount = 3,
}

pub struct SetAuthority<'a> {
    /// Account (Mint or Token)
    pub account: &'a AccountInfo,
    /// Authority of the account
    pub authority: &'a AccountInfo,
}

impl CheckProgramId for SetAuthority<'_> {
    const ID: Pubkey = crate::ID;
}

const DISCRIMINATOR: [u8; 1] = [6];

#[inline(always)]
pub fn set_authority<'a>(
    cpi_ctx: CpiCtx<'a, '_, '_, '_, SetAuthority<'a>>,
    authority_type: AuthorityType,
    new_authority: Option<&'a Pubkey>,
) -> Result<()> {
    let infos = [cpi_ctx.account, cpi_ctx.authority];
    let metas = [
        AccountMeta::writable(cpi_ctx.account.key()),
        AccountMeta::readonly_signer(cpi_ctx.authority.key()),
    ];

    // ix data layout
    // - [0]: discriminator
    // - [1]: authority_type
    // - [2]: new_authority presence flag
    // - [3..35]: new_authority pk
    let mut ix_data = [UNINIT_BYTE; 35];
    let mut length = ix_data.len();

    write_bytes(&mut ix_data, &DISCRIMINATOR);
    write_bytes(&mut ix_data[1..2], &[authority_type as u8]);

    if let Some(new_authority) = new_authority {
        write_bytes(&mut ix_data[2..3], &[1]);
        write_bytes(&mut ix_data[3..], new_authority);
    } else {
        write_bytes(&mut ix_data[2..3], &[0]);

        length = 3;
    }

    let instruction = Instruction {
        program_id: &crate::ID,
        accounts: &metas,
        data: unsafe { from_raw_parts(ix_data.as_ptr() as _, length) },
    };

    if let Some(signers) = cpi_ctx.signers {
        invoke_signed(&instruction, &infos, signers)
    } else {
        invoke(&instruction, &infos)
    }
}