code.vegaprotocol.io/vega@v0.79.0/core/events/ledger_movements.go (about)

     1  // Copyright (C) 2023 Gobalsky Labs Limited
     2  //
     3  // This program is free software: you can redistribute it and/or modify
     4  // it under the terms of the GNU Affero General Public License as
     5  // published by the Free Software Foundation, either version 3 of the
     6  // License, or (at your option) any later version.
     7  //
     8  // This program is distributed in the hope that it will be useful,
     9  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    10  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    11  // GNU Affero General Public License for more details.
    12  //
    13  // You should have received a copy of the GNU Affero General Public License
    14  // along with this program.  If not, see <http://www.gnu.org/licenses/>.
    15  
    16  package events
    17  
    18  import (
    19  	"context"
    20  
    21  	"code.vegaprotocol.io/vega/core/types"
    22  	ptypes "code.vegaprotocol.io/vega/protos/vega"
    23  	eventspb "code.vegaprotocol.io/vega/protos/vega/events/v1"
    24  )
    25  
    26  type LedgerMovements struct {
    27  	*Base
    28  	ledgerMovements []*ptypes.LedgerMovement
    29  }
    30  
    31  // NewLedgerMovements returns an event with transfer responses - this is the replacement of the transfer buffer.
    32  func NewLedgerMovements(ctx context.Context, ledgerMovements []*types.LedgerMovement) *LedgerMovements {
    33  	return &LedgerMovements{
    34  		Base:            newBase(ctx, LedgerMovementsEvent),
    35  		ledgerMovements: types.LedgerMovements(ledgerMovements).IntoProto(),
    36  	}
    37  }
    38  
    39  // LedgerMovements returns the actual event payload.
    40  func (t *LedgerMovements) LedgerMovements() []*ptypes.LedgerMovement {
    41  	return t.ledgerMovements
    42  }
    43  
    44  func (t *LedgerMovements) IsParty(id string) bool {
    45  	isParty := func(owner *string) bool { return owner != nil && *owner == id }
    46  
    47  	for _, r := range t.ledgerMovements {
    48  		for _, e := range r.Entries {
    49  			if isParty(e.FromAccount.Owner) || isParty(e.ToAccount.Owner) {
    50  				return true
    51  			}
    52  		}
    53  	}
    54  	return false
    55  }
    56  
    57  func (t *LedgerMovements) Proto() eventspb.LedgerMovements {
    58  	return eventspb.LedgerMovements{
    59  		LedgerMovements: t.ledgerMovements,
    60  	}
    61  }
    62  
    63  func (t *LedgerMovements) StreamMessage() *eventspb.BusEvent {
    64  	p := t.Proto()
    65  	busEvent := newBusEventFromBase(t.Base)
    66  	busEvent.Event = &eventspb.BusEvent_LedgerMovements{
    67  		LedgerMovements: &p,
    68  	}
    69  
    70  	return busEvent
    71  }
    72  
    73  func TransferResponseEventFromStream(ctx context.Context, be *eventspb.BusEvent) *LedgerMovements {
    74  	return &LedgerMovements{
    75  		Base:            newBaseFromBusEvent(ctx, LedgerMovementsEvent, be),
    76  		ledgerMovements: be.GetLedgerMovements().LedgerMovements,
    77  	}
    78  }