1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
package main
import (
"context"
"encoding/json"
"flag"
"io"
"os"
"github.com/moby/buildkit/client/llb"
"github.com/moby/buildkit/client/llb/imagemetaresolver"
"github.com/moby/buildkit/frontend/dockerfile/dockerfile2llb"
"github.com/moby/buildkit/frontend/dockerui"
"github.com/moby/buildkit/solver/pb"
"github.com/moby/buildkit/util/appcontext"
"github.com/sirupsen/logrus"
)
type buildOpt struct {
target string
partialImageConfigFile string
baseImageConfigFile string
}
func main() {
if err := xmain(); err != nil {
logrus.Fatal(err)
}
}
func xmain() error {
var opt buildOpt
flag.StringVar(&opt.target, "target", "", "target stage")
flag.StringVar(&opt.partialImageConfigFile, "partial-image-config-file", "", "Output partial image config as a JSON file")
flag.StringVar(&opt.baseImageConfigFile, "base-image-config-file", "", "Output base image config as a JSON file")
flag.Parse()
df, err := io.ReadAll(os.Stdin)
if err != nil {
return err
}
caps := pb.Caps.CapSet(pb.Caps.All())
state, img, baseImg, _, err := dockerfile2llb.Dockerfile2LLB(appcontext.Context(), df, dockerfile2llb.ConvertOpt{
MetaResolver: imagemetaresolver.Default(),
LLBCaps: &caps,
Config: dockerui.Config{
Target: opt.target,
},
})
if err != nil {
return err
}
dt, err := state.Marshal(context.TODO())
if err != nil {
return err
}
if err := llb.WriteTo(dt, os.Stdout); err != nil {
return err
}
if opt.partialImageConfigFile != "" {
if err := writeJSON(opt.partialImageConfigFile, img); err != nil {
return err
}
}
if opt.baseImageConfigFile != "" {
if err := writeJSON(opt.baseImageConfigFile, baseImg); err != nil {
return err
}
}
return nil
}
func writeJSON(f string, x interface{}) error {
b, err := json.Marshal(x)
if err != nil {
return err
}
if err := os.RemoveAll(f); err != nil {
return err
}
return os.WriteFile(f, b, 0o644)
}
|
