wip

main.go

@@ -7,11 +7,12 @@ import (
 	"sync"
 
 	"github.com/ben-ha/jcp/logic"
+	"github.com/ben-ha/jcp/state"
 	"github.com/ben-ha/jcp/tui"
 	tea "github.com/charmbracelet/bubbletea"
 )
 
-func updaterFunc(jcp logic.Jcp, ui *tea.Program) {
+func updaterFunc(jcp logic.Jcp, ui *tea.Program, state *state.JcpState) {
 	for {
 		update, more := <-jcp.ProgressChannel
 		if !more {
@@ -26,6 +27,7 @@ func updaterFunc(jcp logic.Jcp, ui *tea.Program) {
 			panic(fmt.Sprintf("An error occurred: %v", update.JcpError.Error()))
 		}
 
+		state.Update(update)
 		ui.Send(tui.UITransferMsg{Progress: update.Progress})
 	}
 
@@ -43,6 +45,13 @@ func main() {
 	src := os.Args[1]
 	dst := os.Args[2]
 
+	state, err := state.InitializeState()
+	if err != nil {
+		panic(fmt.Sprintf("State: %v", err))
+	}
+
+	defer state.SaveState()
+
 	jcp := logic.MakeJcp(10)
 
 	ui := tea.NewProgram(tui.UIModel{})
@@ -51,9 +60,9 @@ func main() {
 
 	go StartUI(ui, &uiComplete)
 
-	go updaterFunc(jcp, ui)
+	go updaterFunc(jcp, ui, state)
 
-	err := jcp.StartCopy(src, dst)
+	err = jcp.StartCopy(src, dst)
 	if err != nil {
 		panic(fmt.Sprintf("Error: %v", err))
 	}

state/state.go

@@ -11,6 +11,7 @@ type CopyDestinationKey = string
 
 type JcpState struct {
 	CopyStates map[CopySourceKey](map[CopyDestinationKey]JcpCopyState)
+	StatePath  string
 }
 
 type CopierType int

state/statemanager.go

@@ -3,6 +3,7 @@ package state
 import (
 	"encoding/json"
 	"os"
+	"path"
 	"sync"
 	"time"
 
@@ -13,19 +14,43 @@ var stateManagerMutex sync.Mutex
 
 const ValidStateWindowInDays = 1
 
-func LoadState(fileName string) *JcpState {
+const JcpStateDirectoryName = "jcp"
+const JcpStateFileName = "state.json"
+
+func InitializeState() (*JcpState, error) {
+	cacheDir, err := os.UserCacheDir()
+	if err != nil {
+		return nil, err
+	}
+
+	jcpDir := path.Join(cacheDir, JcpStateDirectoryName)
+
+	err = os.Mkdir(jcpDir, os.ModePerm)
+	if !os.IsExist(err) {
+		return nil, err
+	}
+
+	return loadState(path.Join(jcpDir, JcpStateFileName)), nil
+}
+
+func loadState(fileName string) *JcpState {
 	data, err := os.ReadFile(fileName)
 	loadedState := &JcpState{}
 	if err == nil {
 		json.Unmarshal(data, loadedState)
 	}
 
 	loadedState.Clean()
+	loadedState.StatePath = fileName
 
 	return loadedState
 }
 
-func (copierState *JcpState) Save(fileName string) {
+func (copierState *JcpState) SaveState() {
+	copierState.saveState(copierState.StatePath)
+}
+
+func (copierState *JcpState) saveState(fileName string) {
 	serialized, err := json.Marshal(copierState)
 
 	if err == nil {