core-sdk come true (#87)

* core-sdk come true * clean up Co-authored-by: chengwenxi <[email protected]>
irisnet · Jun 29, 2021 · b590f5b · b590f5b
1 parent d21cb8e
commit b590f5b
Show file tree

Hide file tree

Showing 241 changed files with 58,168 additions and 0 deletions.
diff --git a/core-sdk/Makefile b/core-sdk/Makefile
@@ -0,0 +1,20 @@
+PACKAGES=$(shell go list ./...)
+PACKAGES_UNITTEST=$(shell go list ./... | grep -v integration_test)
+export GO111MODULE = on
+
+format:
+	find . -name '*.go' -type f -not -path "./vendor*" -not -path "*.git*" -not -path "*.pb.go" | xargs gofmt -w -s
+	find . -name '*.go' -type f -not -path "./vendor*" -not -path "*.git*" -not -path "*.pb.go" | xargs misspell -w
+	find . -name '*.go' -type f -not -path "./vendor*" -not -path "*.git*" -not -path "*.pb.go" | xargs goimports -w -local github.com/irisnet/core-sdk-go
+
+test-unit:
+	@go test -v $(PACKAGES_UNITTEST)
+
+test-integration:
+	cd integration_test/scripts/ && sh build.sh && sh start.sh
+	sleep 2s
+	@go test -v $(PACKAGES)
+	cd integration_test/scripts/ && sh clean.sh
+
+proto-gen:
+	@./third_party/protocgen.sh
diff --git a/core-sdk/bank/bank.go b/core-sdk/bank/bank.go
@@ -0,0 +1,198 @@
+package bank
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"github.com/irisnet/core-sdk-go/common"
+	commoncodec "github.com/irisnet/core-sdk-go/common/codec"
+	"github.com/irisnet/core-sdk-go/common/codec/types"
+	sdk "github.com/irisnet/core-sdk-go/types"
+)
+
+type bankClient struct {
+	sdk.BaseClient
+	commoncodec.Marshaler
+}
+
+//bank NewClient
+func NewClient(bc sdk.BaseClient, cdc commoncodec.Marshaler) Client {
+	return bankClient{
+		BaseClient: bc,
+		Marshaler:  cdc,
+	}
+}
+
+func (b bankClient) Name() string {
+	return ModuleName
+}
+
+func (b bankClient) RegisterInterfaceTypes(registry types.InterfaceRegistry) {
+	RegisterInterfaces(registry)
+}
+
+// QueryAccount return account information specified address
+func (b bankClient) QueryAccount(address string) (sdk.BaseAccount, sdk.Error) {
+	account, err := b.BaseClient.QueryAccount(address)
+	if err != nil {
+		return sdk.BaseAccount{}, sdk.Wrap(err)
+	}
+
+	return account, nil
+}
+
+//  TotalSupply queries the total supply of all coins.
+func (b bankClient) TotalSupply() (sdk.Coins, sdk.Error) {
+	conn, err := b.GenConn()
+	defer func() { _ = conn.Close() }()
+	if err != nil {
+		return nil, sdk.Wrap(err)
+	}
+
+	resp, err := NewQueryClient(conn).TotalSupply(
+		context.Background(),
+		&QueryTotalSupplyRequest{},
+	)
+	if err != nil {
+		return nil, sdk.Wrap(err)
+	}
+	return resp.Supply, nil
+}
+
+// Send is responsible for transferring tokens from `From` to `to` account
+func (b bankClient) Send(to string, amount sdk.DecCoins, baseTx sdk.BaseTx) (sdk.ResultTx, sdk.Error) {
+	sender, err := b.QueryAddress(baseTx.From, baseTx.Password)
+	if err != nil {
+		return sdk.ResultTx{}, sdk.Wrapf("%s not found", baseTx.From)
+	}
+
+	amt, err := b.ToMinCoin(amount...)
+	if err != nil {
+		return sdk.ResultTx{}, sdk.Wrap(err)
+	}
+
+	outAddr, err := sdk.AccAddressFromBech32(to)
+	if err != nil {
+		return sdk.ResultTx{}, sdk.Wrapf(fmt.Sprintf("%s invalid address", to))
+	}
+
+	msg := NewMsgSend(sender, outAddr, amt)
+	return b.BuildAndSend([]sdk.Msg{msg}, baseTx)
+}
+
+func (b bankClient) SendWitchSpecAccountInfo(to string, sequence, accountNumber uint64, amount sdk.DecCoins, baseTx sdk.BaseTx) (sdk.ResultTx, sdk.Error) {
+	sender, err := b.QueryAddress(baseTx.From, baseTx.Password)
+	if err != nil {
+		return sdk.ResultTx{}, sdk.Wrapf("%s not found", baseTx.From)
+	}
+
+	amt, err := b.ToMinCoin(amount...)
+	if err != nil {
+		return sdk.ResultTx{}, sdk.Wrap(err)
+	}
+
+	outAddr, err := sdk.AccAddressFromBech32(to)
+	if err != nil {
+		return sdk.ResultTx{}, sdk.Wrapf(fmt.Sprintf("%s invalid address", to))
+	}
+
+	msg := NewMsgSend(sender, outAddr, amt)
+	return b.BuildAndSendWithAccount(sender.String(), accountNumber, sequence, []sdk.Msg{msg}, baseTx)
+}
+
+func (b bankClient) MultiSend(request MultiSendRequest, baseTx sdk.BaseTx) (resTxs []sdk.ResultTx, err sdk.Error) {
+	sender, err := b.QueryAddress(baseTx.From, baseTx.Password)
+	if err != nil {
+		return nil, sdk.Wrapf("%s not found", baseTx.From)
+	}
+
+	if len(request.Receipts) > maxMsgLen {
+		return b.SendBatch(sender, request, baseTx)
+	}
+
+	var inputs = make([]Input, len(request.Receipts))
+	var outputs = make([]Output, len(request.Receipts))
+	for i, receipt := range request.Receipts {
+		amt, err := b.ToMinCoin(receipt.Amount...)
+		if err != nil {
+			return nil, sdk.Wrap(err)
+		}
+
+		outAddr, e := sdk.AccAddressFromBech32(receipt.Address)
+		if e != nil {
+			return nil, sdk.Wrapf(fmt.Sprintf("%s invalid address", receipt.Address))
+		}
+
+		inputs[i] = NewInput(sender, amt)
+		outputs[i] = NewOutput(outAddr, amt)
+	}
+
+	msg := NewMsgMultiSend(inputs, outputs)
+	res, err := b.BuildAndSend([]sdk.Msg{msg}, baseTx)
+	if err != nil {
+		return nil, sdk.Wrap(err)
+	}
+
+	resTxs = append(resTxs, res)
+	return
+}
+
+func (b bankClient) SendBatch(sender sdk.AccAddress, request MultiSendRequest, baseTx sdk.BaseTx) ([]sdk.ResultTx, sdk.Error) {
+	batchReceipts := common.SubArray(maxMsgLen, request)
+
+	var msgs sdk.Msgs
+	for _, receipts := range batchReceipts {
+
+		req := receipts.(MultiSendRequest)
+		var inputs = make([]Input, len(req.Receipts))
+		var outputs = make([]Output, len(req.Receipts))
+		for i, receipt := range req.Receipts {
+			amt, err := b.ToMinCoin(receipt.Amount...)
+			if err != nil {
+				return nil, sdk.Wrap(err)
+			}
+
+			outAddr, e := sdk.AccAddressFromBech32(receipt.Address)
+			if e != nil {
+				return nil, sdk.Wrapf(fmt.Sprintf("%s invalid address", receipt.Address))
+			}
+
+			inputs[i] = NewInput(sender, amt)
+			outputs[i] = NewOutput(outAddr, amt)
+		}
+		msgs = append(msgs, NewMsgMultiSend(inputs, outputs))
+	}
+	return b.BaseClient.SendBatch(msgs, baseTx)
+}
+
+// SubscribeSendTx Subscribe MsgSend event and return subscription
+func (b bankClient) SubscribeSendTx(from, to string, callback EventMsgSendCallback) sdk.Subscription {
+	var builder = sdk.NewEventQueryBuilder()
+
+	from = strings.TrimSpace(from)
+	if len(from) != 0 {
+		builder.AddCondition(sdk.NewCond(sdk.EventTypeMessage,
+			sdk.AttributeKeySender).EQ(sdk.EventValue(from)))
+	}
+
+	to = strings.TrimSpace(to)
+	if len(to) != 0 {
+		builder.AddCondition(sdk.Cond("transfer.recipient").EQ(sdk.EventValue(to)))
+	}
+
+	subscription, _ := b.SubscribeTx(builder, func(data sdk.EventDataTx) {
+		for _, msg := range data.Tx.GetMsgs() {
+			if value, ok := msg.(*MsgSend); ok {
+				callback(EventDataMsgSend{
+					Height: data.Height,
+					Hash:   data.Hash,
+					From:   value.FromAddress,
+					To:     value.ToAddress,
+					Amount: value.Amount,
+				})
+			}
+		}
+	})
+	return subscription
+}