From d02ef1817441f5c73e51bcd77253d7653cd3187b Mon Sep 17 00:00:00 2001 From: Cassandra Heart Date: Wed, 3 Jan 2024 02:13:51 -0600 Subject: [PATCH] also refer to adjusted balance for rpc --- node/rpc/rpc_server.go | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/node/rpc/rpc_server.go b/node/rpc/rpc_server.go index 6afc7a6..02e4c66 100644 --- a/node/rpc/rpc_server.go +++ b/node/rpc/rpc_server.go @@ -226,6 +226,12 @@ func (r *RPCServer) GetTokenInfo( return nil, errors.Wrap(err, "get token info") } + peerBytes := r.pubSub.GetPeerID() + peerAddr, err := poseidon.HashBytes(peerBytes) + if err != nil { + panic(err) + } + addr, err := poseidon.HashBytes(provingKey.PublicKey) if err != nil { panic(err) @@ -234,6 +240,9 @@ func (r *RPCServer) GetTokenInfo( addrBytes := addr.Bytes() addrBytes = append(make([]byte, 32-len(addrBytes)), addrBytes...) + peerAddrBytes := peerAddr.Bytes() + peerAddrBytes = append(make([]byte, 32-len(peerAddrBytes)), peerAddrBytes...) + frame, err := r.clockStore.GetLatestDataClockFrame( append( p2p.GetBloomFilter(application.CEREMONY_ADDRESS, 256, 3), @@ -303,6 +312,15 @@ func (r *RPCServer) GetTokenInfo( new(big.Int).SetUint64(child.External.Total), ) } + if bytes.Equal( + child.External.Key, + peerAddrBytes, + ) { + ownedTotal = ownedTotal.Add( + ownedTotal, + new(big.Int).SetUint64(child.External.Total), + ) + } } } }