mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-23 19:56:01 +00:00
Merge pull request #1898 from jhadvig/podex_images
Podex handling multiple images
This commit is contained in:
commit
0e8804ee49
@ -40,10 +40,11 @@ import (
|
||||
"gopkg.in/v1/yaml"
|
||||
)
|
||||
|
||||
const usage = "usage: podex [-json|-yaml] <repo/dockerimage>"
|
||||
const usage = "usage: podex [-json|-yaml] -id=ID username/image1 ... username/imageN"
|
||||
|
||||
var generateJSON = flag.Bool("json", false, "generate json manifest")
|
||||
var generateYAML = flag.Bool("yaml", false, "generate yaml manifest")
|
||||
var podName = flag.String("id", "", "set pod name")
|
||||
|
||||
func main() {
|
||||
flag.Parse()
|
||||
@ -51,60 +52,64 @@ func main() {
|
||||
if flag.NArg() < 1 {
|
||||
log.Fatal(usage)
|
||||
}
|
||||
|
||||
imageName := flag.Arg(0)
|
||||
if len(imageName) == 0 {
|
||||
log.Fatal(usage)
|
||||
if *podName == "" {
|
||||
if flag.NArg() > 1 {
|
||||
log.Fatal(usage)
|
||||
}
|
||||
_, *podName = parseDockerImage(flag.Arg(0))
|
||||
}
|
||||
|
||||
if (!*generateJSON && !*generateYAML) || (*generateJSON && *generateYAML) {
|
||||
log.Fatal(usage)
|
||||
}
|
||||
|
||||
// Parse docker image name
|
||||
// IMAGE: [REGISTRYHOST/][USERNAME/]NAME[:TAG]
|
||||
// NAME: [a-z0-9-_.]
|
||||
parts := strings.Split(imageName, "/")
|
||||
baseName := parts[len(parts)-1]
|
||||
|
||||
dockerHost := os.Getenv("DOCKER_HOST")
|
||||
if dockerHost == "" {
|
||||
log.Fatalf("DOCKER_HOST is not set")
|
||||
}
|
||||
|
||||
docker, err := dockerclient.NewClient(dockerHost)
|
||||
if err != nil {
|
||||
log.Fatalf("failed to connect to %q: %v", dockerHost, err)
|
||||
}
|
||||
|
||||
// TODO(proppy): use the regitry API instead of the remote API to get image metadata.
|
||||
img, err := docker.InspectImage(imageName)
|
||||
if err != nil {
|
||||
log.Fatalf("failed to inspect image %q: %v", imageName, err)
|
||||
}
|
||||
// TODO(proppy): add flag to handle multiple version
|
||||
manifest := v1beta1.ContainerManifest{
|
||||
Version: "v1beta1",
|
||||
ID: baseName + "-pod",
|
||||
Containers: []v1beta1.Container{{
|
||||
podContainers := []v1beta1.Container{}
|
||||
|
||||
for _, imageName := range flag.Args() {
|
||||
parts, baseName := parseDockerImage(imageName)
|
||||
container := v1beta1.Container{
|
||||
Name: baseName,
|
||||
Image: imageName,
|
||||
}},
|
||||
}
|
||||
|
||||
// TODO(proppy): use the regitry API instead of the remote API to get image metadata.
|
||||
img, err := docker.InspectImage(imageName)
|
||||
if err != nil {
|
||||
log.Fatalf("failed to inspect image %q: %v", imageName, err)
|
||||
}
|
||||
for p := range img.Config.ExposedPorts {
|
||||
port, err := strconv.Atoi(p.Port())
|
||||
if err != nil {
|
||||
log.Fatalf("failed to parse port %q: %v", parts[0], err)
|
||||
}
|
||||
container.Ports = append(container.Ports, v1beta1.Port{
|
||||
Name: strings.Join([]string{baseName, p.Proto(), p.Port()}, "-"),
|
||||
ContainerPort: port,
|
||||
Protocol: v1beta1.Protocol(strings.ToUpper(p.Proto())),
|
||||
})
|
||||
}
|
||||
podContainers = append(podContainers, container)
|
||||
}
|
||||
|
||||
// TODO(proppy): add flag to handle multiple version
|
||||
manifest := v1beta1.ContainerManifest{
|
||||
Version: "v1beta1",
|
||||
ID: *podName + "-pod",
|
||||
Containers: podContainers,
|
||||
RestartPolicy: v1beta1.RestartPolicy{
|
||||
Always: &v1beta1.RestartPolicyAlways{},
|
||||
},
|
||||
}
|
||||
for p := range img.Config.ExposedPorts {
|
||||
port, err := strconv.Atoi(p.Port())
|
||||
if err != nil {
|
||||
log.Fatalf("failed to parse port %q: %v", parts[0], err)
|
||||
}
|
||||
manifest.Containers[0].Ports = append(manifest.Containers[0].Ports, v1beta1.Port{
|
||||
Name: strings.Join([]string{baseName, p.Proto(), p.Port()}, "-"),
|
||||
ContainerPort: port,
|
||||
Protocol: v1beta1.Protocol(strings.ToUpper(p.Proto())),
|
||||
})
|
||||
}
|
||||
|
||||
if *generateJSON {
|
||||
bs, err := json.MarshalIndent(manifest, "", " ")
|
||||
if err != nil {
|
||||
@ -120,3 +125,15 @@ func main() {
|
||||
os.Stdout.Write(bs)
|
||||
}
|
||||
}
|
||||
|
||||
// parseDockerImage split a docker image name of the form [REGISTRYHOST/][USERNAME/]NAME[:TAG]
|
||||
// TODO: handle the TAG
|
||||
// Returns array of images name parts and base image name
|
||||
func parseDockerImage(imageName string) (parts []string, baseName string) {
|
||||
// Parse docker image name
|
||||
// IMAGE: [REGISTRYHOST/][USERNAME/]NAME[:TAG]
|
||||
// NAME: [a-z0-9-_.]
|
||||
parts = strings.Split(imageName, "/")
|
||||
baseName = parts[len(parts)-1]
|
||||
return
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user