78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
|
// Copyright 2019 Drone.IO Inc. All rights reserved.
|
||
|
|
// Use of this source code is governed by the Drone Non-Commercial License
|
||
|
|
// that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
// +build !oss
|
||
|
|
|
||
|
|
package config
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/drone/drone/core"
|
||
|
|
|
||
|
|
"github.com/google/go-jsonnet"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Jsonnet returns a configuration service that fetches the
|
||
|
|
// jsonnet file directly from the source code management (scm)
|
||
|
|
// system and converts to a yaml file.
|
||
|
|
func Jsonnet(service core.FileService, enabled bool) core.ConfigService {
|
||
|
|
return &jsonnetPlugin{
|
||
|
|
enabled: enabled,
|
||
|
|
repos: &repo{files: service},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type jsonnetPlugin struct {
|
||
|
|
enabled bool
|
||
|
|
repos *repo
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *jsonnetPlugin) Find(ctx context.Context, req *core.ConfigArgs) (*core.Config, error) {
|
||
|
|
if p.enabled == false {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// if the file extension is not jsonnet we can
|
||
|
|
// skip this plugin by returning zero values.
|
||
|
|
if strings.HasSuffix(req.Repo.Config, ".jsonnet") == false {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// get the file contents.
|
||
|
|
config, err := p.repos.Find(ctx, req)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO(bradrydzewski) temporarily disable file imports
|
||
|
|
// TODO(bradrydzewski) handle object vs array output
|
||
|
|
|
||
|
|
// create the jsonnet vm
|
||
|
|
vm := jsonnet.MakeVM()
|
||
|
|
vm.MaxStack = 500
|
||
|
|
vm.StringOutput = false
|
||
|
|
vm.ErrorFormatter.SetMaxStackTraceSize(20)
|
||
|
|
|
||
|
|
// convert the jsonnet file to yaml
|
||
|
|
buf := new(bytes.Buffer)
|
||
|
|
docs, err := vm.EvaluateSnippetStream(req.Repo.Config, config.Data)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// the jsonnet vm returns a stream of yaml documents
|
||
|
|
// that need to be combined into a single yaml file.
|
||
|
|
for _, doc := range docs {
|
||
|
|
buf.WriteString("---")
|
||
|
|
buf.WriteString("\n")
|
||
|
|
buf.WriteString(doc)
|
||
|
|
}
|
||
|
|
|
||
|
|
config.Data = buf.String()
|
||
|
|
return config, nil
|
||
|
|
}
|