package bolt

import (
	"github.com/asdine/storm"
	"github.com/asdine/storm/q"
	"github.com/filebrowser/filebrowser/types"
)

// ShareStore is a shareable links store.
type ShareStore struct {
	DB *storm.DB
}

// Get gets a Share Link from an hash.
func (s ShareStore) Get(hash string) (*types.ShareLink, error) {
	var v types.ShareLink
	err := s.DB.One("Hash", hash, &v)
	if err == storm.ErrNotFound {
		return nil, types.ErrNotExist
	}

	return &v, err
}

// GetPermanent gets the permanent link from a path.
func (s ShareStore) GetPermanent(path string) (*types.ShareLink, error) {
	var v types.ShareLink
	err := s.DB.Select(q.Eq("Path", path), q.Eq("Expires", false)).First(&v)
	if err == storm.ErrNotFound {
		return nil, types.ErrNotExist
	}

	return &v, err
}

// GetByPath gets all the links for a specific path.
func (s ShareStore) GetByPath(hash string) ([]*types.ShareLink, error) {
	var v []*types.ShareLink
	err := s.DB.Find("Path", hash, &v)
	if err == storm.ErrNotFound {
		return v, types.ErrNotExist
	}

	return v, err
}

// Gets retrieves all the shareable links.
func (s ShareStore) Gets() ([]*types.ShareLink, error) {
	var v []*types.ShareLink
	err := s.DB.All(&v)
	if err == storm.ErrNotFound {
		return v, types.ErrNotExist
	}

	return v, err
}

// Save stores a Share Link on the database.
func (s ShareStore) Save(l *types.ShareLink) error {
	return s.DB.Save(l)
}

// Delete deletes a Share Link from the database.
func (s ShareStore) Delete(hash string) error {
	return s.DB.DeleteStruct(&types.ShareLink{Hash: hash})
}