hugolib, media: Make the MediaType available to the templates

diff --git a/hugolib/page_output.go b/hugolib/page_output.go
index 5fdb87057c707ceabd51251ea3ba94d2f979ff88..2d441fd4493fea78599590123b1b069f77e51ce5 100644 (file)
--- a/hugolib/page_output.go
+++ b/hugolib/page_output.go
@@ -19,6 +19,8 @@ import (
        "strings"
        "sync"
 
+       "github.com/spf13/hugo/media"
+
        "github.com/spf13/hugo/output"
 )
 
@@ -133,6 +135,7 @@ type OutputFormat struct {
 
        // It may be tempting to export this, but let us hold on to that horse for a while.
        f output.Format
+
        p *Page
 }
 
@@ -141,6 +144,11 @@ func (o OutputFormat) Name() string {
        return o.f.Name
 }
 
+// MediaType returns this OutputFormat's MediaType (MIME type).
+func (o OutputFormat) MediaType() media.Type {
+       return o.f.MediaType
+}
+
 // TODO(bep) outputs consider just save this wrapper on Page.
 // OutputFormats gives the output formats for this Page.
 func (p *Page) OutputFormats() OutputFormats {

diff --git a/media/mediaType.go b/media/mediaType.go
index 877404ddceecf834fda243cf987087cc28cdb576..45b850077356ead397e90d9fbcfdb2d8407d3fd1 100644 (file)
--- a/media/mediaType.go
+++ b/media/mediaType.go
@@ -26,23 +26,24 @@ type Types []Type
 // If suffix is not provided, the sub type will be used.
 // See // https://en.wikipedia.org/wiki/Media_type
 type Type struct {
-       Type    string // i.e. text
-       SubType string // i.e. html
-       Suffix  string // i.e html
+       MainType string // i.e. text
+       SubType  string // i.e. html
+       Suffix   string // i.e html
 }
 
-// Key return a key used to identify this media type. Hugo will register a set of
-// default media types. These can be overridden by the user in the configuration,
-// by defining a media type with the same Key.
-func (m Type) Key() string {
-       return fmt.Sprintf("%s/%s", m.Type, m.SubType)
+// Type returns a string representing the main- and sub-type of a media type, i.e. "text/css".
+// Hugo will register a set of default media types.
+// These can be overridden by the user in the configuration,
+// by defining a media type with the same Type.
+func (m Type) Type() string {
+       return fmt.Sprintf("%s/%s", m.MainType, m.SubType)
 }
 
 func (m Type) String() string {
        if m.Suffix != "" {
-               return fmt.Sprintf("%s/%s+%s", m.Type, m.SubType, m.Suffix)
+               return fmt.Sprintf("%s/%s+%s", m.MainType, m.SubType, m.Suffix)
        }
-       return fmt.Sprintf("%s/%s", m.Type, m.SubType)
+       return fmt.Sprintf("%s/%s", m.MainType, m.SubType)
 }
 
 var (

diff --git a/media/mediaType_test.go b/media/mediaType_test.go
index 64a0a6567b9f012e497e6495ca0b1e3fe399922d..41cc48369ed571e0a8422d09367a9eb97cf50156 100644 (file)
--- a/media/mediaType_test.go
+++ b/media/mediaType_test.go
@@ -20,18 +20,18 @@ import (
 )
 
 func TestDefaultTypes(t *testing.T) {
-       require.Equal(t, "text", HTMLType.Type)
+       require.Equal(t, "text", HTMLType.MainType)
        require.Equal(t, "html", HTMLType.SubType)
        require.Equal(t, "html", HTMLType.Suffix)
 
-       require.Equal(t, "text/html", HTMLType.Key())
+       require.Equal(t, "text/html", HTMLType.MainType())
        require.Equal(t, "text/html+html", HTMLType.String())
 
-       require.Equal(t, "application", RSSType.Type)
+       require.Equal(t, "application", RSSType.MainType)
        require.Equal(t, "rss", RSSType.SubType)
        require.Equal(t, "xml", RSSType.Suffix)
 
-       require.Equal(t, "application/rss", RSSType.Key())
+       require.Equal(t, "application/rss", RSSType.MainType())
        require.Equal(t, "application/rss+xml", RSSType.String())
 
 }