cmd.go

   1 package main
   2
   3 import (
   4         "fmt"
   5         "io"
   6         "io/ioutil"
   7         "os"
   8         "os/exec"
   9
  10         "git.arvados.org/arvados.git/lib/cmd"
  11 )
  12
  13 var (
  14         handler = cmd.Multi(map[string]cmd.Handler{
  15                 "version":   cmd.Version,
  16                 "-version":  cmd.Version,
  17                 "--version": cmd.Version,
  18
  19                 "import":             &importer{},
  20                 "export-numpy":       &exportNumpy{},
  21                 "build-docker-image": &buildDockerImage{},
  22         })
  23 )
  24
  25 func main() {
  26         os.Exit(handler.RunCommand(os.Args[0], os.Args[1:], os.Stdin, os.Stdout, os.Stderr))
  27 }
  28
  29 type buildDockerImage struct{}
  30
  31 func (cmd *buildDockerImage) RunCommand(prog string, args []string, stdin io.Reader, stdout, stderr io.Writer) int {
  32         tmpdir, err := ioutil.TempDir("", "")
  33         if err != nil {
  34                 fmt.Fprint(stderr, err)
  35                 return 1
  36         }
  37         defer os.RemoveAll(tmpdir)
  38         err = ioutil.WriteFile(tmpdir+"/Dockerfile", []byte(`FROM debian:10
  39 RUN apt-get update
  40 RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends bcftools samtools
  41 `), 0644)
  42         if err != nil {
  43                 fmt.Fprint(stderr, err)
  44                 return 1
  45         }
  46         docker := exec.Command("docker", "build", "--tag=lightning-runtime", tmpdir)
  47         docker.Stdout = os.Stdout
  48         docker.Stderr = os.Stderr
  49         err = docker.Run()
  50         if err != nil {
  51                 return 1
  52         }
  53         return 0
  54 }