作业功能

hydro/handler/misc.js

@@ -0,0 +1,51 @@
+
+const { PERM_JUDGE } = require('../permission');
+const file = require('../model/file');
+const db = require('../service/db');
+const { Route, Handler } = require('../service/server');
+
+const coll = db.collection('status');
+
+class FileDownloadHandler extends Handler {
+    async get({ id, secret, name }) {
+        if (name) name = Buffer.from(name, 'base64').toString();
+        this.response.attachment(name || id);
+        this.response.body = await file.get(id, secret);
+    }
+}
+
+class StatusHandler extends Handler {
+    async get() {
+        const stats = await coll.find().sort({ type: 1, updateAt: -1 }).toArray();
+        for (const i in stats) {
+            let desc = '';
+            const online = new Date(stats[i].updateAt).getTime() > new Date().getTime() - 300000;
+            if (!online) desc = 'Offline';
+            desc = desc || 'Online';
+            stats[i].status = desc;
+        }
+        this.response.body = { stats };
+        this.response.template = 'status.html';
+    }
+}
+
+class StatusUpdateHandler extends Handler {
+    async post(args) {
+        this.checkPerm(PERM_JUDGE);
+        await coll.updateOne({ _id: args._id }, { $set: args }, { upsert: true });
+    }
+}
+
+async function apply() {
+    Route('/fs/:id/:secret', FileDownloadHandler);
+    Route('/fs/:id/:name/:secret', FileDownloadHandler);
+    Route('/status', StatusHandler);
+    Route('/status/update', StatusUpdateHandler);
+}
+
+global.Hydro.handler.misc = module.exports = {
+    FileDownloadHandler,
+    StatusHandler,
+    StatusUpdateHandler,
+    apply,
+};