mirror of
https://github.com/aljazceru/ark.git
synced 2025-12-17 20:24:21 +01:00
Add admin APIs to manage wallet (#226)
* Add admin rpcs to manage wallet * Fix * Fixes * Add sleeping time * Increase sleeping time
This commit is contained in:
committed by
GitHub
parent
fb8a127f4f
commit
1c67c56d9d
129
server/internal/interface/grpc/handlers/walletservice.go
Normal file
129
server/internal/interface/grpc/handlers/walletservice.go
Normal file
@@ -0,0 +1,129 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
arkv1 "github.com/ark-network/ark/api-spec/protobuf/gen/ark/v1"
|
||||
"github.com/ark-network/ark/internal/core/ports"
|
||||
)
|
||||
|
||||
type walletHandler struct {
|
||||
walletService ports.WalletService
|
||||
onUnlock func()
|
||||
}
|
||||
|
||||
func NewWalletHandler(walletService ports.WalletService, onUnlock func()) arkv1.WalletServiceServer {
|
||||
return &walletHandler{walletService, onUnlock}
|
||||
}
|
||||
|
||||
func (a *walletHandler) GenSeed(ctx context.Context, _ *arkv1.GenSeedRequest) (*arkv1.GenSeedResponse, error) {
|
||||
seed, err := a.walletService.GenSeed(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &arkv1.GenSeedResponse{Seed: seed}, nil
|
||||
}
|
||||
|
||||
func (a *walletHandler) Create(ctx context.Context, req *arkv1.CreateRequest) (*arkv1.CreateResponse, error) {
|
||||
if len(req.GetSeed()) <= 0 {
|
||||
return nil, fmt.Errorf("missing wallet seed")
|
||||
}
|
||||
if len(req.GetPassword()) <= 0 {
|
||||
return nil, fmt.Errorf("missing wallet password")
|
||||
}
|
||||
|
||||
if err := a.walletService.Create(
|
||||
ctx, req.GetSeed(), req.GetPassword(),
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &arkv1.CreateResponse{}, nil
|
||||
}
|
||||
|
||||
func (a *walletHandler) Restore(ctx context.Context, req *arkv1.RestoreRequest) (*arkv1.RestoreResponse, error) {
|
||||
if len(req.GetSeed()) <= 0 {
|
||||
return nil, fmt.Errorf("missing wallet seed")
|
||||
}
|
||||
if len(req.GetPassword()) <= 0 {
|
||||
return nil, fmt.Errorf("missing wallet password")
|
||||
}
|
||||
|
||||
if err := a.walletService.Restore(
|
||||
ctx, req.GetSeed(), req.GetPassword(),
|
||||
); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &arkv1.RestoreResponse{}, nil
|
||||
}
|
||||
|
||||
func (a *walletHandler) Unlock(ctx context.Context, req *arkv1.UnlockRequest) (*arkv1.UnlockResponse, error) {
|
||||
if len(req.GetPassword()) <= 0 {
|
||||
return nil, fmt.Errorf("missing wallet password")
|
||||
}
|
||||
if err := a.walletService.Unlock(ctx, req.GetPassword()); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
go a.onUnlock()
|
||||
|
||||
return &arkv1.UnlockResponse{}, nil
|
||||
}
|
||||
|
||||
func (a *walletHandler) Lock(ctx context.Context, req *arkv1.LockRequest) (*arkv1.LockResponse, error) {
|
||||
if len(req.GetPassword()) <= 0 {
|
||||
return nil, fmt.Errorf("missing wallet password")
|
||||
}
|
||||
if err := a.walletService.Lock(ctx, req.GetPassword()); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &arkv1.LockResponse{}, nil
|
||||
}
|
||||
|
||||
func (a *walletHandler) GetStatus(ctx context.Context, _ *arkv1.GetStatusRequest) (*arkv1.GetStatusResponse, error) {
|
||||
status, err := a.walletService.Status(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &arkv1.GetStatusResponse{
|
||||
Initialized: status.IsInitialized(),
|
||||
Unlocked: status.IsUnlocked(),
|
||||
Synced: status.IsSynced(),
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (a *walletHandler) DeriveAddress(ctx context.Context, _ *arkv1.DeriveAddressRequest) (*arkv1.DeriveAddressResponse, error) {
|
||||
addr, err := a.walletService.DeriveAddresses(ctx, 1)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &arkv1.DeriveAddressResponse{Address: addr[0]}, nil
|
||||
}
|
||||
|
||||
func (a *walletHandler) GetBalance(ctx context.Context, _ *arkv1.GetBalanceRequest) (*arkv1.GetBalanceResponse, error) {
|
||||
availableMainBalance, lockedMainBalance, err := a.walletService.MainAccountBalance(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
availableConnectorsBalance, lockedConnectorsBalance, err := a.walletService.ConnectorsAccountBalance(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &arkv1.GetBalanceResponse{
|
||||
MainAccount: &arkv1.Balance{
|
||||
Locked: convertSatoshis(lockedMainBalance),
|
||||
Available: convertSatoshis(availableMainBalance),
|
||||
},
|
||||
ConnectorsAccount: &arkv1.Balance{
|
||||
Locked: convertSatoshis(lockedConnectorsBalance),
|
||||
Available: convertSatoshis(availableConnectorsBalance),
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
Reference in New Issue
Block a user