mirror of
https://github.com/gohugoio/hugo.git
synced 2024-12-29 19:21:24 +00:00
122 lines
2.8 KiB
Go
122 lines
2.8 KiB
Go
|
// Copyright 2019 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 files
|
||
|
|
||
|
import (
|
||
|
"os"
|
||
|
"path/filepath"
|
||
|
"sort"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
// This should be the only list of valid extensions for content files.
|
||
|
contentFileExtensions = []string{
|
||
|
"html", "htm",
|
||
|
"mdown", "markdown", "md",
|
||
|
"asciidoc", "adoc", "ad",
|
||
|
"rest", "rst",
|
||
|
"mmark",
|
||
|
"org",
|
||
|
"pandoc", "pdc"}
|
||
|
|
||
|
contentFileExtensionsSet map[string]bool
|
||
|
)
|
||
|
|
||
|
func init() {
|
||
|
contentFileExtensionsSet = make(map[string]bool)
|
||
|
for _, ext := range contentFileExtensions {
|
||
|
contentFileExtensionsSet[ext] = true
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func IsContentFile(filename string) bool {
|
||
|
return contentFileExtensionsSet[strings.TrimPrefix(filepath.Ext(filename), ".")]
|
||
|
}
|
||
|
|
||
|
func IsContentExt(ext string) bool {
|
||
|
return contentFileExtensionsSet[ext]
|
||
|
}
|
||
|
|
||
|
const (
|
||
|
ContentClassLeaf = "leaf"
|
||
|
ContentClassBranch = "branch"
|
||
|
ContentClassFile = "zfile" // Sort below
|
||
|
ContentClassContent = "zcontent"
|
||
|
)
|
||
|
|
||
|
func ClassifyContentFile(filename string) string {
|
||
|
if !IsContentFile(filename) {
|
||
|
return ContentClassFile
|
||
|
}
|
||
|
if strings.HasPrefix(filename, "_index.") {
|
||
|
return ContentClassBranch
|
||
|
}
|
||
|
|
||
|
if strings.HasPrefix(filename, "index.") {
|
||
|
return ContentClassLeaf
|
||
|
}
|
||
|
|
||
|
return ContentClassContent
|
||
|
}
|
||
|
|
||
|
const (
|
||
|
ComponentFolderArchetypes = "archetypes"
|
||
|
ComponentFolderStatic = "static"
|
||
|
ComponentFolderLayouts = "layouts"
|
||
|
ComponentFolderContent = "content"
|
||
|
ComponentFolderData = "data"
|
||
|
ComponentFolderAssets = "assets"
|
||
|
ComponentFolderI18n = "i18n"
|
||
|
|
||
|
FolderResources = "resources"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
ComponentFolders = []string{
|
||
|
ComponentFolderArchetypes,
|
||
|
ComponentFolderStatic,
|
||
|
ComponentFolderLayouts,
|
||
|
ComponentFolderContent,
|
||
|
ComponentFolderData,
|
||
|
ComponentFolderAssets,
|
||
|
ComponentFolderI18n,
|
||
|
}
|
||
|
|
||
|
componentFoldersSet = make(map[string]bool)
|
||
|
)
|
||
|
|
||
|
func init() {
|
||
|
sort.Strings(ComponentFolders)
|
||
|
for _, f := range ComponentFolders {
|
||
|
componentFoldersSet[f] = true
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// ResolveComponentFolder returns "content" from "content/blog/foo.md" etc.
|
||
|
func ResolveComponentFolder(filename string) string {
|
||
|
filename = strings.TrimPrefix(filename, string(os.PathSeparator))
|
||
|
for _, cf := range ComponentFolders {
|
||
|
if strings.HasPrefix(filename, cf) {
|
||
|
return cf
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return ""
|
||
|
}
|
||
|
|
||
|
func IsComponentFolder(name string) bool {
|
||
|
return componentFoldersSet[name]
|
||
|
}
|