2015-12-07 18:57:01 +00:00
|
|
|
// Copyright 2015 The Hugo Authors. All rights reserved.
|
2014-10-17 00:20:09 +00:00
|
|
|
//
|
2015-11-24 03:16:36 +00:00
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
2014-10-17 00:20:09 +00:00
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
2015-11-24 03:16:36 +00:00
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
2014-10-17 00:20:09 +00:00
|
|
|
//
|
|
|
|
// 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 source
|
|
|
|
|
|
|
|
import (
|
|
|
|
"io"
|
|
|
|
"path/filepath"
|
|
|
|
"strings"
|
2016-01-08 02:48:13 +00:00
|
|
|
|
2017-02-05 03:20:06 +00:00
|
|
|
"github.com/spf13/hugo/hugofs"
|
|
|
|
|
|
|
|
"github.com/spf13/hugo/config"
|
2016-01-08 02:48:13 +00:00
|
|
|
"github.com/spf13/hugo/helpers"
|
2014-10-17 00:20:09 +00:00
|
|
|
)
|
|
|
|
|
2017-02-05 03:20:06 +00:00
|
|
|
type SourceSpec struct {
|
|
|
|
Cfg config.Provider
|
|
|
|
Fs *hugofs.Fs
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewSourceSpec(cfg config.Provider, fs *hugofs.Fs) SourceSpec {
|
|
|
|
return SourceSpec{Cfg: cfg, Fs: fs}
|
|
|
|
}
|
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// File represents a source content file.
|
2016-01-08 02:48:13 +00:00
|
|
|
// All paths are relative from the source directory base
|
2014-10-17 00:20:09 +00:00
|
|
|
type File struct {
|
2016-10-28 08:35:19 +00:00
|
|
|
relpath string // Original relative path, e.g. section/foo.txt
|
2014-10-17 00:20:09 +00:00
|
|
|
logicalName string // foo.txt
|
2016-05-14 04:35:16 +00:00
|
|
|
baseName string // `post` for `post.md`, also `post.en` for `post.en.md`
|
2014-10-17 00:20:09 +00:00
|
|
|
Contents io.Reader
|
|
|
|
section string // The first directory
|
2016-04-07 13:44:53 +00:00
|
|
|
dir string // The relative directory Path (minus file name)
|
2014-10-17 00:20:09 +00:00
|
|
|
ext string // Just the ext (eg txt)
|
2016-12-31 16:55:03 +00:00
|
|
|
uniqueID string // MD5 of the file's path
|
2016-05-14 04:35:16 +00:00
|
|
|
|
|
|
|
translationBaseName string // `post` for `post.es.md` (if `Multilingual` is enabled.)
|
|
|
|
lang string // The language code if `Multilingual` is enabled
|
2014-10-17 00:20:09 +00:00
|
|
|
}
|
|
|
|
|
2016-12-31 16:55:03 +00:00
|
|
|
// UniqueID is the MD5 hash of the file's path and is for most practical applications,
|
2016-04-07 13:44:53 +00:00
|
|
|
// Hugo content files being one of them, considered to be unique.
|
2015-03-11 17:34:57 +00:00
|
|
|
func (f *File) UniqueID() string {
|
|
|
|
return f.uniqueID
|
2014-10-17 00:20:09 +00:00
|
|
|
}
|
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// String returns the file's content as a string.
|
2014-11-04 05:28:20 +00:00
|
|
|
func (f *File) String() string {
|
|
|
|
return helpers.ReaderToString(f.Contents)
|
|
|
|
}
|
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// Bytes returns the file's content as a byte slice.
|
2014-11-04 05:28:20 +00:00
|
|
|
func (f *File) Bytes() []byte {
|
|
|
|
return helpers.ReaderToBytes(f.Contents)
|
|
|
|
}
|
|
|
|
|
2016-11-21 22:01:42 +00:00
|
|
|
// BaseFileName is a filename without extension.
|
2014-10-17 00:20:09 +00:00
|
|
|
func (f *File) BaseFileName() string {
|
2016-05-14 04:35:16 +00:00
|
|
|
return f.baseName
|
|
|
|
}
|
|
|
|
|
2016-11-21 22:01:42 +00:00
|
|
|
// TranslationBaseName is a filename with no extension,
|
|
|
|
// not even the optional language extension part.
|
2016-05-14 04:35:16 +00:00
|
|
|
func (f *File) TranslationBaseName() string {
|
|
|
|
return f.translationBaseName
|
|
|
|
}
|
|
|
|
|
|
|
|
// Lang for this page, if `Multilingual` is enabled on your site.
|
|
|
|
func (f *File) Lang() string {
|
|
|
|
return f.lang
|
2014-10-17 00:20:09 +00:00
|
|
|
}
|
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// Section is first directory below the content root.
|
2014-10-17 00:20:09 +00:00
|
|
|
func (f *File) Section() string {
|
2015-03-07 11:58:39 +00:00
|
|
|
return f.section
|
2014-10-17 00:20:09 +00:00
|
|
|
}
|
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// LogicalName is filename and extension of the file.
|
2014-10-17 00:20:09 +00:00
|
|
|
func (f *File) LogicalName() string {
|
2015-03-07 11:58:39 +00:00
|
|
|
return f.logicalName
|
2014-10-17 00:20:09 +00:00
|
|
|
}
|
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// SetDir sets the relative directory where this file lives.
|
|
|
|
// TODO(bep) Get rid of this.
|
2014-11-17 22:18:40 +00:00
|
|
|
func (f *File) SetDir(dir string) {
|
|
|
|
f.dir = dir
|
|
|
|
}
|
2014-10-17 00:20:09 +00:00
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// Dir gets the name of the directory that contains this file.
|
|
|
|
// The directory is relative to the content root.
|
2014-10-17 00:20:09 +00:00
|
|
|
func (f *File) Dir() string {
|
2015-03-07 11:58:39 +00:00
|
|
|
return f.dir
|
2014-10-17 00:20:09 +00:00
|
|
|
}
|
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// Extension gets the file extension, i.e "myblogpost.md" will return "md".
|
2014-10-17 00:20:09 +00:00
|
|
|
func (f *File) Extension() string {
|
2015-03-07 11:58:39 +00:00
|
|
|
return f.ext
|
2014-10-17 00:20:09 +00:00
|
|
|
}
|
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// Ext is an alias for Extension.
|
2014-10-17 00:20:09 +00:00
|
|
|
func (f *File) Ext() string {
|
|
|
|
return f.Extension()
|
|
|
|
}
|
|
|
|
|
2016-10-28 08:35:19 +00:00
|
|
|
// Path gets the relative path including file name and extension.
|
|
|
|
// The directory is relative to the content root.
|
2014-10-17 00:20:09 +00:00
|
|
|
func (f *File) Path() string {
|
|
|
|
return f.relpath
|
|
|
|
}
|
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// NewFileWithContents creates a new File pointer with the given relative path and
|
2016-07-28 07:30:58 +00:00
|
|
|
// content. The language defaults to "en".
|
2017-02-05 03:20:06 +00:00
|
|
|
func (sp SourceSpec) NewFileWithContents(relpath string, content io.Reader) *File {
|
|
|
|
file := sp.NewFile(relpath)
|
2014-10-17 00:20:09 +00:00
|
|
|
file.Contents = content
|
2016-07-28 07:30:58 +00:00
|
|
|
file.lang = "en"
|
2014-10-17 00:20:09 +00:00
|
|
|
return file
|
|
|
|
}
|
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// NewFile creates a new File pointer with the given relative path.
|
2017-02-05 03:20:06 +00:00
|
|
|
func (sp SourceSpec) NewFile(relpath string) *File {
|
2015-07-21 18:02:42 +00:00
|
|
|
f := &File{
|
2014-10-17 00:20:09 +00:00
|
|
|
relpath: relpath,
|
|
|
|
}
|
2015-07-21 18:02:42 +00:00
|
|
|
|
2016-03-12 18:16:31 +00:00
|
|
|
f.dir, f.logicalName = filepath.Split(f.relpath)
|
2015-07-21 18:02:42 +00:00
|
|
|
f.ext = strings.TrimPrefix(filepath.Ext(f.LogicalName()), ".")
|
2016-05-14 04:35:16 +00:00
|
|
|
f.baseName = helpers.Filename(f.LogicalName())
|
2016-07-28 07:30:58 +00:00
|
|
|
|
2016-11-01 13:18:24 +00:00
|
|
|
lang := strings.TrimPrefix(filepath.Ext(f.baseName), ".")
|
2017-02-05 03:20:06 +00:00
|
|
|
if _, ok := sp.Cfg.GetStringMap("languages")[lang]; lang == "" || !ok {
|
|
|
|
f.lang = sp.Cfg.GetString("defaultContentLanguage")
|
2016-11-01 13:18:24 +00:00
|
|
|
f.translationBaseName = f.baseName
|
|
|
|
} else {
|
|
|
|
f.lang = lang
|
|
|
|
f.translationBaseName = helpers.Filename(f.baseName)
|
2016-05-14 04:35:16 +00:00
|
|
|
}
|
|
|
|
|
2015-07-21 18:02:42 +00:00
|
|
|
f.section = helpers.GuessSection(f.Dir())
|
2016-12-31 16:46:11 +00:00
|
|
|
f.uniqueID = helpers.Md5String(f.Path())
|
2015-07-21 18:02:42 +00:00
|
|
|
|
|
|
|
return f
|
2014-10-17 00:20:09 +00:00
|
|
|
}
|
|
|
|
|
2016-04-07 13:44:53 +00:00
|
|
|
// NewFileFromAbs creates a new File pointer with the given full file path path and
|
|
|
|
// content.
|
2017-02-05 03:20:06 +00:00
|
|
|
func (sp SourceSpec) NewFileFromAbs(base, fullpath string, content io.Reader) (f *File, err error) {
|
2014-10-17 00:20:09 +00:00
|
|
|
var name string
|
|
|
|
if name, err = helpers.GetRelativePath(fullpath, base); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2017-02-05 03:20:06 +00:00
|
|
|
return sp.NewFileWithContents(name, content), nil
|
2014-10-17 00:20:09 +00:00
|
|
|
}
|