ported FetchHTTP to HTTPFetch
Signed-off-by: Richard Jones <richard@dagger.io>
This commit is contained in:
parent
deee606f4e
commit
643d93c35d
78
plan/task/httpfetch.go
Normal file
78
plan/task/httpfetch.go
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
package task
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"io/fs"
|
||||||
|
"net/url"
|
||||||
|
|
||||||
|
"github.com/moby/buildkit/client/llb"
|
||||||
|
"github.com/opencontainers/go-digest"
|
||||||
|
"github.com/rs/zerolog/log"
|
||||||
|
"go.dagger.io/dagger/compiler"
|
||||||
|
"go.dagger.io/dagger/plancontext"
|
||||||
|
"go.dagger.io/dagger/solver"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
Register("HTTPFetch", func() Task { return &httpFetchTask{} })
|
||||||
|
}
|
||||||
|
|
||||||
|
type httpFetchTask struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c httpFetchTask) Run(ctx context.Context, pctx *plancontext.Context, s solver.Solver, v *compiler.Value) (*compiler.Value, error) {
|
||||||
|
var httpFetch struct {
|
||||||
|
Source string
|
||||||
|
Checksum string
|
||||||
|
Dest string
|
||||||
|
Permissions int
|
||||||
|
UID int
|
||||||
|
GID int
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := v.Decode(&httpFetch); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
linkRedacted := httpFetch.Source
|
||||||
|
if u, err := url.Parse(httpFetch.Source); err == nil {
|
||||||
|
linkRedacted = u.Redacted()
|
||||||
|
}
|
||||||
|
|
||||||
|
httpOpts := []llb.HTTPOption{}
|
||||||
|
lg := log.Ctx(ctx)
|
||||||
|
if httpFetch.Checksum != "" {
|
||||||
|
lg.Debug().Str("checksum", httpFetch.Checksum).Msg("adding http option")
|
||||||
|
dgst, err := digest.Parse(httpFetch.Checksum)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
httpOpts = append(httpOpts, llb.Checksum(dgst))
|
||||||
|
}
|
||||||
|
if httpFetch.Dest != "" {
|
||||||
|
lg.Debug().Str("dest", httpFetch.Dest).Msg("adding http option")
|
||||||
|
httpOpts = append(httpOpts, llb.Filename(httpFetch.Dest))
|
||||||
|
}
|
||||||
|
if httpFetch.Permissions != 0 {
|
||||||
|
lg.Debug().Str("permissions", string(httpFetch.Permissions)).Msg("adding http option")
|
||||||
|
httpOpts = append(httpOpts, llb.Chmod(fs.FileMode(httpFetch.Permissions)))
|
||||||
|
}
|
||||||
|
if httpFetch.UID != 0 && httpFetch.GID != 0 {
|
||||||
|
lg.Debug().Str("uid", string(httpFetch.UID)).Str("gid", string(httpFetch.GID)).Msg("adding http option")
|
||||||
|
httpOpts = append(httpOpts, llb.Chown(httpFetch.UID, httpFetch.GID))
|
||||||
|
}
|
||||||
|
|
||||||
|
httpOpts = append(httpOpts, withCustomName(v, "FetchHTTP %s", linkRedacted))
|
||||||
|
|
||||||
|
st := llb.HTTP(httpFetch.Source, httpOpts...)
|
||||||
|
|
||||||
|
result, err := s.Solve(ctx, st, pctx.Platform.Get())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
fs := pctx.FS.New(result)
|
||||||
|
return compiler.NewValue().FillFields(map[string]interface{}{
|
||||||
|
"output": fs.MarshalCUE(),
|
||||||
|
})
|
||||||
|
}
|
@ -18,7 +18,6 @@ package engine
|
|||||||
|
|
||||||
// Fetch a file over HTTP
|
// Fetch a file over HTTP
|
||||||
#HTTPFetch: {
|
#HTTPFetch: {
|
||||||
@dagger(notimplemented)
|
|
||||||
$dagger: task: _name: "HTTPFetch"
|
$dagger: task: _name: "HTTPFetch"
|
||||||
|
|
||||||
// Source url
|
// Source url
|
||||||
|
@ -105,6 +105,11 @@ setup() {
|
|||||||
assert_failure
|
assert_failure
|
||||||
run "$DAGGER" --europa up ./tasks/gitPull/bad_ref.cue
|
run "$DAGGER" --europa up ./tasks/gitPull/bad_ref.cue
|
||||||
assert_failure
|
assert_failure
|
||||||
|
}
|
||||||
|
|
||||||
|
@test "task: #HTTPFetch" {
|
||||||
|
cd "$TESTDIR"
|
||||||
|
"$DAGGER" --europa up ./tasks/httpfetch/exist.cue
|
||||||
|
run "$DAGGER" --europa up ./tasks/httpfetch/not_exist.cue
|
||||||
|
assert_failure
|
||||||
}
|
}
|
||||||
|
10
tests/tasks/httpfetch/exist.cue
Normal file
10
tests/tasks/httpfetch/exist.cue
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import "alpha.dagger.io/europa/dagger/engine"
|
||||||
|
|
||||||
|
engine.#Plan & {
|
||||||
|
actions: fetch: engine.#HTTPFetch & {
|
||||||
|
source: "https://releases.dagger.io/dagger/latest_version"
|
||||||
|
dest: "/latest.html"
|
||||||
|
}
|
||||||
|
}
|
10
tests/tasks/httpfetch/not_exist.cue
Normal file
10
tests/tasks/httpfetch/not_exist.cue
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import "alpha.dagger.io/europa/dagger/engine"
|
||||||
|
|
||||||
|
engine.#Plan & {
|
||||||
|
actions: fetch: engine.#HTTPFetch & {
|
||||||
|
source: "https://releases.dagger.io/dagger/asfgdsfg"
|
||||||
|
dest: "/latest.html"
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user