146 lines
2.8 KiB
Go
146 lines
2.8 KiB
Go
// Copyright 2023 Mikhail Klementev. All rights reserved.
|
|
// Use of this source code is governed by a AGPLv3 license
|
|
// (or later) that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"os/user"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
type container struct {
|
|
name string
|
|
timeout time.Duration
|
|
Volumes struct {
|
|
LibModules string
|
|
UsrSrc string
|
|
Boot string
|
|
}
|
|
// Additional arguments
|
|
Args []string
|
|
}
|
|
|
|
func NewContainer(name string, timeout time.Duration) (c container, err error) {
|
|
c.name = name
|
|
c.timeout = timeout
|
|
|
|
usr, err := user.Current()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
c.Volumes.LibModules = fmt.Sprintf(
|
|
"%s/.out-of-tree/volumes/%s/lib/modules", usr.HomeDir, name)
|
|
os.MkdirAll(c.Volumes.LibModules, 0777)
|
|
|
|
c.Volumes.UsrSrc = fmt.Sprintf(
|
|
"%s/.out-of-tree/volumes/%s/usr/src", usr.HomeDir, name)
|
|
os.MkdirAll(c.Volumes.UsrSrc, 0777)
|
|
|
|
c.Volumes.Boot = fmt.Sprintf(
|
|
"%s/.out-of-tree/volumes/%s/boot", usr.HomeDir, name)
|
|
os.MkdirAll(c.Volumes.Boot, 0777)
|
|
|
|
return
|
|
}
|
|
|
|
func (c container) Build(imagePath string) (output string, err error) {
|
|
args := []string{"build"}
|
|
args = append(args, "-t", c.name, imagePath)
|
|
|
|
cmd := exec.Command("docker", args...)
|
|
|
|
flog := log.With().
|
|
Str("command", fmt.Sprintf("%v", cmd)).
|
|
Logger()
|
|
|
|
stdout, err := cmd.StdoutPipe()
|
|
if err != nil {
|
|
return
|
|
}
|
|
cmd.Stderr = cmd.Stdout
|
|
|
|
err = cmd.Start()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
scanner := bufio.NewScanner(stdout)
|
|
for scanner.Scan() {
|
|
m := scanner.Text()
|
|
output += m + "\n"
|
|
flog.Trace().Str("stdout", m).Msg("")
|
|
}
|
|
}()
|
|
|
|
err = cmd.Wait()
|
|
return
|
|
}
|
|
|
|
func (c container) Run(workdir string, command string) (output string, err error) {
|
|
flog := log.With().
|
|
Str("container", c.name).
|
|
Str("workdir", workdir).
|
|
Str("command", command).
|
|
Logger()
|
|
|
|
var args []string
|
|
args = append(args, "run", "--rm")
|
|
args = append(args, c.Args...)
|
|
args = append(args,
|
|
"-v", workdir+":/work",
|
|
"-v", c.Volumes.LibModules+":/lib/modules",
|
|
"-v", c.Volumes.UsrSrc+":/usr/src",
|
|
"-v", c.Volumes.Boot+":/boot")
|
|
args = append(args, c.name, "bash", "-c", "cd /work && "+command)
|
|
|
|
cmd := exec.Command("docker", args...)
|
|
|
|
log.Debug().Msgf("%v", cmd)
|
|
|
|
stdout, err := cmd.StdoutPipe()
|
|
if err != nil {
|
|
return
|
|
}
|
|
cmd.Stderr = cmd.Stdout
|
|
|
|
timer := time.AfterFunc(c.timeout, func() {
|
|
flog.Info().Msg("killing container by timeout")
|
|
cmd.Process.Kill()
|
|
})
|
|
defer timer.Stop()
|
|
|
|
err = cmd.Start()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
scanner := bufio.NewScanner(stdout)
|
|
for scanner.Scan() {
|
|
m := scanner.Text()
|
|
output += m + "\n"
|
|
flog.Trace().Str("stdout", m).Msg("")
|
|
}
|
|
}()
|
|
|
|
err = cmd.Wait()
|
|
if err != nil {
|
|
e := fmt.Sprintf("error `%v` for cmd `%v` with output `%v`",
|
|
err, command, output)
|
|
err = errors.New(e)
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|