@aeriajs/builtins 0.0.0

package/dist/collections/file/download.mjs

@@ -0,0 +1,232 @@
+function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) {
+    try {
+        var info = gen[key](arg);
+        var value = info.value;
+    } catch (error) {
+        reject(error);
+        return;
+    }
+    if (info.done) {
+        resolve(value);
+    } else {
+        Promise.resolve(value).then(_next, _throw);
+    }
+}
+function _async_to_generator(fn) {
+    return function() {
+        var self = this, args = arguments;
+        return new Promise(function(resolve, reject) {
+            var gen = fn.apply(self, args);
+            function _next(value) {
+                asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value);
+            }
+            function _throw(err) {
+                asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err);
+            }
+            _next(undefined);
+        });
+    };
+}
+function _ts_generator(thisArg, body) {
+    var f, y, t, g, _ = {
+        label: 0,
+        sent: function() {
+            if (t[0] & 1) throw t[1];
+            return t[1];
+        },
+        trys: [],
+        ops: []
+    };
+    return g = {
+        next: verb(0),
+        "throw": verb(1),
+        "return": verb(2)
+    }, typeof Symbol === "function" && (g[Symbol.iterator] = function() {
+        return this;
+    }), g;
+    function verb(n) {
+        return function(v) {
+            return step([
+                n,
+                v
+            ]);
+        };
+    }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while(_)try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [
+                op[0] & 2,
+                t.value
+            ];
+            switch(op[0]){
+                case 0:
+                case 1:
+                    t = op;
+                    break;
+                case 4:
+                    _.label++;
+                    return {
+                        value: op[1],
+                        done: false
+                    };
+                case 5:
+                    _.label++;
+                    y = op[1];
+                    op = [
+                        0
+                    ];
+                    continue;
+                case 7:
+                    op = _.ops.pop();
+                    _.trys.pop();
+                    continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) {
+                        _ = 0;
+                        continue;
+                    }
+                    if (op[0] === 3 && (!t || op[1] > t[0] && op[1] < t[3])) {
+                        _.label = op[1];
+                        break;
+                    }
+                    if (op[0] === 6 && _.label < t[1]) {
+                        _.label = t[1];
+                        t = op;
+                        break;
+                    }
+                    if (t && _.label < t[2]) {
+                        _.label = t[2];
+                        _.ops.push(op);
+                        break;
+                    }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop();
+                    continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) {
+            op = [
+                6,
+                e
+            ];
+            y = 0;
+        } finally{
+            f = t = 0;
+        }
+        if (op[0] & 5) throw op[1];
+        return {
+            value: op[0] ? op[1] : void 0,
+            done: true
+        };
+    }
+}
+import { ObjectId } from "mongodb";
+import { left } from "@aeriajs/common";
+import fs from "fs";
+export var FileReadError;
+(function(FileReadError) {
+    FileReadError["DocumentNotFound"] = "DOCUMENT_NOT_FOUND";
+    FileReadError["FileNotFound"] = "FILE_NOT_FOUND";
+})(FileReadError || (FileReadError = {}));
+export var download = function() {
+    var _ref = _async_to_generator(function(payload, context) {
+        var fileId, _payload_options, options, file, stat, e, range, parts, start, end, chunkSize;
+        return _ts_generator(this, function(_state) {
+            switch(_state.label){
+                case 0:
+                    fileId = payload.fileId, _payload_options = payload.options, options = _payload_options === void 0 ? [] : _payload_options;
+                    return [
+                        4,
+                        context.collection.model.findOne({
+                            _id: new ObjectId(fileId)
+                        }, {
+                            projection: {
+                                absolute_path: 1,
+                                filename: 1,
+                                mime: 1
+                            }
+                        })
+                    ];
+                case 1:
+                    file = _state.sent();
+                    if (!file) {
+                        if (!payload.noHeaders) {
+                            context.response.writeHead(404, {
+                                "content-type": "application/json"
+                            });
+                        }
+                        return [
+                            2,
+                            left("DOCUMENT_NOT_FOUND")
+                        ];
+                    }
+                    _state.label = 2;
+                case 2:
+                    _state.trys.push([
+                        2,
+                        4,
+                        ,
+                        5
+                    ]);
+                    return [
+                        4,
+                        fs.promises.stat(file.absolute_path)
+                    ];
+                case 3:
+                    stat = _state.sent();
+                    return [
+                        3,
+                        5
+                    ];
+                case 4:
+                    e = _state.sent();
+                    context.response.writeHead(404, {
+                        "content-type": "application/json"
+                    });
+                    return [
+                        2,
+                        left("FILE_NOT_FOUND")
+                    ];
+                case 5:
+                    range = context.request.headers.range;
+                    if (range) {
+                        parts = range.replace(/bytes=/, "").split("-");
+                        start = parseInt(parts[0]);
+                        end = parts[1] ? parseInt(parts[1]) : stat.size - 1;
+                        chunkSize = end - start + 1;
+                        if (!payload.noHeaders) {
+                            context.response.writeHead(206, {
+                                "accept-ranges": "bytes",
+                                "content-range": "bytes ".concat(start, "-").concat(end, "/").concat(stat.size),
+                                "content-length": chunkSize,
+                                "content-type": file.mime,
+                                "content-disposition": "".concat(options.includes("download") ? "attachment; " : "", "filename=").concat(encodeURI(file.filename))
+                            });
+                        }
+                        return [
+                            2,
+                            fs.createReadStream(file.absolute_path, {
+                                start: start,
+                                end: end
+                            })
+                        ];
+                    }
+                    if (!payload.noHeaders) {
+                        context.response.writeHead(200, {
+                            "content-type": file.mime,
+                            "content-disposition": "".concat(options.includes("download") ? "attachment; " : "", "filename=").concat(encodeURI(file.filename))
+                        });
+                    }
+                    return [
+                        2,
+                        fs.createReadStream(file.absolute_path)
+                    ];
+            }
+        });
+    });
+    return function download(payload, context) {
+        return _ref.apply(this, arguments);
+    };
+}();