generate.go

package overflow

import (
	"fmt"
	"path/filepath"
	"strings"
)

func (o *OverflowState) GenerateStub(network, filePath string, standalone bool) (string, error) {
	solution, err := o.ParseAll()
	if err != nil {
		return "", err
	}

	interactionName := strings.TrimSuffix(filepath.Base(filePath), filepath.Ext(filePath))
	var interaction *OverflowDeclarationInfo
	var commandName string
	if strings.HasPrefix(filePath, o.TransactionBasePath) || strings.HasPrefix("./"+filePath, o.TransactionBasePath) {
		interaction = solution.Transactions[interactionName]
		commandName = "Tx"
	} else {
		interaction = solution.Scripts[interactionName]
		commandName = "Script"
	}
	if interaction == nil {
		return "", fmt.Errorf("could not find interaction of type %s with name %s", commandName, interaction)
	}
	lines := []string{
		fmt.Sprintf(`  o.%s("%s",`, commandName, interactionName),
	}

	if commandName == "Tx" {
		lines = append(lines, "    WithSigner(\"<>\"),")
	}
	for name, value := range interaction.Parameters {
		lines = append(lines, fmt.Sprintf("    WithArg(\"%s\", <>), //%s", name, value))
	}
	var stub string
	if len(lines) > 1 {
		lines = append(lines, "  )")
		stub = strings.Join(lines, "\n")
	} else {
		stub = strings.ReplaceAll(lines[0], ",", ")")
	}

	if !standalone {
		return stub, nil
	}

	return fmt.Sprintf(`package main

import (
   . "github.com/bjartek/overflow/v2"
)

func main() {
  o := Overflow(WithNetwork("%s"), WithPrintResults())
%s
}`, network, stub), nil
}