pinoc 0.3.6

A CLI tool for setting up pinocchio program project
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

use shank_idl::idl::Idl;
use shank_idl::idl_type::IdlType;
use shank_idl::idl_type_definition::IdlTypeDefinitionTy;

/// Rewrites `{"defined": "Address"}` and `{"defined": "publicKey"}` to the
/// literal `"publicKey"` IDL type, since `@codama/nodes-from-anchor` mishandles
/// both (see docs/codama-comparison.md).
pub fn to_codama_compatible(idl: &Idl) -> Idl {
    let mut idl = idl.clone();

    for account in &mut idl.accounts {
        rewrite_type_definition(&mut account.ty);
    }
    for ty_def in &mut idl.types {
        rewrite_type_definition(&mut ty_def.ty);
    }
    for ix in &mut idl.instructions {
        for field in &mut ix.args {
            field.ty = rewrite_type(&field.ty);
        }
    }

    idl
}

fn rewrite_type_definition(ty: &mut IdlTypeDefinitionTy) {
    match ty {
        IdlTypeDefinitionTy::Struct { fields } => {
            for field in fields {
                field.ty = rewrite_type(&field.ty);
            }
        }
        IdlTypeDefinitionTy::Enum { .. } => {
            // pinoc's templates never emit enum accounts/types today.
        }
    }
}

fn rewrite_type(ty: &IdlType) -> IdlType {
    match ty {
        IdlType::Defined(name) if name == "Address" || name.eq_ignore_ascii_case("publicKey") => {
            IdlType::PublicKey
        }
        IdlType::Array(inner, size) => IdlType::Array(Box::new(rewrite_type(inner)), *size),
        IdlType::Vec(inner) => IdlType::Vec(Box::new(rewrite_type(inner))),
        IdlType::Option(inner) => IdlType::Option(Box::new(rewrite_type(inner))),
        other => other.clone(),
    }
}