-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cache.go
69 lines (57 loc) · 1.05 KB
/
cache.go
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
69
package main
import (
"encoding/hex"
"encoding/json"
"os"
"path/filepath"
)
func cacheClear() {
// Remove the cache directory
err := os.RemoveAll(cacheDir())
if err != nil {
exit(err)
}
if isAgentServerRunning() {
err = agent().ClearCache()
if err != nil {
exit(err)
}
}
}
func cacheDir() string {
return filepath.Join(os.TempDir(), "whisper")
}
func cacheFilePath(key string) string {
return filepath.Join(cacheDir(), hex.EncodeToString([]byte(key)))
}
func cache(key string, data any) {
b, err := json.Marshal(data)
if err != nil {
exit(err)
}
// Ensure the cache directory exists
err = os.MkdirAll(cacheDir(), 0o755)
if err != nil {
exit(err)
}
// Write the data to the cache file
err = os.WriteFile(cacheFilePath(key), b, 0o644)
if err != nil {
exit(err)
}
}
func getCache(key string, data any) bool {
p := cacheFilePath(key)
if _, err := os.Stat(p); err != nil {
return false
}
b, err := os.ReadFile(p)
if err != nil {
exit(err)
}
err = json.Unmarshal(b, data)
if err != nil {
exit(err)
}
return true
}