generateblock.go

package generateblock

import (
	"context"

	"github.com/cryptogarageinc/generate-block-for-testing/internal/domain/service"
	"github.com/cryptogarageinc/generate-block-for-testing/internal/handler"
	"github.com/cryptogarageinc/generate-block-for-testing/internal/infrastructure/repository"
	"github.com/cryptogarageinc/generate-block-for-testing/internal/usecase"
)

type Connection struct {
	Host        string
	RpcUserID   string
	RpcPassword string
}

type Generator struct {
	handle  handler.Handler
	network string

	ignoreEmptyMempool   bool
	hasCheckInitialBlkDl bool
}

func NewGenerator(
	nodeInfo *Connection,
	network string,
) *Generator {
	return &Generator{
		handle:  newHandler(nodeInfo.Host, nodeInfo.RpcUserID, nodeInfo.RpcPassword),
		network: network,
	}
}

func (g *Generator) WithIgnoreEmptyMempool(ignoreEmptyMempool bool) *Generator {
	g.ignoreEmptyMempool = ignoreEmptyMempool
	return g
}

func (g *Generator) WithCheckInitialBlockDownload(hasCheckInitialBlkDl bool) *Generator {
	g.hasCheckInitialBlkDl = hasCheckInitialBlkDl
	return g
}

func (g *Generator) GenerateBlock(
	ctx context.Context,
	address string,
	generateCount uint,
) error {
	return g.handle.GenerateBlock(
		ctx, g.network, "", []string{}, address, generateCount,
		g.ignoreEmptyMempool, g.hasCheckInitialBlkDl)
}

func (g *Generator) GenerateElementsDynafedBlock(
	ctx context.Context,
	fedpegScript string,
	pakEntries []string,
	generateCount uint,
) error {
	return g.handle.GenerateBlock(
		ctx, g.network, fedpegScript, pakEntries, "", generateCount,
		g.ignoreEmptyMempool, g.hasCheckInitialBlkDl)
}

func GenerateBlock(
	ctx context.Context,
	nodeInfo *Connection,
	network string,
	address string,
	generateCount uint,
) error {
	handle := newHandler(nodeInfo.Host, nodeInfo.RpcUserID, nodeInfo.RpcPassword)
	return handle.GenerateBlock(ctx, network, "", []string{}, address,
		generateCount, false, false)
}

func GenerateElementsDynafedBlock(
	ctx context.Context,
	nodeInfo *Connection,
	network string,
	fedpegScript string,
	pakEntries []string,
	generateCount uint,
) error {
	handle := newHandler(nodeInfo.Host, nodeInfo.RpcUserID, nodeInfo.RpcPassword)
	return handle.GenerateBlock(ctx, network, fedpegScript, pakEntries, "",
		generateCount, false, false)
}

func newHandler(
	host, rpcUserID, rpcPassword string,
) handler.Handler {
	blockchainConfig := repository.NewBlockchainRpcConfig(
		host, rpcUserID, rpcPassword)
	blockchainRepo := repository.NewBlockchainRpc(blockchainConfig)

	genBlockService := service.NewGenerateBlock(blockchainRepo)
	genBlockUsecase := usecase.NewGenerateBlock(genBlockService)

	return handler.NewHandler(genBlockUsecase)
}