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
use solana_program::{
account_info::{next_account_info, AccountInfo},
entrypoint::ProgramResult,
pubkey::Pubkey,
};
use crate::{
assertions::{assert_owned_by, collection::assert_has_collection_authority},
error::MetadataError,
instruction::SetCollectionSizeArgs,
state::{CollectionDetails, Metadata, TokenMetadataAccount},
utils::clean_write_metadata,
};
pub fn set_collection_size(
program_id: &Pubkey,
accounts: &[AccountInfo],
args: SetCollectionSizeArgs,
) -> ProgramResult {
let size = args.size;
let account_info_iter = &mut accounts.iter();
let parent_nft_metadata_account_info = next_account_info(account_info_iter)?;
let collection_update_authority_account_info = next_account_info(account_info_iter)?;
let collection_mint_account_info = next_account_info(account_info_iter)?;
let using_delegated_collection_authority = accounts.len() == 4;
assert_owned_by(parent_nft_metadata_account_info, program_id)?;
assert_owned_by(collection_mint_account_info, &spl_token::id())?;
let mut metadata = Metadata::from_account_info(parent_nft_metadata_account_info)?;
if !collection_update_authority_account_info.is_signer {
return Err(MetadataError::UpdateAuthorityIsNotSigner.into());
}
if using_delegated_collection_authority {
let collection_authority_record = next_account_info(account_info_iter)?;
assert_has_collection_authority(
collection_update_authority_account_info,
&metadata,
collection_mint_account_info.key,
Some(collection_authority_record),
)?;
} else {
assert_has_collection_authority(
collection_update_authority_account_info,
&metadata,
collection_mint_account_info.key,
None,
)?;
}
if metadata.collection_details.is_some() {
return Err(MetadataError::SizedCollection.into());
} else {
metadata.collection_details = Some(CollectionDetails::V1 { size });
}
clean_write_metadata(&mut metadata, parent_nft_metadata_account_info)?;
Ok(())
}