drone-yaml/yaml/parse.go

176 lines
3.6 KiB
Go
Raw Normal View History

2019-02-10 19:00:16 +00:00
// Copyright 2019 Drone IO, Inc.
2019-04-24 14:52:00 +00:00
//
2019-02-10 19:00:16 +00:00
// 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
2019-04-24 14:52:00 +00:00
//
2019-02-10 19:00:16 +00:00
// http://www.apache.org/licenses/LICENSE-2.0
2019-04-24 14:52:00 +00:00
//
2019-02-10 19:00:16 +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.
2019-01-22 23:44:17 +00:00
package yaml
import (
"bufio"
"bytes"
2019-05-30 15:14:47 +00:00
"errors"
2019-01-22 23:44:17 +00:00
"io"
"os"
"strings"
2019-05-30 01:21:23 +00:00
"github.com/buildkite/yaml"
2019-01-22 23:44:17 +00:00
)
2019-05-30 15:14:47 +00:00
var errorMissingKind = errors.New("yaml: missing kind attribute")
2019-01-22 23:44:17 +00:00
// Parse parses the configuration from io.Reader r.
func Parse(r io.Reader) (*Manifest, error) {
resources, err := ParseRaw(r)
if err != nil {
return nil, err
}
manifest := new(Manifest)
for _, raw := range resources {
if raw == nil {
continue
}
resource, err := parseRaw(raw)
if err != nil {
return nil, err
}
2019-05-30 15:14:47 +00:00
if resource.GetKind() == "" {
return nil, errorMissingKind
}
2019-01-22 23:44:17 +00:00
manifest.Resources = append(
manifest.Resources,
resource,
)
}
return manifest, nil
}
// ParseBytes parses the configuration from bytes b.
func ParseBytes(b []byte) (*Manifest, error) {
return Parse(
bytes.NewBuffer(b),
)
}
// ParseString parses the configuration from string s.
func ParseString(s string) (*Manifest, error) {
return ParseBytes(
[]byte(s),
)
}
// ParseFile parses the configuration from path p.
func ParseFile(p string) (*Manifest, error) {
f, err := os.Open(p)
if err != nil {
return nil, err
}
defer f.Close()
return Parse(f)
}
func parseRaw(r *RawResource) (Resource, error) {
var obj Resource
switch r.Kind {
case "cron":
obj = new(Cron)
case "secret":
obj = new(Secret)
case "signature":
obj = new(Signature)
case "registry":
obj = new(Registry)
default:
obj = new(Pipeline)
}
err := yaml.Unmarshal(r.Data, obj)
return obj, err
}
// ParseRaw parses the multi-document yaml from the
// io.Reader and returns a slice of raw resources.
func ParseRaw(r io.Reader) ([]*RawResource, error) {
const newline = '\n'
var resources []*RawResource
var resource *RawResource
scanner := bufio.NewScanner(r)
for scanner.Scan() {
line := scanner.Text()
if isSeparator(line) {
resource = nil
}
if resource == nil {
resource = &RawResource{}
resources = append(resources, resource)
}
if isSeparator(line) {
continue
}
if isTerminator(line) {
break
}
if scanner.Err() == io.EOF {
break
}
resource.Data = append(
resource.Data,
line...,
)
resource.Data = append(
resource.Data,
newline,
)
}
for _, resource := range resources {
err := yaml.Unmarshal(resource.Data, resource)
if err != nil {
return nil, err
}
}
return resources, nil
}
// ParseRawString parses the multi-document yaml from s
// and returns a slice of raw resources.
func ParseRawString(s string) ([]*RawResource, error) {
return ParseRaw(
strings.NewReader(s),
)
}
// ParseRawBytes parses the multi-document yaml from b
// and returns a slice of raw resources.
func ParseRawBytes(b []byte) ([]*RawResource, error) {
return ParseRaw(
bytes.NewReader(b),
)
}
// ParseRawFile parses the multi-document yaml from path p
// and returns a slice of raw resources.
func ParseRawFile(p string) ([]*RawResource, error) {
f, err := os.Open(p)
if err != nil {
return nil, err
}
defer f.Close()
return ParseRaw(f)
}
func isSeparator(s string) bool {
return strings.HasPrefix(s, "---")
}
func isTerminator(s string) bool {
return strings.HasPrefix(s, "...")
}