installer: Register gob types in init() function

livingsilver94 · Nov 2, 2023 · 57c28bd · 57c28bd
1 parent 3b17976
commit 57c28bd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/cli/privilege.go b/cli/privilege.go
@@ -3,14 +3,13 @@ package cli
 import (
 	"os"
 
-	"github.com/livingsilver94/backee/installer"
+	_ "github.com/livingsilver94/backee/installer"
 	priv "github.com/livingsilver94/backee/privilege"
 )
 
 type privilege struct{}
 
 func (p privilege) Run() error {
-	installer.RegisterPrivilegedTypes()
 	run, err := priv.ReceiveRunner(os.Stdin)
 	if err != nil {
 		return err

diff --git a/installer/fs.go b/installer/fs.go
@@ -13,6 +13,12 @@ import (
 	"github.com/livingsilver94/backee/service"
 )
 
+func init() {
+	privilege.RegisterInterfaceImpl(SymlinkWriter{})
+	privilege.RegisterInterfaceImpl(CopyWriter{})
+	privilege.RegisterInterfaceImpl(privilegedPathWriter{})
+}
+
 type FileWriter interface {
 	loadSource(src string) error
 	writeDestination(dst string) error
@@ -47,12 +53,6 @@ func WritePathPrivileged(dst service.FilePath, src string, wr PrivilegedFileWrit
 	return privilege.Run(&r)
 }
 
-func RegisterPrivilegedTypes() {
-	privilege.RegisterInterfaceImpl(SymlinkWriter{})
-	privilege.RegisterInterfaceImpl(CopyWriter{})
-	privilege.RegisterInterfaceImpl(privilegedPathWriter{})
-}
-
 type SymlinkWriter struct {
 	srcPath string
 }