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
70
71
72
73
74
|
package timer
import (
"encoding/json"
"os"
"path/filepath"
"time"
)
const (
stateFile = ".timr_state"
)
// StateFilePathOverride is used by tests to override the state file path.
var StateFilePathOverride string
func SetStateFilePathOverride(path string) {
StateFilePathOverride = path
}
type State struct {
StartTime time.Time
ElapsedTime time.Duration
Running bool
}
func GetStateFile() (string, error) {
if StateFilePathOverride != "" {
return StateFilePathOverride, nil
}
configDir, err := os.UserConfigDir()
if err != nil {
return "", err
}
return filepath.Join(configDir, "timr", stateFile), nil
}
func LoadState() (State, error) {
var state State
stateFilePath, err := GetStateFile()
if err != nil {
return state, err
}
data, err := os.ReadFile(stateFilePath)
if err != nil {
if os.IsNotExist(err) {
return State{}, nil
}
return state, err
}
err = json.Unmarshal(data, &state)
return state, err
}
func (s *State) Save() error {
data, err := json.Marshal(s)
if err != nil {
return err
}
stateFilePath, err := GetStateFile()
if err != nil {
return err
}
dir := filepath.Dir(stateFilePath)
if err := os.MkdirAll(dir, 0755); err != nil {
return err
}
return os.WriteFile(stateFilePath, data, 0644)
}
|
