package vain

import (
	"errors"
	"sync"
)

type MemStore struct {
	l sync.RWMutex
	p map[string]Package
}

func NewMemStore() *MemStore {
	return &MemStore{
		p: make(map[string]Package),
	}
}

func (ms MemStore) Add(p Package) error {
	ms.l.Lock()
	ms.p[p.Path] = p
	ms.l.Unlock()
	return nil
}

func (ms MemStore) Remove(path string) error {
	ms.l.Lock()
	delete(ms.p, path)
	ms.l.Unlock()
	return nil
}

func (ms MemStore) Save() error {
	return errors.New("save is not implemented")
}

func (ms MemStore) All() []Package {
	r := []Package{}
	ms.l.RLock()
	for _, p := range ms.p {
		r = append(r, p)
	}
	ms.l.RUnlock()
	return r
}