shithub: hugo

Info  •  Files  •  Log  •  Branches

resource: Make GetByPrefix work for Page resources

Fixes #4264

--- a/hugolib/page_bundler_test.go

+++ b/hugolib/page_bundler_test.go

@@ -111,6 +111,10 @@

 				assert.Contains(firstPage.Content, "TheContent")

 				assert.Len(leafBundle1.Resources, 6) // 2 pages 3 images 1 custom mime type

+				assert.Equal(firstPage, pageResources.GetByPrefix("1"))

+				assert.Equal(secondPage, pageResources.GetByPrefix("2"))

+				assert.Nil(pageResources.GetByPrefix("doesnotexist"))

+

 				imageResources := leafBundle1.Resources.ByType("image")

 				assert.Len(imageResources, 3)

 				image := imageResources[0]

--- a/resource/resource.go

+++ b/resource/resource.go

@@ -72,8 +72,15 @@

 func (r Resources) GetByPrefix(prefix string) Resource {

 	prefix = strings.ToLower(prefix)

 	for _, resource := range r {

-		_, name := filepath.Split(resource.RelPermalink())

+		var name string

+		f, ok := resource.(source.File)

+		if ok {

+			name = f.BaseFileName()

+		} else {

+			_, name = filepath.Split(resource.RelPermalink())

+		}

 		name = strings.ToLower(name)

+

 		if strings.HasPrefix(name, prefix) {

 			return resource

 		}