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         "github.com/mattn/go-isatty"
  12         "github.com/sirupsen/logrus"
  13 )
  14
  15 var (
  16         handler = cmd.Multi(map[string]cmd.Handler{
  17                 "version":   cmd.Version,
  18                 "-version":  cmd.Version,
  19                 "--version": cmd.Version,
  20
  21                 "ref2genome":         &ref2genome{},
  22                 "vcf2fasta":          &vcf2fasta{},
  23                 "import":             &importer{},
  24                 "import-stats-plot":  &importstatsplot{},
  25                 "export":             &exporter{},
  26                 "export-numpy":       &exportNumpy{},
  27                 "filter":             &filterer{},
  28                 "build-docker-image": &buildDockerImage{},
  29                 "pca-go":             &goPCA{},
  30                 "pca-py":             &pythonPCA{},
  31                 "plot":               &pythonPlot{},
  32                 "diff-fasta":         &diffFasta{},
  33                 "stats":              &statscmd{},
  34                 "merge":              &merger{},
  35         })
  36 )
  37
  38 func main() {
  39         if !isatty.IsTerminal(os.Stderr.Fd()) {
  40                 logrus.StandardLogger().Formatter = &logrus.TextFormatter{DisableTimestamp: true}
  41         }
  42         os.Exit(handler.RunCommand(os.Args[0], os.Args[1:], os.Stdin, os.Stdout, os.Stderr))
  43 }
  44
  45 type buildDockerImage struct{}
  46
  47 func (cmd *buildDockerImage) RunCommand(prog string, args []string, stdin io.Reader, stdout, stderr io.Writer) int {
  48         tmpdir, err := ioutil.TempDir("", "")
  49         if err != nil {
  50                 fmt.Fprint(stderr, err)
  51                 return 1
  52         }
  53         defer os.RemoveAll(tmpdir)
  54         err = ioutil.WriteFile(tmpdir+"/Dockerfile", []byte(`FROM debian:10
  55 RUN DEBIAN_FRONTEND=noninteractive \
  56   apt-get update && \
  57   apt-get dist-upgrade -y && \
  58   apt-get install -y --no-install-recommends bcftools bedtools samtools python2 python3-sklearn python3-matplotlib && \
  59   apt-get clean
  60 `), 0644)
  61         if err != nil {
  62                 fmt.Fprint(stderr, err)
  63                 return 1
  64         }
  65         docker := exec.Command("docker", "build", "--tag=lightning-runtime", tmpdir)
  66         docker.Stdout = stdout
  67         docker.Stderr = stderr
  68         err = docker.Run()
  69         if err != nil {
  70                 return 1
  71         }
  72         fmt.Fprintf(stderr, "built and tagged new docker image, lightning-runtime\n")
  73         return 0
  74 }