mirror of
https://github.com/criyle/go-sandbox.git
synced 2025-09-26 23:19:11 +08:00
109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
package container
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/criyle/go-sandbox/pkg/unixsocket"
|
|
)
|
|
|
|
func (c *containerServer) handlePing() error {
|
|
return c.sendReply(reply{}, unixsocket.Msg{})
|
|
}
|
|
|
|
func (c *containerServer) handleConf(conf *confCmd) error {
|
|
if conf != nil {
|
|
c.containerConfig = conf.Conf
|
|
if err := initContainer(conf.Conf); err != nil {
|
|
return err
|
|
}
|
|
if c.ContainerUID == 0 {
|
|
c.ContainerUID = containerUID
|
|
}
|
|
if c.ContainerGID == 0 {
|
|
c.ContainerGID = containerGID
|
|
}
|
|
env, err := readDotEnv()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
c.defaultEnv = env
|
|
}
|
|
return c.sendReply(reply{}, unixsocket.Msg{})
|
|
}
|
|
|
|
func (c *containerServer) handleOpen(open []OpenCmd) error {
|
|
if len(open) == 0 {
|
|
return c.sendErrorReply("open: no open parameter received")
|
|
}
|
|
|
|
// open files
|
|
fds := make([]int, 0, len(open))
|
|
fileToClose := make([]*os.File, 0, len(open)) // let sendMsg close these files
|
|
for _, o := range open {
|
|
outFile, err := os.OpenFile(o.Path, o.Flag, o.Perm)
|
|
if err != nil {
|
|
for _, f := range fileToClose {
|
|
f.Close()
|
|
}
|
|
return c.sendErrorReply("open: %v", err)
|
|
}
|
|
fileToClose = append(fileToClose, outFile)
|
|
fds = append(fds, int(outFile.Fd()))
|
|
}
|
|
|
|
return c.sendReplyFiles(reply{}, unixsocket.Msg{Fds: fds}, fileToClose)
|
|
}
|
|
|
|
func (c *containerServer) handleDelete(delete *deleteCmd) error {
|
|
if delete == nil {
|
|
return c.sendErrorReply("delete: no parameter provided")
|
|
}
|
|
if err := os.Remove(delete.Path); err != nil {
|
|
return c.sendErrorReply("delete: %v", err)
|
|
}
|
|
return c.sendReply(reply{}, unixsocket.Msg{})
|
|
}
|
|
|
|
func (c *containerServer) handleReset() error {
|
|
for _, m := range c.Mounts {
|
|
if !m.IsTmpFs() {
|
|
continue
|
|
}
|
|
if err := removeContents(filepath.Join("/", m.Target)); err != nil {
|
|
return c.sendErrorReply("reset: %v %v", m.Target, err)
|
|
}
|
|
}
|
|
return c.sendReply(reply{}, unixsocket.Msg{})
|
|
}
|
|
|
|
// readDotEnv attempts to read /.env file and save as default environment variables
|
|
func readDotEnv() ([]string, error) {
|
|
f, err := os.Open("/.env")
|
|
if err != nil {
|
|
if errors.Is(err, os.ErrNotExist) {
|
|
return nil, nil
|
|
}
|
|
return nil, fmt.Errorf("dotenv: open /.env: %w", err)
|
|
}
|
|
defer f.Close()
|
|
|
|
var ret []string
|
|
scanner := bufio.NewScanner(f)
|
|
for scanner.Scan() {
|
|
line := strings.TrimSpace(scanner.Text())
|
|
if len(line) == 0 || strings.HasPrefix(line, "#") {
|
|
continue
|
|
}
|
|
if !strings.Contains(line, "=") {
|
|
return nil, fmt.Errorf("dotenv: invalid line: %s", line)
|
|
}
|
|
ret = append(ret, line)
|
|
}
|
|
return ret, nil
|
|
}
|