Minor changes

backend/src/restful/index.js

@@ -0,0 +1,118 @@
+import {
+    SETTINGS_KEY,
+    GIST_BACKUP_KEY,
+    GIST_BACKUP_FILE_NAME,
+} from './constants';
+import { ENV } from '../utils/open-api';
+import express from '../utils/express';
+import { IP_API } from '../utils/geo';
+import Gist from '../utils/gist';
+import $ from '../core/app';
+
+import registerSubscriptionRoutes from './subscriptions';
+import registerCollectionRoutes from './collections';
+import registerArtifactRoutes from './artifacts';
+import registerSettingRoutes from './settings';
+
+export default function serve() {
+    const $app = express();
+
+    // register routes
+    registerCollectionRoutes($app);
+    registerSubscriptionRoutes($app);
+    registerSettingRoutes($app);
+    registerArtifactRoutes($app);
+
+    // utils
+    $app.get('/api/utils/IP_API/:server', IP_API); // IP-API reverse proxy
+    $app.get('/api/utils/env', getEnv); // get runtime environment
+    $app.get('/api/utils/backup', gistBackup); // gist backup actions
+
+    // Redirect sub.store to vercel webpage
+    $app.get('/', async (req, res) => {
+        // 302 redirect
+        res.set('location', 'https://sub-store.vercel.app/').status(302).end();
+    });
+
+    // handle preflight request for QX
+    if (ENV().isQX) {
+        $app.options('/', async (req, res) => {
+            res.status(200).end();
+        });
+    }
+
+    $app.all('/', (_, res) => {
+        res.send('Hello from sub-store, made with ❤️ by Peng-YM');
+    });
+
+    $app.start();
+}
+
+function getEnv(req, res) {
+    const { isNode, isQX, isLoon, isSurge } = ENV();
+    let backend = 'Node';
+    if (isNode) backend = 'Node';
+    if (isQX) backend = 'QX';
+    if (isLoon) backend = 'Loon';
+    if (isSurge) backend = 'Surge';
+    res.json({
+        backend,
+    });
+}
+
+async function gistBackup(req, res) {
+    const { action } = req.query;
+    // read token
+    const { gistToken } = $.read(SETTINGS_KEY);
+    if (!gistToken) {
+        res.status(500).json({
+            status: 'failed',
+            message: '未找到Gist备份Token!',
+        });
+    } else {
+        const gist = new Gist({
+            token: gistToken,
+            key: GIST_BACKUP_KEY,
+        });
+        try {
+            let content;
+            const settings = $.read(SETTINGS_KEY);
+            switch (action) {
+                case 'upload':
+                    // update syncTime.
+                    settings.syncTime = new Date().getTime();
+                    $.write(settings, SETTINGS_KEY);
+                    content = $.read('#sub-store');
+                    if ($.env.isNode)
+                        content = JSON.stringify($.cache, null, `  `);
+                    $.info(`上传备份中...`);
+                    await gist.upload({ [GIST_BACKUP_FILE_NAME]: { content } });
+                    break;
+                case 'download':
+                    $.info(`还原备份中...`);
+                    content = await gist.download(GIST_BACKUP_FILE_NAME);
+                    // restore settings
+                    $.write(content, '#sub-store');
+                    if ($.env.isNode) {
+                        content = JSON.parse(content);
+                        Object.keys(content).forEach((key) => {
+                            $.write(content[key], key);
+                        });
+                    }
+                    break;
+            }
+            res.json({
+                status: 'success',
+            });
+        } catch (err) {
+            const msg = `${
+                action === 'upload' ? '上传' : '下载'
+            }备份失败！${err}`;
+            $.error(msg);
+            res.status(500).json({
+                status: 'failed',
+                message: msg,
+            });
+        }
+    }
+}