go.work and *.sql files can be also solution files

Author: roblaszczak

trainings/files/read.go

@@ -2,6 +2,7 @@ package files
 
 import (
 	"os"
+	"path"
 	"path/filepath"
 
 	"github.com/pkg/errors"
@@ -25,7 +26,7 @@ func (f Files) ReadSolutionFiles(trainingRootFs afero.Fs, dir string) ([]*genpro
 			if info.IsDir() {
 				return nil
 			}
-			if filepath.Ext(info.Name()) != ".go" && info.Name() != "go.mod" {
+			if !isSolutionFile(info) {
 				return nil
 			}
 
@@ -60,3 +61,20 @@ func (f Files) ReadSolutionFiles(trainingRootFs afero.Fs, dir string) ([]*genpro
 
 	return files, nil
 }
+
+func isSolutionFile(info os.FileInfo) bool {
+	if filepath.Ext(info.Name()) == ".go" {
+		return true
+	}
+	if info.Name() == "go.mod" {
+		return true
+	}
+	if info.Name() == "go.work" {
+		return true
+	}
+	if path.Ext(info.Name()) != ".sql" {
+		return true
+	}
+
+	return false
+}