Improved builder globals

Author: cmaglie

arduino/globals/globals.go

@@ -16,33 +16,31 @@
 package globals
 
 var (
-	empty struct{}
-
 	// MainFileValidExtension is the extension that must be used for files in new sketches
 	MainFileValidExtension string = ".ino"
 
 	// MainFileValidExtensions lists valid extensions for a sketch file
-	MainFileValidExtensions = map[string]struct{}{
-		MainFileValidExtension: empty,
+	MainFileValidExtensions = map[string]bool{
+		MainFileValidExtension: true,
 		// .pde extension is deprecated and must not be used for new sketches
-		".pde": empty,
+		".pde": true,
 	}
 
 	// AdditionalFileValidExtensions lists any file extension the builder considers as valid
-	AdditionalFileValidExtensions = map[string]struct{}{
-		".h":    empty,
-		".c":    empty,
-		".hpp":  empty,
-		".hh":   empty,
-		".cpp":  empty,
-		".cxx":  empty,
-		".cc":   empty,
-		".S":    empty,
-		".adoc": empty,
-		".md":   empty,
-		".json": empty,
-		".tpp":  empty,
-		".ipp":  empty,
+	AdditionalFileValidExtensions = map[string]bool{
+		".h":    true,
+		".c":    true,
+		".hpp":  true,
+		".hh":   true,
+		".cpp":  true,
+		".cxx":  true,
+		".cc":   true,
+		".S":    true,
+		".adoc": true,
+		".md":   true,
+		".json": true,
+		".tpp":  true,
+		".ipp":  true,
 	}
 
 	// SourceFilesValidExtensions lists valid extensions for source files (no headers).
@@ -57,10 +55,10 @@ var (
 	}
 
 	// HeaderFilesValidExtensions lists valid extensions for header files
-	HeaderFilesValidExtensions = map[string]struct{}{
-		".h":   empty,
-		".hpp": empty,
-		".hh":  empty,
+	HeaderFilesValidExtensions = map[string]bool{
+		".h":   true,
+		".hpp": true,
+		".hh":  true,
 	}
 
 	// DefaultIndexURL is the default index url

arduino/sketch/sketch.go

@@ -54,7 +54,7 @@ func New(path *paths.Path) (*Sketch, error) {
 	} else if !exist {
 		return nil, fmt.Errorf("%s: %s", tr("no such file or directory"), path)
 	}
-	if _, validIno := globals.MainFileValidExtensions[path.Ext()]; validIno && !path.IsDir() {
+	if globals.MainFileValidExtensions[path.Ext()] && !path.IsDir() {
 		path = path.Parent()
 	}
 
@@ -121,7 +121,7 @@ func New(path *paths.Path) (*Sketch, error) {
 		f.Close()
 
 		ext := p.Ext()
-		if _, found := globals.MainFileValidExtensions[ext]; found {
+		if globals.MainFileValidExtensions[ext] {
 			if p.EqualsTo(mainFile) {
 				// The main file must not be included in the lists of other files
 				continue
@@ -132,7 +132,7 @@ func New(path *paths.Path) (*Sketch, error) {
 				sketch.OtherSketchFiles.Add(p)
 				sketch.RootFolderFiles.Add(p)
 			}
-		} else if _, found := globals.AdditionalFileValidExtensions[ext]; found {
+		} else if globals.AdditionalFileValidExtensions[ext] {
 			// If the user exported the compiles binaries to the Sketch "build" folder
 			// they would be picked up but we don't want them, so we skip them like so
 			if p.IsInsideDir(sketch.FullPath.Join("build")) {
@@ -158,12 +158,8 @@ func New(path *paths.Path) (*Sketch, error) {
 // supportedFiles reads all files recursively contained in Sketch and
 // filter out unneded or unsupported ones and returns them
 func (s *Sketch) supportedFiles() (*paths.PathList, error) {
-	validExtensions := []string{}
-	for ext := range globals.MainFileValidExtensions {
-		validExtensions = append(validExtensions, ext)
-	}
-	for ext := range globals.AdditionalFileValidExtensions {
-		validExtensions = append(validExtensions, ext)
+	filterValidExtensions := func(p *paths.Path) bool {
+		return globals.MainFileValidExtensions[p.Ext()] || globals.AdditionalFileValidExtensions[p.Ext()]
 	}
 
 	filterOutBuildPaths := func(p *paths.Path) bool {
@@ -174,7 +170,7 @@ func (s *Sketch) supportedFiles() (*paths.PathList, error) {
 		filterOutBuildPaths,
 		paths.AndFilter(
 			paths.FilterOutPrefixes("."),
-			paths.FilterSuffixes(validExtensions...),
+			filterValidExtensions,
 			paths.FilterOutDirectories(),
 		),
 	)

commands/upload/upload.go

@@ -752,7 +752,7 @@ func detectSketchNameFromBuildPath(buildPath *paths.Path) (string, error) {
 
 		// Sometimes we may have particular files like:
 		// Blink.ino.with_bootloader.bin
-		if _, ok := globals.MainFileValidExtensions[filepath.Ext(name)]; !ok {
+		if globals.MainFileValidExtensions[filepath.Ext(name)] {
 			// just ignore those files
 			continue
 		}