Cleanup internal packages
This commit is contained in:
parent
86a4334796
commit
abafab9f0a
7 changed files with 93 additions and 145 deletions
|
|
@ -142,7 +142,7 @@ func (si systemupdate) Run(context wisski_distillery.Context) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := logging.LogOperation(func() error {
|
if err := logging.LogOperation(func() error {
|
||||||
return unpack.InstallResource(dis.RuntimeDir(), "runtime", core.Runtime, func(dst, src string) {
|
return unpack.InstallDir(dis.RuntimeDir(), "runtime", core.Runtime, func(dst, src string) {
|
||||||
context.Printf("[copy] %s\n", dst)
|
context.Printf("[copy] %s\n", dst)
|
||||||
})
|
})
|
||||||
}, context.IOStream, "Unpacking Runtime Components"); err != nil {
|
}, context.IOStream, "Unpacking Runtime Components"); err != nil {
|
||||||
|
|
|
||||||
|
|
@ -45,7 +45,7 @@ type InstallationContext map[string]string
|
||||||
func (is Installable) Install(io stream.IOStream, context InstallationContext) error {
|
func (is Installable) Install(io stream.IOStream, context InstallationContext) error {
|
||||||
if is.ContextPath != "" {
|
if is.ContextPath != "" {
|
||||||
// setup the base files
|
// setup the base files
|
||||||
if err := unpack.InstallResource(
|
if err := unpack.InstallDir(
|
||||||
is.Dir,
|
is.Dir,
|
||||||
is.ContextPath,
|
is.ContextPath,
|
||||||
is.Resources,
|
is.Resources,
|
||||||
|
|
|
||||||
|
|
@ -1,3 +1,4 @@
|
||||||
|
// Package targz provides facilities for packaging tar.gz files
|
||||||
package targz
|
package targz
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
|
@ -9,7 +10,9 @@ import (
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Package packages the directory src into dst.
|
// Package packages the source directory into a 'tar.gz' file into destination.
|
||||||
|
// If the destination already exists, it is truncated.
|
||||||
|
//
|
||||||
// onCopy, when not nil, is called for each file being copied into the archive.
|
// onCopy, when not nil, is called for each file being copied into the archive.
|
||||||
func Package(dst, src string, onCopy func(rel string, src string)) (count int64, err error) {
|
func Package(dst, src string, onCopy func(rel string, src string)) (count int64, err error) {
|
||||||
// create the target archive
|
// create the target archive
|
||||||
|
|
|
||||||
|
|
@ -1,76 +0,0 @@
|
||||||
package unpack
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io"
|
|
||||||
"io/fs"
|
|
||||||
"os"
|
|
||||||
|
|
||||||
"github.com/pkg/errors"
|
|
||||||
)
|
|
||||||
|
|
||||||
// InstallFile installs the file from src into dst.
|
|
||||||
//
|
|
||||||
// If the destination path does not exist, it is created.
|
|
||||||
func InstallFile(dst string, src fs.File) error {
|
|
||||||
// stat it!
|
|
||||||
srcInfo, err := src.Stat()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// if this is a directory, something went wrong!
|
|
||||||
if srcInfo.IsDir() {
|
|
||||||
return errExpectedFileButGotDirectory
|
|
||||||
}
|
|
||||||
|
|
||||||
// and store it there!
|
|
||||||
return installFile(dst, srcInfo, src)
|
|
||||||
}
|
|
||||||
|
|
||||||
func installFile(dst string, srcInfo fs.FileInfo, src fs.File) error {
|
|
||||||
// create the file using the right mode!
|
|
||||||
file, err := os.OpenFile(dst, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, srcInfo.Mode())
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer file.Close()
|
|
||||||
|
|
||||||
// copy over the content!
|
|
||||||
_, err = io.Copy(file, src)
|
|
||||||
return errors.Wrapf(err, "Error writing to destination %s", dst)
|
|
||||||
}
|
|
||||||
|
|
||||||
// InstallTemplate unpacks the resource located at src in fsys, then processes it as a template, and eventually writes it to dst.
|
|
||||||
// Any existing file is truncated and overwritten.
|
|
||||||
//
|
|
||||||
// See [WriteTemplate] for possible errors.
|
|
||||||
func InstallTemplate(dst string, context map[string]string, src string, fsys fs.FS) error {
|
|
||||||
|
|
||||||
// open the srcFile
|
|
||||||
srcFile, err := fsys.Open(src)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer srcFile.Close()
|
|
||||||
|
|
||||||
// stat it
|
|
||||||
srcInfo, err := srcFile.Stat()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// check if it is a directory
|
|
||||||
if srcInfo.IsDir() {
|
|
||||||
return errExpectedFileButGotDirectory
|
|
||||||
}
|
|
||||||
|
|
||||||
// open the destination file
|
|
||||||
file, err := os.OpenFile(dst, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, srcInfo.Mode())
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer file.Close()
|
|
||||||
|
|
||||||
// write the file!
|
|
||||||
return WriteTemplate(file, context, srcFile)
|
|
||||||
}
|
|
||||||
|
|
@ -1,6 +1,7 @@
|
||||||
package unpack
|
package unpack
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"io"
|
||||||
"io/fs"
|
"io/fs"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
|
@ -8,6 +9,9 @@ import (
|
||||||
"github.com/pkg/errors"
|
"github.com/pkg/errors"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var errExpectedFileButGotDirectory = errors.New("expected a file, but got a directory")
|
||||||
|
var errExpectedDirectoryButGotFile = errors.New("expected a directory, but got a file")
|
||||||
|
|
||||||
// InstallDir installs the directory at src within fsys to dst.
|
// InstallDir installs the directory at src within fsys to dst.
|
||||||
//
|
//
|
||||||
// onInstallFile is called for each file or directory being installed.
|
// onInstallFile is called for each file or directory being installed.
|
||||||
|
|
@ -42,6 +46,38 @@ func InstallDir(dst string, src string, fsys fs.FS, onInstallFile func(dst, src
|
||||||
return installDir(dst, srcInfo, srcFile.(fs.ReadDirFile), src, fsys, onInstallFile)
|
return installDir(dst, srcInfo, srcFile.(fs.ReadDirFile), src, fsys, onInstallFile)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// installResource installs the resource at src within fsys to dst.
|
||||||
|
//
|
||||||
|
// OnInstallFile is called for each source and destination file.
|
||||||
|
// OnInstallFile may be nil.
|
||||||
|
func installResource(dst string, src string, fsys fs.FS, onInstallFile func(dst, src string)) error {
|
||||||
|
// open the srcFile
|
||||||
|
srcFile, err := fsys.Open(src)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer srcFile.Close()
|
||||||
|
|
||||||
|
// stat it!
|
||||||
|
srcInfo, err := srcFile.Stat()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// call the hook (if any)
|
||||||
|
if onInstallFile != nil {
|
||||||
|
onInstallFile(dst, src)
|
||||||
|
}
|
||||||
|
|
||||||
|
// this is a directory, so the cast is safe!
|
||||||
|
if srcInfo.IsDir() {
|
||||||
|
return installDir(dst, srcInfo, srcFile.(fs.ReadDirFile), src, fsys, onInstallFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
// this is a regular file!
|
||||||
|
return installFile(dst, srcInfo, srcFile)
|
||||||
|
}
|
||||||
|
|
||||||
func installDir(dst string, srcInfo fs.FileInfo, srcFile fs.ReadDirFile, src string, fsys fs.FS, onInstallFile func(dst, src string)) error {
|
func installDir(dst string, srcInfo fs.FileInfo, srcFile fs.ReadDirFile, src string, fsys fs.FS, onInstallFile func(dst, src string)) error {
|
||||||
// create the destination
|
// create the destination
|
||||||
dstStat, dstErr := os.Stat(dst)
|
dstStat, dstErr := os.Stat(dst)
|
||||||
|
|
@ -68,7 +104,7 @@ func installDir(dst string, srcInfo fs.FileInfo, srcFile fs.ReadDirFile, src str
|
||||||
|
|
||||||
// iterate over all the children
|
// iterate over all the children
|
||||||
for _, entry := range entries {
|
for _, entry := range entries {
|
||||||
if err := InstallResource(
|
if err := installResource(
|
||||||
filepath.Join(dst, entry.Name()),
|
filepath.Join(dst, entry.Name()),
|
||||||
filepath.Join(src, entry.Name()),
|
filepath.Join(src, entry.Name()),
|
||||||
fsys,
|
fsys,
|
||||||
|
|
@ -80,3 +116,16 @@ func installDir(dst string, srcInfo fs.FileInfo, srcFile fs.ReadDirFile, src str
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func installFile(dst string, srcInfo fs.FileInfo, src fs.File) error {
|
||||||
|
// create the file using the right mode!
|
||||||
|
file, err := os.OpenFile(dst, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, srcInfo.Mode())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
// copy over the content!
|
||||||
|
_, err = io.Copy(file, src)
|
||||||
|
return errors.Wrapf(err, "Error writing to destination %s", dst)
|
||||||
|
}
|
||||||
|
|
@ -5,6 +5,8 @@ import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
|
"io/fs"
|
||||||
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"golang.org/x/exp/maps"
|
"golang.org/x/exp/maps"
|
||||||
|
|
@ -194,3 +196,38 @@ parseloop:
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// InstallTemplate unpacks the resource located at src in fsys, then processes it as a template, and eventually writes it to dst.
|
||||||
|
// Any existing file is truncated and overwritten.
|
||||||
|
//
|
||||||
|
// See [WriteTemplate] for possible errors.
|
||||||
|
func InstallTemplate(dst string, context map[string]string, src string, fsys fs.FS) error {
|
||||||
|
|
||||||
|
// open the srcFile
|
||||||
|
srcFile, err := fsys.Open(src)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer srcFile.Close()
|
||||||
|
|
||||||
|
// stat it
|
||||||
|
srcInfo, err := srcFile.Stat()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// check if it is a directory
|
||||||
|
if srcInfo.IsDir() {
|
||||||
|
return errExpectedFileButGotDirectory
|
||||||
|
}
|
||||||
|
|
||||||
|
// open the destination file
|
||||||
|
file, err := os.OpenFile(dst, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, srcInfo.Mode())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
// write the file!
|
||||||
|
return WriteTemplate(file, context, srcFile)
|
||||||
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,65 +0,0 @@
|
||||||
// Package unpack unpacks files and templates to a target directory
|
|
||||||
package unpack
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"errors"
|
|
||||||
"io/fs"
|
|
||||||
)
|
|
||||||
|
|
||||||
var errExpectedFileButGotDirectory = errors.New("expected a file, but got a directory")
|
|
||||||
var errExpectedDirectoryButGotFile = errors.New("expected a directory, but got a file")
|
|
||||||
|
|
||||||
// InstallResource installs the resource at src within fsys to dst.
|
|
||||||
//
|
|
||||||
// OnInstallFile is called for each source and destination file.
|
|
||||||
// OnInstallFile may be nil.
|
|
||||||
//
|
|
||||||
// See [InstallDir] or [InstallFile].
|
|
||||||
func InstallResource(dst string, src string, fsys fs.FS, onInstallFile func(dst, src string)) error {
|
|
||||||
// open the srcFile
|
|
||||||
srcFile, err := fsys.Open(src)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer srcFile.Close()
|
|
||||||
|
|
||||||
// stat it!
|
|
||||||
srcInfo, err := srcFile.Stat()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// call the hook (if any)
|
|
||||||
if onInstallFile != nil {
|
|
||||||
onInstallFile(dst, src)
|
|
||||||
}
|
|
||||||
|
|
||||||
// this is a directory, so the cast is safe!
|
|
||||||
if srcInfo.IsDir() {
|
|
||||||
return installDir(dst, srcInfo, srcFile.(fs.ReadDirFile), src, fsys, onInstallFile)
|
|
||||||
}
|
|
||||||
|
|
||||||
// this is a regular file!
|
|
||||||
return installFile(dst, srcInfo, srcFile)
|
|
||||||
}
|
|
||||||
|
|
||||||
// UnpackTemplate unpacks the given file template and template.
|
|
||||||
// See [WriteTemplate] for possible errors.
|
|
||||||
func UnpackTemplate(context map[string]string, src fs.File) ([]byte, fs.FileMode, error) {
|
|
||||||
// stat the source file to install
|
|
||||||
srcStat, srcErr := src.Stat()
|
|
||||||
if srcErr != nil {
|
|
||||||
return nil, 0, srcErr
|
|
||||||
}
|
|
||||||
|
|
||||||
// should not be a directory
|
|
||||||
if srcStat.IsDir() {
|
|
||||||
return nil, 0, errExpectedFileButGotDirectory
|
|
||||||
}
|
|
||||||
|
|
||||||
// read all the bytes into a buffer
|
|
||||||
var buffer bytes.Buffer
|
|
||||||
err := WriteTemplate(&buffer, context, src)
|
|
||||||
return buffer.Bytes(), srcStat.Mode(), err
|
|
||||||
}
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue