socket 断线自动重连，文件上传功能

hydro/handler/home.ts

@@ -308,6 +308,7 @@ class HomeMessagesHandler extends Handler {
     async postSend(domainId: string, uid: number, content: string) {
         const udoc = await user.getById('system', uid);
         if (!udoc) throw new UserNotFoundError(uid);
+        if (udoc.gravatar) udoc.gravatar = misc.gravatar(udoc.gravatar);
         const mdoc = await message.send(this.user._id, uid, content);
         // TODO(twd2): improve here: projection
         if (this.user._id !== uid) {
@@ -330,7 +331,7 @@ class HomeMessagesConnectionHandler extends ConnectionHandler {
         bus.subscribe([`user_message-${this.user._id}`], this, 'onMessageReceived');
     }
 
-    async onMessageReceived(e) {
+    async onMessageReceived(e: any) {
         this.send(e.value);
     }
 
@@ -342,17 +343,17 @@ class HomeMessagesConnectionHandler extends ConnectionHandler {
 class HomeFileHandler extends Handler {
     async get() {
         const ufdocs = await file.getMulti({ owner: this.user._id }).toArray();
-        const fdict = await file.getMetaDict(ufdocs.map((ufdoc) => ufdoc._id));
         this.response.template = 'home_file.html';
-        this.response.body = { ufdocs, fdict };
+        this.response.body = { ufdocs };
     }
 
-    async postDelete(ufid) {
+    @param('ufid', Types.ObjectID)
+    async postDelete(domainId: string, ufid: ObjectID) {
         const ufdoc = await file.getMeta(ufid);
         if (ufdoc.owner !== this.user._id) this.checkPriv(PRIV.PRIV_DELETE_FILE);
         else this.checkPriv(PRIV.PRIV_DELETE_FILE_SELF);
         const result = await file.del(ufdoc._id);
-        if (result) await user.inc(this.user._id, 'usage', ufdoc.length);
+        if (result) await user.inc(this.user._id, 'usage', -ufdoc.size);
         this.back();
     }
 }

hydro/handler/misc.ts

@@ -1,4 +1,5 @@
 /* eslint-disable camelcase */
+import fs from 'fs';
 import { ObjectID } from 'mongodb';
 import { PRIV } from '../model/builtin';
 import * as system from '../model/system';
@@ -9,6 +10,7 @@ import * as db from '../service/db';
 import {
     Route, Handler, Types, param,
 } from '../service/server';
+import { BadRequestError } from '../error';
 
 const coll = db.collection('status');
 
@@ -23,6 +25,41 @@ class FileDownloadHandler extends Handler {
     }
 }
 
+class FileUploadHandler extends Handler {
+    async getQuota() {
+        let quota = await system.get('user.quota');
+        if (this.user.hasPriv(PRIV.PRIV_UNLIMITED_QUOTA)) {
+            quota = 2 ** 63 - 1;
+        }
+        return quota;
+    }
+
+    async get() {
+        this.response.template = 'fs_upload.html';
+        this.response.body = { fdoc: null, usage: this.user.usage, quota: await this.getQuota() };
+    }
+
+    @param('title', Types.String)
+    async post(domainId: string, title: string) {
+        if (!this.request.files.file) throw new BadRequestError();
+        const quota = await this.getQuota();
+        const lfdoc = await fs.promises.stat(this.request.files.file.path);
+        let ufid: ObjectID;
+        const udoc = await user.inc(this.user._id, 'usage', lfdoc.size);
+        try {
+            ufid = await file.add(this.request.files.file.path, title, this.user._id);
+        } catch (e) {
+            await user.inc(this.user._id, 'usage', -lfdoc.size);
+            throw e;
+        }
+        const fdoc = await file.getMeta(ufid);
+        this.response.template = 'fs_upload.html';
+        this.response.body = {
+            fdoc, ufid, usage: udoc.usage, quota,
+        };
+    }
+}
+
 class StatusHandler extends Handler {
     async get() {
         const stats = await coll.find().sort({ type: 1, updateAt: -1 }).toArray();
@@ -109,6 +146,7 @@ class UiSettingsHandler extends Handler {
 export async function apply() {
     Route('file_download', '/fs/:fileId/:secret', FileDownloadHandler);
     Route('file_download_with_name', '/fs/:fileId/:name/:secret', FileDownloadHandler);
+    Route('file_upload', '/fs/upload', FileUploadHandler, PRIV.PRIV_CREATE_FILE);
     Route('status', '/status', StatusHandler);
     Route('status_update', '/status/update', StatusUpdateHandler);
     Route('switch_language', '/language/:lang', SwitchLanguageHandler);