1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
package file
import (
"crypto/sha256"
"log"
"os"
)
func getChecksum(path string) [32]byte {
var checksum [32]byte
data, err := os.ReadFile(path)
if err != nil {
log.Printf("reading %s: %v (file does not exist or cannot be read)", path, err)
return checksum
}
checksum = sha256.Sum256(data)
log.Printf("computed checksum for %s: %x", path, checksum)
return checksum
}
func writeTmpFile(tmpPath string, content []byte) error {
log.Printf("writing %d bytes to temporary file %s", len(content), tmpPath)
if err := os.WriteFile(tmpPath, content, 0o644); err != nil {
log.Printf("failed to write temporary file %s: %v", tmpPath, err)
return err
}
log.Printf("successfully wrote temporary file %s", tmpPath)
return nil
}
func updateFromTmp(tmpPath, path string, checksumChanged bool) error {
if !checksumChanged {
log.Printf("checksums match, removing temporary file %s", tmpPath)
if err := os.Remove(tmpPath); err != nil {
log.Printf("failed to remove temporary file %s: %v", tmpPath, err)
return err
}
log.Printf("no changes needed for %s", path)
return nil
}
log.Printf("checksums differ, renaming %s to %s", tmpPath, path)
if err := os.Rename(tmpPath, path); err != nil {
log.Printf("failed to rename %s to %s: %v", tmpPath, path, err)
os.Remove(tmpPath)
return err
}
log.Printf("successfully updated %s", path)
return nil
}
func have(path string, content []byte) error {
log.Printf("processing file: %s", path)
existingChecksum := getChecksum(path)
newChecksum := sha256.Sum256(content)
log.Printf("computed checksum for new content: %x", newChecksum)
tmpPath := path + ".tmp"
if err := writeTmpFile(tmpPath, content); err != nil {
return err
}
return updateFromTmp(tmpPath, path, existingChecksum != newChecksum)
}
func HaveString(path, content string) error {
return have(path, []byte(content))
}
|
