Start to add zap logger everywhere
This commit is contained in:
@@ -4,6 +4,7 @@ import (
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"github.com/debridmediamanager.com/zurg/pkg/logutil"
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
@@ -19,10 +20,14 @@ type ConfigInterface interface {
|
||||
MeetsConditions(directory, torrentID, torrentName string, fileNames []string) bool
|
||||
GetOnLibraryUpdate() string
|
||||
GetNetworkBufferSize() int
|
||||
GetMountPath() string
|
||||
GetMountPoint() string
|
||||
}
|
||||
|
||||
func LoadZurgConfig(filename string) (ConfigInterface, error) {
|
||||
rlog := logutil.NewLogger()
|
||||
log := rlog.Named("config")
|
||||
|
||||
log.Debug("Loading config file", filename)
|
||||
content, err := os.ReadFile(filename)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
@@ -35,6 +40,7 @@ func LoadZurgConfig(filename string) (ConfigInterface, error) {
|
||||
|
||||
switch initialConfig.Version {
|
||||
case "v1":
|
||||
log.Debug("Detected config version: v1")
|
||||
return loadV1Config(content)
|
||||
|
||||
default:
|
||||
|
||||
@@ -10,7 +10,7 @@ type ZurgConfig struct {
|
||||
CanRepair bool `yaml:"enable_repair"`
|
||||
OnLibraryUpdate string `yaml:"on_library_update"`
|
||||
NetworkBufferSize int `yaml:"network_buffer_size"`
|
||||
MountPath string `yaml:"mount_path"`
|
||||
MountPoint string `yaml:"mount_point"`
|
||||
}
|
||||
|
||||
func (z *ZurgConfig) GetToken() string {
|
||||
@@ -57,6 +57,6 @@ func (z *ZurgConfig) GetNetworkBufferSize() int {
|
||||
return z.NetworkBufferSize
|
||||
}
|
||||
|
||||
func (z *ZurgConfig) GetMountPath() string {
|
||||
return z.MountPath
|
||||
func (z *ZurgConfig) GetMountPoint() string {
|
||||
return z.MountPoint
|
||||
}
|
||||
|
||||
@@ -5,14 +5,19 @@ import (
|
||||
"sort"
|
||||
"strings"
|
||||
|
||||
"github.com/debridmediamanager.com/zurg/pkg/logutil"
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
func loadV1Config(content []byte) (*ZurgConfigV1, error) {
|
||||
rlog := logutil.NewLogger()
|
||||
log := rlog.Named("config")
|
||||
|
||||
var configV1 ZurgConfigV1
|
||||
if err := yaml.Unmarshal(content, &configV1); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
log.Debug("V1 config loaded successfully")
|
||||
|
||||
return &configV1, nil
|
||||
}
|
||||
@@ -22,9 +27,13 @@ func (z *ZurgConfigV1) GetVersion() string {
|
||||
}
|
||||
|
||||
func (z *ZurgConfigV1) GetDirectories() []string {
|
||||
rlog := logutil.NewLogger()
|
||||
log := rlog.Named("config")
|
||||
|
||||
rootDirectories := make([]string, len(z.Directories))
|
||||
i := 0
|
||||
for directory := range z.Directories {
|
||||
log.Debug("Adding to rootDirectories", directory)
|
||||
rootDirectories[i] = directory
|
||||
i++
|
||||
}
|
||||
@@ -32,6 +41,12 @@ func (z *ZurgConfigV1) GetDirectories() []string {
|
||||
}
|
||||
|
||||
func (z *ZurgConfigV1) GetGroupMap() map[string][]string {
|
||||
// Obtain a new sugared logger instance
|
||||
rlog := logutil.NewLogger()
|
||||
log := rlog.Named("config")
|
||||
|
||||
log.Debug("Starting to build the group map from directories")
|
||||
|
||||
var groupMap = make(map[string][]string)
|
||||
var groupOrderMap = make(map[string]int) // To store GroupOrder for each directory
|
||||
|
||||
@@ -39,6 +54,7 @@ func (z *ZurgConfigV1) GetGroupMap() map[string][]string {
|
||||
for directory, val := range z.Directories {
|
||||
groupMap[val.Group] = append(groupMap[val.Group], directory)
|
||||
groupOrderMap[directory] = val.GroupOrder
|
||||
log.Debugf("Added directory to group: %s, group: %s, groupOrder: %d", directory, val.Group, val.GroupOrder)
|
||||
}
|
||||
|
||||
// Sort the slice based on GroupOrder and then directory name for deterministic order
|
||||
@@ -50,6 +66,7 @@ func (z *ZurgConfigV1) GetGroupMap() map[string][]string {
|
||||
return groupOrderMap[dirs[i]] < groupOrderMap[dirs[j]]
|
||||
})
|
||||
groupMap[group] = dirs
|
||||
log.Debugf("Sorted directories within a group: %s, sortedDirectories: %v", group, dirs)
|
||||
}
|
||||
|
||||
// Return a deep copy of the map
|
||||
|
||||
@@ -1,38 +0,0 @@
|
||||
package mount
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"bazil.org/fuse"
|
||||
"github.com/mdlayher/sdnotify"
|
||||
)
|
||||
|
||||
func Mount(mountpoint string) error {
|
||||
n, err := sdnotify.New()
|
||||
if err != nil && !errors.Is(err, os.ErrNotExist) {
|
||||
log.Fatalf("failed to open systemd notifier: %v", err)
|
||||
}
|
||||
err = n.Notify(
|
||||
sdnotify.Statusf("service started successfully"),
|
||||
sdnotify.Ready,
|
||||
)
|
||||
if err != nil {
|
||||
log.Fatalf("failed to send ready notification: %v", err)
|
||||
}
|
||||
return Unmount(mountpoint, n)
|
||||
}
|
||||
|
||||
func Unmount(mountpoint string, n *sdnotify.Notifier) error {
|
||||
log.Println("Unmounting...")
|
||||
err := n.Notify(
|
||||
sdnotify.Statusf("service stopped successfully"),
|
||||
sdnotify.Ready,
|
||||
)
|
||||
if err != nil {
|
||||
log.Fatalf("failed to send stop notification: %v", err)
|
||||
}
|
||||
fuse.Unmount(mountpoint)
|
||||
return nil
|
||||
}
|
||||
21
internal/zfs/fs.go
Normal file
21
internal/zfs/fs.go
Normal file
@@ -0,0 +1,21 @@
|
||||
package zfs
|
||||
|
||||
import (
|
||||
"os"
|
||||
"sync"
|
||||
|
||||
"bazil.org/fuse/fs"
|
||||
)
|
||||
|
||||
type FS struct {
|
||||
uid uint32
|
||||
gid uint32
|
||||
umask os.FileMode
|
||||
directIO bool
|
||||
lock sync.RWMutex
|
||||
}
|
||||
|
||||
// Root returns the root path
|
||||
func (f *FS) Root() (fs.Node, error) {
|
||||
return nil, nil
|
||||
}
|
||||
45
internal/zfs/mount.go
Normal file
45
internal/zfs/mount.go
Normal file
@@ -0,0 +1,45 @@
|
||||
package zfs
|
||||
|
||||
import (
|
||||
"os"
|
||||
|
||||
"bazil.org/fuse"
|
||||
"bazil.org/fuse/fs"
|
||||
|
||||
"golang.org/x/sys/unix"
|
||||
)
|
||||
|
||||
func Mount(mountpoint string) error {
|
||||
options := []fuse.MountOption{
|
||||
fuse.AllowOther(),
|
||||
fuse.AllowNonEmptyMount(),
|
||||
fuse.MaxReadahead(uint32(128 << 10)),
|
||||
fuse.DefaultPermissions(),
|
||||
fuse.FSName("zurgfs"),
|
||||
}
|
||||
c, err := fuse.Mount(mountpoint, options...)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer c.Close()
|
||||
|
||||
srv := fs.New(c, nil)
|
||||
|
||||
filesys := &FS{
|
||||
uid: uint32(unix.Geteuid()),
|
||||
gid: uint32(unix.Getegid()),
|
||||
umask: os.FileMode(0),
|
||||
}
|
||||
|
||||
// Serve our tree via FUSE.
|
||||
if err := srv.Serve(filesys); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func Unmount(mountpoint string) error {
|
||||
fuse.Unmount(mountpoint)
|
||||
return nil
|
||||
}
|
||||
50
internal/zfs/object.go
Normal file
50
internal/zfs/object.go
Normal file
@@ -0,0 +1,50 @@
|
||||
package zfs
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"bazil.org/fuse"
|
||||
"bazil.org/fuse/fs"
|
||||
"github.com/debridmediamanager.com/zurg/internal/torrent"
|
||||
)
|
||||
|
||||
type Object struct {
|
||||
fs *FS
|
||||
objectID string
|
||||
}
|
||||
|
||||
func (o Object) GetObject() (object *torrent.Torrent, err error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func (o Object) Attr(ctx context.Context, attr *fuse.Attr) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (o Object) ReadDirAll(ctx context.Context) ([]fuse.Dirent, error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func (o Object) Lookup(ctx context.Context, name string) (fs.Node, error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func (o Object) Read(ctx context.Context, req *fuse.ReadRequest, resp *fuse.ReadResponse) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (o Object) Open(ctx context.Context, req *fuse.OpenRequest, resp *fuse.OpenResponse) (fs.Handle, error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func (o Object) Remove(ctx context.Context, req *fuse.RemoveRequest) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (o Object) Mkdir(ctx context.Context, req *fuse.MkdirRequest) (fs.Node, error) {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func (o Object) Rename(ctx context.Context, req *fuse.RenameRequest, newDir fs.Node) error {
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user