feat: support office document nodes
This commit is contained in:
@@ -73,6 +73,63 @@ func (h *NodeHandler) CreateFolder(w http.ResponseWriter, r *http.Request) {
|
||||
writeJSON(w, http.StatusCreated, node)
|
||||
}
|
||||
|
||||
func (h *NodeHandler) ListOfficeLinks(w http.ResponseWriter, r *http.Request) {
|
||||
userID := commonmw.GetUserID(r.Context())
|
||||
links, err := h.repo.ListOfficeExternalURLs(r.Context(), userID, subordinates(r))
|
||||
if err != nil {
|
||||
writeInternalError(w, r, err, "failed to list office links")
|
||||
return
|
||||
}
|
||||
writeJSON(w, http.StatusOK, links)
|
||||
}
|
||||
|
||||
func (h *NodeHandler) CreateOfficeDocument(w http.ResponseWriter, r *http.Request) {
|
||||
var req model.CreateOfficeDocumentRequest
|
||||
if err := decodeJSON(r, &req); err != nil {
|
||||
writeError(w, http.StatusBadRequest, "invalid json")
|
||||
return
|
||||
}
|
||||
req.Title = strings.TrimSpace(req.Title)
|
||||
req.OfficeID = strings.TrimSpace(req.OfficeID)
|
||||
req.OfficeFormat = strings.Trim(strings.ToLower(req.OfficeFormat), ". ")
|
||||
if req.Title == "" {
|
||||
writeError(w, http.StatusBadRequest, "title is required")
|
||||
return
|
||||
}
|
||||
if req.OfficeID == "" {
|
||||
writeError(w, http.StatusBadRequest, "office_id is required")
|
||||
return
|
||||
}
|
||||
if !allowedOfficeFormat(req.OfficeFormat) {
|
||||
writeError(w, http.StatusBadRequest, "office_format is not allowed")
|
||||
return
|
||||
}
|
||||
userID := commonmw.GetUserID(r.Context())
|
||||
if !h.requireWritableParent(w, r, userID, req.ParentID) {
|
||||
return
|
||||
}
|
||||
externalURL := "/office/" + req.OfficeID
|
||||
originalFilename := req.Title + "." + req.OfficeFormat
|
||||
mimeType := req.OfficeFormat
|
||||
node, err := h.repo.CreateOfficeDocument(r.Context(), &model.Node{
|
||||
ParentID: req.ParentID,
|
||||
Title: req.Title,
|
||||
OwnerUserID: userID,
|
||||
OriginalFilename: &originalFilename,
|
||||
MimeType: &mimeType,
|
||||
Extension: &req.OfficeFormat,
|
||||
SizeBytes: req.SizeBytes,
|
||||
OfficeFormat: &req.OfficeFormat,
|
||||
ExternalURL: &externalURL,
|
||||
})
|
||||
if err != nil {
|
||||
writeInternalError(w, r, err, "failed to create office document")
|
||||
return
|
||||
}
|
||||
h.repo.Audit(r.Context(), userID, "files.office_document_create", "files_node", node.ID, "{}")
|
||||
writeJSON(w, http.StatusCreated, node)
|
||||
}
|
||||
|
||||
func (h *NodeHandler) UploadFile(w http.ResponseWriter, r *http.Request) {
|
||||
if !h.store.Configured() {
|
||||
writeError(w, http.StatusServiceUnavailable, "storage not configured")
|
||||
@@ -376,6 +433,15 @@ func emptyToNil(v string) *string {
|
||||
return &v
|
||||
}
|
||||
|
||||
func allowedOfficeFormat(format string) bool {
|
||||
switch format {
|
||||
case "doc", "docx", "odt", "xls", "xlsx", "xlsm", "ods", "ppt", "pptx", "odp":
|
||||
return true
|
||||
default:
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
func subordinates(r *http.Request) []string {
|
||||
ids := csvHeader(r, "X-User-Subordinates")
|
||||
if len(ids) == 0 {
|
||||
|
||||
@@ -54,6 +54,14 @@ type CreateFolderRequest struct {
|
||||
Title string `json:"title"`
|
||||
}
|
||||
|
||||
type CreateOfficeDocumentRequest struct {
|
||||
ParentID *string `json:"parent_id"`
|
||||
Title string `json:"title"`
|
||||
OfficeID string `json:"office_id"`
|
||||
OfficeFormat string `json:"office_format"`
|
||||
SizeBytes int64 `json:"size_bytes"`
|
||||
}
|
||||
|
||||
type UpdateNodeRequest struct {
|
||||
ParentID *string `json:"parent_id"`
|
||||
Title *string `json:"title"`
|
||||
|
||||
@@ -127,6 +127,49 @@ func (r *NodeRepository) CreateFile(ctx context.Context, n *model.Node) (*model.
|
||||
`, n.ParentID, n.Title, n.OwnerUserID, n.StorageKey, n.OriginalFilename, n.MimeType, n.Extension, n.SizeBytes).Scan)
|
||||
}
|
||||
|
||||
func (r *NodeRepository) CreateOfficeDocument(ctx context.Context, n *model.Node) (*model.Node, error) {
|
||||
return scanNode(r.pool.QueryRow(ctx, `
|
||||
INSERT INTO files_nodes
|
||||
(parent_id, node_type, title, owner_user_id, created_by, original_filename,
|
||||
mime_type, extension, size_bytes, office_format, external_url)
|
||||
VALUES ($1, 'office_document', $2, $3, $3, $4, $5, $6, $7, $8, $9)
|
||||
RETURNING id, parent_id, node_type, title, owner_user_id, owner_department_id,
|
||||
created_by, updated_by, storage_key, original_filename, mime_type,
|
||||
extension, size_bytes, office_format, external_url, version,
|
||||
'edit' AS effective_access, created_at, updated_at, trashed_at, purge_after, deleted_at
|
||||
`, n.ParentID, n.Title, n.OwnerUserID, n.OriginalFilename, n.MimeType, n.Extension, n.SizeBytes, n.OfficeFormat, n.ExternalURL).Scan)
|
||||
}
|
||||
|
||||
func (r *NodeRepository) ListOfficeExternalURLs(ctx context.Context, userID string, subordinateIDs []string) ([]string, error) {
|
||||
rows, err := r.pool.Query(ctx, `
|
||||
SELECT DISTINCT n.external_url
|
||||
FROM files_nodes n
|
||||
WHERE n.deleted_at IS NULL
|
||||
AND n.node_type = 'office_document'
|
||||
AND n.external_url IS NOT NULL
|
||||
AND (
|
||||
n.owner_user_id = $1
|
||||
OR has_node_access(n.id, $1)
|
||||
OR n.owner_user_id::text = ANY($2::text[])
|
||||
)
|
||||
ORDER BY n.external_url
|
||||
`, userID, subordinateIDs)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
out := make([]string, 0)
|
||||
for rows.Next() {
|
||||
var url string
|
||||
if err := rows.Scan(&url); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out = append(out, url)
|
||||
}
|
||||
return out, rows.Err()
|
||||
}
|
||||
|
||||
func (r *NodeRepository) Update(ctx context.Context, id, actorID string, req model.UpdateNodeRequest) (*model.Node, error) {
|
||||
return scanNode(r.pool.QueryRow(ctx, `
|
||||
UPDATE files_nodes
|
||||
|
||||
Reference in New Issue
Block a user