0f312113 |
package docker
import (
"bufio"
"encoding/json"
"fmt"
"github.com/dotcloud/docker/utils"
"io"
"net/url"
"os"
"reflect"
"strings"
)
|
b06784b0 |
type builderClient struct { |
c2a14bb1 |
cli *DockerCli |
0f312113 |
image string
maintainer string
config *Config
tmpContainers map[string]struct{}
tmpImages map[string]struct{}
needCommit bool
}
|
f35f0840 |
func (b *builderClient) clearTmp(containers, images map[string]struct{}) { |
0f312113 |
for i := range images { |
c2a14bb1 |
if _, _, err := b.cli.call("DELETE", "/images/"+i, nil); err != nil {
utils.Debugf("%s", err)
} |
0f312113 |
utils.Debugf("Removing image %s", i)
}
}
|
f35f0840 |
func (b *builderClient) CmdFrom(name string) error { |
0f312113 |
obj, statusCode, err := b.cli.call("GET", "/images/"+name+"/json", nil)
if statusCode == 404 { |
49505c59 |
remote := name
var tag string
if strings.Contains(remote, ":") {
remoteParts := strings.Split(remote, ":")
tag = remoteParts[1]
remote = remoteParts[0]
}
var out io.Writer
if os.Getenv("DEBUG") != "" {
out = os.Stdout
} else {
out = &utils.NopWriter{}
}
if err := b.cli.stream("POST", "/images/create?fromImage="+remote+"&tag="+tag, nil, out); err != nil { |
0f312113 |
return err
}
obj, _, err = b.cli.call("GET", "/images/"+name+"/json", nil)
if err != nil {
return err
}
}
if err != nil {
return err
}
|
fd224ee5 |
img := &APIID{} |
0f312113 |
if err := json.Unmarshal(obj, img); err != nil {
return err
} |
fd224ee5 |
b.image = img.ID |
f35f0840 |
utils.Debugf("Using image %s", b.image) |
0f312113 |
return nil
}
|
f35f0840 |
func (b *builderClient) CmdMaintainer(name string) error { |
0f312113 |
b.needCommit = true
b.maintainer = name
return nil
}
|
f35f0840 |
func (b *builderClient) CmdRun(args string) error { |
0f312113 |
if b.image == "" {
return fmt.Errorf("Please provide a source image with `from` prior to run")
} |
c2a14bb1 |
config, _, err := ParseRun([]string{b.image, "/bin/sh", "-c", args}, nil) |
0f312113 |
if err != nil {
return err
} |
c6bc90d0 |
cmd, env := b.config.Cmd, b.config.Env
b.config.Cmd = nil |
0f312113 |
MergeConfig(b.config, config) |
c6bc90d0 |
|
fd224ee5 |
body, statusCode, err := b.cli.call("POST", "/images/getCache", &APIImageConfig{ID: b.image, Config: b.config}) |
0f312113 |
if err != nil {
if statusCode != 404 {
return err
}
}
if statusCode != 404 { |
fd224ee5 |
apiID := &APIID{}
if err := json.Unmarshal(body, apiID); err != nil { |
0f312113 |
return err
} |
c6bc90d0 |
utils.Debugf("Use cached version") |
fd224ee5 |
b.image = apiID.ID |
0f312113 |
return nil
} |
c6bc90d0 |
cid, err := b.run()
if err != nil {
return err
}
b.config.Cmd, b.config.Env = cmd, env
return b.commit(cid) |
c2a14bb1 |
}
|
f35f0840 |
func (b *builderClient) CmdEnv(args string) error { |
c2a14bb1 |
b.needCommit = true
tmp := strings.SplitN(args, " ", 2)
if len(tmp) != 2 {
return fmt.Errorf("Invalid ENV format")
}
key := strings.Trim(tmp[0], " ")
value := strings.Trim(tmp[1], " ")
for i, elem := range b.config.Env {
if strings.HasPrefix(elem, key+"=") {
b.config.Env[i] = key + "=" + value
return nil
}
}
b.config.Env = append(b.config.Env, key+"="+value)
return nil
} |
0f312113 |
|
f35f0840 |
func (b *builderClient) CmdCmd(args string) error { |
c2a14bb1 |
b.needCommit = true |
13e687e5 |
var cmd []string
if err := json.Unmarshal([]byte(args), &cmd); err != nil { |
f35f0840 |
utils.Debugf("Error unmarshalling: %s, using /bin/sh -c", err) |
13e687e5 |
b.config.Cmd = []string{"/bin/sh", "-c", args}
} else {
b.config.Cmd = cmd
} |
c2a14bb1 |
return nil
}
|
f35f0840 |
func (b *builderClient) CmdExpose(args string) error { |
c2a14bb1 |
ports := strings.Split(args, " ")
b.config.PortSpecs = append(ports, b.config.PortSpecs...)
return nil
}
|
f35f0840 |
func (b *builderClient) CmdInsert(args string) error { |
0f135ad7 |
// tmp := strings.SplitN(args, "\t ", 2)
// sourceUrl, destPath := tmp[0], tmp[1]
// v := url.Values{}
// v.Set("url", sourceUrl)
// v.Set("path", destPath)
// body, _, err := b.cli.call("POST", "/images/insert?"+v.Encode(), nil)
// if err != nil {
// return err
// }
|
fd224ee5 |
// apiId := &APIId{} |
0f135ad7 |
// if err := json.Unmarshal(body, apiId); err != nil {
// return err
// }
// FIXME: Reimplement this, we need to retrieve the resulting Id |
c2a14bb1 |
return fmt.Errorf("INSERT not implemented")
}
|
f35f0840 |
func (b *builderClient) run() (string, error) { |
c6bc90d0 |
if b.image == "" {
return "", fmt.Errorf("Please provide a source image with `from` prior to run") |
b51303cd |
} |
c6bc90d0 |
b.config.Image = b.image |
c2a14bb1 |
body, _, err := b.cli.call("POST", "/containers/create", b.config) |
0f312113 |
if err != nil { |
c6bc90d0 |
return "", err |
0f312113 |
}
|
fd224ee5 |
apiRun := &APIRun{} |
c6bc90d0 |
if err := json.Unmarshal(body, apiRun); err != nil {
return "", err |
0f312113 |
} |
c6bc90d0 |
for _, warning := range apiRun.Warnings { |
0f312113 |
fmt.Fprintln(os.Stderr, "WARNING: ", warning)
}
//start the container |
fd224ee5 |
_, _, err = b.cli.call("POST", "/containers/"+apiRun.ID+"/start", nil) |
0f312113 |
if err != nil { |
c6bc90d0 |
return "", err |
0f312113 |
} |
fd224ee5 |
b.tmpContainers[apiRun.ID] = struct{}{} |
0f312113 |
// Wait for it to finish |
fd224ee5 |
body, _, err = b.cli.call("POST", "/containers/"+apiRun.ID+"/wait", nil) |
0f312113 |
if err != nil { |
c6bc90d0 |
return "", err
} |
fd224ee5 |
apiWait := &APIWait{} |
c6bc90d0 |
if err := json.Unmarshal(body, apiWait); err != nil {
return "", err
}
if apiWait.StatusCode != 0 {
return "", fmt.Errorf("The command %v returned a non-zero code: %d", b.config.Cmd, apiWait.StatusCode)
}
|
fd224ee5 |
return apiRun.ID, nil |
c6bc90d0 |
}
|
f35f0840 |
func (b *builderClient) commit(id string) error { |
c6bc90d0 |
if b.image == "" {
return fmt.Errorf("Please provide a source image with `from` prior to run")
}
b.config.Image = b.image
if id == "" {
cmd := b.config.Cmd
b.config.Cmd = []string{"true"} |
86ada2fa |
cid, err := b.run()
if err != nil { |
c6bc90d0 |
return err
} |
86ada2fa |
id = cid |
c6bc90d0 |
b.config.Cmd = cmd |
0f312113 |
}
// Commit the container
v := url.Values{} |
c6bc90d0 |
v.Set("container", id) |
0f312113 |
v.Set("author", b.maintainer) |
c6bc90d0 |
body, _, err := b.cli.call("POST", "/commit?"+v.Encode(), b.config) |
0f312113 |
if err != nil {
return err
} |
fd224ee5 |
apiID := &APIID{}
if err := json.Unmarshal(body, apiID); err != nil { |
0f312113 |
return err
} |
fd224ee5 |
b.tmpImages[apiID.ID] = struct{}{}
b.image = apiID.ID |
c6bc90d0 |
b.needCommit = false |
0f312113 |
return nil
}
|
d42c10aa |
func (b *builderClient) Build(dockerfile, context io.Reader) (string, error) { |
49505c59 |
defer b.clearTmp(b.tmpContainers, b.tmpImages) |
0f312113 |
file := bufio.NewReader(dockerfile)
for {
line, err := file.ReadString('\n')
if err != nil {
if err == io.EOF {
break
}
return "", err
}
line = strings.Replace(strings.TrimSpace(line), " ", " ", 1)
// Skip comments and empty line
if len(line) == 0 || line[0] == '#' {
continue
}
tmp := strings.SplitN(line, " ", 2)
if len(tmp) != 2 {
return "", fmt.Errorf("Invalid Dockerfile format")
}
instruction := strings.ToLower(strings.Trim(tmp[0], " "))
arguments := strings.Trim(tmp[1], " ")
|
0f135ad7 |
fmt.Fprintf(os.Stderr, "%s %s (%s)\n", strings.ToUpper(instruction), arguments, b.image) |
0f312113 |
|
c2a14bb1 |
method, exists := reflect.TypeOf(b).MethodByName("Cmd" + strings.ToUpper(instruction[:1]) + strings.ToLower(instruction[1:])) |
0f312113 |
if !exists { |
0f135ad7 |
fmt.Fprintf(os.Stderr, "Skipping unknown instruction %s\n", strings.ToUpper(instruction)) |
0f312113 |
}
ret := method.Func.Call([]reflect.Value{reflect.ValueOf(b), reflect.ValueOf(arguments)})[0].Interface()
if ret != nil {
return "", ret.(error)
}
|
0f135ad7 |
fmt.Fprintf(os.Stderr, "===> %v\n", b.image) |
0f312113 |
}
if b.needCommit { |
c6bc90d0 |
if err := b.commit(""); err != nil {
return "", err
} |
0f312113 |
}
if b.image != "" {
// The build is successful, keep the temporary containers and images
for i := range b.tmpImages {
delete(b.tmpImages, i)
}
for i := range b.tmpContainers {
delete(b.tmpContainers, i)
} |
0f135ad7 |
fmt.Fprintf(os.Stderr, "Build finished. image id: %s\n", b.image) |
0f312113 |
return b.image, nil
}
return "", fmt.Errorf("An error occured during the build\n")
} |
c2a14bb1 |
|
3adf9ce0 |
func NewBuilderClient(proto, addr string) BuildFile { |
b06784b0 |
return &builderClient{ |
3adf9ce0 |
cli: NewDockerCli(proto, addr), |
c2a14bb1 |
config: &Config{},
tmpContainers: make(map[string]struct{}),
tmpImages: make(map[string]struct{}),
}
} |