fix: handle no data no fallback

microlinkhq · Sep 15, 2024 · 3ddeb1a · 3ddeb1a
1 parent 8cf7365
commit 3ddeb1a
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/src/send/index.js b/src/send/index.js
@@ -14,9 +14,16 @@ const pickHeaders = headers =>
 module.exports = ({ type, data, req, res }) => {
   const { query } = req
   const statusCode = data ? 200 : 404
-  return query.json
-    ? send(res, statusCode, { url: data })
-    : type === 'buffer'
-      ? send(res, statusCode, dataUriToBuffer(data))
-      : got.stream(data, { headers: pickHeaders(req.headers) }).pipe(res)
+
+  if (query.json) {
+    return send(res, statusCode, { url: data })
+  }
+
+  if (type === 'buffer' || data === undefined) {
+    const responseData = data === undefined ? data : dataUriToBuffer(data)
+    return send(res, statusCode, responseData)
+  }
+
+  const stream = got.stream(data, { headers: pickHeaders(req.headers) })
+  return stream.pipe(res)
 }