11package dockerexec
22
33import (
4+ "bufio"
45 "context"
56 "encoding/hex"
7+ "encoding/json"
68 "io"
9+ "math/rand"
710 "time"
811
12+ "github.com/docker/docker/api/types"
913 "github.com/docker/docker/api/types/filters"
1014 "github.com/docker/docker/api/types/image"
1115 "github.com/docker/docker/client"
16+ "github.com/docker/docker/pkg/archive"
1217 "github.com/pkg/errors"
1318 "go.uber.org/zap"
14- "golang.org/x/exp/rand"
1519)
1620
1721type Options struct {
@@ -34,7 +38,7 @@ func New(opts *Options) (*Docker, error) {
3438 logger = zap .NewNop ()
3539 }
3640
37- rnd := rand .New (rand .NewSource (uint64 ( time .Now ().UnixNano () )))
41+ rnd := rand .New (rand .NewSource (time .Now ().UnixNano ()))
3842
3943 d := & Docker {
4044 client : c ,
@@ -77,6 +81,14 @@ func (d *Docker) CommandContext(ctx context.Context, program string, args ...str
7781 }
7882}
7983
84+ func (d * Docker ) RemoveImage (ctx context.Context ) error {
85+ _ , err := d .client .ImageRemove (ctx , d .image , image.RemoveOptions {Force : true , PruneChildren : true })
86+ if err != nil {
87+ return errors .WithStack (err )
88+ }
89+ return nil
90+ }
91+
8092func (d * Docker ) containerUniqueName () string {
8193 var hash [4 ]byte
8294 _ , _ = d .rnd .Read (hash [:])
@@ -90,8 +102,62 @@ func (d *Docker) buildOrPullImage(ctx context.Context) error {
90102 return d .pullImage (ctx )
91103}
92104
93- func (d * Docker ) buildImage (context.Context ) error {
94- return errors .New ("not implemented" )
105+ func (d * Docker ) buildImage (ctx context.Context ) error {
106+ tar , err := archive .TarWithOptions (d .buildContext , & archive.TarOptions {})
107+ if err != nil {
108+ return errors .WithMessage (err , "failed to create tar archive" )
109+ }
110+
111+ resp , err := d .client .ImageBuild (
112+ ctx ,
113+ tar ,
114+ types.ImageBuildOptions {
115+ Dockerfile : d .dockerfile ,
116+ Tags : []string {d .image },
117+ Remove : true ,
118+ ForceRemove : true ,
119+ NoCache : true ,
120+ },
121+ )
122+ if err != nil {
123+ return errors .WithMessage (err , "failed to build image" )
124+ }
125+ defer resp .Body .Close ()
126+
127+ return errors .WithStack (
128+ printLogs (resp .Body , d .logger ),
129+ )
130+ }
131+
132+ func printLogs (r io.Reader , logger * zap.Logger ) error {
133+ type errorLine struct {
134+ Error string `json:"error"`
135+ ErrorDetail struct {
136+ Message string `json:"message"`
137+ } `json:"errorDetail"`
138+ }
139+
140+ var lastLine string
141+
142+ scanner := bufio .NewScanner (r )
143+
144+ for scanner .Scan () {
145+ lastLine = scanner .Text ()
146+ logger .Debug ("docker build" , zap .String ("log" , lastLine ))
147+ }
148+
149+ if err := scanner .Err (); err != nil {
150+ return errors .WithMessage (err , "docker build" )
151+ }
152+
153+ errLine := errorLine {}
154+ if err := json .Unmarshal ([]byte (lastLine ), & errLine ); err != nil {
155+ return errors .WithMessage (err , "docker build" )
156+ }
157+ if errLine .Error != "" {
158+ return errors .Errorf ("docker build: %s" , errLine .Error )
159+ }
160+ return nil
95161}
96162
97163func (d * Docker ) pullImage (ctx context.Context ) error {
0 commit comments