@teamkeel/functions-runtime 0.389.7 → 0.390.0-next.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@teamkeel/functions-runtime",
-  "version": "0.389.7",
+  "version": "0.390.0-next.2",
   "description": "Internal package used by @teamkeel/sdk",
   "main": "src/index.js",
   "scripts": {
@@ -19,6 +19,8 @@
     "vitest": "^0.34.6"
   },
   "dependencies": {
+    "@aws-sdk/client-s3": "^3.617.0",
+    "@aws-sdk/credential-providers": "^3.617.0",
     "@neondatabase/serverless": "^0.9.3",
     "@opentelemetry/api": "^1.7.0",
     "@opentelemetry/exporter-trace-otlp-proto": "^0.46.0",

package/src/InlineFile.js

@@ -1,3 +1,9 @@
+const { S3Client, PutObjectCommand } = require("@aws-sdk/client-s3");
+const { fromEnv } = require("@aws-sdk/credential-providers");
+const { useDatabase } = require("./database");
+const { DatabaseError } = require("./errors");
+const KSUID = require("ksuid");
+
 class InlineFile {
   constructor(filename, contentType, size, url) {
     this.filename = filename;
@@ -41,14 +47,64 @@ class InlineFile {
 
     if (this._dataURL) {
       var data = this._dataURL.split(",")[1];
-      var byteString = Buffer.from(data, "base64");
-      return new Blob([byteString], { type: this.contentType });
+      return Buffer.from(data, "base64");
     }
   }
 
-  store() {
-    //TODO: actually store and generate a key
-    this.key = uuidv4();
+  async store() {
+    const content = this.read();
+    this.key = KSUID.randomSync().string;
+
+    if (isS3Storage()) {
+      const s3Client = new S3Client({
+        credentials: fromEnv(),
+        region: process.env.KEEL_REGION,
+      });
+
+      const params = {
+        Bucket: process.env.KEEL_FILES_BUCKET_NAME,
+        Key: key,
+        Body: content,
+        ContentType: this.contentType,
+      };
+
+      const command = new PutObjectCommand(params);
+      try {
+        const result = await s3Client.send(command);
+        console.log(`File uploaded successfully. ETag: ${result.ETag}`);
+        return {
+          key: this.key,
+          size: this.size,
+          filename: this.filename,
+          contentType: this.contentType,
+        };
+      } catch (error) {
+        console.error("Error uploading file:", error);
+        throw error;
+      }
+    }
+
+    // default to db storage
+    const db = useDatabase();
+
+    try {
+      let query = db.insertInto("keel_storage").values({
+        id: this.key,
+        filename: this.filename,
+        content_type: this.contentType,
+        data: content,
+      });
+
+      await query.returningAll().executeTakeFirstOrThrow();
+      return {
+        key: this.key,
+        size: this.size,
+        filename: this.filename,
+        contentType: this.contentType,
+      };
+    } catch (e) {
+      throw new DatabaseError(e);
+    }
   }
 
   toJSON() {
@@ -66,3 +122,7 @@ class InlineFile {
 module.exports = {
   InlineFile,
 };
+
+function isS3Storage() {
+  return "KEEL_FILES_BUCKET_NAME" in process.env;
+}