80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
package full
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.uber.org/fx"
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
"github.com/filecoin-project/go-state-types/big"
|
|
"github.com/filecoin-project/go-state-types/crypto"
|
|
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/chain/stmgr"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/wallet"
|
|
"github.com/filecoin-project/lotus/lib/sigs"
|
|
)
|
|
|
|
type WalletAPI struct {
|
|
fx.In
|
|
|
|
StateManager *stmgr.StateManager
|
|
Default wallet.Default
|
|
api.WalletAPI
|
|
}
|
|
|
|
func (a *WalletAPI) WalletBalance(ctx context.Context, addr address.Address) (types.BigInt, error) {
|
|
act, err := a.StateManager.LoadActorTsk(ctx, addr, types.EmptyTSK)
|
|
if xerrors.Is(err, types.ErrActorNotFound) {
|
|
return big.Zero(), nil
|
|
} else if err != nil {
|
|
return big.Zero(), err
|
|
}
|
|
return act.Balance, nil
|
|
}
|
|
|
|
func (a *WalletAPI) WalletSign(ctx context.Context, k address.Address, msg []byte) (*crypto.Signature, error) {
|
|
keyAddr, err := a.StateManager.ResolveToKeyAddress(ctx, k, nil)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("failed to resolve ID address: %w", keyAddr)
|
|
}
|
|
return a.WalletSign(ctx, keyAddr, msg)
|
|
}
|
|
|
|
func (a *WalletAPI) WalletSignMessage(ctx context.Context, k address.Address, msg *types.Message) (*types.SignedMessage, error) {
|
|
keyAddr, err := a.StateManager.ResolveToKeyAddress(ctx, k, nil)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("failed to resolve ID address: %w", keyAddr)
|
|
}
|
|
|
|
mcid := msg.Cid()
|
|
|
|
sig, err := a.WalletAPI.WalletSign(ctx, k, mcid.Bytes())
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("failed to sign message: %w", err)
|
|
}
|
|
|
|
return &types.SignedMessage{
|
|
Message: *msg,
|
|
Signature: *sig,
|
|
}, nil
|
|
}
|
|
|
|
func (a *WalletAPI) WalletVerify(ctx context.Context, k address.Address, msg []byte, sig *crypto.Signature) (bool, error) {
|
|
return sigs.Verify(sig, k, msg) == nil, nil
|
|
}
|
|
|
|
func (a *WalletAPI) WalletDefaultAddress(ctx context.Context) (address.Address, error) {
|
|
return a.Default.GetDefault()
|
|
}
|
|
|
|
func (a *WalletAPI) WalletSetDefault(ctx context.Context, addr address.Address) error {
|
|
return a.Default.SetDefault(addr)
|
|
}
|
|
|
|
func (a *WalletAPI) WalletValidateAddress(ctx context.Context, str string) (address.Address, error) {
|
|
return address.NewFromString(str)
|
|
}
|