fix(indexer): add validation to slug

internal/indexer/indexer.go

@@ -5,6 +5,7 @@ import (
 	"html/template"
 	"os"
 	"path/filepath"
+	"regexp"
 	"sort"
 
 	"github.com/waynezhang/foto/internal/config"
@@ -35,6 +36,9 @@ func Build(metadata []config.SectionMetadata, option config.ExtractOption) ([]Se
 
 	for _, val := range metadata {
 		slug := val.Slug
+		if !validSlug(slug) {
+			return nil, fmt.Errorf("Slug \"%s\" is invalid. Only letters([a-zA-Z]), numbers([09-]), underscore(_) and hyphen(-) can be used.", slug)
+		}
 		if slugs[slug] {
 			return nil, fmt.Errorf("Slug \"%s\" already exists. Slug needs to be unique.", slug)
 		}
@@ -113,3 +117,8 @@ func buildImageSet(path string, option config.ExtractOption) (*ImageSet, error)
 		},
 	}, nil
 }
+
+func validSlug(slug string) bool {
+	matched, _ := regexp.MatchString("^[a-zA-Z0-9-_]+$", slug)
+	return matched
+}

internal/indexer/indexer_test.go

@@ -18,6 +18,14 @@ var (
 	}
 )
 
+func TestValidSlug(t *testing.T) {
+	assert.True(t, validSlug("abcde-efg_9999"))
+	assert.False(t, validSlug("abcde efg_9999"))
+	assert.False(t, validSlug("abcde-efgかたかな"))
+	assert.False(t, validSlug("abcde.efg_999"))
+	assert.False(t, validSlug(""))
+}
+
 func TestBuild(t *testing.T) {
 	var meta1 config.SectionMetadata
 	var meta2 config.SectionMetadata