package main

import (
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"os"
)

type PersistentData struct {
	QuestradeRefreshToken string `json:"questradeRefreshToken"`
}

func loadPersistentData() (*PersistentData, error) {
	persistentData := &PersistentData{}

	f, err := os.Open("data/persistentData.json")
	if errors.Is(err, os.ErrNotExist) {
		// handle the case where the file doesn't exist
		persistentData.QuestradeRefreshToken = os.Getenv("questrade_refresh_token")
		return persistentData, nil
	}

	defer f.Close()

	b, err := io.ReadAll(f)
	if err != nil {
		return nil, fmt.Errorf("failed to read file data/persistentData.json: %v", err)
	}

	err = json.Unmarshal(b, persistentData)
	if err != nil {
		return nil, fmt.Errorf("failed to unmarshal data/persistentData.json to PersistentData struct: %v", err)
	}

	return persistentData, nil

}

func savePersistentData(PersistentData) error {
	b, err := json.Marshal(persistentData)
	if err != nil {
		return fmt.Errorf("failed to marshal persistentData to bytes: %v", err)
	}
	err = os.WriteFile("data/persistentData.json", b, 0644)
	if err != nil {
		return fmt.Errorf("failed to write file data/persistentData.json: %v", err)
	}
	return nil
}