From 46eaa2548f86c2681e6b2af0435f42d82376f22e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bj=C3=B8rn=20Erik=20Pedersen?= Date: Thu, 29 Oct 2020 17:14:04 +0100 Subject: [PATCH] Add module.replacements Fixes #7904 Fixes #7906 --- modules/collect.go | 14 +++++--- modules/config.go | 49 ++++++++++++++++++++++++++++ modules/config_test.go | 72 ++++++++++++++++++++++++++++++------------ 3 files changed, 111 insertions(+), 24 deletions(-) diff --git a/modules/collect.go b/modules/collect.go index 3059d3f99f5..60d3445bb5c 100644 --- a/modules/collect.go +++ b/modules/collect.go @@ -274,10 +274,17 @@ func (c *collector) add(owner *moduleAdapter, moduleImport Import, disabled bool } } - // Fall back to /themes/ + if moduleDir == "" && strings.HasPrefix(modulePath, ".") { + // Try first relative to the project. + moduleDir = filepath.Join(c.ccfg.WorkingDir, modulePath) + if found, _ := afero.Exists(c.fs, moduleDir); !found { + moduleDir = "" + } + } + if moduleDir == "" { + // Finally: Try themes dir moduleDir = filepath.Join(c.ccfg.ThemesDir, modulePath) - if found, _ := afero.Exists(c.fs, moduleDir); !found { c.err = c.wrapModuleNotFound(errors.Errorf(`module %q not found; either add it as a Hugo Module or store it in %q.`, modulePath, c.ccfg.ThemesDir)) return nil, nil @@ -441,7 +448,7 @@ func (c *collector) applyThemeConfig(tc *moduleAdapter) error { tc.cfg = cfg } - config, err := DecodeConfig(cfg) + config, err := decodeConfig(cfg, c.moduleConfig.replacementsMap) if err != nil { return err } @@ -605,7 +612,6 @@ func (c *collector) normalizeMounts(owner *moduleAdapter, mounts []Mount) ([]Mou mnt.Source = filepath.Clean(mnt.Source) mnt.Target = filepath.Clean(mnt.Target) - var sourceDir string if owner.projectMod && filepath.IsAbs(mnt.Source) { diff --git a/modules/config.go b/modules/config.go index e0a0ea060cd..1ce8c9f02be 100644 --- a/modules/config.go +++ b/modules/config.go @@ -18,6 +18,8 @@ import ( "path/filepath" "strings" + "github.com/pkg/errors" + "github.com/gohugoio/hugo/common/hugo" "github.com/gohugoio/hugo/config" @@ -40,6 +42,14 @@ var DefaultModuleConfig = Config{ // Comma separated glob list matching paths that should be // treated as private. Private: "*.*", + + // A list of replacement directives mapping a module path to a directory + // or a theme component in the themes folder. + // Note that this will turn the component into a traditional theme component + // that does not partake in vendoring etc. + // The syntax is the similar to the replacement directives used in go.mod, e.g: + // github.com/mod1 -> ../mod1,github.com/mod2 -> ../mod2 + Replacements: nil, } // ApplyProjectConfigDefaults applies default/missing module configuration for @@ -182,7 +192,12 @@ func ApplyProjectConfigDefaults(cfg config.Provider, mod Module) error { // DecodeConfig creates a modules Config from a given Hugo configuration. func DecodeConfig(cfg config.Provider) (Config, error) { + return decodeConfig(cfg, nil) +} + +func decodeConfig(cfg config.Provider, pathReplacements map[string]string) (Config, error) { c := DefaultModuleConfig + c.replacementsMap = pathReplacements if cfg == nil { return c, nil @@ -197,6 +212,37 @@ func DecodeConfig(cfg config.Provider) (Config, error) { return c, err } + if c.replacementsMap == nil { + + if len(c.Replacements) == 1 { + c.Replacements = strings.Split(c.Replacements[0], ",") + } + + for i, repl := range c.Replacements { + c.Replacements[i] = strings.TrimSpace(repl) + } + + c.replacementsMap = make(map[string]string) + for _, repl := range c.Replacements { + parts := strings.Split(repl, "->") + if len(parts) != 2 { + return c, errors.Errorf(`invalid module.replacements: %q; configure replacement pairs on the form "oldpath->newpath" `, repl) + } + + c.replacementsMap[strings.TrimSpace(parts[0])] = strings.TrimSpace(parts[1]) + } + } + + if c.replacementsMap != nil && c.Imports != nil { + for i, imp := range c.Imports { + if newImp, found := c.replacementsMap[imp.Path]; found { + imp.Path = newImp + c.Imports[i] = imp + } + } + + } + for i, mnt := range c.Mounts { mnt.Source = filepath.Clean(mnt.Source) mnt.Target = filepath.Clean(mnt.Target) @@ -233,6 +279,9 @@ type Config struct { // "github.com/**". NoVendor string + Replacements []string + replacementsMap map[string]string + // Configures GOPROXY. Proxy string // Configures GONOPROXY. diff --git a/modules/config_test.go b/modules/config_test.go index 60fa9586ea3..dd9dbc22f86 100644 --- a/modules/config_test.go +++ b/modules/config_test.go @@ -41,7 +41,9 @@ func TestConfigHugoVersionIsValid(t *testing.T) { func TestDecodeConfig(t *testing.T) { c := qt.New(t) - tomlConfig := ` + + c.Run("Basic", func(c *qt.C) { + tomlConfig := ` [module] [module.hugoVersion] @@ -63,31 +65,61 @@ source="src/markdown/blog" target="content/blog" lang="en" ` - cfg, err := config.FromConfigString(tomlConfig, "toml") - c.Assert(err, qt.IsNil) + cfg, err := config.FromConfigString(tomlConfig, "toml") + c.Assert(err, qt.IsNil) - mcfg, err := DecodeConfig(cfg) - c.Assert(err, qt.IsNil) + mcfg, err := DecodeConfig(cfg) + c.Assert(err, qt.IsNil) - v056 := hugo.VersionString("0.56.0") + v056 := hugo.VersionString("0.56.0") - hv := mcfg.HugoVersion + hv := mcfg.HugoVersion - c.Assert(v056.Compare(hv.Min), qt.Equals, -1) - c.Assert(v056.Compare(hv.Max), qt.Equals, 1) - c.Assert(hv.Extended, qt.Equals, true) + c.Assert(v056.Compare(hv.Min), qt.Equals, -1) + c.Assert(v056.Compare(hv.Max), qt.Equals, 1) + c.Assert(hv.Extended, qt.Equals, true) - if hugo.IsExtended { - c.Assert(hv.IsValid(), qt.Equals, true) - } + if hugo.IsExtended { + c.Assert(hv.IsValid(), qt.Equals, true) + } + + c.Assert(len(mcfg.Mounts), qt.Equals, 1) + c.Assert(len(mcfg.Imports), qt.Equals, 1) + imp := mcfg.Imports[0] + imp.Path = "github.com/bep/mycomponent" + c.Assert(imp.Mounts[1].Source, qt.Equals, "src/markdown/blog") + c.Assert(imp.Mounts[1].Target, qt.Equals, "content/blog") + c.Assert(imp.Mounts[1].Lang, qt.Equals, "en") + }) + + c.Run("Replacements", func(c *qt.C) { + for _, tomlConfig := range []string{` +[module] +replacements="a->b,github.com/bep/mycomponent->c" +[[module.imports]] +path="github.com/bep/mycomponent" +`, ` +[module] +replacements=["a->b","github.com/bep/mycomponent->c"] +[[module.imports]] +path="github.com/bep/mycomponent" +`} { + + cfg, err := config.FromConfigString(tomlConfig, "toml") + c.Assert(err, qt.IsNil) + + mcfg, err := DecodeConfig(cfg) + c.Assert(err, qt.IsNil) + c.Assert(mcfg.Replacements, qt.DeepEquals, []string{"a->b", "github.com/bep/mycomponent->c"}) + c.Assert(mcfg.replacementsMap, qt.DeepEquals, map[string]string{ + "a": "b", + "github.com/bep/mycomponent": "c", + }) + + c.Assert(mcfg.Imports[0].Path, qt.Equals, "c") - c.Assert(len(mcfg.Mounts), qt.Equals, 1) - c.Assert(len(mcfg.Imports), qt.Equals, 1) - imp := mcfg.Imports[0] - imp.Path = "github.com/bep/mycomponent" - c.Assert(imp.Mounts[1].Source, qt.Equals, "src/markdown/blog") - c.Assert(imp.Mounts[1].Target, qt.Equals, "content/blog") - c.Assert(imp.Mounts[1].Lang, qt.Equals, "en") + } + }) }