104 lines
2.2 KiB
Go
104 lines
2.2 KiB
Go
package pods
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/containers/podman/v4/pkg/bindings"
|
|
"github.com/containers/podman/v4/pkg/bindings/containers"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// CopySkelToContainer copies an Archive into the containers root
|
|
func CopySkelToContainer(id string) error {
|
|
chownBool := viper.GetBool("skel_chown")
|
|
log.Println("Copying skel Archive to container")
|
|
var err error
|
|
conn, err := bindings.GetClient(Socket)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
options := containers.CopyOptions{
|
|
Chown: &chownBool,
|
|
}
|
|
|
|
fileInfo, err := os.Stat(viper.GetString("skel"))
|
|
|
|
if err != nil {
|
|
log.Println("Could not access Skel archive")
|
|
return err
|
|
}
|
|
log.Println("Skel Archive is ", fileInfo.Size(), "b")
|
|
|
|
file, err := os.Open(viper.GetString("skel"))
|
|
if err != nil {
|
|
log.Println("Could not open skel archive", err)
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
params, err := options.ToParams()
|
|
if err != nil {
|
|
log.Println("Could not parse skel options", err)
|
|
return err
|
|
}
|
|
params.Set("path", viper.GetString("skel_target"))
|
|
|
|
fmt.Println("Container", id, " Params: ", params.Encode())
|
|
response, err := conn.DoRequest(
|
|
Socket,
|
|
file,
|
|
http.MethodPut,
|
|
"/containers/%s/archive",
|
|
params,
|
|
nil,
|
|
id,
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if response.StatusCode != http.StatusOK {
|
|
log.Println("Error putting archive into container: ", response.StatusCode, response.Body)
|
|
return err
|
|
}
|
|
log.Println(response.StatusCode, response.Body)
|
|
if viper.GetString("skel_user") != "" {
|
|
chownSkel(id)
|
|
|
|
}
|
|
|
|
// Manually chown the Directory, because podmans chown is sometimes unreliable
|
|
return response.Process(nil)
|
|
}
|
|
|
|
func chownSkel(id string) {
|
|
command := fmt.Sprintf(
|
|
"chown -R %s %s \n\n",
|
|
viper.GetString("skel_user"),
|
|
viper.GetString("skel_target"),
|
|
)
|
|
done := make(chan bool)
|
|
detachKeys := "\n\n"
|
|
commandReader := strings.NewReader(command)
|
|
var output bytes.Buffer
|
|
outputWriter := io.Writer(&output)
|
|
options := &containers.AttachOptions{
|
|
DetachKeys: &detachKeys,
|
|
}
|
|
|
|
err := containers.Attach(Socket, id, commandReader, outputWriter, outputWriter, done, options)
|
|
|
|
if err != nil {
|
|
log.Printf("Error: %v \n %s\n", err, output.String())
|
|
} else {
|
|
log.Println(output.String())
|
|
}
|
|
|
|
}
|