Move file backend into own file

This commit is contained in:
Sam W 2023-05-04 22:38:00 +01:00
parent 7ae7166ac1
commit d0238d3faa
2 changed files with 59 additions and 53 deletions

View File

@ -2,14 +2,7 @@ package debanator
import ( import (
"io" "io"
"io/fs"
"net/http" "net/http"
"os"
"path"
"path/filepath"
"strings"
log "github.com/sirupsen/logrus"
) )
// A backend to search for packages in // A backend to search for packages in
@ -30,49 +23,3 @@ type DebFile interface {
GetName() string GetName() string
} }
type FileBackend struct {
path string
}
// A deb file existing on the local filesystem
type fsDebFile struct {
path string
}
func (f fsDebFile) GetReader() (ReaderAtCloser, error) {
return os.Open(f.path)
}
func (f fsDebFile) GetName() string {
_, name := filepath.Split(f.path)
return name
}
func NewFileBackend(path string) FileBackend {
return FileBackend{path}
}
func (fb FileBackend) ServeFiles(prefix string) http.Handler {
return http.StripPrefix(path.Join(prefix, "pool"), http.FileServer(http.Dir(fb.path)))
}
func (fb FileBackend) GetFiles() ([]DebFile, error) {
var debs []DebFile
fs.WalkDir(os.DirFS(fb.path), ".", func(dirpath string, dir fs.DirEntry, err error) error {
if err != nil {
log.WithFields(log.Fields{
"path": dirpath,
"error": err,
}).Warn("Error scanning for debs")
return nil
}
if !dir.IsDir() && strings.HasSuffix(dir.Name(), ".deb") {
debs = append(debs, DebFile(fsDebFile{
filepath.Join(fb.path, dirpath),
}))
}
return nil
})
log.Infof("got files: %v", debs)
return debs, nil
}

59
backend_file.go Normal file
View File

@ -0,0 +1,59 @@
package debanator
import (
"io/fs"
"net/http"
"os"
"path"
"path/filepath"
"strings"
log "github.com/sirupsen/logrus"
)
// A deb file existing on the local filesystem
type fsDebFile struct {
path string
}
func (f fsDebFile) GetReader() (ReaderAtCloser, error) {
return os.Open(f.path)
}
func (f fsDebFile) GetName() string {
_, name := filepath.Split(f.path)
return name
}
type FileBackend struct {
path string
}
func NewFileBackend(path string) FileBackend {
return FileBackend{path}
}
func (fb FileBackend) ServeFiles(prefix string) http.Handler {
return http.StripPrefix(path.Join(prefix, "pool"), http.FileServer(http.Dir(fb.path)))
}
func (fb FileBackend) GetFiles() ([]DebFile, error) {
var debs []DebFile
fs.WalkDir(os.DirFS(fb.path), ".", func(dirpath string, dir fs.DirEntry, err error) error {
if err != nil {
log.WithFields(log.Fields{
"path": dirpath,
"error": err,
}).Warn("Error scanning for debs")
return nil
}
if !dir.IsDir() && strings.HasSuffix(dir.Name(), ".deb") {
debs = append(debs, DebFile(fsDebFile{
filepath.Join(fb.path, dirpath),
}))
}
return nil
})
log.Infof("got files: %v", debs)
return debs, nil
}