package store

import (
	"errors"
	"os"

	mc "github.com/bradfitz/gomemcache/memcache"
)

type MemcacheStore struct {
	Client *mc.Client
}

func (m *MemcacheStore) Get(key string) (Item, error) {
	kv, err := m.Client.Get(key)
	if err != nil && err == mc.ErrCacheMiss {
		return nil, errors.New("key not found")
	} else if err != nil {
		return nil, err
	}

	if kv == nil {
		return nil, errors.New("key not found")
	}

	return &MemcacheItem{
		key:   kv.Key,
		value: kv.Value,
	}, nil
}

func (m *MemcacheStore) Del(key string) error {
	return m.Client.Delete(key)
}

func (m *MemcacheStore) Put(item Item) error {
	return m.Client.Set(&mc.Item{
		Key:   item.Key(),
		Value: item.Value(),
	})
}

func (m *MemcacheStore) NewItem(key string, value []byte) Item {
	return &MemcacheItem{
		key:   key,
		value: value,
	}
}

func NewMemcacheStore() Store {
	server := os.Getenv("MEMCACHED_SERVICE_HOST")
	port := os.Getenv("MEMCACHED_SERVICE_PORT")

	if len(server) == 0 {
		server = "127.0.0.1"
	}

	if len(port) == 0 {
		port = "11211"
	}

	return &MemcacheStore{
		Client: mc.New(server + ":" + port),
	}
}