synapsecns/sanguine

package types

import (
    "fmt"
    "math/big"
)

// NotaryAttestation is the notary attestation interface.
type NotaryAttestation interface {
    // AttPayload is raw bytes of the attestation payload.
    AttPayload() []byte
    // AgentRoot is the agent root.
    AgentRoot() [32]byte
    // SnapGas is the snap gas.
    SnapGas() []*big.Int
    // Attestation is the decoded attestation from the AttPayload
    Attestation() Attestation
}

type notaryAttestation struct {
    attPayload  []byte
    agentRoot   [32]byte
    snapGas     []*big.Int
    attestation Attestation
}

// NewNotaryAttestation creates a new notary attestation.
func NewNotaryAttestation(attPayload []byte, agentRoot [32]byte, snapGas []*big.Int) (NotaryAttestation, error) {
    attestation, err := DecodeAttestation(attPayload)
    if err != nil {
        return nil, fmt.Errorf("could not decode attestation: %w", err)
    }

    return &notaryAttestation{
        attPayload:  attPayload,
        agentRoot:   agentRoot,
        snapGas:     snapGas,
        attestation: attestation,
    }, nil
}

func (a notaryAttestation) AttPayload() []byte {
    return a.attPayload
}

func (a notaryAttestation) AgentRoot() [32]byte {
    return a.agentRoot
}

func (a notaryAttestation) SnapGas() []*big.Int {
    return a.snapGas
}

func (a notaryAttestation) Attestation() Attestation {
    return a.attestation
}