57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
// Copyright (C) 2017, 2023 Opsmate, Inc.
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla
|
|
// Public License, v. 2.0. If a copy of the MPL was not distributed
|
|
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
//
|
|
// This software is distributed WITHOUT A WARRANTY OF ANY KIND.
|
|
// See the Mozilla Public License for details.
|
|
|
|
package monitor
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
)
|
|
|
|
func randomFileSuffix() string {
|
|
var randomBytes [12]byte
|
|
if _, err := rand.Read(randomBytes[:]); err != nil {
|
|
panic(err)
|
|
}
|
|
return hex.EncodeToString(randomBytes[:])
|
|
}
|
|
|
|
func writeFile(filename string, data []byte, perm os.FileMode) error {
|
|
tempname := filename + ".tmp." + randomFileSuffix()
|
|
if err := os.WriteFile(tempname, data, perm); err != nil {
|
|
return fmt.Errorf("error writing %s: %w", filename, err)
|
|
}
|
|
if err := os.Rename(tempname, filename); err != nil {
|
|
os.Remove(tempname)
|
|
return fmt.Errorf("error writing %s: %w", filename, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func writeTextFile(filename string, fileText string, perm os.FileMode) error {
|
|
return writeFile(filename, []byte(fileText), perm)
|
|
}
|
|
|
|
func writeJSONFile(filename string, data any, perm os.FileMode) error {
|
|
fileBytes, err := json.Marshal(data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fileBytes = append(fileBytes, '\n')
|
|
return writeFile(filename, fileBytes, perm)
|
|
}
|
|
|
|
func fileExists(filename string) bool {
|
|
_, err := os.Lstat(filename)
|
|
return err == nil
|
|
}
|