67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package bolt
|
|
|
|
import (
|
|
"github.com/asdine/storm"
|
|
"github.com/asdine/storm/q"
|
|
fb "github.com/filebrowser/filebrowser"
|
|
)
|
|
|
|
// 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) (*fb.ShareLink, error) {
|
|
var v fb.ShareLink
|
|
err := s.DB.One("Hash", hash, &v)
|
|
if err == storm.ErrNotFound {
|
|
return nil, fb.ErrNotExist
|
|
}
|
|
|
|
return &v, err
|
|
}
|
|
|
|
// GetPermanent gets the permanent link from a path.
|
|
func (s ShareStore) GetPermanent(path string) (*fb.ShareLink, error) {
|
|
var v fb.ShareLink
|
|
err := s.DB.Select(q.Eq("Path", path), q.Eq("Expires", false)).First(&v)
|
|
if err == storm.ErrNotFound {
|
|
return nil, fb.ErrNotExist
|
|
}
|
|
|
|
return &v, err
|
|
}
|
|
|
|
// GetByPath gets all the links for a specific path.
|
|
func (s ShareStore) GetByPath(hash string) ([]*fb.ShareLink, error) {
|
|
var v []*fb.ShareLink
|
|
err := s.DB.Find("Path", hash, &v)
|
|
if err == storm.ErrNotFound {
|
|
return v, fb.ErrNotExist
|
|
}
|
|
|
|
return v, err
|
|
}
|
|
|
|
// Gets retrieves all the shareable links.
|
|
func (s ShareStore) Gets() ([]*fb.ShareLink, error) {
|
|
var v []*fb.ShareLink
|
|
err := s.DB.All(&v)
|
|
if err == storm.ErrNotFound {
|
|
return v, fb.ErrNotExist
|
|
}
|
|
|
|
return v, err
|
|
}
|
|
|
|
// Save stores a Share Link on the database.
|
|
func (s ShareStore) Save(l *fb.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(&fb.ShareLink{Hash: hash})
|
|
}
|