ficsit-cli-flake/cli/disk/main.go

67 lines
1.5 KiB
Go
Raw Normal View History

2022-06-22 22:24:35 +00:00
package disk
import (
"io"
"net/url"
"github.com/pkg/errors"
"github.com/rs/zerolog/log"
)
type Disk interface {
// Exists checks if the provided file or directory exists
2022-06-22 22:24:35 +00:00
Exists(path string) error
// Read returns the entire file as a byte buffer
//
// Returns error if provided path is not a file
2022-06-22 22:24:35 +00:00
Read(path string) ([]byte, error)
// Write writes provided byte buffer to the path
2022-06-22 22:24:35 +00:00
Write(path string, data []byte) error
// Remove deletes the provided file or directory recursively
2022-06-22 22:24:35 +00:00
Remove(path string) error
// MkDir creates the provided directory recursively
2022-06-22 22:24:35 +00:00
MkDir(path string) error
// ReadDir returns all entries within the directory
//
// Returns error if provided path is not a directory
2022-06-22 22:24:35 +00:00
ReadDir(path string) ([]Entry, error)
// IsNotExist returns true if provided error is a not-exist type error
2022-06-22 22:24:35 +00:00
IsNotExist(err error) bool
// IsExist returns true if provided error is a does-exist type error
2022-06-22 22:24:35 +00:00
IsExist(err error) bool
// Open opens provided path for writing
2022-06-22 22:24:35 +00:00
Open(path string, flag int) (io.WriteCloser, error)
}
type Entry interface {
IsDir() bool
Name() string
}
func FromPath(path string) (Disk, error) {
parsed, err := url.Parse(path)
if err != nil {
return nil, errors.Wrap(err, "failed to parse path")
}
switch parsed.Scheme {
case "ftp":
log.Info().Str("path", path).Msg("connecting to ftp")
return newFTP(path)
case "sftp":
log.Info().Str("path", path).Msg("connecting to sftp")
return newSFTP(path)
}
log.Info().Str("path", path).Msg("using local disk")
return newLocal(path)
}