103 lines
2.2 KiB
Go
103 lines
2.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
lcli "github.com/filecoin-project/lotus/cli"
|
||
|
|
|
||
|
|
"github.com/filecoin-project/go-address"
|
||
|
|
"github.com/filecoin-project/go-state-types/abi"
|
||
|
|
"github.com/filecoin-project/go-state-types/big"
|
||
|
|
"github.com/urfave/cli/v2"
|
||
|
|
"golang.org/x/xerrors"
|
||
|
|
)
|
||
|
|
|
||
|
|
var marketCmd = &cli.Command{
|
||
|
|
Name: "market",
|
||
|
|
Usage: "Interact with the market actor",
|
||
|
|
Flags: []cli.Flag{},
|
||
|
|
Subcommands: []*cli.Command{
|
||
|
|
marketDealFeesCmd,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
var marketDealFeesCmd = &cli.Command{
|
||
|
|
Name: "get-deal-fees",
|
||
|
|
Usage: "View the storage fees associated with a particular deal or storage provider",
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "provider",
|
||
|
|
Usage: "provider whose outstanding fees you'd like to calculate",
|
||
|
|
},
|
||
|
|
&cli.IntFlag{
|
||
|
|
Name: "dealId",
|
||
|
|
Usage: "deal whose outstanding fees you'd like to calculate",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Action: func(cctx *cli.Context) error {
|
||
|
|
api, closer, err := lcli.GetFullNodeAPI(cctx)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
defer closer()
|
||
|
|
|
||
|
|
ctx := lcli.ReqContext(cctx)
|
||
|
|
|
||
|
|
ts, err := lcli.LoadTipSet(ctx, cctx, api)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
ht := ts.Height()
|
||
|
|
|
||
|
|
if cctx.IsSet("provider") {
|
||
|
|
p, err := address.NewFromString(cctx.String("provider"))
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("failed to parse provider: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
deals, err := api.StateMarketDeals(ctx, ts.Key())
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
ef := big.Zero()
|
||
|
|
pf := big.Zero()
|
||
|
|
count := 0
|
||
|
|
|
||
|
|
for _, deal := range deals {
|
||
|
|
if deal.Proposal.Provider == p {
|
||
|
|
e, p := deal.Proposal.GetDealFees(ht)
|
||
|
|
ef = big.Add(ef, e)
|
||
|
|
pf = big.Add(pf, p)
|
||
|
|
count++
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fmt.Println("Total deals: ", count)
|
||
|
|
fmt.Println("Total earned fees: ", ef)
|
||
|
|
fmt.Println("Total pending fees: ", pf)
|
||
|
|
fmt.Println("Total fees: ", big.Add(ef, pf))
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
if dealid := cctx.Int("dealId"); dealid != 0 {
|
||
|
|
deal, err := api.StateMarketStorageDeal(ctx, abi.DealID(dealid), ts.Key())
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
ef, pf := deal.Proposal.GetDealFees(ht)
|
||
|
|
|
||
|
|
fmt.Println("Earned fees: ", ef)
|
||
|
|
fmt.Println("Pending fees: ", pf)
|
||
|
|
fmt.Println("Total fees: ", big.Add(ef, pf))
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return xerrors.New("must provide either --provider or --dealId flag")
|
||
|
|
},
|
||
|
|
}
|