hugo/hugolib/handler_meta.go

118 lines
2.4 KiB
Go
Raw Normal View History

// Copyright 2015 The Hugo Authors. All rights reserved.
2014-10-17 20:57:48 +00:00
//
2015-11-24 03:16:36 +00:00
// Licensed under the Apache License, Version 2.0 (the "License");
2014-10-17 20:57:48 +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 20:57:48 +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 hugolib
import (
"errors"
2014-10-20 21:42:16 +00:00
"fmt"
"github.com/spf13/hugo/source"
)
var handlers []Handler
type MetaHandler interface {
2014-11-01 16:05:37 +00:00
// Read the Files in and register
2014-10-20 21:51:53 +00:00
Read(*source.File, *Site, HandleResults)
2014-11-01 16:05:37 +00:00
// Generic Convert Function with coordination
2014-10-21 00:15:33 +00:00
Convert(interface{}, *Site, HandleResults)
2014-11-01 16:05:37 +00:00
Handle() Handler
2014-10-20 21:42:16 +00:00
}
type HandleResults chan<- HandledResult
func NewMetaHandler(in string) *MetaHandle {
x := &MetaHandle{ext: in}
x.Handler()
return x
}
2014-10-20 21:42:16 +00:00
type MetaHandle struct {
handler Handler
ext string
2014-10-17 20:57:48 +00:00
}
func (mh *MetaHandle) Read(f *source.File, s *Site, results HandleResults) {
if h := mh.Handler(); h != nil {
results <- h.Read(f, s)
return
}
2014-10-17 20:57:48 +00:00
results <- HandledResult{err: errors.New("No handler found"), file: f}
2014-10-20 21:42:16 +00:00
}
func (mh *MetaHandle) Convert(i interface{}, s *Site, results HandleResults) {
h := mh.Handler()
2014-10-21 00:15:33 +00:00
if f, ok := i.(*source.File); ok {
results <- h.FileConvert(f, s)
return
2014-10-21 00:15:33 +00:00
}
2014-10-20 21:42:16 +00:00
if p, ok := i.(*Page); ok {
if p == nil {
results <- HandledResult{err: errors.New("file resulted in a nil page")}
return
}
if h == nil {
results <- HandledResult{err: fmt.Errorf("No handler found for page '%s'. Verify the markup is supported by Hugo.", p.FullFilePath())}
return
}
results <- h.PageConvert(p)
}
2014-10-17 20:57:48 +00:00
}
func (mh *MetaHandle) Handler() Handler {
if mh.handler == nil {
mh.handler = FindHandler(mh.ext)
// if no handler found, use default handler
if mh.handler == nil {
mh.handler = FindHandler("*")
}
}
return mh.handler
2014-10-17 20:57:48 +00:00
}
2014-10-20 21:42:16 +00:00
func FindHandler(ext string) Handler {
2014-10-17 20:57:48 +00:00
for _, h := range Handlers() {
2014-10-20 21:42:16 +00:00
if HandlerMatch(h, ext) {
2014-10-17 20:57:48 +00:00
return h
}
}
return nil
}
2014-10-20 21:42:16 +00:00
func HandlerMatch(h Handler, ext string) bool {
2014-10-17 20:57:48 +00:00
for _, x := range h.Extensions() {
if ext == x {
return true
}
}
return false
}
func RegisterHandler(h Handler) {
handlers = append(handlers, h)
}
func Handlers() []Handler {
return handlers
}