49 lines
1.8 KiB
Go
49 lines
1.8 KiB
Go
package retrievaladapter
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/ipfs/go-cid"
|
|
blockstore "github.com/ipfs/go-ipfs-blockstore"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
"github.com/filecoin-project/go-fil-markets/retrievalmarket"
|
|
retrievaltoken "github.com/filecoin-project/go-fil-markets/shared/tokenamount"
|
|
retrievaltypes "github.com/filecoin-project/go-fil-markets/shared/types"
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/markets/utils"
|
|
"github.com/filecoin-project/lotus/storage/sectorblocks"
|
|
)
|
|
|
|
type retrievalProviderNode struct {
|
|
sectorBlocks *sectorblocks.SectorBlocks
|
|
full api.FullNode
|
|
}
|
|
|
|
// NewRetrievalProviderNode returns a new node adapter for a retrieval provider that talks to the
|
|
// Lotus Node
|
|
func NewRetrievalProviderNode(sectorBlocks *sectorblocks.SectorBlocks, full api.FullNode) retrievalmarket.RetrievalProviderNode {
|
|
return &retrievalProviderNode{sectorBlocks, full}
|
|
}
|
|
|
|
func (rpn *retrievalProviderNode) GetPieceSize(pieceCid []byte) (uint64, error) {
|
|
asCid, err := cid.Cast(pieceCid)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return rpn.sectorBlocks.GetSize(asCid)
|
|
}
|
|
|
|
func (rpn *retrievalProviderNode) SealedBlockstore(approveUnseal func() error) blockstore.Blockstore {
|
|
return rpn.sectorBlocks.SealedBlockstore(approveUnseal)
|
|
}
|
|
|
|
func (rpn *retrievalProviderNode) SavePaymentVoucher(ctx context.Context, paymentChannel address.Address, voucher *retrievaltypes.SignedVoucher, proof []byte, expectedAmount retrievaltoken.TokenAmount) (retrievaltoken.TokenAmount, error) {
|
|
localVoucher, err := utils.FromSharedSignedVoucher(voucher)
|
|
if err != nil {
|
|
return retrievaltoken.FromInt(0), err
|
|
}
|
|
added, err := rpn.full.PaychVoucherAdd(ctx, paymentChannel, localVoucher, proof, utils.FromSharedTokenAmount(expectedAmount))
|
|
return utils.ToSharedTokenAmount(added), err
|
|
}
|