constraints.rs 64 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673
  1. use crate::*;
  2. use syn::parse::{Error as ParseError, Result as ParseResult};
  3. use syn::{bracketed, Token};
  4. pub fn parse(f: &syn::Field, f_ty: Option<&Ty>) -> ParseResult<ConstraintGroup> {
  5. let mut constraints = ConstraintGroupBuilder::new(f_ty);
  6. for attr in f.attrs.iter().filter(is_account) {
  7. for c in attr.parse_args_with(Punctuated::<ConstraintToken, Comma>::parse_terminated)? {
  8. constraints.add(c)?;
  9. }
  10. }
  11. let account_constraints = constraints.build()?;
  12. Ok(account_constraints)
  13. }
  14. pub fn is_account(attr: &&syn::Attribute) -> bool {
  15. attr.path
  16. .get_ident()
  17. .map_or(false, |ident| ident == "account")
  18. }
  19. // Parses a single constraint from a parse stream for `#[account(<STREAM>)]`.
  20. pub fn parse_token(stream: ParseStream) -> ParseResult<ConstraintToken> {
  21. let ident = stream.call(Ident::parse_any)?;
  22. let kw = ident.to_string();
  23. let c = match kw.as_str() {
  24. "init" => ConstraintToken::Init(Context::new(
  25. ident.span(),
  26. ConstraintInit { if_needed: false },
  27. )),
  28. "init_if_needed" => ConstraintToken::Init(Context::new(
  29. ident.span(),
  30. ConstraintInit { if_needed: true },
  31. )),
  32. "zero" => ConstraintToken::Zeroed(Context::new(ident.span(), ConstraintZeroed {})),
  33. "mut" => ConstraintToken::Mut(Context::new(
  34. ident.span(),
  35. ConstraintMut {
  36. error: parse_optional_custom_error(&stream)?,
  37. },
  38. )),
  39. "signer" => ConstraintToken::Signer(Context::new(
  40. ident.span(),
  41. ConstraintSigner {
  42. error: parse_optional_custom_error(&stream)?,
  43. },
  44. )),
  45. "executable" => {
  46. ConstraintToken::Executable(Context::new(ident.span(), ConstraintExecutable {}))
  47. }
  48. "mint" => {
  49. stream.parse::<Token![:]>()?;
  50. stream.parse::<Token![:]>()?;
  51. let kw = stream.call(Ident::parse_any)?.to_string();
  52. stream.parse::<Token![=]>()?;
  53. let span = ident
  54. .span()
  55. .join(stream.span())
  56. .unwrap_or_else(|| ident.span());
  57. match kw.as_str() {
  58. "authority" => ConstraintToken::MintAuthority(Context::new(
  59. span,
  60. ConstraintMintAuthority {
  61. mint_auth: stream.parse()?,
  62. },
  63. )),
  64. "freeze_authority" => ConstraintToken::MintFreezeAuthority(Context::new(
  65. span,
  66. ConstraintMintFreezeAuthority {
  67. mint_freeze_auth: stream.parse()?,
  68. },
  69. )),
  70. "decimals" => ConstraintToken::MintDecimals(Context::new(
  71. span,
  72. ConstraintMintDecimals {
  73. decimals: stream.parse()?,
  74. },
  75. )),
  76. "token_program" => ConstraintToken::MintTokenProgram(Context::new(
  77. span,
  78. ConstraintTokenProgram {
  79. token_program: stream.parse()?,
  80. },
  81. )),
  82. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  83. }
  84. }
  85. "extensions" => {
  86. stream.parse::<Token![:]>()?;
  87. stream.parse::<Token![:]>()?;
  88. let kw = stream.call(Ident::parse_any)?.to_string();
  89. match kw.as_str() {
  90. "group_pointer" => {
  91. stream.parse::<Token![:]>()?;
  92. stream.parse::<Token![:]>()?;
  93. let kw = stream.call(Ident::parse_any)?.to_string();
  94. stream.parse::<Token![=]>()?;
  95. let span = ident
  96. .span()
  97. .join(stream.span())
  98. .unwrap_or_else(|| ident.span());
  99. match kw.as_str() {
  100. "authority" => {
  101. ConstraintToken::ExtensionGroupPointerAuthority(Context::new(
  102. span,
  103. ConstraintExtensionAuthority {
  104. authority: stream.parse()?,
  105. },
  106. ))
  107. }
  108. "group_address" => {
  109. ConstraintToken::ExtensionGroupPointerGroupAddress(Context::new(
  110. span,
  111. ConstraintExtensionGroupPointerGroupAddress {
  112. group_address: stream.parse()?,
  113. },
  114. ))
  115. }
  116. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  117. }
  118. }
  119. "group_member_pointer" => {
  120. stream.parse::<Token![:]>()?;
  121. stream.parse::<Token![:]>()?;
  122. let kw = stream.call(Ident::parse_any)?.to_string();
  123. stream.parse::<Token![=]>()?;
  124. let span = ident
  125. .span()
  126. .join(stream.span())
  127. .unwrap_or_else(|| ident.span());
  128. match kw.as_str() {
  129. "authority" => {
  130. ConstraintToken::ExtensionGroupMemberPointerAuthority(Context::new(
  131. span,
  132. ConstraintExtensionAuthority {
  133. authority: stream.parse()?,
  134. },
  135. ))
  136. }
  137. "member_address" => {
  138. ConstraintToken::ExtensionGroupMemberPointerMemberAddress(Context::new(
  139. span,
  140. ConstraintExtensionGroupMemberPointerMemberAddress {
  141. member_address: stream.parse()?,
  142. },
  143. ))
  144. }
  145. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  146. }
  147. }
  148. "metadata_pointer" => {
  149. stream.parse::<Token![:]>()?;
  150. stream.parse::<Token![:]>()?;
  151. let kw = stream.call(Ident::parse_any)?.to_string();
  152. stream.parse::<Token![=]>()?;
  153. let span = ident
  154. .span()
  155. .join(stream.span())
  156. .unwrap_or_else(|| ident.span());
  157. match kw.as_str() {
  158. "authority" => {
  159. ConstraintToken::ExtensionMetadataPointerAuthority(Context::new(
  160. span,
  161. ConstraintExtensionAuthority {
  162. authority: stream.parse()?,
  163. },
  164. ))
  165. }
  166. "metadata_address" => {
  167. ConstraintToken::ExtensionMetadataPointerMetadataAddress(Context::new(
  168. span,
  169. ConstraintExtensionMetadataPointerMetadataAddress {
  170. metadata_address: stream.parse()?,
  171. },
  172. ))
  173. }
  174. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  175. }
  176. }
  177. "close_authority" => {
  178. stream.parse::<Token![:]>()?;
  179. stream.parse::<Token![:]>()?;
  180. let kw = stream.call(Ident::parse_any)?.to_string();
  181. stream.parse::<Token![=]>()?;
  182. let span = ident
  183. .span()
  184. .join(stream.span())
  185. .unwrap_or_else(|| ident.span());
  186. match kw.as_str() {
  187. "authority" => ConstraintToken::ExtensionCloseAuthority(Context::new(
  188. span,
  189. ConstraintExtensionAuthority {
  190. authority: stream.parse()?,
  191. },
  192. )),
  193. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  194. }
  195. }
  196. "permanent_delegate" => {
  197. stream.parse::<Token![:]>()?;
  198. stream.parse::<Token![:]>()?;
  199. let kw = stream.call(Ident::parse_any)?.to_string();
  200. stream.parse::<Token![=]>()?;
  201. let span = ident
  202. .span()
  203. .join(stream.span())
  204. .unwrap_or_else(|| ident.span());
  205. match kw.as_str() {
  206. "delegate" => ConstraintToken::ExtensionPermanentDelegate(Context::new(
  207. span,
  208. ConstraintExtensionPermanentDelegate {
  209. permanent_delegate: stream.parse()?,
  210. },
  211. )),
  212. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  213. }
  214. }
  215. "transfer_hook" => {
  216. stream.parse::<Token![:]>()?;
  217. stream.parse::<Token![:]>()?;
  218. let kw = stream.call(Ident::parse_any)?.to_string();
  219. stream.parse::<Token![=]>()?;
  220. let span = ident
  221. .span()
  222. .join(stream.span())
  223. .unwrap_or_else(|| ident.span());
  224. match kw.as_str() {
  225. "authority" => ConstraintToken::ExtensionTokenHookAuthority(Context::new(
  226. span,
  227. ConstraintExtensionAuthority {
  228. authority: stream.parse()?,
  229. },
  230. )),
  231. "program_id" => ConstraintToken::ExtensionTokenHookProgramId(Context::new(
  232. span,
  233. ConstraintExtensionTokenHookProgramId {
  234. program_id: stream.parse()?,
  235. },
  236. )),
  237. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  238. }
  239. }
  240. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  241. }
  242. }
  243. "token" => {
  244. stream.parse::<Token![:]>()?;
  245. stream.parse::<Token![:]>()?;
  246. let kw = stream.call(Ident::parse_any)?.to_string();
  247. stream.parse::<Token![=]>()?;
  248. let span = ident
  249. .span()
  250. .join(stream.span())
  251. .unwrap_or_else(|| ident.span());
  252. match kw.as_str() {
  253. "mint" => ConstraintToken::TokenMint(Context::new(
  254. span,
  255. ConstraintTokenMint {
  256. mint: stream.parse()?,
  257. },
  258. )),
  259. "authority" => ConstraintToken::TokenAuthority(Context::new(
  260. span,
  261. ConstraintTokenAuthority {
  262. auth: stream.parse()?,
  263. },
  264. )),
  265. "token_program" => ConstraintToken::TokenTokenProgram(Context::new(
  266. span,
  267. ConstraintTokenProgram {
  268. token_program: stream.parse()?,
  269. },
  270. )),
  271. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  272. }
  273. }
  274. "associated_token" => {
  275. stream.parse::<Token![:]>()?;
  276. stream.parse::<Token![:]>()?;
  277. let kw = stream.call(Ident::parse_any)?.to_string();
  278. stream.parse::<Token![=]>()?;
  279. let span = ident
  280. .span()
  281. .join(stream.span())
  282. .unwrap_or_else(|| ident.span());
  283. match kw.as_str() {
  284. "mint" => ConstraintToken::AssociatedTokenMint(Context::new(
  285. span,
  286. ConstraintTokenMint {
  287. mint: stream.parse()?,
  288. },
  289. )),
  290. "authority" => ConstraintToken::AssociatedTokenAuthority(Context::new(
  291. span,
  292. ConstraintTokenAuthority {
  293. auth: stream.parse()?,
  294. },
  295. )),
  296. "token_program" => ConstraintToken::AssociatedTokenTokenProgram(Context::new(
  297. span,
  298. ConstraintTokenProgram {
  299. token_program: stream.parse()?,
  300. },
  301. )),
  302. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  303. }
  304. }
  305. "bump" => {
  306. let bump = {
  307. if stream.peek(Token![=]) {
  308. stream.parse::<Token![=]>()?;
  309. Some(stream.parse()?)
  310. } else {
  311. None
  312. }
  313. };
  314. ConstraintToken::Bump(Context::new(ident.span(), ConstraintTokenBump { bump }))
  315. }
  316. "seeds" => {
  317. if stream.peek(Token![:]) {
  318. stream.parse::<Token![:]>()?;
  319. stream.parse::<Token![:]>()?;
  320. let kw = stream.call(Ident::parse_any)?.to_string();
  321. stream.parse::<Token![=]>()?;
  322. let span = ident
  323. .span()
  324. .join(stream.span())
  325. .unwrap_or_else(|| ident.span());
  326. match kw.as_str() {
  327. "program" => ConstraintToken::ProgramSeed(Context::new(
  328. span,
  329. ConstraintProgramSeed {
  330. program_seed: stream.parse()?,
  331. },
  332. )),
  333. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  334. }
  335. } else {
  336. stream.parse::<Token![=]>()?;
  337. let span = ident
  338. .span()
  339. .join(stream.span())
  340. .unwrap_or_else(|| ident.span());
  341. let seeds;
  342. let bracket = bracketed!(seeds in stream);
  343. ConstraintToken::Seeds(Context::new(
  344. span.join(bracket.span).unwrap_or(span),
  345. ConstraintSeeds {
  346. seeds: seeds.parse_terminated(Expr::parse)?,
  347. },
  348. ))
  349. }
  350. }
  351. "realloc" => {
  352. if stream.peek(Token![=]) {
  353. stream.parse::<Token![=]>()?;
  354. let span = ident
  355. .span()
  356. .join(stream.span())
  357. .unwrap_or_else(|| ident.span());
  358. ConstraintToken::Realloc(Context::new(
  359. span,
  360. ConstraintRealloc {
  361. space: stream.parse()?,
  362. },
  363. ))
  364. } else {
  365. stream.parse::<Token![:]>()?;
  366. stream.parse::<Token![:]>()?;
  367. let kw = stream.call(Ident::parse_any)?.to_string();
  368. stream.parse::<Token![=]>()?;
  369. let span = ident
  370. .span()
  371. .join(stream.span())
  372. .unwrap_or_else(|| ident.span());
  373. match kw.as_str() {
  374. "payer" => ConstraintToken::ReallocPayer(Context::new(
  375. span,
  376. ConstraintReallocPayer {
  377. target: stream.parse()?,
  378. },
  379. )),
  380. "zero" => ConstraintToken::ReallocZero(Context::new(
  381. span,
  382. ConstraintReallocZero {
  383. zero: stream.parse()?,
  384. },
  385. )),
  386. _ => return Err(ParseError::new(ident.span(), "Invalid attribute. realloc::payer and realloc::zero are the only valid attributes")),
  387. }
  388. }
  389. }
  390. _ => {
  391. stream.parse::<Token![=]>()?;
  392. let span = ident
  393. .span()
  394. .join(stream.span())
  395. .unwrap_or_else(|| ident.span());
  396. match kw.as_str() {
  397. "has_one" => ConstraintToken::HasOne(Context::new(
  398. span,
  399. ConstraintHasOne {
  400. join_target: stream.parse()?,
  401. error: parse_optional_custom_error(&stream)?,
  402. },
  403. )),
  404. "owner" => ConstraintToken::Owner(Context::new(
  405. span,
  406. ConstraintOwner {
  407. owner_address: stream.parse()?,
  408. error: parse_optional_custom_error(&stream)?,
  409. },
  410. )),
  411. "rent_exempt" => ConstraintToken::RentExempt(Context::new(
  412. span,
  413. match stream.parse::<Ident>()?.to_string().as_str() {
  414. "skip" => ConstraintRentExempt::Skip,
  415. "enforce" => ConstraintRentExempt::Enforce,
  416. _ => {
  417. return Err(ParseError::new(
  418. span,
  419. "rent_exempt must be either skip or enforce",
  420. ))
  421. }
  422. },
  423. )),
  424. "payer" => ConstraintToken::Payer(Context::new(
  425. span,
  426. ConstraintPayer {
  427. target: stream.parse()?,
  428. },
  429. )),
  430. "space" => ConstraintToken::Space(Context::new(
  431. span,
  432. ConstraintSpace {
  433. space: stream.parse()?,
  434. },
  435. )),
  436. "constraint" => ConstraintToken::Raw(Context::new(
  437. span,
  438. ConstraintRaw {
  439. raw: stream.parse()?,
  440. error: parse_optional_custom_error(&stream)?,
  441. },
  442. )),
  443. "close" => ConstraintToken::Close(Context::new(
  444. span,
  445. ConstraintClose {
  446. sol_dest: stream.parse()?,
  447. },
  448. )),
  449. "address" => ConstraintToken::Address(Context::new(
  450. span,
  451. ConstraintAddress {
  452. address: stream.parse()?,
  453. error: parse_optional_custom_error(&stream)?,
  454. },
  455. )),
  456. _ => return Err(ParseError::new(ident.span(), "Invalid attribute")),
  457. }
  458. }
  459. };
  460. Ok(c)
  461. }
  462. fn parse_optional_custom_error(stream: &ParseStream) -> ParseResult<Option<Expr>> {
  463. if stream.peek(Token![@]) {
  464. stream.parse::<Token![@]>()?;
  465. stream.parse().map(Some)
  466. } else {
  467. Ok(None)
  468. }
  469. }
  470. #[derive(Default)]
  471. pub struct ConstraintGroupBuilder<'ty> {
  472. pub f_ty: Option<&'ty Ty>,
  473. pub init: Option<Context<ConstraintInit>>,
  474. pub zeroed: Option<Context<ConstraintZeroed>>,
  475. pub mutable: Option<Context<ConstraintMut>>,
  476. pub signer: Option<Context<ConstraintSigner>>,
  477. pub has_one: Vec<Context<ConstraintHasOne>>,
  478. pub raw: Vec<Context<ConstraintRaw>>,
  479. pub owner: Option<Context<ConstraintOwner>>,
  480. pub rent_exempt: Option<Context<ConstraintRentExempt>>,
  481. pub seeds: Option<Context<ConstraintSeeds>>,
  482. pub executable: Option<Context<ConstraintExecutable>>,
  483. pub payer: Option<Context<ConstraintPayer>>,
  484. pub space: Option<Context<ConstraintSpace>>,
  485. pub close: Option<Context<ConstraintClose>>,
  486. pub address: Option<Context<ConstraintAddress>>,
  487. pub token_mint: Option<Context<ConstraintTokenMint>>,
  488. pub token_authority: Option<Context<ConstraintTokenAuthority>>,
  489. pub token_token_program: Option<Context<ConstraintTokenProgram>>,
  490. pub associated_token_mint: Option<Context<ConstraintTokenMint>>,
  491. pub associated_token_authority: Option<Context<ConstraintTokenAuthority>>,
  492. pub associated_token_token_program: Option<Context<ConstraintTokenProgram>>,
  493. pub mint_authority: Option<Context<ConstraintMintAuthority>>,
  494. pub mint_freeze_authority: Option<Context<ConstraintMintFreezeAuthority>>,
  495. pub mint_decimals: Option<Context<ConstraintMintDecimals>>,
  496. pub mint_token_program: Option<Context<ConstraintTokenProgram>>,
  497. pub extension_group_pointer_authority: Option<Context<ConstraintExtensionAuthority>>,
  498. pub extension_group_pointer_group_address:
  499. Option<Context<ConstraintExtensionGroupPointerGroupAddress>>,
  500. pub extension_group_member_pointer_authority: Option<Context<ConstraintExtensionAuthority>>,
  501. pub extension_group_member_pointer_member_address:
  502. Option<Context<ConstraintExtensionGroupMemberPointerMemberAddress>>,
  503. pub extension_metadata_pointer_authority: Option<Context<ConstraintExtensionAuthority>>,
  504. pub extension_metadata_pointer_metadata_address:
  505. Option<Context<ConstraintExtensionMetadataPointerMetadataAddress>>,
  506. pub extension_close_authority: Option<Context<ConstraintExtensionAuthority>>,
  507. pub extension_transfer_hook_authority: Option<Context<ConstraintExtensionAuthority>>,
  508. pub extension_transfer_hook_program_id: Option<Context<ConstraintExtensionTokenHookProgramId>>,
  509. pub extension_permanent_delegate: Option<Context<ConstraintExtensionPermanentDelegate>>,
  510. pub bump: Option<Context<ConstraintTokenBump>>,
  511. pub program_seed: Option<Context<ConstraintProgramSeed>>,
  512. pub realloc: Option<Context<ConstraintRealloc>>,
  513. pub realloc_payer: Option<Context<ConstraintReallocPayer>>,
  514. pub realloc_zero: Option<Context<ConstraintReallocZero>>,
  515. }
  516. impl<'ty> ConstraintGroupBuilder<'ty> {
  517. pub fn new(f_ty: Option<&'ty Ty>) -> Self {
  518. Self {
  519. f_ty,
  520. init: None,
  521. zeroed: None,
  522. mutable: None,
  523. signer: None,
  524. has_one: Vec::new(),
  525. raw: Vec::new(),
  526. owner: None,
  527. rent_exempt: None,
  528. seeds: None,
  529. executable: None,
  530. payer: None,
  531. space: None,
  532. close: None,
  533. address: None,
  534. token_mint: None,
  535. token_authority: None,
  536. token_token_program: None,
  537. associated_token_mint: None,
  538. associated_token_authority: None,
  539. associated_token_token_program: None,
  540. mint_authority: None,
  541. mint_freeze_authority: None,
  542. mint_decimals: None,
  543. mint_token_program: None,
  544. extension_group_pointer_authority: None,
  545. extension_group_pointer_group_address: None,
  546. extension_group_member_pointer_authority: None,
  547. extension_group_member_pointer_member_address: None,
  548. extension_metadata_pointer_authority: None,
  549. extension_metadata_pointer_metadata_address: None,
  550. extension_close_authority: None,
  551. extension_transfer_hook_authority: None,
  552. extension_transfer_hook_program_id: None,
  553. extension_permanent_delegate: None,
  554. bump: None,
  555. program_seed: None,
  556. realloc: None,
  557. realloc_payer: None,
  558. realloc_zero: None,
  559. }
  560. }
  561. pub fn build(mut self) -> ParseResult<ConstraintGroup> {
  562. // Init.
  563. if let Some(i) = &self.init {
  564. if cfg!(not(feature = "init-if-needed")) && i.if_needed {
  565. return Err(ParseError::new(
  566. i.span(),
  567. "init_if_needed requires that anchor-lang be imported \
  568. with the init-if-needed cargo feature enabled. \
  569. Carefully read the init_if_needed docs before using this feature \
  570. to make sure you know how to protect yourself against \
  571. re-initialization attacks.",
  572. ));
  573. }
  574. match self.mutable {
  575. Some(m) => {
  576. return Err(ParseError::new(
  577. m.span(),
  578. "mut cannot be provided with init",
  579. ))
  580. }
  581. None => self
  582. .mutable
  583. .replace(Context::new(i.span(), ConstraintMut { error: None })),
  584. };
  585. // Rent exempt if not explicitly skipped.
  586. if self.rent_exempt.is_none() {
  587. self.rent_exempt
  588. .replace(Context::new(i.span(), ConstraintRentExempt::Enforce));
  589. }
  590. if self.payer.is_none() {
  591. return Err(ParseError::new(
  592. i.span(),
  593. "payer must be provided when initializing an account",
  594. ));
  595. }
  596. // When initializing a non-PDA account, the account being
  597. // initialized must sign to invoke the system program's create
  598. // account instruction.
  599. if self.signer.is_none() && self.seeds.is_none() && self.associated_token_mint.is_none()
  600. {
  601. self.signer
  602. .replace(Context::new(i.span(), ConstraintSigner { error: None }));
  603. }
  604. // Assert a bump target is not given on init.
  605. if let Some(b) = &self.bump {
  606. if b.bump.is_some() {
  607. return Err(ParseError::new(
  608. b.span(),
  609. "bump targets should not be provided with init. Please use bump without a target."
  610. ));
  611. }
  612. }
  613. // TokenAccount.
  614. if let Some(token_mint) = &self.token_mint {
  615. if self.token_authority.is_none() {
  616. return Err(ParseError::new(
  617. token_mint.span(),
  618. "when initializing, token authority must be provided if token mint is",
  619. ));
  620. }
  621. }
  622. if let Some(token_authority) = &self.token_authority {
  623. if self.token_mint.is_none() {
  624. return Err(ParseError::new(
  625. token_authority.span(),
  626. "when initializing, token mint must be provided if token authority is",
  627. ));
  628. }
  629. }
  630. // Mint.
  631. if let Some(mint_decimals) = &self.mint_decimals {
  632. if self.mint_authority.is_none() {
  633. return Err(ParseError::new(
  634. mint_decimals.span(),
  635. "when initializing, mint authority must be provided if mint decimals is",
  636. ));
  637. }
  638. }
  639. if let Some(mint_authority) = &self.mint_authority {
  640. if self.mint_decimals.is_none() {
  641. return Err(ParseError::new(
  642. mint_authority.span(),
  643. "when initializing, mint decimals must be provided if mint authority is",
  644. ));
  645. }
  646. }
  647. }
  648. // Realloc.
  649. if let Some(r) = &self.realloc {
  650. if self.realloc_payer.is_none() {
  651. return Err(ParseError::new(
  652. r.span(),
  653. "realloc::payer must be provided when using realloc",
  654. ));
  655. }
  656. if self.realloc_zero.is_none() {
  657. return Err(ParseError::new(
  658. r.span(),
  659. "realloc::zero must be provided when using realloc",
  660. ));
  661. }
  662. }
  663. // Zero.
  664. if let Some(z) = &self.zeroed {
  665. match self.mutable {
  666. Some(m) => {
  667. return Err(ParseError::new(
  668. m.span(),
  669. "mut cannot be provided with zeroed",
  670. ))
  671. }
  672. None => self
  673. .mutable
  674. .replace(Context::new(z.span(), ConstraintMut { error: None })),
  675. };
  676. // Rent exempt if not explicitly skipped.
  677. if self.rent_exempt.is_none() {
  678. self.rent_exempt
  679. .replace(Context::new(z.span(), ConstraintRentExempt::Enforce));
  680. }
  681. }
  682. // Seeds.
  683. if let Some(i) = &self.seeds {
  684. if self.init.is_some() && self.payer.is_none() {
  685. return Err(ParseError::new(
  686. i.span(),
  687. "payer must be provided when creating a program derived address",
  688. ));
  689. }
  690. if self.bump.is_none() {
  691. return Err(ParseError::new(
  692. i.span(),
  693. "bump must be provided with seeds",
  694. ));
  695. }
  696. }
  697. // Space.
  698. if let Some(i) = &self.init {
  699. let initializing_token_program_acc = self.token_mint.is_some()
  700. || self.mint_authority.is_some()
  701. || self.token_authority.is_some()
  702. || self.associated_token_authority.is_some();
  703. match (self.space.is_some(), initializing_token_program_acc) {
  704. (true, true) => {
  705. return Err(ParseError::new(
  706. self.space.as_ref().unwrap().span(),
  707. "space is not required for initializing an spl account",
  708. ));
  709. }
  710. (false, false) => {
  711. return Err(ParseError::new(
  712. i.span(),
  713. "space must be provided with init",
  714. ));
  715. }
  716. _ => (),
  717. }
  718. }
  719. let ConstraintGroupBuilder {
  720. f_ty: _,
  721. init,
  722. zeroed,
  723. mutable,
  724. signer,
  725. has_one,
  726. raw,
  727. owner,
  728. rent_exempt,
  729. seeds,
  730. executable,
  731. payer,
  732. space,
  733. close,
  734. address,
  735. token_mint,
  736. token_authority,
  737. token_token_program,
  738. associated_token_mint,
  739. associated_token_authority,
  740. associated_token_token_program,
  741. mint_authority,
  742. mint_freeze_authority,
  743. mint_decimals,
  744. mint_token_program,
  745. extension_group_pointer_authority,
  746. extension_group_pointer_group_address,
  747. extension_group_member_pointer_authority,
  748. extension_group_member_pointer_member_address,
  749. extension_metadata_pointer_authority,
  750. extension_metadata_pointer_metadata_address,
  751. extension_close_authority,
  752. extension_transfer_hook_authority,
  753. extension_transfer_hook_program_id,
  754. extension_permanent_delegate,
  755. bump,
  756. program_seed,
  757. realloc,
  758. realloc_payer,
  759. realloc_zero,
  760. } = self;
  761. // Converts Option<Context<T>> -> Option<T>.
  762. macro_rules! into_inner {
  763. ($opt:ident) => {
  764. $opt.map(|c| c.into_inner())
  765. };
  766. ($opt:expr) => {
  767. $opt.map(|c| c.into_inner())
  768. };
  769. }
  770. // Converts Vec<Context<T>> - Vec<T>.
  771. macro_rules! into_inner_vec {
  772. ($opt:ident) => {
  773. $opt.into_iter().map(|c| c.into_inner()).collect()
  774. };
  775. }
  776. let is_init = init.is_some();
  777. let seeds = seeds.map(|c| ConstraintSeedsGroup {
  778. is_init,
  779. seeds: c.seeds.clone(),
  780. bump: into_inner!(bump)
  781. .map(|b| b.bump)
  782. .expect("bump must be provided with seeds"),
  783. program_seed: into_inner!(program_seed).map(|id| id.program_seed),
  784. });
  785. let associated_token = match (
  786. associated_token_mint,
  787. associated_token_authority,
  788. &associated_token_token_program,
  789. ) {
  790. (Some(mint), Some(auth), _) => Some(ConstraintAssociatedToken {
  791. wallet: auth.into_inner().auth,
  792. mint: mint.into_inner().mint,
  793. token_program: associated_token_token_program
  794. .as_ref()
  795. .map(|a| a.clone().into_inner().token_program),
  796. }),
  797. (Some(mint), None, _) => return Err(ParseError::new(
  798. mint.span(),
  799. "authority must be provided to specify an associated token program derived address",
  800. )),
  801. (None, Some(auth), _) => {
  802. return Err(ParseError::new(
  803. auth.span(),
  804. "mint must be provided to specify an associated token program derived address",
  805. ))
  806. },
  807. (None, None, Some(token_program)) => {
  808. return Err(ParseError::new(
  809. token_program.span(),
  810. "mint and authority must be provided to specify an associated token program derived address",
  811. ))
  812. }
  813. _ => None,
  814. };
  815. if let Some(associated_token) = &associated_token {
  816. if seeds.is_some() {
  817. return Err(ParseError::new(
  818. associated_token.mint.span(),
  819. "'associated_token' constraints cannot be used with the 'seeds' constraint",
  820. ));
  821. }
  822. }
  823. let token_account = match (&token_mint, &token_authority, &token_token_program) {
  824. (None, None, None) => None,
  825. _ => Some(ConstraintTokenAccountGroup {
  826. mint: token_mint.as_ref().map(|a| a.clone().into_inner().mint),
  827. authority: token_authority
  828. .as_ref()
  829. .map(|a| a.clone().into_inner().auth),
  830. token_program: token_token_program
  831. .as_ref()
  832. .map(|a| a.clone().into_inner().token_program),
  833. }),
  834. };
  835. let mint = match (
  836. &mint_decimals,
  837. &mint_authority,
  838. &mint_freeze_authority,
  839. &mint_token_program,
  840. &extension_group_pointer_authority,
  841. &extension_group_pointer_group_address,
  842. &extension_group_member_pointer_authority,
  843. &extension_group_member_pointer_member_address,
  844. &extension_metadata_pointer_authority,
  845. &extension_metadata_pointer_metadata_address,
  846. &extension_close_authority,
  847. &extension_transfer_hook_authority,
  848. &extension_transfer_hook_program_id,
  849. &extension_permanent_delegate,
  850. ) {
  851. (
  852. None,
  853. None,
  854. None,
  855. None,
  856. None,
  857. None,
  858. None,
  859. None,
  860. None,
  861. None,
  862. None,
  863. None,
  864. None,
  865. None,
  866. ) => None,
  867. _ => Some(ConstraintTokenMintGroup {
  868. decimals: mint_decimals
  869. .as_ref()
  870. .map(|a| a.clone().into_inner().decimals),
  871. mint_authority: mint_authority
  872. .as_ref()
  873. .map(|a| a.clone().into_inner().mint_auth),
  874. freeze_authority: mint_freeze_authority
  875. .as_ref()
  876. .map(|a| a.clone().into_inner().mint_freeze_auth),
  877. token_program: mint_token_program
  878. .as_ref()
  879. .map(|a| a.clone().into_inner().token_program),
  880. // extensions
  881. group_pointer_authority: extension_group_pointer_authority
  882. .as_ref()
  883. .map(|a| a.clone().into_inner().authority),
  884. group_pointer_group_address: extension_group_pointer_group_address
  885. .as_ref()
  886. .map(|a| a.clone().into_inner().group_address),
  887. group_member_pointer_authority: extension_group_member_pointer_authority
  888. .as_ref()
  889. .map(|a| a.clone().into_inner().authority),
  890. group_member_pointer_member_address: extension_group_member_pointer_member_address
  891. .as_ref()
  892. .map(|a| a.clone().into_inner().member_address),
  893. metadata_pointer_authority: extension_metadata_pointer_authority
  894. .as_ref()
  895. .map(|a| a.clone().into_inner().authority),
  896. metadata_pointer_metadata_address: extension_metadata_pointer_metadata_address
  897. .as_ref()
  898. .map(|a| a.clone().into_inner().metadata_address),
  899. close_authority: extension_close_authority
  900. .as_ref()
  901. .map(|a| a.clone().into_inner().authority),
  902. permanent_delegate: extension_permanent_delegate
  903. .as_ref()
  904. .map(|a| a.clone().into_inner().permanent_delegate),
  905. transfer_hook_authority: extension_transfer_hook_authority
  906. .as_ref()
  907. .map(|a| a.clone().into_inner().authority),
  908. transfer_hook_program_id: extension_transfer_hook_program_id
  909. .as_ref()
  910. .map(|a| a.clone().into_inner().program_id),
  911. }),
  912. };
  913. Ok(ConstraintGroup {
  914. init: init.as_ref().map(|i| Ok(ConstraintInitGroup {
  915. if_needed: i.if_needed,
  916. seeds: seeds.clone(),
  917. payer: into_inner!(payer.clone()).unwrap().target,
  918. space: space.clone().map(|s| s.space.clone()),
  919. kind: if let Some(tm) = &token_mint {
  920. InitKind::Token {
  921. mint: tm.clone().into_inner().mint,
  922. owner: match &token_authority {
  923. Some(a) => a.clone().into_inner().auth,
  924. None => return Err(ParseError::new(
  925. tm.span(),
  926. "authority must be provided to initialize a token program derived address"
  927. )),
  928. },
  929. token_program: token_token_program.map(|tp| tp.into_inner().token_program),
  930. }
  931. } else if let Some(at) = &associated_token {
  932. InitKind::AssociatedToken {
  933. mint: at.mint.clone(),
  934. owner: at.wallet.clone(),
  935. token_program: associated_token_token_program.map(|tp| tp.into_inner().token_program),
  936. }
  937. } else if let Some(d) = &mint_decimals {
  938. InitKind::Mint {
  939. decimals: d.clone().into_inner().decimals,
  940. owner: match &mint_authority {
  941. Some(a) => a.clone().into_inner().mint_auth,
  942. None => return Err(ParseError::new(
  943. d.span(),
  944. "authority must be provided to initialize a mint program derived address"
  945. ))
  946. },
  947. freeze_authority: mint_freeze_authority.map(|fa| fa.into_inner().mint_freeze_auth),
  948. token_program: mint_token_program.map(|tp| tp.into_inner().token_program),
  949. // extensions
  950. group_pointer_authority: extension_group_pointer_authority.map(|gpa| gpa.into_inner().authority),
  951. group_pointer_group_address: extension_group_pointer_group_address.map(|gpga| gpga.into_inner().group_address),
  952. group_member_pointer_authority: extension_group_member_pointer_authority.map(|gmpa| gmpa.into_inner().authority),
  953. group_member_pointer_member_address: extension_group_member_pointer_member_address.map(|gmpma| gmpma.into_inner().member_address),
  954. metadata_pointer_authority: extension_metadata_pointer_authority.map(|mpa| mpa.into_inner().authority),
  955. metadata_pointer_metadata_address: extension_metadata_pointer_metadata_address.map(|mpma| mpma.into_inner().metadata_address),
  956. close_authority: extension_close_authority.map(|ca| ca.into_inner().authority),
  957. permanent_delegate: extension_permanent_delegate.map(|pd| pd.into_inner().permanent_delegate),
  958. transfer_hook_authority: extension_transfer_hook_authority.map(|tha| tha.into_inner().authority),
  959. transfer_hook_program_id: extension_transfer_hook_program_id.map(|thpid| thpid.into_inner().program_id),
  960. }
  961. } else {
  962. InitKind::Program {
  963. owner: owner.as_ref().map(|o| o.owner_address.clone()),
  964. }
  965. },
  966. })).transpose()?,
  967. realloc: realloc.as_ref().map(|r| ConstraintReallocGroup {
  968. payer: into_inner!(realloc_payer).unwrap().target,
  969. space: r.space.clone(),
  970. zero: into_inner!(realloc_zero).unwrap().zero,
  971. }),
  972. zeroed: into_inner!(zeroed),
  973. mutable: into_inner!(mutable),
  974. signer: into_inner!(signer),
  975. has_one: into_inner_vec!(has_one),
  976. raw: into_inner_vec!(raw),
  977. owner: into_inner!(owner),
  978. rent_exempt: into_inner!(rent_exempt),
  979. executable: into_inner!(executable),
  980. close: into_inner!(close),
  981. address: into_inner!(address),
  982. associated_token: if !is_init { associated_token } else { None },
  983. seeds,
  984. token_account: if !is_init {token_account} else {None},
  985. mint: if !is_init {mint} else {None},
  986. })
  987. }
  988. pub fn add(&mut self, c: ConstraintToken) -> ParseResult<()> {
  989. match c {
  990. ConstraintToken::Init(c) => self.add_init(c),
  991. ConstraintToken::Zeroed(c) => self.add_zeroed(c),
  992. ConstraintToken::Mut(c) => self.add_mut(c),
  993. ConstraintToken::Signer(c) => self.add_signer(c),
  994. ConstraintToken::HasOne(c) => self.add_has_one(c),
  995. ConstraintToken::Raw(c) => self.add_raw(c),
  996. ConstraintToken::Owner(c) => self.add_owner(c),
  997. ConstraintToken::RentExempt(c) => self.add_rent_exempt(c),
  998. ConstraintToken::Seeds(c) => self.add_seeds(c),
  999. ConstraintToken::Executable(c) => self.add_executable(c),
  1000. ConstraintToken::Payer(c) => self.add_payer(c),
  1001. ConstraintToken::Space(c) => self.add_space(c),
  1002. ConstraintToken::Close(c) => self.add_close(c),
  1003. ConstraintToken::Address(c) => self.add_address(c),
  1004. ConstraintToken::TokenAuthority(c) => self.add_token_authority(c),
  1005. ConstraintToken::TokenMint(c) => self.add_token_mint(c),
  1006. ConstraintToken::TokenTokenProgram(c) => self.add_token_token_program(c),
  1007. ConstraintToken::AssociatedTokenAuthority(c) => self.add_associated_token_authority(c),
  1008. ConstraintToken::AssociatedTokenMint(c) => self.add_associated_token_mint(c),
  1009. ConstraintToken::AssociatedTokenTokenProgram(c) => {
  1010. self.add_associated_token_token_program(c)
  1011. }
  1012. ConstraintToken::MintAuthority(c) => self.add_mint_authority(c),
  1013. ConstraintToken::MintFreezeAuthority(c) => self.add_mint_freeze_authority(c),
  1014. ConstraintToken::MintDecimals(c) => self.add_mint_decimals(c),
  1015. ConstraintToken::MintTokenProgram(c) => self.add_mint_token_program(c),
  1016. ConstraintToken::Bump(c) => self.add_bump(c),
  1017. ConstraintToken::ProgramSeed(c) => self.add_program_seed(c),
  1018. ConstraintToken::Realloc(c) => self.add_realloc(c),
  1019. ConstraintToken::ReallocPayer(c) => self.add_realloc_payer(c),
  1020. ConstraintToken::ReallocZero(c) => self.add_realloc_zero(c),
  1021. ConstraintToken::ExtensionGroupPointerAuthority(c) => {
  1022. self.add_extension_group_pointer_authority(c)
  1023. }
  1024. ConstraintToken::ExtensionGroupPointerGroupAddress(c) => {
  1025. self.add_extension_group_pointer_group_address(c)
  1026. }
  1027. ConstraintToken::ExtensionGroupMemberPointerAuthority(c) => {
  1028. self.add_extension_group_member_pointer_authority(c)
  1029. }
  1030. ConstraintToken::ExtensionGroupMemberPointerMemberAddress(c) => {
  1031. self.add_extension_group_member_pointer_member_address(c)
  1032. }
  1033. ConstraintToken::ExtensionMetadataPointerAuthority(c) => {
  1034. self.add_extension_metadata_pointer_authority(c)
  1035. }
  1036. ConstraintToken::ExtensionMetadataPointerMetadataAddress(c) => {
  1037. self.add_extension_metadata_pointer_metadata_address(c)
  1038. }
  1039. ConstraintToken::ExtensionCloseAuthority(c) => self.add_extension_close_authority(c),
  1040. ConstraintToken::ExtensionTokenHookAuthority(c) => self.add_extension_authority(c),
  1041. ConstraintToken::ExtensionTokenHookProgramId(c) => {
  1042. self.add_extension_transfer_hook_program_id(c)
  1043. }
  1044. ConstraintToken::ExtensionPermanentDelegate(c) => {
  1045. self.add_extension_permanent_delegate(c)
  1046. }
  1047. }
  1048. }
  1049. fn add_init(&mut self, c: Context<ConstraintInit>) -> ParseResult<()> {
  1050. if self.init.is_some() {
  1051. return Err(ParseError::new(c.span(), "init already provided"));
  1052. }
  1053. if self.zeroed.is_some() {
  1054. return Err(ParseError::new(c.span(), "zeroed already provided"));
  1055. }
  1056. if self.token_mint.is_some() {
  1057. return Err(ParseError::new(
  1058. c.span(),
  1059. "init must be provided before token mint",
  1060. ));
  1061. }
  1062. if self.token_authority.is_some() {
  1063. return Err(ParseError::new(
  1064. c.span(),
  1065. "init must be provided before token authority",
  1066. ));
  1067. }
  1068. if self.token_token_program.is_some() {
  1069. return Err(ParseError::new(
  1070. c.span(),
  1071. "init must be provided before token account token program",
  1072. ));
  1073. }
  1074. if self.mint_authority.is_some() {
  1075. return Err(ParseError::new(
  1076. c.span(),
  1077. "init must be provided before mint authority",
  1078. ));
  1079. }
  1080. if self.mint_freeze_authority.is_some() {
  1081. return Err(ParseError::new(
  1082. c.span(),
  1083. "init must be provided before mint freeze authority",
  1084. ));
  1085. }
  1086. if self.mint_decimals.is_some() {
  1087. return Err(ParseError::new(
  1088. c.span(),
  1089. "init must be provided before mint decimals",
  1090. ));
  1091. }
  1092. if self.mint_token_program.is_some() {
  1093. return Err(ParseError::new(
  1094. c.span(),
  1095. "init must be provided before mint token program",
  1096. ));
  1097. }
  1098. if self.associated_token_mint.is_some() {
  1099. return Err(ParseError::new(
  1100. c.span(),
  1101. "init must be provided before associated token mint",
  1102. ));
  1103. }
  1104. if self.associated_token_authority.is_some() {
  1105. return Err(ParseError::new(
  1106. c.span(),
  1107. "init must be provided before associated token authority",
  1108. ));
  1109. }
  1110. if self.associated_token_token_program.is_some() {
  1111. return Err(ParseError::new(
  1112. c.span(),
  1113. "init must be provided before associated token account token program",
  1114. ));
  1115. }
  1116. self.init.replace(c);
  1117. Ok(())
  1118. }
  1119. fn add_zeroed(&mut self, c: Context<ConstraintZeroed>) -> ParseResult<()> {
  1120. if self.zeroed.is_some() {
  1121. return Err(ParseError::new(c.span(), "zeroed already provided"));
  1122. }
  1123. if self.init.is_some() {
  1124. return Err(ParseError::new(c.span(), "init already provided"));
  1125. }
  1126. // Require a known account type that implements the `Discriminator` trait so that we can
  1127. // get the discriminator length dynamically
  1128. if !matches!(&self.f_ty, Some(Ty::Account(_) | Ty::AccountLoader(_))) {
  1129. return Err(ParseError::new(
  1130. c.span(),
  1131. "`zero` constraint requires the type to implement the `Discriminator` trait",
  1132. ));
  1133. }
  1134. self.zeroed.replace(c);
  1135. Ok(())
  1136. }
  1137. fn add_realloc(&mut self, c: Context<ConstraintRealloc>) -> ParseResult<()> {
  1138. if !matches!(self.f_ty, Some(Ty::Account(_)))
  1139. && !matches!(self.f_ty, Some(Ty::AccountLoader(_)))
  1140. {
  1141. return Err(ParseError::new(
  1142. c.span(),
  1143. "realloc must be on an Account or AccountLoader",
  1144. ));
  1145. }
  1146. if self.mutable.is_none() {
  1147. return Err(ParseError::new(
  1148. c.span(),
  1149. "mut must be provided before realloc",
  1150. ));
  1151. }
  1152. if self.realloc.is_some() {
  1153. return Err(ParseError::new(c.span(), "realloc already provided"));
  1154. }
  1155. self.realloc.replace(c);
  1156. Ok(())
  1157. }
  1158. fn add_realloc_payer(&mut self, c: Context<ConstraintReallocPayer>) -> ParseResult<()> {
  1159. if self.realloc.is_none() {
  1160. return Err(ParseError::new(
  1161. c.span(),
  1162. "realloc must be provided before realloc::payer",
  1163. ));
  1164. }
  1165. if self.realloc_payer.is_some() {
  1166. return Err(ParseError::new(c.span(), "realloc::payer already provided"));
  1167. }
  1168. self.realloc_payer.replace(c);
  1169. Ok(())
  1170. }
  1171. fn add_realloc_zero(&mut self, c: Context<ConstraintReallocZero>) -> ParseResult<()> {
  1172. if self.realloc.is_none() {
  1173. return Err(ParseError::new(
  1174. c.span(),
  1175. "realloc must be provided before realloc::zero",
  1176. ));
  1177. }
  1178. if self.realloc_zero.is_some() {
  1179. return Err(ParseError::new(c.span(), "realloc::zero already provided"));
  1180. }
  1181. self.realloc_zero.replace(c);
  1182. Ok(())
  1183. }
  1184. fn add_close(&mut self, c: Context<ConstraintClose>) -> ParseResult<()> {
  1185. if !matches!(self.f_ty, Some(Ty::Account(_)))
  1186. && !matches!(self.f_ty, Some(Ty::AccountLoader(_)))
  1187. {
  1188. return Err(ParseError::new(
  1189. c.span(),
  1190. "close must be on an Account, AccountLoader",
  1191. ));
  1192. }
  1193. if self.mutable.is_none() {
  1194. return Err(ParseError::new(
  1195. c.span(),
  1196. "mut must be provided before close",
  1197. ));
  1198. }
  1199. if self.close.is_some() {
  1200. return Err(ParseError::new(c.span(), "close already provided"));
  1201. }
  1202. self.close.replace(c);
  1203. Ok(())
  1204. }
  1205. fn add_address(&mut self, c: Context<ConstraintAddress>) -> ParseResult<()> {
  1206. if self.address.is_some() {
  1207. return Err(ParseError::new(c.span(), "address already provided"));
  1208. }
  1209. self.address.replace(c);
  1210. Ok(())
  1211. }
  1212. fn add_token_mint(&mut self, c: Context<ConstraintTokenMint>) -> ParseResult<()> {
  1213. if self.token_mint.is_some() {
  1214. return Err(ParseError::new(c.span(), "token mint already provided"));
  1215. }
  1216. if self.associated_token_mint.is_some() {
  1217. return Err(ParseError::new(
  1218. c.span(),
  1219. "associated token mint already provided",
  1220. ));
  1221. }
  1222. self.token_mint.replace(c);
  1223. Ok(())
  1224. }
  1225. fn add_associated_token_mint(&mut self, c: Context<ConstraintTokenMint>) -> ParseResult<()> {
  1226. if self.associated_token_mint.is_some() {
  1227. return Err(ParseError::new(
  1228. c.span(),
  1229. "associated token mint already provided",
  1230. ));
  1231. }
  1232. if self.token_mint.is_some() {
  1233. return Err(ParseError::new(c.span(), "token mint already provided"));
  1234. }
  1235. self.associated_token_mint.replace(c);
  1236. Ok(())
  1237. }
  1238. fn add_bump(&mut self, c: Context<ConstraintTokenBump>) -> ParseResult<()> {
  1239. if self.bump.is_some() {
  1240. return Err(ParseError::new(c.span(), "bump already provided"));
  1241. }
  1242. if self.seeds.is_none() {
  1243. return Err(ParseError::new(
  1244. c.span(),
  1245. "seeds must be provided before bump",
  1246. ));
  1247. }
  1248. self.bump.replace(c);
  1249. Ok(())
  1250. }
  1251. fn add_program_seed(&mut self, c: Context<ConstraintProgramSeed>) -> ParseResult<()> {
  1252. if self.program_seed.is_some() {
  1253. return Err(ParseError::new(c.span(), "seeds::program already provided"));
  1254. }
  1255. if self.seeds.is_none() {
  1256. return Err(ParseError::new(
  1257. c.span(),
  1258. "seeds must be provided before seeds::program",
  1259. ));
  1260. }
  1261. if let Some(ref init) = self.init {
  1262. if init.if_needed {
  1263. return Err(ParseError::new(
  1264. c.span(),
  1265. "seeds::program cannot be used with init_if_needed",
  1266. ));
  1267. } else {
  1268. return Err(ParseError::new(
  1269. c.span(),
  1270. "seeds::program cannot be used with init",
  1271. ));
  1272. }
  1273. }
  1274. self.program_seed.replace(c);
  1275. Ok(())
  1276. }
  1277. fn add_token_authority(&mut self, c: Context<ConstraintTokenAuthority>) -> ParseResult<()> {
  1278. if self.token_authority.is_some() {
  1279. return Err(ParseError::new(
  1280. c.span(),
  1281. "token authority already provided",
  1282. ));
  1283. }
  1284. self.token_authority.replace(c);
  1285. Ok(())
  1286. }
  1287. fn add_associated_token_authority(
  1288. &mut self,
  1289. c: Context<ConstraintTokenAuthority>,
  1290. ) -> ParseResult<()> {
  1291. if self.associated_token_authority.is_some() {
  1292. return Err(ParseError::new(
  1293. c.span(),
  1294. "associated token authority already provided",
  1295. ));
  1296. }
  1297. if self.token_authority.is_some() {
  1298. return Err(ParseError::new(
  1299. c.span(),
  1300. "token authority already provided",
  1301. ));
  1302. }
  1303. self.associated_token_authority.replace(c);
  1304. Ok(())
  1305. }
  1306. fn add_token_token_program(&mut self, c: Context<ConstraintTokenProgram>) -> ParseResult<()> {
  1307. if self.token_token_program.is_some() {
  1308. return Err(ParseError::new(
  1309. c.span(),
  1310. "token token_program already provided",
  1311. ));
  1312. }
  1313. self.token_token_program.replace(c);
  1314. Ok(())
  1315. }
  1316. fn add_associated_token_token_program(
  1317. &mut self,
  1318. c: Context<ConstraintTokenProgram>,
  1319. ) -> ParseResult<()> {
  1320. if self.associated_token_token_program.is_some() {
  1321. return Err(ParseError::new(
  1322. c.span(),
  1323. "associated token token_program already provided",
  1324. ));
  1325. }
  1326. self.associated_token_token_program.replace(c);
  1327. Ok(())
  1328. }
  1329. fn add_mint_authority(&mut self, c: Context<ConstraintMintAuthority>) -> ParseResult<()> {
  1330. if self.mint_authority.is_some() {
  1331. return Err(ParseError::new(c.span(), "mint authority already provided"));
  1332. }
  1333. self.mint_authority.replace(c);
  1334. Ok(())
  1335. }
  1336. fn add_mint_freeze_authority(
  1337. &mut self,
  1338. c: Context<ConstraintMintFreezeAuthority>,
  1339. ) -> ParseResult<()> {
  1340. if self.mint_freeze_authority.is_some() {
  1341. return Err(ParseError::new(
  1342. c.span(),
  1343. "mint freeze_authority already provided",
  1344. ));
  1345. }
  1346. self.mint_freeze_authority.replace(c);
  1347. Ok(())
  1348. }
  1349. fn add_mint_decimals(&mut self, c: Context<ConstraintMintDecimals>) -> ParseResult<()> {
  1350. if self.mint_decimals.is_some() {
  1351. return Err(ParseError::new(c.span(), "mint decimals already provided"));
  1352. }
  1353. self.mint_decimals.replace(c);
  1354. Ok(())
  1355. }
  1356. fn add_mint_token_program(&mut self, c: Context<ConstraintTokenProgram>) -> ParseResult<()> {
  1357. if self.mint_token_program.is_some() {
  1358. return Err(ParseError::new(
  1359. c.span(),
  1360. "mint token_program already provided",
  1361. ));
  1362. }
  1363. self.mint_token_program.replace(c);
  1364. Ok(())
  1365. }
  1366. fn add_mut(&mut self, c: Context<ConstraintMut>) -> ParseResult<()> {
  1367. if self.mutable.is_some() {
  1368. return Err(ParseError::new(c.span(), "mut already provided"));
  1369. }
  1370. self.mutable.replace(c);
  1371. Ok(())
  1372. }
  1373. fn add_signer(&mut self, c: Context<ConstraintSigner>) -> ParseResult<()> {
  1374. if self.signer.is_some() {
  1375. return Err(ParseError::new(c.span(), "signer already provided"));
  1376. }
  1377. self.signer.replace(c);
  1378. Ok(())
  1379. }
  1380. fn add_has_one(&mut self, c: Context<ConstraintHasOne>) -> ParseResult<()> {
  1381. if self
  1382. .has_one
  1383. .iter()
  1384. .filter(|item| item.join_target == c.join_target)
  1385. .count()
  1386. > 0
  1387. {
  1388. return Err(ParseError::new(c.span(), "has_one target already provided"));
  1389. }
  1390. self.has_one.push(c);
  1391. Ok(())
  1392. }
  1393. fn add_raw(&mut self, c: Context<ConstraintRaw>) -> ParseResult<()> {
  1394. self.raw.push(c);
  1395. Ok(())
  1396. }
  1397. fn add_owner(&mut self, c: Context<ConstraintOwner>) -> ParseResult<()> {
  1398. if self.owner.is_some() {
  1399. return Err(ParseError::new(c.span(), "owner already provided"));
  1400. }
  1401. self.owner.replace(c);
  1402. Ok(())
  1403. }
  1404. fn add_rent_exempt(&mut self, c: Context<ConstraintRentExempt>) -> ParseResult<()> {
  1405. if self.rent_exempt.is_some() {
  1406. return Err(ParseError::new(c.span(), "rent already provided"));
  1407. }
  1408. self.rent_exempt.replace(c);
  1409. Ok(())
  1410. }
  1411. fn add_seeds(&mut self, c: Context<ConstraintSeeds>) -> ParseResult<()> {
  1412. if self.seeds.is_some() {
  1413. return Err(ParseError::new(c.span(), "seeds already provided"));
  1414. }
  1415. self.seeds.replace(c);
  1416. Ok(())
  1417. }
  1418. fn add_executable(&mut self, c: Context<ConstraintExecutable>) -> ParseResult<()> {
  1419. if self.executable.is_some() {
  1420. return Err(ParseError::new(c.span(), "executable already provided"));
  1421. }
  1422. self.executable.replace(c);
  1423. Ok(())
  1424. }
  1425. fn add_payer(&mut self, c: Context<ConstraintPayer>) -> ParseResult<()> {
  1426. if self.init.is_none() {
  1427. return Err(ParseError::new(
  1428. c.span(),
  1429. "init must be provided before payer",
  1430. ));
  1431. }
  1432. if self.payer.is_some() {
  1433. return Err(ParseError::new(c.span(), "payer already provided"));
  1434. }
  1435. self.payer.replace(c);
  1436. Ok(())
  1437. }
  1438. fn add_space(&mut self, c: Context<ConstraintSpace>) -> ParseResult<()> {
  1439. if self.init.is_none() {
  1440. return Err(ParseError::new(
  1441. c.span(),
  1442. "init must be provided before space",
  1443. ));
  1444. }
  1445. if self.space.is_some() {
  1446. return Err(ParseError::new(c.span(), "space already provided"));
  1447. }
  1448. self.space.replace(c);
  1449. Ok(())
  1450. }
  1451. // extensions
  1452. fn add_extension_group_pointer_authority(
  1453. &mut self,
  1454. c: Context<ConstraintExtensionAuthority>,
  1455. ) -> ParseResult<()> {
  1456. if self.extension_group_pointer_authority.is_some() {
  1457. return Err(ParseError::new(
  1458. c.span(),
  1459. "extension group pointer authority already provided",
  1460. ));
  1461. }
  1462. self.extension_group_pointer_authority.replace(c);
  1463. Ok(())
  1464. }
  1465. fn add_extension_group_pointer_group_address(
  1466. &mut self,
  1467. c: Context<ConstraintExtensionGroupPointerGroupAddress>,
  1468. ) -> ParseResult<()> {
  1469. if self.extension_group_pointer_group_address.is_some() {
  1470. return Err(ParseError::new(
  1471. c.span(),
  1472. "extension group pointer group address already provided",
  1473. ));
  1474. }
  1475. self.extension_group_pointer_group_address.replace(c);
  1476. Ok(())
  1477. }
  1478. fn add_extension_group_member_pointer_authority(
  1479. &mut self,
  1480. c: Context<ConstraintExtensionAuthority>,
  1481. ) -> ParseResult<()> {
  1482. if self.extension_group_member_pointer_authority.is_some() {
  1483. return Err(ParseError::new(
  1484. c.span(),
  1485. "extension group member pointer authority already provided",
  1486. ));
  1487. }
  1488. self.extension_group_member_pointer_authority.replace(c);
  1489. Ok(())
  1490. }
  1491. fn add_extension_group_member_pointer_member_address(
  1492. &mut self,
  1493. c: Context<ConstraintExtensionGroupMemberPointerMemberAddress>,
  1494. ) -> ParseResult<()> {
  1495. if self.extension_group_member_pointer_member_address.is_some() {
  1496. return Err(ParseError::new(
  1497. c.span(),
  1498. "extension group member pointer member address already provided",
  1499. ));
  1500. }
  1501. self.extension_group_member_pointer_member_address
  1502. .replace(c);
  1503. Ok(())
  1504. }
  1505. fn add_extension_metadata_pointer_authority(
  1506. &mut self,
  1507. c: Context<ConstraintExtensionAuthority>,
  1508. ) -> ParseResult<()> {
  1509. if self.extension_metadata_pointer_authority.is_some() {
  1510. return Err(ParseError::new(
  1511. c.span(),
  1512. "extension metadata pointer authority already provided",
  1513. ));
  1514. }
  1515. self.extension_metadata_pointer_authority.replace(c);
  1516. Ok(())
  1517. }
  1518. fn add_extension_metadata_pointer_metadata_address(
  1519. &mut self,
  1520. c: Context<ConstraintExtensionMetadataPointerMetadataAddress>,
  1521. ) -> ParseResult<()> {
  1522. if self.extension_metadata_pointer_metadata_address.is_some() {
  1523. return Err(ParseError::new(
  1524. c.span(),
  1525. "extension metadata pointer metadata address already provided",
  1526. ));
  1527. }
  1528. self.extension_metadata_pointer_metadata_address.replace(c);
  1529. Ok(())
  1530. }
  1531. fn add_extension_close_authority(
  1532. &mut self,
  1533. c: Context<ConstraintExtensionAuthority>,
  1534. ) -> ParseResult<()> {
  1535. if self.extension_close_authority.is_some() {
  1536. return Err(ParseError::new(
  1537. c.span(),
  1538. "extension close authority already provided",
  1539. ));
  1540. }
  1541. self.extension_close_authority.replace(c);
  1542. Ok(())
  1543. }
  1544. fn add_extension_authority(
  1545. &mut self,
  1546. c: Context<ConstraintExtensionAuthority>,
  1547. ) -> ParseResult<()> {
  1548. if self.extension_transfer_hook_authority.is_some() {
  1549. return Err(ParseError::new(
  1550. c.span(),
  1551. "extension transfer hook authority already provided",
  1552. ));
  1553. }
  1554. self.extension_transfer_hook_authority.replace(c);
  1555. Ok(())
  1556. }
  1557. fn add_extension_transfer_hook_program_id(
  1558. &mut self,
  1559. c: Context<ConstraintExtensionTokenHookProgramId>,
  1560. ) -> ParseResult<()> {
  1561. if self.extension_transfer_hook_program_id.is_some() {
  1562. return Err(ParseError::new(
  1563. c.span(),
  1564. "extension transfer hook program id already provided",
  1565. ));
  1566. }
  1567. self.extension_transfer_hook_program_id.replace(c);
  1568. Ok(())
  1569. }
  1570. fn add_extension_permanent_delegate(
  1571. &mut self,
  1572. c: Context<ConstraintExtensionPermanentDelegate>,
  1573. ) -> ParseResult<()> {
  1574. if self.extension_permanent_delegate.is_some() {
  1575. return Err(ParseError::new(
  1576. c.span(),
  1577. "extension permanent delegate already provided",
  1578. ));
  1579. }
  1580. self.extension_permanent_delegate.replace(c);
  1581. Ok(())
  1582. }
  1583. }