all

Downloads/Voltaserve/webdav/src/handler/handle-put.ts

@@ -0,0 +1,71 @@
+import fs from 'fs'
+import { readFile } from 'fs/promises'
+import { IncomingMessage, ServerResponse } from 'http'
+import os from 'os'
+import path from 'path'
+import { v4 as uuidv4 } from 'uuid'
+import { Token } from '@/client/idp'
+import { File, FileAPI } from '@/client/api'
+import { handleError } from '@/infra/error'
+
+/*
+  This method creates or updates a resource with the provided content.
+
+  Example implementation:
+
+  - Extract the file path from the URL.
+  - Create a write stream to the file.
+  - Listen for the data event to write the incoming data to the file.
+  - Listen for the end event to indicate the completion of the write stream.
+  - Set the response status code to 201 if created or 204 if updated.
+  - Return the response.
+ */
+async function handlePut(
+  req: IncomingMessage,
+  res: ServerResponse,
+  token: Token
+) {
+  const api = new FileAPI(token)
+  try {
+    const directory = await api.getByPath(
+      decodeURIComponent(path.dirname(req.url))
+    )
+    const outputPath = path.join(os.tmpdir(), uuidv4())
+    const ws = fs.createWriteStream(outputPath)
+    req.pipe(ws)
+    ws.on('error', (err) => {
+      console.error(err)
+      res.statusCode = 500
+      res.end()
+    })
+    ws.on('finish', async () => {
+      try {
+        res.statusCode = 201
+        res.end()
+        const blob = new Blob([await readFile(outputPath)])
+        let existingFile: File | null = null
+        try {
+          existingFile = await api.getByPath(decodeURIComponent(req.url))
+          // Delete existing file to simulate an overwrite
+          await api.delete(existingFile.id)
+        } catch {
+          // Ignored
+        }
+        await api.upload({
+          workspaceId: directory.workspaceId,
+          parentId: directory.id,
+          name: decodeURIComponent(path.basename(req.url)),
+          blob,
+        })
+      } catch (err) {
+        handleError(err, res)
+      } finally {
+        fs.rmSync(outputPath)
+      }
+    })
+  } catch (err) {
+    handleError(err, res)
+  }
+}
+
+export default handlePut