mirror of
https://git.sr.ht/~phw/scotty
synced 2025-06-01 19:38:34 +02:00
Moved archive package to public pkg/
This commit is contained in:
parent
28c618ffce
commit
1244405747
10 changed files with 160 additions and 104 deletions
|
@ -21,7 +21,7 @@ import (
|
||||||
"errors"
|
"errors"
|
||||||
"sort"
|
"sort"
|
||||||
|
|
||||||
"go.uploadedlobster.com/scotty/internal/archive"
|
"go.uploadedlobster.com/scotty/pkg/archive"
|
||||||
)
|
)
|
||||||
|
|
||||||
var historyFileGlobs = []string{
|
var historyFileGlobs = []string{
|
||||||
|
|
|
@ -32,7 +32,7 @@ import (
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/simonfrey/jsonl"
|
"github.com/simonfrey/jsonl"
|
||||||
"go.uploadedlobster.com/scotty/internal/archive"
|
"go.uploadedlobster.com/scotty/pkg/archive"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Represents a ListenBrainz export archive.
|
// Represents a ListenBrainz export archive.
|
||||||
|
|
|
@ -27,12 +27,10 @@ THE SOFTWARE.
|
||||||
package archive
|
package archive
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"archive/zip"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"io/fs"
|
"io/fs"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Generic interface to access files inside an archive.
|
// Generic interface to access files inside an archive.
|
||||||
|
@ -101,102 +99,3 @@ type filesystemFile struct {
|
||||||
func (f *filesystemFile) Open() (io.ReadCloser, error) {
|
func (f *filesystemFile) Open() (io.ReadCloser, error) {
|
||||||
return os.Open(f.path)
|
return os.Open(f.path)
|
||||||
}
|
}
|
||||||
|
|
||||||
// An implementation of the archiveBackend interface for zip files.
|
|
||||||
type zipArchive struct {
|
|
||||||
zip *zip.ReadCloser
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *zipArchive) OpenArchive(path string) error {
|
|
||||||
zip, err := zip.OpenReader(path)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
a.zip = zip
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *zipArchive) Close() error {
|
|
||||||
if a.zip == nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return a.zip.Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *zipArchive) Glob(pattern string) ([]FileInfo, error) {
|
|
||||||
result := make([]FileInfo, 0)
|
|
||||||
for _, file := range a.zip.File {
|
|
||||||
if file.FileInfo().IsDir() {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
if matched, err := filepath.Match(pattern, file.Name); matched {
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
info := FileInfo{
|
|
||||||
Name: file.Name,
|
|
||||||
File: file,
|
|
||||||
}
|
|
||||||
result = append(result, info)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return result, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *zipArchive) Open(path string) (fs.File, error) {
|
|
||||||
file, err := a.zip.Open(path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return file, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// An implementation of the archiveBackend interface for directories.
|
|
||||||
type dirArchive struct {
|
|
||||||
path string
|
|
||||||
dirFS fs.FS
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *dirArchive) OpenArchive(path string) error {
|
|
||||||
a.path = filepath.Clean(path)
|
|
||||||
a.dirFS = os.DirFS(path)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *dirArchive) Close() error {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Open opens the named file in the archive.
|
|
||||||
// [fs.File.Close] must be called to release any associated resources.
|
|
||||||
func (a *dirArchive) Open(path string) (fs.File, error) {
|
|
||||||
return a.dirFS.Open(path)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *dirArchive) Glob(pattern string) ([]FileInfo, error) {
|
|
||||||
files, err := fs.Glob(a.dirFS, pattern)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
result := make([]FileInfo, 0)
|
|
||||||
for _, name := range files {
|
|
||||||
stat, err := fs.Stat(a.dirFS, name)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if stat.IsDir() {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
fullPath := filepath.Join(a.path, name)
|
|
||||||
info := FileInfo{
|
|
||||||
Name: name,
|
|
||||||
File: &filesystemFile{path: fullPath},
|
|
||||||
}
|
|
||||||
result = append(result, info)
|
|
||||||
}
|
|
||||||
|
|
||||||
return result, nil
|
|
||||||
}
|
|
|
@ -29,7 +29,7 @@ import (
|
||||||
"slices"
|
"slices"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"go.uploadedlobster.com/scotty/internal/archive"
|
"go.uploadedlobster.com/scotty/pkg/archive"
|
||||||
)
|
)
|
||||||
|
|
||||||
func ExampleOpenArchive() {
|
func ExampleOpenArchive() {
|
77
pkg/archive/dir.go
Normal file
77
pkg/archive/dir.go
Normal file
|
@ -0,0 +1,77 @@
|
||||||
|
/*
|
||||||
|
Copyright © 2025 Philipp Wolfer <phw@uploadedlobster.com>
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in
|
||||||
|
all copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
THE SOFTWARE.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package archive
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io/fs"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
// An implementation of the [ArchiveReader] interface for directories.
|
||||||
|
type dirArchive struct {
|
||||||
|
path string
|
||||||
|
dirFS fs.FS
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *dirArchive) OpenArchive(path string) error {
|
||||||
|
a.path = filepath.Clean(path)
|
||||||
|
a.dirFS = os.DirFS(path)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *dirArchive) Close() error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open opens the named file in the archive.
|
||||||
|
// [fs.File.Close] must be called to release any associated resources.
|
||||||
|
func (a *dirArchive) Open(path string) (fs.File, error) {
|
||||||
|
return a.dirFS.Open(path)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *dirArchive) Glob(pattern string) ([]FileInfo, error) {
|
||||||
|
files, err := fs.Glob(a.dirFS, pattern)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
result := make([]FileInfo, 0)
|
||||||
|
for _, name := range files {
|
||||||
|
stat, err := fs.Stat(a.dirFS, name)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if stat.IsDir() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
fullPath := filepath.Join(a.path, name)
|
||||||
|
info := FileInfo{
|
||||||
|
Name: name,
|
||||||
|
File: &filesystemFile{path: fullPath},
|
||||||
|
}
|
||||||
|
result = append(result, info)
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
80
pkg/archive/zip.go
Normal file
80
pkg/archive/zip.go
Normal file
|
@ -0,0 +1,80 @@
|
||||||
|
/*
|
||||||
|
Copyright © 2025 Philipp Wolfer <phw@uploadedlobster.com>
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in
|
||||||
|
all copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
THE SOFTWARE.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package archive
|
||||||
|
|
||||||
|
import (
|
||||||
|
"archive/zip"
|
||||||
|
"io/fs"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
// An implementation of the [ArchiveReader] interface for zip files.
|
||||||
|
type zipArchive struct {
|
||||||
|
zip *zip.ReadCloser
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *zipArchive) OpenArchive(path string) error {
|
||||||
|
zip, err := zip.OpenReader(path)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.zip = zip
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *zipArchive) Close() error {
|
||||||
|
if a.zip == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return a.zip.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *zipArchive) Glob(pattern string) ([]FileInfo, error) {
|
||||||
|
result := make([]FileInfo, 0)
|
||||||
|
for _, file := range a.zip.File {
|
||||||
|
if file.FileInfo().IsDir() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if matched, err := filepath.Match(pattern, file.Name); matched {
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
info := FileInfo{
|
||||||
|
Name: file.Name,
|
||||||
|
File: file,
|
||||||
|
}
|
||||||
|
result = append(result, info)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *zipArchive) Open(path string) (fs.File, error) {
|
||||||
|
file, err := a.zip.Open(path)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return file, nil
|
||||||
|
}
|
Loading…
Add table
Add a link
Reference in a new issue