/
utils.go
63 lines (53 loc) · 1.32 KB
/
utils.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package net9p
import (
"github.com/kennylevinsen/qp"
"github.com/kennylevinsen/qptools/fileserver/trees"
)
type openCounterHandle struct {
trees.ReadWriteAtCloser
closeHook func()
}
func (h *openCounterHandle) Close() error {
err := h.ReadWriteAtCloser.Close()
h.closeHook()
return err
}
type openCounterFile struct {
trees.File
OpenHook func()
CloseHook func()
}
func (f *openCounterFile) Open(user string, mode qp.OpenMode) (trees.ReadWriteAtCloser, error) {
res, err := f.File.Open(user, mode)
if err != nil {
return res, err
}
f.OpenHook()
return &openCounterHandle{
ReadWriteAtCloser: res,
closeHook: f.CloseHook,
}, nil
}
func newOpenCounterFile(f trees.File, open func(), close func()) trees.File {
return &openCounterFile{
File: f,
OpenHook: open,
CloseHook: close,
}
}
type MagicWalkFile struct {
*trees.SyntheticFile
MagicWalkHook func(user string) (trees.File, error)
}
func (f *MagicWalkFile) Arrived(user string) (trees.File, error) {
if f.MagicWalkHook == nil {
return nil, nil
}
return f.MagicWalkHook(user)
}
func NewMagicWalkFile(name string, permissions qp.FileMode, user, group string, hook func(user string) (trees.File, error)) *MagicWalkFile {
return &MagicWalkFile{
SyntheticFile: trees.NewSyntheticFile(name, permissions, user, group),
MagicWalkHook: hook,
}
}