1 files changed, 89 insertions, 0 deletions
diff --git a/js/timers.ts b/js/timers.ts
new file mode 100644
index 000000000..da2cccd89
--- /dev/null
+++ b/js/timers.ts
@@ -0,0 +1,89 @@
+// Copyright 2018 Ryan Dahl <ry@tinyclouds.org>
+// All rights reserved. MIT License.
+import { deno as pb } from "./msg.pb";
+import { pubInternal, sub } from "./dispatch";
+import { assert } from "./util";
+
+let nextTimerId = 1;
+
+// tslint:disable-next-line:no-any
+export type TimerCallback = (...args: any[]) => void;
+
+interface Timer {
+  id: number;
+  cb: TimerCallback;
+  interval: boolean;
+  // tslint:disable-next-line:no-any
+  args: any[];
+  delay: number; // milliseconds
+}
+
+const timers = new Map<number, Timer>();
+
+export function initTimers() {
+  sub("timers", onMessage);
+}
+
+function onMessage(payload: Uint8Array) {
+  const msg = pb.Msg.decode(payload);
+  assert(msg.command === pb.Msg.Command.TIMER_READY);
+  const { timerReadyId, timerReadyDone } = msg;
+  const timer = timers.get(timerReadyId);
+  if (!timer) {
+    return;
+  }
+  timer.cb(...timer.args);
+  if (timerReadyDone) {
+    timers.delete(timerReadyId);
+  }
+}
+
+function setTimer(
+  cb: TimerCallback,
+  delay: number,
+  interval: boolean,
+  // tslint:disable-next-line:no-any
+  args: any[]
+): number {
+  const timer = {
+    id: nextTimerId++,
+    interval,
+    delay,
+    args,
+    cb
+  };
+  timers.set(timer.id, timer);
+  pubInternal("timers", {
+    command: pb.Msg.Command.TIMER_START,
+    timerStartId: timer.id,
+    timerStartInterval: timer.interval,
+    timerStartDelay: timer.delay
+  });
+  return timer.id;
+}
+
+export function setTimeout(
+  cb: TimerCallback,
+  delay: number,
+  // tslint:disable-next-line:no-any
+  ...args: any[]
+): number {
+  return setTimer(cb, delay, false, args);
+}
+
+export function setInterval(
+  cb: TimerCallback,
+  delay: number,
+  // tslint:disable-next-line:no-any
+  ...args: any[]
+): number {
+  return setTimer(cb, delay, true, args);
+}
+
+export function clearTimer(id: number) {
+  timers.delete(id);
+  pubInternal("timers", {
+    command: pb.Msg.Command.TIMER_CLEAR,
+    timerClearId: id
+  });
+}