cmd/arvados-package/install.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 package main
   6
   7 import (
   8         "context"
   9         "fmt"
  10         "io"
  11         "io/ioutil"
  12         "os"
  13         "os/exec"
  14         "path/filepath"
  15         "strings"
  16
  17         "github.com/docker/docker/api/types"
  18         "github.com/docker/docker/client"
  19 )
  20
  21 func testinstall(ctx context.Context, opts opts, stdin io.Reader, stdout, stderr io.Writer) error {
  22         depsImageName := "arvados-package-deps-" + opts.TargetOS
  23         depsCtrName := strings.Replace(depsImageName, ":", "-", -1)
  24
  25         _, prog := filepath.Split(os.Args[0])
  26         tmpdir, err := ioutil.TempDir("", prog+".")
  27         if err != nil {
  28                 return fmt.Errorf("TempDir: %w", err)
  29         }
  30         defer os.RemoveAll(tmpdir)
  31
  32         sourcesFile := tmpdir + "/arvados-local.list"
  33         err = ioutil.WriteFile(sourcesFile, []byte("deb [trusted=yes] file:///pkg ./\n"), 0644)
  34         if err != nil {
  35                 return fmt.Errorf("Write %s: %w", sourcesFile, err)
  36         }
  37
  38         if exists, err := dockerImageExists(ctx, depsImageName); err != nil {
  39                 return err
  40         } else if !exists || opts.RebuildImage {
  41                 err = dockerRm(ctx, depsCtrName)
  42                 if err != nil {
  43                         return err
  44                 }
  45                 defer dockerRm(ctx, depsCtrName)
  46                 cmd := exec.CommandContext(ctx, "docker", "run",
  47                         "--name", depsCtrName,
  48                         "--tmpfs", "/tmp:exec,mode=01777",
  49                         "-v", opts.PackageDir+":/pkg:ro",
  50                         "-v", sourcesFile+":/etc/apt/sources.list.d/arvados-local.list:ro",
  51                         "--env", "DEBIAN_FRONTEND=noninteractive",
  52                         opts.TargetOS,
  53                         "bash", "-c", `
  54 set -e
  55 apt-get update
  56 apt-get install -y eatmydata
  57 eatmydata apt-get install -y --no-install-recommends arvados-server-easy postgresql
  58 eatmydata apt-get remove -y arvados-server-easy
  59 `)
  60                 cmd.Stdout = stdout
  61                 cmd.Stderr = stderr
  62                 err = cmd.Run()
  63                 if err != nil {
  64                         return fmt.Errorf("docker run: %w", err)
  65                 }
  66
  67                 cmd = exec.CommandContext(ctx, "docker", "commit", depsCtrName, depsImageName)
  68                 cmd.Stdout = stdout
  69                 cmd.Stderr = stderr
  70                 err = cmd.Run()
  71                 if err != nil {
  72                         return fmt.Errorf("docker commit: %w", err)
  73                 }
  74         }
  75
  76         versionsuffix := ""
  77         if opts.PackageVersion != "" {
  78                 versionsuffix = "=" + opts.PackageVersion
  79         }
  80         cmd := exec.CommandContext(ctx, "docker", "run", "--rm",
  81                 "--tmpfs", "/tmp:exec,mode=01777",
  82                 "-v", opts.PackageDir+":/pkg:ro",
  83                 "-v", sourcesFile+":/etc/apt/sources.list.d/arvados-local.list:ro",
  84                 "--env", "DEBIAN_FRONTEND=noninteractive",
  85                 depsImageName,
  86                 "bash", "-c", `
  87 set -e
  88 PATH="/var/lib/arvados/bin:$PATH"
  89 apt-get update
  90 eatmydata apt-get install --reinstall -y --no-install-recommends arvados-server-easy`+versionsuffix+`
  91 apt-get -y autoremove
  92 /etc/init.d/postgresql start
  93 arvados-server init -cluster-id x1234
  94 exec arvados-server boot -listen-host 0.0.0.0 -shutdown
  95 `)
  96         cmd.Stdout = stdout
  97         cmd.Stderr = stderr
  98         err = cmd.Run()
  99         if err != nil {
 100                 return fmt.Errorf("docker run: %w", err)
 101         }
 102         return nil
 103 }
 104
 105 func dockerImageExists(ctx context.Context, name string) (bool, error) {
 106         cli, err := client.NewEnvClient()
 107         if err != nil {
 108                 return false, err
 109         }
 110         imgs, err := cli.ImageList(ctx, types.ImageListOptions{All: true})
 111         if err != nil {
 112                 return false, err
 113         }
 114         for _, img := range imgs {
 115                 for _, tag := range img.RepoTags {
 116                         if tag == name {
 117                                 return true, nil
 118                         }
 119                 }
 120         }
 121         return false, nil
 122 }