交互式安装

hydro/service/db.js

@@ -18,4 +18,5 @@ Mongo.MongoClient.connect(mongourl, { useNewUrlParser: true, useUnifiedTopology:
 
 global.Hydro.service.db = module.exports = {
     collection: (c) => db.collection(c),
+    dropDatabase: () => db.dropDatabase(),
 };

hydro/service/server.js

@@ -8,10 +8,10 @@ const cache = require('koa-static-cache');
 const sockjs = require('sockjs');
 const http = require('http');
 const https = require('https');
-const options = require('../options');
 const validator = require('../lib/validator');
 const template = require('../lib/template');
 const user = require('../model/user');
+const system = require('../model/system');
 const blacklist = require('../model/blacklist');
 const token = require('../model/token');
 const opcount = require('../model/opcount');
@@ -21,19 +21,24 @@ const {
 } = require('../error');
 
 const app = new Koa();
-const server = (options.listen.https ? https : http).createServer(app.callback());
-app.keys = options.session.keys;
-app.use(cache(path.join(process.cwd(), '.uibuild'), {
-    maxAge: 365 * 24 * 60 * 60,
-}));
-app.use(Body({
-    multipart: true,
-    formidable: {
-        maxFileSize: 256 * 1024 * 1024,
-    },
-}));
+let server;
 const router = new Router();
 
+async function prepare() {
+    const useHttps = await system.get('listen.https');
+    server = (useHttps ? https : http).createServer(app.callback());
+    app.keys = await system.get('session.keys');
+    app.use(cache(path.join(process.cwd(), '.uibuild'), {
+        maxAge: 365 * 24 * 60 * 60,
+    }));
+    app.use(Body({
+        multipart: true,
+        formidable: {
+            maxFileSize: 256 * 1024 * 1024,
+        },
+    }));
+}
+
 class Handler {
     /**
      * @param {import('koa').Context} ctx
@@ -128,8 +133,8 @@ class Handler {
         this.now = new Date();
         this._handler.sid = this.request.cookies.get('sid');
         this._handler.save = this.request.cookies.get('save');
-        if (this._handler.save) this._handler.expireSeconds = options.session.saved_expire_seconds;
-        else this._handler.expireSeconds = options.session.unsaved_expire_seconds;
+        if (this._handler.save) this._handler.expireSeconds = await system.get('session.saved_expire_seconds');
+        else this._handler.expireSeconds = await system.get('session.unsaved_expire_seconds');
         this.session = this._handler.sid
             ? await token.update(
                 this._handler.sid,
@@ -213,7 +218,7 @@ class Handler {
                 },
             );
         }
-        const cookie = { secure: options.session.secure };
+        const cookie = { secure: await system.get('session.secure') };
         if (this._handler.save) {
             cookie.expires = this.session.expireAt;
             cookie.maxAge = this._handler.expireSeconds;
@@ -354,6 +359,7 @@ class ConnectionHandler {
         if (!this.user) throw new UserNotFoundError(this.session.uid);
     }
 }
+
 function Connection(prefix, RouteConnHandler) {
     const sock = sockjs.createServer({ prefix });
     sock.on('connection', async (conn) => {
@@ -389,14 +395,14 @@ function Connection(prefix, RouteConnHandler) {
     sock.installHandlers(server);
 }
 
-function start() {
+async function start() {
     app.use(morgan(':method :url :status :res[content-length] - :response-time ms'));
     app.use(router.routes()).use(router.allowedMethods());
     Route('*', Handler);
-    server.listen(options.listen.port);
-    console.log('Server listening at: %s', options.listen.port);
+    server.listen(await system.get('listen.port'));
+    console.log('Server listening at: %s', await system.get('listen.port'));
 }
 
 global.Hydro.service.server = module.exports = {
-    Handler, ConnectionHandler, Route, Connection, start,
+    Handler, ConnectionHandler, Route, Connection, prepare, start,
 };

hydro/service/setup.js

@@ -0,0 +1,73 @@
+const fs = require('fs');
+const path = require('path');
+const Koa = require('koa');
+const morgan = require('koa-morgan');
+const Body = require('koa-body');
+const Router = require('koa-router');
+const cache = require('koa-static-cache');
+const http = require('http');
+const nunjucks = require('nunjucks');
+
+function Loader() { }
+Loader.prototype.getSource = function getSource(name) {
+    if (!global.Hydro.template[name]) throw new Error(`Cannot get template ${name}`);
+    return {
+        src: global.Hydro.template[name],
+        path: name,
+    };
+};
+
+class Nunjucks extends nunjucks.Environment {
+    constructor() {
+        super(new Loader(), { autoescape: true, trimBlocks: true });
+        this.addFilter('json', (self) => JSON.stringify(self), false);
+        this.addFilter('base64_encode', (s) => Buffer.from(s).toString('base64'));
+    }
+}
+const env = new Nunjucks();
+function render(name) {
+    return new Promise((resolve, reject) => {
+        env.render(name, {
+            typeof: (o) => typeof o,
+            static_url: (str) => `/${str}`,
+            handler: { renderTitle: (str) => str },
+            _: (str) => str,
+        }, (err, res) => {
+            if (err) reject(err);
+            else resolve(res);
+        });
+    });
+}
+const app = new Koa();
+const server = http.createServer(app.callback());
+const router = new Router();
+app.keys = ['Hydro'];
+app.use(cache(path.join(process.cwd(), '.uibuild'), {
+    maxAge: 365 * 24 * 60 * 60,
+}));
+app.use(Body({
+    multipart: true,
+    formidable: {
+        maxFileSize: 256 * 1024 * 1024,
+    },
+}));
+
+router.get('/', async (ctx) => {
+    ctx.body = await render('setup.html');
+    ctx.response.type = 'text/html';
+});
+router.post('/', async (ctx) => {
+    const {
+        host, port, name, username, password,
+    } = ctx.request.body;
+    fs.writeFileSync(path.resolve(process.cwd(), 'config.json'), JSON.stringify({
+        host, port, name, username, password,
+    }));
+    ctx.body = await render('setup_done.html');
+    ctx.response.type = 'text/html';
+});
+
+app.use(morgan(':method :url :status :res[content-length] - :response-time ms'));
+app.use(router.routes()).use(router.allowedMethods());
+server.listen(8888);
+console.log('Server listening at: 8888');