internal/lockedfile: optionally force write lock

For acquiring a write lock on a directory.

Signed-off-by: Ophestra <cat@gensokyo.uk>

1 files changed, 14 insertions, 5 deletions

diff --git a/internal/lockedfile/lockedfile.go b/internal/lockedfile/lockedfile.go
index f48124ff..c7baff60 100644
--- a/internal/lockedfile/lockedfile.go
+++ b/internal/lockedfile/lockedfile.go
@@ -34,15 +34,12 @@ type osFile struct {
 	*os.File
 }
 
-// OpenFile is like os.OpenFile, but returns a locked file.
-// If flag includes os.O_WRONLY or os.O_RDWR, the file is write-locked;
-// otherwise, it is read-locked.
-func OpenFile(name string, flag int, perm fs.FileMode) (*File, error) {
+func openFile(name string, flag int, perm fs.FileMode, w bool) (*File, error) {
 	var (
 		f   = new(File)
 		err error
 	)
-	f.osFile.File, err = openFile(name, flag, perm)
+	f.osFile.File, err = osOpenFile(name, flag, perm, w)
 	if err != nil {
 		return nil, err
 	}
@@ -59,11 +56,23 @@ func OpenFile(name string, flag int, perm fs.FileMode) (*File, error) {
 	return f, nil
 }
 
+// OpenFile is like os.OpenFile, but returns a locked file.
+// If flag includes os.O_WRONLY or os.O_RDWR, the file is write-locked;
+// otherwise, it is read-locked.
+func OpenFile(name string, flag int, perm fs.FileMode) (*File, error) {
+	return openFile(name, flag, perm, false)
+}
+
 // Open is like os.Open, but returns a read-locked file.
 func Open(name string) (*File, error) {
 	return OpenFile(name, os.O_RDONLY, 0)
 }
 
+// OpenDir is like Open, but returns a write-locked directory.
+func OpenDir(name string) (*File, error) {
+	return openFile(name, os.O_RDONLY, 0, true)
+}
+
 // Create is like os.Create, but returns a write-locked file.
 func Create(name string) (*File, error) {
 	return OpenFile(name, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0666)