next-tinacms-dos 0.0.0-20240902032558 → 0.0.0-20240903031924

package/dist/handlers.mjs

@@ -0,0 +1,215 @@
+// src/handlers.ts
+import {
+  S3Client,
+  ListObjectsCommand,
+  PutObjectCommand,
+  DeleteObjectCommand
+} from "@aws-sdk/client-s3";
+import path from "path";
+import fs from "fs";
+import multer from "multer";
+import { promisify } from "util";
+var mediaHandlerConfig = {
+  api: {
+    bodyParser: false
+  }
+};
+var createMediaHandler = (config, options) => {
+  const client = new S3Client(config.config);
+  const bucket = config.bucket;
+  let mediaRoot = config.mediaRoot || "";
+  if (mediaRoot) {
+    if (!mediaRoot.endsWith("/")) {
+      mediaRoot = mediaRoot + "/";
+    }
+    if (mediaRoot.startsWith("/")) {
+      mediaRoot = mediaRoot.substr(1);
+    }
+  }
+  let cdnUrl = options?.cdnUrl || config.config.endpoint.toString().replace(/http(s|):\/\//i, `https://${bucket}.`);
+  cdnUrl = cdnUrl + (cdnUrl.endsWith("/") ? "" : "/");
+  return async (req, res) => {
+    const isAuthorized = await config.authorized(req, res);
+    if (!isAuthorized) {
+      res.status(401).json({ message: "sorry this user is unauthorized" });
+      return;
+    }
+    switch (req.method) {
+      case "GET":
+        return listMedia(req, res, client, bucket, mediaRoot, cdnUrl);
+      case "POST":
+        return uploadMedia(req, res, client, bucket, mediaRoot, cdnUrl);
+      case "DELETE":
+        return deleteAsset(req, res, client, bucket);
+      default:
+        res.end(404);
+    }
+  };
+};
+async function uploadMedia(req, res, client, bucket, mediaRoot, cdnUrl) {
+  const upload = promisify(
+    multer({
+      storage: multer.diskStorage({
+        directory: (req2, file, cb) => {
+          cb(null, "/tmp");
+        },
+        filename: (req2, file, cb) => {
+          cb(null, file.originalname);
+        }
+      })
+    }).single("file")
+  );
+  await upload(req, res);
+  const { directory } = req.body;
+  let prefix = directory.replace(/^\//, "").replace(/\/$/, "");
+  if (prefix)
+    prefix = prefix + "/";
+  const filePath = req.file.path;
+  const fileType = req.file?.mimetype;
+  const blob = fs.readFileSync(filePath);
+  const filename = path.basename(filePath);
+  const params = {
+    Bucket: bucket,
+    Key: mediaRoot ? path.join(mediaRoot, prefix + filename) : prefix + filename,
+    Body: blob,
+    ACL: "public-read",
+    ContentType: fileType || "application/octet-stream"
+  };
+  const command = new PutObjectCommand(params);
+  try {
+    const src = cdnUrl + prefix + filename;
+    await client.send(command);
+    res.json({
+      type: "file",
+      id: prefix + filename,
+      filename,
+      directory: prefix,
+      thumbnails: {
+        "75x75": src,
+        "400x400": src,
+        "1000x1000": src
+      },
+      src: cdnUrl + (mediaRoot ? path.join(mediaRoot, prefix + filename) : prefix + filename)
+    });
+  } catch (e) {
+    console.error("Error uploading media");
+    console.error(e);
+    res.status(500).send(findErrorMessage(e));
+  }
+}
+function stripMediaRoot(mediaRoot, key) {
+  if (!mediaRoot) {
+    return key;
+  }
+  const mediaRootParts = mediaRoot.split("/").filter((part) => part);
+  if (!mediaRootParts || !mediaRootParts[0]) {
+    return key;
+  }
+  const keyParts = key.split("/").filter((part) => part);
+  for (let i = 0; i < mediaRootParts.length; i++) {
+    if (keyParts[0] === mediaRootParts[i]) {
+      keyParts.shift();
+    }
+  }
+  return keyParts.join("/");
+}
+async function listMedia(req, res, client, bucket, mediaRoot, cdnUrl) {
+  try {
+    const {
+      directory = "",
+      limit = 500,
+      offset
+    } = req.query;
+    let prefix = directory.replace(/^\//, "").replace(/\/$/, "");
+    if (prefix)
+      prefix = prefix + "/";
+    const params = {
+      Bucket: bucket,
+      Delimiter: "/",
+      Prefix: mediaRoot ? path.join(mediaRoot, prefix) : prefix,
+      Marker: offset?.toString(),
+      MaxKeys: directory && !offset ? +limit + 1 : +limit
+    };
+    const response = await client.send(new ListObjectsCommand(params));
+    const items = [];
+    response.CommonPrefixes?.forEach(({ Prefix }) => {
+      const strippedPrefix = stripMediaRoot(mediaRoot, Prefix);
+      if (!strippedPrefix) {
+        return;
+      }
+      items.push({
+        id: Prefix,
+        type: "dir",
+        filename: path.basename(strippedPrefix),
+        directory: path.dirname(strippedPrefix)
+      });
+    });
+    items.push(
+      ...(response.Contents || []).filter((file) => {
+        const strippedKey = stripMediaRoot(mediaRoot, file.Key);
+        return strippedKey !== prefix;
+      }).map(getDOSToTinaFunc(cdnUrl, mediaRoot))
+    );
+    res.json({
+      items,
+      offset: response.NextMarker
+    });
+  } catch (e) {
+    console.error("Error listing media");
+    console.error(e);
+    res.status(500);
+    const message = findErrorMessage(e);
+    res.json({ e: message });
+  }
+}
+var findErrorMessage = (e) => {
+  if (typeof e == "string")
+    return e;
+  if (e.message)
+    return e.message;
+  if (e.error && e.error.message)
+    return e.error.message;
+  return "an error occurred";
+};
+async function deleteAsset(req, res, client, bucket) {
+  const { media } = req.query;
+  const [, objectKey] = media;
+  const params = {
+    Bucket: bucket,
+    Key: objectKey
+  };
+  try {
+    const data = await client.send(new DeleteObjectCommand(params));
+    res.json(data);
+  } catch (err) {
+    console.error("Error deleting media");
+    console.error(err);
+    res.status(500).json({
+      message: err.message || "Something went wrong"
+    });
+  }
+}
+function getDOSToTinaFunc(cdnUrl, mediaRoot) {
+  return function dosToTina(file) {
+    const strippedKey = stripMediaRoot(mediaRoot, file.Key);
+    const filename = path.basename(strippedKey);
+    const directory = path.dirname(strippedKey) + "/";
+    const src = cdnUrl + file.Key;
+    return {
+      id: file.Key,
+      filename,
+      directory,
+      src,
+      thumbnails: {
+        "75x75": src,
+        "400x400": src,
+        "1000x1000": src
+      },
+      type: "file"
+    };
+  };
+}
+export {
+  createMediaHandler,
+  mediaHandlerConfig
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "next-tinacms-dos",
-  "version": "0.0.0-20240902032558",
+  "version": "0.0.0-20240903031924",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "files": [
@@ -18,7 +18,7 @@
     ]
   },
   "peerDependencies": {
-    "tinacms": "0.0.0-20240902032558"
+    "tinacms": "0.0.0-20240903031924"
   },
   "devDependencies": {
     "@types/crypto-js": "^3.1.47",
@@ -28,8 +28,8 @@
     "react": "^18.3.1",
     "react-dom": "^18.3.1",
     "typescript": "4.6.4",
-    "@tinacms/scripts": "0.0.0-20240902032558",
-    "tinacms": "0.0.0-20240902032558"
+    "@tinacms/scripts": "0.0.0-20240903031924",
+    "tinacms": "0.0.0-20240903031924"
   },
   "publishConfig": {
     "registry": "https://registry.npmjs.org"