feat: share CRUD with cascade delete
This commit is contained in:
@@ -0,0 +1,67 @@
|
|||||||
|
package store
|
||||||
|
|
||||||
|
import "database/sql"
|
||||||
|
|
||||||
|
type Share struct {
|
||||||
|
ID string
|
||||||
|
FileID string
|
||||||
|
Token string
|
||||||
|
PasswordHash string
|
||||||
|
ExpiresAt int64
|
||||||
|
CreatedAt int64
|
||||||
|
LastAccessedAt int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) ShareCreate(sh *Share) error {
|
||||||
|
_, err := s.db.Exec(
|
||||||
|
`INSERT INTO shares (id, file_id, token, password_hash, expires_at, created_at, last_accessed_at)
|
||||||
|
VALUES (?, ?, ?, ?, ?, ?, ?)`,
|
||||||
|
sh.ID, sh.FileID, sh.Token, sh.PasswordHash, sh.ExpiresAt, sh.CreatedAt, sh.LastAccessedAt,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) ShareGetByToken(token string) (*Share, error) {
|
||||||
|
row := s.db.QueryRow(
|
||||||
|
`SELECT id, file_id, token, password_hash, expires_at, created_at, last_accessed_at
|
||||||
|
FROM shares WHERE token = ?`, token,
|
||||||
|
)
|
||||||
|
return scanShare(row)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) ShareList() ([]*Share, error) {
|
||||||
|
rows, err := s.db.Query(
|
||||||
|
`SELECT id, file_id, token, password_hash, expires_at, created_at, last_accessed_at
|
||||||
|
FROM shares ORDER BY created_at DESC`,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
|
||||||
|
var shares []*Share
|
||||||
|
for rows.Next() {
|
||||||
|
sh, err := scanShare(rows)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
shares = append(shares, sh)
|
||||||
|
}
|
||||||
|
return shares, rows.Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) ShareDelete(id string) error {
|
||||||
|
_, err := s.db.Exec(`DELETE FROM shares WHERE id = ?`, id)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func scanShare(row interface {
|
||||||
|
Scan(dest ...any) error
|
||||||
|
}) (*Share, error) {
|
||||||
|
sh := &Share{}
|
||||||
|
err := row.Scan(&sh.ID, &sh.FileID, &sh.Token, &sh.PasswordHash, &sh.ExpiresAt, &sh.CreatedAt, &sh.LastAccessedAt)
|
||||||
|
if err == sql.ErrNoRows {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return sh, err
|
||||||
|
}
|
||||||
@@ -0,0 +1,70 @@
|
|||||||
|
package store
|
||||||
|
|
||||||
|
import (
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestShareCreate(t *testing.T) {
|
||||||
|
dir := t.TempDir()
|
||||||
|
s, err := Open(filepath.Join(dir, "test.db"))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer s.Close()
|
||||||
|
|
||||||
|
f := &File{
|
||||||
|
ID: "file-1",
|
||||||
|
OriginalName: "doc.pdf",
|
||||||
|
StorageName: "uuid-1",
|
||||||
|
Dir: "inbox",
|
||||||
|
Size: 2048,
|
||||||
|
CreatedAt: 1750000000,
|
||||||
|
}
|
||||||
|
if err := s.FileCreate(f); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
share := &Share{
|
||||||
|
ID: "share-1",
|
||||||
|
FileID: "file-1",
|
||||||
|
Token: "token-abc",
|
||||||
|
CreatedAt: 1750000000,
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := s.ShareCreate(share); err != nil {
|
||||||
|
t.Fatalf("ShareCreate failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
got, err := s.ShareGetByToken("token-abc")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ShareGetByToken failed: %v", err)
|
||||||
|
}
|
||||||
|
if got.FileID != "file-1" {
|
||||||
|
t.Errorf("FileID = %q, want %q", got.FileID, "file-1")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFileDeleteCascadesShares(t *testing.T) {
|
||||||
|
dir := t.TempDir()
|
||||||
|
s, err := Open(filepath.Join(dir, "test.db"))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer s.Close()
|
||||||
|
|
||||||
|
f := &File{ID: "file-2", OriginalName: "x.pdf", StorageName: "uuid-2", Dir: "inbox", Size: 100, CreatedAt: 1750000000}
|
||||||
|
s.FileCreate(f)
|
||||||
|
|
||||||
|
sh := &Share{ID: "share-2", FileID: "file-2", Token: "tok-2", CreatedAt: 1750000000}
|
||||||
|
s.ShareCreate(sh)
|
||||||
|
|
||||||
|
if err := s.FileDelete("file-2"); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = s.ShareGetByToken("tok-2")
|
||||||
|
if err == nil {
|
||||||
|
t.Error("Share should be deleted after file deletion")
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -15,6 +15,11 @@ func Open(dbPath string) (*Store, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if _, err := db.Exec("PRAGMA foreign_keys = ON"); err != nil {
|
||||||
|
db.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
if err := migrate(db); err != nil {
|
if err := migrate(db); err != nil {
|
||||||
db.Close()
|
db.Close()
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|||||||
Reference in New Issue
Block a user