status-keycard-go/pkg/pairing/store.go

package pairing

import (
	"encoding/json"
	"os"
	"path/filepath"
)

type Store struct {
	path   string
	values map[string]*Info
}

func NewStore(storage string) (*Store, error) {
	p := &Store{path: storage}
	b, err := os.ReadFile(p.path)

	if err != nil {
		if os.IsNotExist(err) {
			parent := filepath.Dir(p.path)
			err = os.MkdirAll(parent, 0750)

			if err != nil {
				return nil, err
			}

			p.values = map[string]*Info{}
		} else {
			return nil, err
		}
	} else {
		err = json.Unmarshal(b, &p.values)

		if err != nil {
			return nil, err
		}
	}

	return p, nil
}

func (p *Store) save() error {
	b, err := json.Marshal(p.values)

	if err != nil {
		return err
	}

	err = os.WriteFile(p.path, b, 0640)

	if err != nil {
		return err
	}

	return nil
}

func (p *Store) Store(instanceUID string, pairing *Info) error {
	p.values[instanceUID] = pairing
	return p.save()
}

func (p *Store) Get(instanceUID string) *Info {
	return p.values[instanceUID]
}

func (p *Store) Delete(instanceUID string) {
	delete(p.values, instanceUID)
}