66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package client
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/bndr/gojenkins"
|
|
"github.com/jenkinsci/kubernetes-operator/pkg/log"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// GroovyScriptExecutionFailed is custom error type which indicates passed groovy script is invalid
|
|
type GroovyScriptExecutionFailed struct {
|
|
ConfigurationType string
|
|
Source string
|
|
Name string
|
|
Logs string
|
|
}
|
|
|
|
func (e GroovyScriptExecutionFailed) Error() string {
|
|
return "script execution failed"
|
|
}
|
|
|
|
func (jenkins *jenkins) ExecuteScript(script string) (string, error) {
|
|
now := time.Now().Unix()
|
|
verifier := fmt.Sprintf("verifier-%d", now)
|
|
return jenkins.executeScript(script, verifier)
|
|
}
|
|
|
|
func (jenkins *jenkins) executeScript(script string, verifier string) (string, error) {
|
|
output := ""
|
|
fullScript := fmt.Sprintf("%s\nprint println('%s')", script, verifier)
|
|
|
|
data := url.Values{}
|
|
data.Set("script", fullScript)
|
|
|
|
ar := gojenkins.NewAPIRequest("POST", "/scriptText", bytes.NewBufferString(data.Encode()))
|
|
if err := jenkins.Requester.SetCrumb(context.TODO(), ar); err != nil {
|
|
return output, err
|
|
}
|
|
ar.SetHeader("Content-Type", "application/x-www-form-urlencoded")
|
|
ar.Suffix = ""
|
|
|
|
r, err := jenkins.Requester.Do(context.TODO(), ar, &output, nil)
|
|
if err != nil {
|
|
return "", errors.Wrapf(err, "couldn't execute groovy script, logs '%s'", output)
|
|
}
|
|
if err := r.Body.Close(); err != nil {
|
|
log.Log.Error(err, "failed to close jenkins.executeScript.Requester")
|
|
}
|
|
if r.StatusCode != http.StatusOK {
|
|
return output, errors.Errorf("invalid status code '%d', logs '%s'", r.StatusCode, output)
|
|
}
|
|
|
|
if !strings.Contains(output, verifier) {
|
|
return output, &GroovyScriptExecutionFailed{}
|
|
}
|
|
|
|
return output, nil
|
|
}
|