diff options
Diffstat (limited to 'hugolib/page_unwrap.go')
-rw-r--r-- | hugolib/page_unwrap.go | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/hugolib/page_unwrap.go b/hugolib/page_unwrap.go new file mode 100644 index 000000000..eda6636d1 --- /dev/null +++ b/hugolib/page_unwrap.go @@ -0,0 +1,50 @@ +// 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 hugolib + +import ( + "github.com/pkg/errors" + + "github.com/gohugoio/hugo/resources/page" +) + +// Wraps a Page. +type pageWrapper interface { + page() page.Page +} + +// unwrapPage is used in equality checks and similar. +func unwrapPage(in interface{}) (page.Page, error) { + switch v := in.(type) { + case *pageState: + return v, nil + case pageWrapper: + return v.page(), nil + case page.Page: + return v, nil + case nil: + return nil, nil + default: + return nil, errors.Errorf("unwrapPage: %T not supported", in) + } +} + +func mustUnwrapPage(in interface{}) page.Page { + p, err := unwrapPage(in) + if err != nil { + panic(err) + } + + return p +} |