cmd.go

   1 // Copyright (C) The Lightning Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 package lightning
   6
   7 import (
   8         "fmt"
   9         "io"
  10         "io/ioutil"
  11         "os"
  12         "os/exec"
  13         "runtime/debug"
  14         "strings"
  15
  16         "git.arvados.org/arvados.git/lib/cmd"
  17         "github.com/mattn/go-isatty"
  18         "github.com/sirupsen/logrus"
  19 )
  20
  21 var (
  22         handler = cmd.Multi(map[string]cmd.Handler{
  23                 "version":   cmd.Version,
  24                 "-version":  cmd.Version,
  25                 "--version": cmd.Version,
  26
  27                 "ref2genome":         &ref2genome{},
  28                 "vcf2fasta":          &vcf2fasta{},
  29                 "import":             &importer{},
  30                 "annotate":           &annotatecmd{},
  31                 "export":             &exporter{},
  32                 "export-numpy":       &exportNumpy{},
  33                 "flake":              &flakecmd{},
  34                 "slice":              &slicecmd{},
  35                 "slice-numpy":        &sliceNumpy{},
  36                 "tiling-stats":       &tilingStats{},
  37                 "anno2vcf":           &anno2vcf{},
  38                 "numpy-comvar":       &numpyComVar{},
  39                 "filter":             &filtercmd{},
  40                 "build-docker-image": &buildDockerImage{},
  41                 "pca-go":             &goPCA{},
  42                 "pca-py":             &pythonPCA{},
  43                 "plot":               &pythonPlot{},
  44                 "diff-fasta":         &diffFasta{},
  45                 "stats":              &statscmd{},
  46                 "merge":              &merger{},
  47                 "dump":               &dump{},
  48                 "dumpgob":            &dumpGob{},
  49         })
  50 )
  51
  52 func init() {
  53         if os.Getenv("GOGC") == "" {
  54                 debug.SetGCPercent(30)
  55         }
  56 }
  57
  58 func Main() {
  59         if !isatty.IsTerminal(os.Stderr.Fd()) {
  60                 logrus.StandardLogger().Formatter = &logrus.TextFormatter{DisableTimestamp: true}
  61         }
  62         if len(os.Args) >= 2 && !strings.HasSuffix(os.Args[1], "version") {
  63                 cmd.Version.RunCommand("lightning", nil, nil, os.Stderr, os.Stderr)
  64         }
  65         os.Exit(handler.RunCommand(os.Args[0], os.Args[1:], os.Stdin, os.Stdout, os.Stderr))
  66 }
  67
  68 type buildDockerImage struct{}
  69
  70 func (cmd *buildDockerImage) RunCommand(prog string, args []string, stdin io.Reader, stdout, stderr io.Writer) int {
  71         tmpdir, err := ioutil.TempDir("", "")
  72         if err != nil {
  73                 fmt.Fprint(stderr, err)
  74                 return 1
  75         }
  76         defer os.RemoveAll(tmpdir)
  77         err = ioutil.WriteFile(tmpdir+"/Dockerfile", []byte(`FROM debian:bullseye
  78 RUN DEBIAN_FRONTEND=noninteractive \
  79   apt-get update && \
  80   apt-get dist-upgrade -y && \
  81   apt-get install -y --no-install-recommends bcftools bedtools samtools python2 python3-sklearn python3-matplotlib ca-certificates && \
  82   apt-get clean
  83 `), 0644)
  84         if err != nil {
  85                 fmt.Fprint(stderr, err)
  86                 return 1
  87         }
  88         docker := exec.Command("docker", "build", "--tag=lightning-runtime", tmpdir)
  89         docker.Stdout = stdout
  90         docker.Stderr = stderr
  91         err = docker.Run()
  92         if err != nil {
  93                 return 1
  94         }
  95         fmt.Fprintf(stderr, "built and tagged new docker image, lightning-runtime\n")
  96         return 0
  97 }