126 lines
2.3 KiB
Go
126 lines
2.3 KiB
Go
package fs
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/badjware/gitlabfs/git"
|
|
"github.com/badjware/gitlabfs/gitlab"
|
|
|
|
"github.com/hanwen/go-fuse/v2/fs"
|
|
"github.com/hanwen/go-fuse/v2/fuse"
|
|
)
|
|
|
|
const (
|
|
staticInodeStart = uint64(int(^(uint(0))>>1)) + 1
|
|
)
|
|
|
|
type staticNode interface {
|
|
fs.InodeEmbedder
|
|
Ino() uint64
|
|
Mode() uint32
|
|
}
|
|
|
|
type FSParam struct {
|
|
Git git.GitClonerPuller
|
|
Gitlab gitlab.GitlabFetcher
|
|
|
|
staticInoChan chan uint64
|
|
}
|
|
|
|
type rootNode struct {
|
|
fs.Inode
|
|
param *FSParam
|
|
rootGroupIds []int
|
|
userIds []int
|
|
}
|
|
|
|
var _ = (fs.NodeOnAdder)((*rootNode)(nil))
|
|
|
|
func (n *rootNode) OnAdd(ctx context.Context) {
|
|
groupsInode := n.NewPersistentInode(
|
|
ctx,
|
|
newGroupsNode(
|
|
n.rootGroupIds,
|
|
n.param,
|
|
),
|
|
fs.StableAttr{
|
|
Ino: <-n.param.staticInoChan,
|
|
Mode: fuse.S_IFDIR,
|
|
},
|
|
)
|
|
n.AddChild("groups", groupsInode, false)
|
|
|
|
usersInode := n.NewPersistentInode(
|
|
ctx,
|
|
newUsersNode(
|
|
n.userIds,
|
|
n.param,
|
|
),
|
|
fs.StableAttr{
|
|
Ino: <-n.param.staticInoChan,
|
|
Mode: fuse.S_IFDIR,
|
|
},
|
|
)
|
|
n.AddChild("users", usersInode, false)
|
|
|
|
fmt.Println("Mounted and ready to use")
|
|
}
|
|
|
|
func Start(mountpoint string, rootGroupIds []int, userIds []int, param *FSParam, debug bool) error {
|
|
fmt.Printf("Mounting in %v\n", mountpoint)
|
|
|
|
opts := &fs.Options{}
|
|
opts.Debug = debug
|
|
|
|
param.staticInoChan = make(chan uint64)
|
|
root := &rootNode{
|
|
param: param,
|
|
rootGroupIds: rootGroupIds,
|
|
userIds: userIds,
|
|
}
|
|
|
|
go staticInoGenerator(root.param.staticInoChan)
|
|
|
|
server, err := fs.Mount(mountpoint, root, opts)
|
|
if err != nil {
|
|
return fmt.Errorf("mount failed: %v", err)
|
|
}
|
|
|
|
signalChan := make(chan os.Signal)
|
|
go signalHandler(signalChan, server)
|
|
signal.Notify(signalChan, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
// server.Serve() is already called in fs.Mount() so we shouldn't call it ourself. We wait for the server to terminate.
|
|
server.Wait()
|
|
|
|
return nil
|
|
}
|
|
|
|
func staticInoGenerator(staticInoChan chan<- uint64) {
|
|
i := staticInodeStart
|
|
for {
|
|
staticInoChan <- i
|
|
i++
|
|
}
|
|
}
|
|
|
|
func signalHandler(signalChan <-chan os.Signal, server *fuse.Server) {
|
|
err := server.WaitMount()
|
|
if err != nil {
|
|
fmt.Printf("failed to start exit signal handler: %v\n", err)
|
|
return
|
|
}
|
|
for {
|
|
s := <-signalChan
|
|
fmt.Printf("Caught %v: stopping\n", s)
|
|
err := server.Unmount()
|
|
if err != nil {
|
|
fmt.Printf("Failed to unmount: %v\n", err)
|
|
}
|
|
}
|
|
}
|