Skip to content

Commit

Permalink
resource: Implement Resource.Content
Browse files Browse the repository at this point in the history
Fixes #4622
  • Loading branch information
bep committed Apr 15, 2018
1 parent 417c5e2 commit 0e7716a
Show file tree
Hide file tree
Showing 3 changed files with 48 additions and 0 deletions.
1 change: 1 addition & 0 deletions resource/image.go
Expand Up @@ -568,6 +568,7 @@ func (i *Image) encodeToDestinations(img image.Image, conf imageConfig, resource

func (i *Image) clone() *Image {
g := *i.genericResource
g.resourceContent = &resourceContent{}

return &Image{
imaging: i.imaging,
Expand Down
12 changes: 12 additions & 0 deletions resource/image_test.go
Expand Up @@ -322,6 +322,18 @@ func TestSVGImage(t *testing.T) {
assert.NotNil(svg)
}

func TestSVGImageContent(t *testing.T) {
assert := require.New(t)
spec := newTestResourceSpec(assert)
svg := fetchResourceForSpec(spec, assert, "circle.svg")
assert.NotNil(svg)

content, err := svg.Content()
assert.NoError(err)
assert.IsType("", content)
assert.Contains(content.(string), `<svg height="100" width="100">`)
}

func BenchmarkResizeParallel(b *testing.B) {
assert := require.New(b)
img := fetchSunset(assert)
Expand Down
35 changes: 35 additions & 0 deletions resource/resource.go
Expand Up @@ -87,6 +87,14 @@ type Resource interface {

// Params set in front matter for this resource.
Params() map[string]interface{}

// Content returns this resource's content. It will be equivalent to reading the content
// that RelPermalink points to in the published folder.
// The return type will be contextual, and should be what you would expect:
// * Page: template.HTML
// * JSON: String
// * Etc.
Content() (interface{}, error)
}

// Resources represents a slice of resources, which can be a mix of different types.
Expand Down Expand Up @@ -360,6 +368,11 @@ func (d dirFile) path() string {
return path.Join(d.dir, d.file)
}

type resourceContent struct {
content string
contentInit sync.Once
}

// genericResource represents a generic linkable resource.
type genericResource struct {
// The relative path to this resource.
Expand Down Expand Up @@ -390,6 +403,26 @@ type genericResource struct {
osFileInfo os.FileInfo

targetPathBuilder func(rel string) string

// We create copies of this struct, so this needs to be a pointer.
*resourceContent
}

func (l *genericResource) Content() (interface{}, error) {
var err error
l.contentInit.Do(func() {
var b []byte

b, err := afero.ReadFile(l.sourceFs(), l.AbsSourceFilename())
if err != nil {
return
}

l.content = string(b)

})

return l.content, err
}

func (l *genericResource) sourceFs() afero.Fs {
Expand Down Expand Up @@ -444,6 +477,7 @@ func (l *genericResource) updateParams(params map[string]interface{}) {
// Implement the Cloner interface.
func (l genericResource) WithNewBase(base string) Resource {
l.base = base
l.resourceContent = &resourceContent{}
return &l
}

Expand Down Expand Up @@ -611,5 +645,6 @@ func (r *Spec) newGenericResource(
params: make(map[string]interface{}),
name: baseFilename,
title: baseFilename,
resourceContent: &resourceContent{},
}
}

0 comments on commit 0e7716a

Please sign in to comment.