52c0fa6b6d
This commit introduces a significant refactoring of the `cmd` package to improve testability and increases test coverage across the application. Key changes include: - Refactored Cobra commands to use `RunE` for better error handling and testing. - Extracted business logic from command handlers into separate, testable functions. - Added comprehensive unit tests for the `cmd`, `compress`, `github`, `logger`, and `pwa` packages. - Added tests for missing command-line arguments, as requested. - Implemented the `borg all` command to clone all public repositories for a GitHub user or organization. - Restored and improved the `collect pwa` functionality. - Removed duplicate code and fixed various bugs. - Addressed a resource leak in the `all` command. - Improved error handling in the `pwa` package. - Refactored `main.go` to remove duplicated logic. - Fixed several other minor bugs and inconsistencies.
100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
package tarfs
|
|
|
|
import (
|
|
"archive/tar"
|
|
"bytes"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// TarFS is a http.FileSystem that serves files from a tar archive.
|
|
type TarFS struct {
|
|
files map[string]*tarFile
|
|
}
|
|
|
|
// New creates a new TarFS from a tar archive.
|
|
func New(data []byte) (*TarFS, error) {
|
|
fs := &TarFS{
|
|
files: make(map[string]*tarFile),
|
|
}
|
|
|
|
tr := tar.NewReader(bytes.NewReader(data))
|
|
for {
|
|
hdr, err := tr.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if strings.HasPrefix(hdr.Name, "rootfs/") {
|
|
content, err := io.ReadAll(tr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
fs.files[strings.TrimPrefix(hdr.Name, "rootfs/")] = &tarFile{
|
|
header: hdr,
|
|
content: bytes.NewReader(content),
|
|
modTime: hdr.ModTime,
|
|
}
|
|
}
|
|
}
|
|
|
|
return fs, nil
|
|
}
|
|
|
|
// Open opens a file from the tar archive.
|
|
func (fs *TarFS) Open(name string) (http.File, error) {
|
|
name = strings.TrimPrefix(name, "/")
|
|
if file, ok := fs.files[name]; ok {
|
|
// Reset the reader to the beginning of the file
|
|
file.content.Seek(0, 0)
|
|
return file, nil
|
|
}
|
|
|
|
return nil, os.ErrNotExist
|
|
}
|
|
|
|
// tarFile is a http.File that represents a file in a tar archive.
|
|
type tarFile struct {
|
|
header *tar.Header
|
|
content *bytes.Reader
|
|
modTime time.Time
|
|
}
|
|
|
|
func (f *tarFile) Close() error { return nil }
|
|
func (f *tarFile) Read(p []byte) (int, error) { return f.content.Read(p) }
|
|
func (f *tarFile) Seek(offset int64, whence int) (int64, error) {
|
|
return f.content.Seek(offset, whence)
|
|
}
|
|
|
|
func (f *tarFile) Readdir(count int) ([]os.FileInfo, error) {
|
|
return nil, os.ErrInvalid
|
|
}
|
|
|
|
func (f *tarFile) Stat() (os.FileInfo, error) {
|
|
return &tarFileInfo{
|
|
name: path.Base(f.header.Name),
|
|
size: f.header.Size,
|
|
modTime: f.modTime,
|
|
}, nil
|
|
}
|
|
|
|
// tarFileInfo is a os.FileInfo that represents a file in a tar archive.
|
|
type tarFileInfo struct {
|
|
name string
|
|
size int64
|
|
modTime time.Time
|
|
}
|
|
|
|
func (i *tarFileInfo) Name() string { return i.name }
|
|
func (i *tarFileInfo) Size() int64 { return i.size }
|
|
func (i *tarFileInfo) Mode() os.FileMode { return 0444 }
|
|
func (i *tarFileInfo) ModTime() time.Time { return i.modTime }
|
|
func (i *tarFileInfo) IsDir() bool { return false }
|
|
func (i *tarFileInfo) Sys() interface{} { return nil }
|