forked from mirrors/homebox
31b34241e0
* change /content/ -> /homebox/ * add cache to code generators * update env variables to set data storage * update env variables * set env variables in prod container * implement attachment post route (WIP) * get attachment endpoint * attachment download * implement string utilities lib * implement generic drop zone * use explicit truncate * remove clean dir * drop strings composable for lib * update item types and add attachments * add attachment API * implement service context * consolidate API code * implement editing attachments * implement upload limit configuration * improve error handling * add docs for max upload size * fix test cases
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package repo
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/hay-kot/homebox/backend/ent"
|
|
"github.com/hay-kot/homebox/backend/ent/attachment"
|
|
)
|
|
|
|
// AttachmentRepo is a repository for Attachments table that links Items to Documents
|
|
// While also specifying the type of the attachment. This _ONLY_ provides basic Create Update
|
|
// And Delete operations. For accessing the actual documents, use the Items repository since it
|
|
// provides the attachments with the documents.
|
|
type AttachmentRepo struct {
|
|
db *ent.Client
|
|
}
|
|
|
|
func (r *AttachmentRepo) Create(ctx context.Context, itemId, docId uuid.UUID, typ attachment.Type) (*ent.Attachment, error) {
|
|
return r.db.Attachment.Create().
|
|
SetType(typ).
|
|
SetDocumentID(docId).
|
|
SetItemID(itemId).
|
|
Save(ctx)
|
|
}
|
|
|
|
func (r *AttachmentRepo) Get(ctx context.Context, id uuid.UUID) (*ent.Attachment, error) {
|
|
return r.db.Attachment.
|
|
Query().
|
|
Where(attachment.ID(id)).
|
|
WithItem().
|
|
WithDocument().
|
|
Only(ctx)
|
|
}
|
|
|
|
func (r *AttachmentRepo) Update(ctx context.Context, itemId uuid.UUID, typ attachment.Type) (*ent.Attachment, error) {
|
|
itm, err := r.db.Attachment.UpdateOneID(itemId).
|
|
SetType(typ).
|
|
Save(ctx)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return r.Get(ctx, itm.ID)
|
|
}
|
|
|
|
func (r *AttachmentRepo) Delete(ctx context.Context, id uuid.UUID) error {
|
|
return r.db.Attachment.DeleteOneID(id).Exec(ctx)
|
|
}
|