use lockedfile to write hosts

Author: adrianliechti

go.mod

@@ -12,6 +12,7 @@ require (
 	github.com/olekukonko/tablewriter v0.0.5
 	github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c
 	github.com/pkg/sftp v1.13.7
+	github.com/rogpeppe/go-internal v1.13.1
 	github.com/urfave/cli/v3 v3.0.0-alpha9.7
 	golang.org/x/crypto v0.33.0
 	golang.org/x/exp v0.0.0-20250210185358-939b2ce775ac
@@ -73,7 +74,6 @@ require (
 	github.com/opencontainers/go-digest v1.0.0 // indirect
 	github.com/pkg/errors v0.9.1 // indirect
 	github.com/rivo/uniseg v0.4.7 // indirect
-	github.com/rogpeppe/go-internal v1.13.1 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect
 	github.com/stretchr/testify v1.10.0 // indirect
 	github.com/x448/float16 v0.8.4 // indirect

pkg/system/hosts.go

@@ -2,11 +2,12 @@ package system
 
 import (
 	"fmt"
-	"io"
 	"os"
 	"runtime"
 	"slices"
 	"strings"
+
+	"github.com/rogpeppe/go-internal/lockedfile"
 )
 
 type HostsSection struct {
@@ -54,15 +55,7 @@ func (s *HostsSection) Flush() error {
 		ln = "\r\n"
 	}
 
-	file, err := os.OpenFile(s.path, os.O_RDWR, 0644)
-
-	if err != nil {
-		return err
-	}
-
-	defer file.Close()
-
-	data, err := io.ReadAll(file)
+	data, err := lockedfile.Read(s.path)
 
 	if err != nil {
 		return err
@@ -97,17 +90,7 @@ func (s *HostsSection) Flush() error {
 
 	text = strings.TrimRight(text, ln) + ln
 
-	if _, err := file.Seek(0, 0); err != nil {
-		return err
-	}
-
-	if err := file.Truncate(0); err != nil {
-		return err
-	}
-
-	if _, err := file.WriteString(text); err != nil {
-		return err
-	}
+	lockedfile.Write(s.path, strings.NewReader(text), 0644)
 
 	return nil
 }