core: ctx.setInterval support

packages/hydrooj/src/service/module.ts

@@ -31,9 +31,9 @@ export class Runtime {
         if (this.loaded) throw new Error(`Module ${this.filename} was already loaded`);
         this.sideEffect = false;
         if (typeof module.dispose === 'function') this.disposables.push(module.dispose);
-        const T = (origFunc) => (origFunc ? (...args) => {
+        const T = (origFunc, disposeFunc?) => (origFunc ? (...args) => {
             const res = origFunc(...args);
-            this.disposables.push(res);
+            this.disposables.push(disposeFunc ? () => disposeFunc(res) : res);
             return res;
         } : null);
         module.apply({
@@ -43,6 +43,9 @@ export class Runtime {
             once: T(bus.once),
             off: bus.off,
             addScript: T(addScript),
+            setInterval: T(setInterval, clearInterval),
+            setTimeout: T(setTimeout, clearTimeout),
+            setImmediate: T(setImmediate, clearImmediate),
         } as Context);
         if (module.sideEffect) this.sideEffect = true;
         this.loaded = true;

packages/prom-client/handler.ts

@@ -1,7 +1,7 @@
 import { AggregatorRegistry, metric } from 'prom-client';
 import * as system from 'hydrooj/src/model/system';
 import * as bus from 'hydrooj/src/service/bus';
-import { Handler, Route } from 'hydrooj/src/service/server';
+import { Handler } from 'hydrooj/src/service/server';
 import { registry } from './metrics';
 
 declare module 'hydrooj/src/service/bus' {
@@ -42,11 +42,10 @@ class MetricsHandler extends Handler {
     }
 }
 
-bus.on('metrics', (id, metrics) => { instances[id] = metrics; });
-setInterval(async () => {
-    bus.broadcast('metrics', process.env.NODE_APP_INSTANCE!, await registry.getMetricsAsJSON());
-}, 5000 * (+system.get('prom-client.collect_rate') || 1));
-
-global.Hydro.handler.prom = () => {
-    Route('metrics', '/metrics', MetricsHandler);
-};
+export function apply(ctx) {
+    ctx.on('metrics', (id, metrics) => { instances[id] = metrics; });
+    ctx.setInterval(async () => {
+        bus.broadcast('metrics', process.env.NODE_APP_INSTANCE!, await registry.getMetricsAsJSON());
+    }, 5000 * (+system.get('prom-client.collect_rate') || 1));
+    ctx.Route('metrics', '/metrics', MetricsHandler);
+}

packages/ui-default/components/preview/preview.page.ts

@@ -162,4 +162,5 @@ const dataPreviewPage = new AutoloadPage('dataPreview', () => {
   $(document).on('click', '[data-preview]', previewFile);
 });
 
+window.Hydro.components.preview = { startEdit, previewFile };
 export default dataPreviewPage;