Skip to content

Commit

Permalink
Add an unified .Site.Config with a services section
Browse files Browse the repository at this point in the history
  • Loading branch information
bep committed May 22, 2018
1 parent 966f68a commit e94f9fd
Show file tree
Hide file tree
Showing 4 changed files with 159 additions and 8 deletions.
61 changes: 61 additions & 0 deletions config/services/servicesConfig.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
// Copyright 2018 The Hugo Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package services

import (
"github.com/gohugoio/hugo/config"
"github.com/mitchellh/mapstructure"
)

const (
servicesConfigKey = "services"

disqusShortnameKey = "disqusshortname"
googleAnalyticsKey = "googleanalytics"
)

// Config is a privacy configuration for all the relevant services in Hugo.
type Config struct {
Disqus Disqus
GoogleAnalytics GoogleAnalytics
}

// Disqus holds the functional configuration settings related to the Disqus template.
type Disqus struct {
// A Shortname is the unique identifier assigned to a Disqus site.
Shortname string
}

// GoogleAnalytics holds the functional configuration settings related to the Google Analytics template.
type GoogleAnalytics struct {
// The GA tracking ID.
ID string
}

func DecodeConfig(cfg config.Provider) (c Config, err error) {
m := cfg.GetStringMap(servicesConfigKey)

err = mapstructure.WeakDecode(m, &c)

// Keep backwards compability.
if c.GoogleAnalytics.ID == "" {
// Try the global config
c.GoogleAnalytics.ID = cfg.GetString(googleAnalyticsKey)
}
if c.Disqus.Shortname == "" {
c.Disqus.Shortname = cfg.GetString(disqusShortnameKey)
}

return
}
64 changes: 64 additions & 0 deletions config/services/servicesConfig_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,64 @@
// Copyright 2018 The Hugo Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package services

import (
"testing"

"github.com/gohugoio/hugo/config"
"github.com/spf13/viper"
"github.com/stretchr/testify/require"
)

func TestDecodeConfigFromTOML(t *testing.T) {
assert := require.New(t)

tomlConfig := `
someOtherValue = "foo"
[services]
[services.disqus]
shortname = "DS"
[services.googleAnalytics]
id = "ga_id"
`
cfg, err := config.FromConfigString(tomlConfig, "toml")
assert.NoError(err)

config, err := DecodeConfig(cfg)
assert.NoError(err)
assert.NotNil(config)

assert.Equal("DS", config.Disqus.Shortname)
assert.Equal("ga_id", config.GoogleAnalytics.ID)

}

// Support old root-level GA settings etc.
func TestUseSettingsFromRootIfSet(t *testing.T) {
assert := require.New(t)

cfg := viper.New()
cfg.Set("disqusShortname", "root_short")
cfg.Set("googleAnalytics", "ga_root")

config, err := DecodeConfig(cfg)
assert.NoError(err)
assert.NotNil(config)

assert.Equal("root_short", config.Disqus.Shortname)
assert.Equal("ga_root", config.GoogleAnalytics.ID)

}
21 changes: 21 additions & 0 deletions hugolib/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ import (

"github.com/gohugoio/hugo/config"
"github.com/gohugoio/hugo/config/privacy"
"github.com/gohugoio/hugo/config/services"
"github.com/gohugoio/hugo/helpers"
"github.com/spf13/afero"
"github.com/spf13/viper"
Expand All @@ -33,6 +34,26 @@ type SiteConfig struct {
// This contains all privacy related settings that can be used to
// make the YouTube template etc. GDPR compliant.
Privacy privacy.Config

// Services contains config for services such as Google Analytics etc.
Services services.Config
}

func loadSiteConfig(cfg config.Provider) (scfg SiteConfig, err error) {
privacyConfig, err := privacy.DecodeConfig(cfg)
if err != nil {
return
}

servicesConfig, err := services.DecodeConfig(cfg)
if err != nil {
return
}

scfg.Privacy = privacyConfig
scfg.Services = servicesConfig

return
}

// ConfigSourceDescriptor describes where to find the config (e.g. config.toml etc.).
Expand Down
21 changes: 13 additions & 8 deletions hugolib/site.go
Original file line number Diff line number Diff line change
Expand Up @@ -27,8 +27,6 @@ import (
"strings"
"time"

"github.com/gohugoio/hugo/config/privacy"

"github.com/gohugoio/hugo/resource"

"golang.org/x/sync/errgroup"
Expand Down Expand Up @@ -375,8 +373,6 @@ type SiteInfo struct {
RSSLink string
Author map[string]interface{}
LanguageCode string
DisqusShortname string
GoogleAnalytics string
Copyright string
LastChange time.Time
Permalinks PermalinkOverrides
Expand Down Expand Up @@ -421,6 +417,17 @@ func (s *SiteInfo) ServerPort() int {
return p
}

// GoogleAnalytics is kept here for historic reasons.
func (s *SiteInfo) GoogleAnalytics() string {
return s.Config.Services.GoogleAnalytics.ID

}

// DisqusShortname is kept here for historic reasons.
func (s *SiteInfo) DisqusShortname() string {
return s.Config.Services.Disqus.Shortname
}

// Used in tests.

type siteBuilderCfg struct {
Expand Down Expand Up @@ -1116,7 +1123,7 @@ func (s *Site) initializeSiteInfo() error {
}
}

privacyConfig, err := privacy.DecodeConfig(lang)
siteConfig, err := loadSiteConfig(lang)
if err != nil {
return err
}
Expand All @@ -1127,14 +1134,12 @@ func (s *Site) initializeSiteInfo() error {
Social: lang.GetStringMapString("social"),
LanguageCode: lang.GetString("languageCode"),
Copyright: lang.GetString("copyright"),
DisqusShortname: lang.GetString("disqusShortname"),
multilingual: multilingual,
Language: lang,
LanguagePrefix: languagePrefix,
Languages: languages,
defaultContentLanguageInSubdir: defaultContentInSubDir,
sectionPagesMenu: lang.GetString("sectionPagesMenu"),
GoogleAnalytics: lang.GetString("googleAnalytics"),
BuildDrafts: s.Cfg.GetBool("buildDrafts"),
canonifyURLs: s.Cfg.GetBool("canonifyURLs"),
relativeURLs: s.Cfg.GetBool("relativeURLs"),
Expand All @@ -1147,7 +1152,7 @@ func (s *Site) initializeSiteInfo() error {
Data: &s.Data,
owner: s.owner,
s: s,
Config: SiteConfig{Privacy: privacyConfig},
Config: siteConfig,
}

rssOutputFormat, found := s.outputFormats[KindHome].GetByName(output.RSSFormat.Name)
Expand Down

0 comments on commit e94f9fd

Please sign in to comment.