Skip to content

Commit

Permalink
netgoal: allow unknown template tokens (#6091)
Browse files Browse the repository at this point in the history
  • Loading branch information
algorandskiy authored Aug 1, 2024
1 parent 05a52e2 commit 94f1355
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 10 deletions.
6 changes: 4 additions & 2 deletions cmd/netgoal/network.go
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@ var networkRecipeFile string
var networkName string
var networkGenesisVersionModifier string
var miscStringStringTokens []string
var ignoreUnknownTokens bool

var cpuprofilePath string

Expand All @@ -56,7 +57,8 @@ func init() {
networkBuildCmd.Flags().BoolVarP(&networkUseGenesisFiles, "use-existing-files", "e", false, "Use existing genesis files.")
networkBuildCmd.Flags().BoolVarP(&bootstrapLoadingFile, "gen-db-files", "b", false, "Generate database files.")
networkBuildCmd.Flags().BoolVarP(&networkIgnoreExistingDir, "force", "f", false, "Force generation into existing directory.")
networkBuildCmd.Flags().StringSliceVarP(&miscStringStringTokens, "val", "v", nil, "name=value, may be reapeated")
networkBuildCmd.Flags().StringSliceVarP(&miscStringStringTokens, "val", "v", nil, "name=value, may be repeated")
networkBuildCmd.Flags().BoolVarP(&ignoreUnknownTokens, "ignore", "i", false, "Ignore unknown tokens in network template file")
networkBuildCmd.Flags().StringVar(&cpuprofilePath, "cpuprofile", "", "write cpu profile to path")

rootCmd.PersistentFlags().StringVarP(&networkGenesisVersionModifier, "modifier", "m", "", "Override Genesis Version Modifier (eg 'v1')")
Expand Down Expand Up @@ -136,7 +138,7 @@ func runBuildNetwork() error {
return fmt.Errorf("error resolving network template file '%s' to full path: %v", networkTemplateFile, err)
}

netCfg, err := remote.InitDeployedNetworkConfig(networkTemplateFile, buildConfig)
netCfg, err := remote.InitDeployedNetworkConfig(networkTemplateFile, buildConfig, ignoreUnknownTokens)
if err != nil {
return fmt.Errorf("error loading Network Config file '%s': %v", networkTemplateFile, err)
}
Expand Down
20 changes: 12 additions & 8 deletions netdeploy/remote/deployedNetwork.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ package remote
import (
"encoding/binary"
"encoding/json"
"errors"
"fmt"
"io/fs"
"math/rand"
Expand Down Expand Up @@ -58,14 +59,14 @@ var ErrDeployedNetworkInsufficientHosts = fmt.Errorf("target network requires mo
// ErrDeployedNetworkNameCantIncludeWildcard is returned by Validate if network name contains '*'
var ErrDeployedNetworkNameCantIncludeWildcard = fmt.Errorf("network name cannont include wild-cards")

// ErrDeployedNetworkTemplate A template file contained {{Field}} sections that were not handled by a corresponding Field value in configuration.
type ErrDeployedNetworkTemplate struct {
UnhandledTemplate string
// deployedNetworkTemplateError A template file contained {{Field}} sections that were not handled by a corresponding Field value in configuration.
type deployedNetworkTemplateError struct {
unhandledTemplate string
}

// Error satisfies error interface
func (ednt ErrDeployedNetworkTemplate) Error() string {
return fmt.Sprintf("config file contains unrecognized token: %s", ednt.UnhandledTemplate)
func (dnte deployedNetworkTemplateError) Error() string {
return fmt.Sprintf("config file contains unrecognized token: %s", dnte.unhandledTemplate)
}

// DeployedNetworkConfig represents the complete configuration specification for a deployed network
Expand Down Expand Up @@ -123,10 +124,13 @@ int 1
`

// InitDeployedNetworkConfig loads the DeployedNetworkConfig from a file
func InitDeployedNetworkConfig(file string, buildConfig BuildConfig) (cfg DeployedNetworkConfig, err error) {
func InitDeployedNetworkConfig(file string, buildConfig BuildConfig, ignoreUnkTokens bool) (cfg DeployedNetworkConfig, err error) {
processedFile, err := loadAndProcessConfig(file, buildConfig)
if err != nil {
return
var dnte deployedNetworkTemplateError
if !errors.As(err, &dnte) || !ignoreUnkTokens {
return
}
}

err = json.Unmarshal([]byte(processedFile), &cfg)
Expand Down Expand Up @@ -178,7 +182,7 @@ func replaceTokens(original string, buildConfig BuildConfig) (expanded string, e
if closeIndex < 0 {
closeIndex = len(expanded) - 2
}
return "", ErrDeployedNetworkTemplate{expanded[openIndex : closeIndex+2]}
return expanded, deployedNetworkTemplateError{expanded[openIndex : closeIndex+2]}
}

return
Expand Down

0 comments on commit 94f1355

Please sign in to comment.