dfx 0.0.3 → 0.1.0

package/DiscordShard/heartbeats.ts

@@ -0,0 +1,65 @@
+import * as T from "@effect-ts/core/Effect"
+import * as R from "@effect-ts/core/Effect/Ref"
+import * as SC from "@effect-ts/core/Effect/Schedule"
+import { pipe } from "@effect-ts/core/Function"
+import * as O from "@effect-ts/core/Option"
+import * as CB from "callbag-effect-ts"
+import * as DWS from "../DiscordWS"
+import { GatewayOpcode, GatewayPayload, HelloEvent } from "../types"
+import { Reconnect } from "../WS"
+import * as Commands from "./commands"
+import * as Utils from "./utils"
+
+const send = (ref: R.Ref<boolean>, seqRef: R.Ref<O.Option<number>>) =>
+  pipe(
+    R.get(seqRef),
+    T.map((o) => Commands.heartbeat(O.toNullable(o))),
+    T.tap(() => R.set_(ref, false)),
+  )
+
+const maybeSend = (ref: R.Ref<boolean>, seqRef: R.Ref<O.Option<number>>) =>
+  pipe(
+    R.get(ref),
+    T.chain(
+      (acked): T.UIO<DWS.Message> =>
+        acked ? send(ref, seqRef) : T.succeed(Reconnect),
+    ),
+  )
+
+export const fromRaw = <R, E>(
+  source: CB.EffectSource<R, E, GatewayPayload>,
+  seqRef: R.Ref<O.Option<number>>,
+) =>
+  pipe(
+    R.makeRef(true),
+
+    T.map((ackedRef) => {
+      const heartbeats = pipe(
+        Utils.opCode(source)<HelloEvent>(GatewayOpcode.HELLO),
+
+        // Reset ack state for each hello
+        CB.tap(() => R.set_(ackedRef, true)),
+
+        CB.switchMap((p) =>
+          CB.fromSchedule(
+            SC.duration(p.d!.heartbeat_interval * Math.random())["++"](
+              SC.spaced(p.d!.heartbeat_interval),
+            ),
+          ),
+        ),
+
+        // Map to gateway message depending on ack state
+        CB.mapEffect(() => maybeSend(ackedRef, seqRef)),
+      )
+
+      const acks = pipe(
+        Utils.opCode(source)(GatewayOpcode.HEARTBEAT_ACK),
+        CB.tap(() => R.set_(ackedRef, true)),
+        CB.drain,
+      )
+
+      return CB.merge_(heartbeats, acks)
+    }),
+
+    CB.unwrap,
+  )

package/DiscordShard/identify.d.ts

@@ -0,0 +1,15 @@
+import * as R from "@effect-ts/core/Effect/Ref";
+import * as O from "@effect-ts/core/Option";
+import * as CB from "callbag-effect-ts";
+import { GatewayPayload, ReadyEvent, UpdatePresence } from "../types";
+export interface Options {
+    token: string;
+    intents: number;
+    shard: [number, number];
+    presence?: UpdatePresence;
+}
+export interface Requirements {
+    latestReady: R.Ref<O.Option<ReadyEvent>>;
+    latestSequence: R.Ref<O.Option<number>>;
+}
+export declare const fromRaw: <R, E>(source: CB.EffectSource<R, E, GatewayPayload<any>>, { latestReady, latestSequence, ...opts }: Options & Requirements) => CB.EffectSource<R, E, GatewayPayload<any>>;

package/DiscordShard/identify.js

@@ -0,0 +1,35 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fromRaw = void 0;
+const tslib_1 = require("tslib");
+const tracing_1 = tslib_1.__importStar(require("@effect-ts/core/Tracing"));
+const fileName_1 = "DiscordShard/identify.ts";
+const T = tslib_1.__importStar(require("@effect-ts/core/Effect"));
+const R = tslib_1.__importStar(require("@effect-ts/core/Effect/Ref"));
+const Function_1 = require("@effect-ts/core/Function");
+const O = tslib_1.__importStar(require("@effect-ts/core/Option"));
+const CB = tslib_1.__importStar(require("callbag-effect-ts"));
+const OS = tslib_1.__importStar(require("os"));
+const types_1 = require("../types");
+const Commands = tslib_1.__importStar(require("./commands"));
+const Utils = tslib_1.__importStar(require("./utils"));
+const identify = ({ token, intents, shard, presence }) => Commands.identify({
+    token,
+    intents,
+    properties: {
+        $os: OS.platform(),
+        $browser: "dfx",
+        $device: "dfx",
+    },
+    shard,
+    presence,
+});
+const resume = (token, ready, seq) => Commands.resume({
+    token,
+    session_id: ready.session_id,
+    seq,
+});
+const identifyOrResume = (opts, ready, seq) => (T.map_(T.zip_(R.get(ready), R.get(seq), fileName_1 + ":56:10"), ({ tuple }) => (O.fold_(O.zip_(...tuple), () => identify(opts), ({ tuple }) => resume(opts.token, ...tuple))), fileName_1 + ":57:10"));
+const fromRaw = (source, { latestReady, latestSequence, ...opts }) => (CB.mapEffect_(Utils.opCode(source)(types_1.GatewayOpcode.HELLO), (_) => identifyOrResume(opts, latestReady, latestSequence)));
+exports.fromRaw = fromRaw;
+//# sourceMappingURL=identify.js.map

package/DiscordShard/identify.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"identify.js","sourceRoot":"","sources":["identify.ts"],"names":[],"mappings":";;;;;;AAAA,kEAA2C;AAC3C,sEAA+C;AAC/C,uDAA+C;AAC/C,kEAA2C;AAC3C,8DAAuC;AACvC,+CAAwB;AACxB,oCAMiB;AACjB,6DAAsC;AACtC,uDAAgC;AAchC,MAAM,QAAQ,GAAG,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAW,EAAE,EAAE,CAChE,QAAQ,CAAC,QAAQ,CAAC;IAChB,KAAK;IACL,OAAO;IACP,UAAU,EAAE;QACV,GAAG,EAAE,EAAE,CAAC,QAAQ,EAAE;QAClB,QAAQ,EAAE,KAAK;QACf,OAAO,EAAE,KAAK;KACf;IACD,KAAK;IACL,QAAQ;CACT,CAAC,CAAA;AAEJ,MAAM,MAAM,GAAG,CAAC,KAAa,EAAE,KAAiB,EAAE,GAAW,EAAE,EAAE,CAC/D,QAAQ,CAAC,MAAM,CAAC;IACd,KAAK;IACL,UAAU,EAAE,KAAK,CAAC,UAAU;IAC5B,GAAG;CACJ,CAAC,CAAA;AAEJ,MAAM,gBAAgB,GAAG,CACvB,IAAa,EACb,KAAkC,EAClC,GAA4B,EAC5B,EAAE,EAIA,CAAC,MADD,CAAC,MADD,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EACN,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,0BACV,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAGhB,CAAC,OADD,CAAC,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,EAEd,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,EACpB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,KAAK,CAAC,EAE9C,yBAEJ,CAAA;AAEI,MAAM,OAAO,GAAG,CACrB,MAA6C,EAC7C,EAAE,WAAW,EAAE,cAAc,EAAE,GAAG,IAAI,EAA0B,EAChE,EAAE,EAGA,EAAE,YADF,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAa,qBAAa,CAAC,KAAK,CAAC,EACxC,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC,IAAI,EAAE,WAAW,EAAE,cAAc,CAAC,EACxE,CAAA;AAPU,QAAA,OAAO,WAOjB"}

package/DiscordShard/identify.ts

@@ -0,0 +1,75 @@
+import * as T from "@effect-ts/core/Effect"
+import * as R from "@effect-ts/core/Effect/Ref"
+import { pipe } from "@effect-ts/core/Function"
+import * as O from "@effect-ts/core/Option"
+import * as CB from "callbag-effect-ts"
+import * as OS from "os"
+import {
+  GatewayOpcode,
+  GatewayPayload,
+  HelloEvent,
+  ReadyEvent,
+  UpdatePresence,
+} from "../types"
+import * as Commands from "./commands"
+import * as Utils from "./utils"
+
+export interface Options {
+  token: string
+  intents: number
+  shard: [number, number]
+  presence?: UpdatePresence
+}
+
+export interface Requirements {
+  latestReady: R.Ref<O.Option<ReadyEvent>>
+  latestSequence: R.Ref<O.Option<number>>
+}
+
+const identify = ({ token, intents, shard, presence }: Options) =>
+  Commands.identify({
+    token,
+    intents,
+    properties: {
+      $os: OS.platform(),
+      $browser: "dfx",
+      $device: "dfx",
+    },
+    shard,
+    presence,
+  })
+
+const resume = (token: string, ready: ReadyEvent, seq: number) =>
+  Commands.resume({
+    token,
+    session_id: ready.session_id,
+    seq,
+  })
+
+const identifyOrResume = (
+  opts: Options,
+  ready: R.Ref<O.Option<ReadyEvent>>,
+  seq: R.Ref<O.Option<number>>,
+) =>
+  pipe(
+    R.get(ready),
+    T.zip(R.get(seq)),
+    T.map(({ tuple }) =>
+      pipe(
+        O.zip_(...tuple),
+        O.fold(
+          () => identify(opts),
+          ({ tuple }) => resume(opts.token, ...tuple),
+        ),
+      ),
+    ),
+  )
+
+export const fromRaw = <R, E>(
+  source: CB.EffectSource<R, E, GatewayPayload>,
+  { latestReady, latestSequence, ...opts }: Options & Requirements,
+) =>
+  pipe(
+    Utils.opCode(source)<HelloEvent>(GatewayOpcode.HELLO),
+    CB.mapEffect((_) => identifyOrResume(opts, latestReady, latestSequence)),
+  )

package/DiscordShard/index.d.ts

@@ -0,0 +1,32 @@
+import * as T from "@effect-ts/core/Effect";
+import * as L from "@effect-ts/core/Effect/Layer";
+import * as M from "@effect-ts/core/Effect/Managed";
+import * as CB from "callbag-effect-ts";
+import * as Config from "../DiscordConfig";
+import * as DWS from "../DiscordWS";
+import { GatewayEvent, GatewayPayload } from "../types";
+export interface Options {
+    shard: [number, number];
+}
+declare const makeService: () => {
+    readonly _tag: "DiscordShardService";
+    readonly make: (shard: [id: number, count: number]) => M.Managed<import("@effect-ts/system/Has").Has<Config.DiscordConfig>, never, {
+        readonly run: T.RIO<import("@effect-ts/system/Has").Has<DWS.DiscordWS> & import("@effect-ts/system/Has").Has<import("../WS").WS> & import("@effect-ts/system/Has").Has<import("../Log").Log> & import("@effect-ts/system/Clock").HasClock, void>;
+        readonly raw: CB.EffectSource<import("@effect-ts/system/Has").Has<DWS.DiscordWS> & import("@effect-ts/system/Has").Has<import("../WS").WS> & import("@effect-ts/system/Has").Has<import("../Log").Log> & import("@effect-ts/system/Clock").HasClock, unknown, GatewayPayload<any>>;
+        readonly dispatch: CB.EffectSource<import("@effect-ts/system/Has").Has<DWS.DiscordWS> & import("@effect-ts/system/Has").Has<import("../WS").WS> & import("@effect-ts/system/Has").Has<import("../Log").Log> & import("@effect-ts/system/Clock").HasClock, unknown, GatewayPayload<GatewayEvent>>;
+        readonly send: (p: GatewayPayload) => T.Effect<unknown, never, boolean>;
+        readonly reconnect: () => T.Effect<unknown, never, boolean>;
+    }>;
+};
+export interface DiscordShard extends ReturnType<typeof makeService> {
+}
+export declare const DiscordShard: import("@effect-ts/system/Has").Tag<DiscordShard>;
+export declare const LiveDiscordShard: L.Layer<unknown, never, import("@effect-ts/system/Has").Has<DiscordShard>>;
+export declare const make: (shard: [number, number]) => M.Managed<import("@effect-ts/system/Has").Has<Config.DiscordConfig> & import("@effect-ts/system/Has").Has<DiscordShard>, never, {
+    readonly run: T.RIO<import("@effect-ts/system/Has").Has<DWS.DiscordWS> & import("@effect-ts/system/Has").Has<import("../WS").WS> & import("@effect-ts/system/Has").Has<import("../Log").Log> & import("@effect-ts/system/Clock").HasClock, void>;
+    readonly raw: CB.EffectSource<import("@effect-ts/system/Has").Has<DWS.DiscordWS> & import("@effect-ts/system/Has").Has<import("../WS").WS> & import("@effect-ts/system/Has").Has<import("../Log").Log> & import("@effect-ts/system/Clock").HasClock, unknown, GatewayPayload<any>>;
+    readonly dispatch: CB.EffectSource<import("@effect-ts/system/Has").Has<DWS.DiscordWS> & import("@effect-ts/system/Has").Has<import("../WS").WS> & import("@effect-ts/system/Has").Has<import("../Log").Log> & import("@effect-ts/system/Clock").HasClock, unknown, GatewayPayload<GatewayEvent>>;
+    readonly send: (p: GatewayPayload) => T.Effect<unknown, never, boolean>;
+    readonly reconnect: () => T.Effect<unknown, never, boolean>;
+}>;
+export {};

package/DiscordShard/index.js

@@ -0,0 +1,63 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.make = exports.LiveDiscordShard = exports.DiscordShard = void 0;
+const tslib_1 = require("tslib");
+const tracing_1 = tslib_1.__importStar(require("@effect-ts/core/Tracing"));
+const fileName_1 = "DiscordShard/index.ts";
+const T = tslib_1.__importStar(require("@effect-ts/core/Effect"));
+const L = tslib_1.__importStar(require("@effect-ts/core/Effect/Layer"));
+const M = tslib_1.__importStar(require("@effect-ts/core/Effect/Managed"));
+const Q = tslib_1.__importStar(require("@effect-ts/core/Effect/Queue"));
+const Function_1 = require("@effect-ts/core/Function");
+const Has_1 = require("@effect-ts/core/Has");
+const O = tslib_1.__importStar(require("@effect-ts/core/Option"));
+const CB = tslib_1.__importStar(require("callbag-effect-ts"));
+const Config = tslib_1.__importStar(require("../DiscordConfig"));
+const DWS = tslib_1.__importStar(require("../DiscordWS"));
+const types_1 = require("../types");
+const WS_1 = require("../WS");
+const Heartbeats = tslib_1.__importStar(require("./heartbeats"));
+const Identify = tslib_1.__importStar(require("./identify"));
+const Invalid = tslib_1.__importStar(require("./invalidSession"));
+const Utils = tslib_1.__importStar(require("./utils"));
+const makeImpl = (shard) => M.gen(function* (_) {
+    const token = yield* _(Config.token, fileName_1 + ":29:27");
+    const gateway = yield* _(Config.gateway, fileName_1 + ":30:29");
+    const outbound = yield* _(Q.makeUnbounded(), fileName_1 + ":31:30");
+    const [latestReady, updateLatestReady] = yield* _(Utils.latest((0, Function_1.flow)(O.fromPredicate((p) => p.op === types_1.GatewayOpcode.DISPATCH && p.t === "READY"), O.map((p) => p.d))), fileName_1 + ":33:54");
+    const [latestSequence, updateLatestSequence] = yield* _(Utils.latest((p) => O.fromNullable(p.s)), fileName_1 + ":44:60");
+    const raw = (CB.share(CB.unwrap(DWS.open({
+        outgoingQueue: outbound,
+    }))));
+    const updateRefs = (CB.runDrain(updateLatestReady(updateLatestSequence(raw))));
+    const dispatch = (CB.share(CB.filter_(raw, (p) => p.op === types_1.GatewayOpcode.DISPATCH)));
+    // heartbeats
+    const heartbeatEffects = (CB.forEach_(Heartbeats.fromRaw(raw, latestSequence), (p) => Q.offer_(outbound, p)));
+    // identify
+    const identifyEffects = (CB.forEach_(Identify.fromRaw(raw, {
+        token,
+        shard,
+        intents: gateway.intents,
+        presence: gateway.presence,
+        latestSequence,
+        latestReady,
+    }), (p) => Q.offer_(outbound, p)));
+    // invalid session
+    const invalidEffects = (CB.forEach_(Invalid.fromRaw(raw, latestReady), (p) => Q.offer_(outbound, p)));
+    return {
+        run: (T.ignore(T.zipPar(invalidEffects, fileName_1 + ":101:17")(T.zipPar(identifyEffects, fileName_1 + ":100:17")(T.zipPar(heartbeatEffects, fileName_1 + ":99:17")(updateRefs))), fileName_1 + ":102:17")),
+        raw,
+        dispatch,
+        send: (p) => Q.offer_(outbound, p),
+        reconnect: () => Q.offer_(outbound, WS_1.Reconnect),
+    };
+});
+const makeService = () => ({
+    _tag: "DiscordShardService",
+    make: makeImpl,
+});
+exports.DiscordShard = (0, Has_1.tag)();
+exports.LiveDiscordShard = L.fromFunction(exports.DiscordShard)(makeService);
+const make = (shard) => M.accessServiceM(exports.DiscordShard)(({ make }) => make(shard));
+exports.make = make;
+//# sourceMappingURL=index.js.map

package/DiscordShard/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["index.ts"],"names":[],"mappings":";;;;;;AAAA,kEAA2C;AAC3C,wEAAiD;AACjD,0EAAmD;AACnD,wEAAiD;AACjD,uDAAqD;AACrD,6CAAyC;AACzC,kEAA2C;AAC3C,8DAAuC;AACvC,iEAA0C;AAC1C,0DAAmC;AACnC,oCAKiB;AACjB,8BAAiC;AACjC,iEAA0C;AAC1C,6DAAsC;AACtC,kEAA2C;AAC3C,uDAAgC;AAMhC,MAAM,QAAQ,GAAG,CAAC,KAAkC,EAAE,EAAE,CACtD,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;IAChB,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,wBAAC,CAAA;IACpC,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,wBAAC,CAAA;IACxC,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,EAAe,wBAAC,CAAA;IAEzD,MAAM,CAAC,WAAW,EAAE,iBAAiB,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAC/C,KAAK,CAAC,MAAM,CACV,IAAA,eAAI,EACF,CAAC,CAAC,aAAa,CACb,CAAC,CAAC,EAAmC,EAAE,CACrC,CAAC,CAAC,EAAE,KAAK,qBAAa,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC,KAAK,OAAO,CACrD,EACD,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC,CACnB,CACF,wBACF,CAAA;IACD,MAAM,CAAC,cAAc,EAAE,oBAAoB,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CACrD,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,wBACzC,CAAA;IAED,MAAM,GAAG,IAKP,EAAE,CAAC,KAAK,CADR,EAAE,CAAC,MAAM,CAHT,GAAG,CAAC,IAAI,CAAC;QACP,aAAa,EAAE,QAAQ;KACxB,CAAC,GAGH,CAAA;IACD,MAAM,UAAU,IAId,EAAE,CAAC,QAAQ,CADX,iBAAiB,CADjB,oBAAoB,CADpB,GAAG,IAIJ,CAAA;IAED,MAAM,QAAQ,IAMZ,EAAE,CAAC,KAAK,CAJR,EAAE,SADF,GAAG,EAED,CAAC,CAAC,EAAqC,EAAE,CACvC,CAAC,CAAC,EAAE,KAAK,qBAAa,CAAC,QAAQ,GAGpC,CAAA;IAED,aAAa;IACb,MAAM,gBAAgB,IAEpB,EAAE,UADF,UAAU,CAAC,OAAO,CAAC,GAAG,EAAE,cAAc,CAAC,EAC5B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,EACxC,CAAA;IAED,WAAW;IACX,MAAM,eAAe,IASnB,EAAE,UARF,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE;QACpB,KAAK;QACL,KAAK;QACL,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,cAAc;QACd,WAAW;KACZ,CAAC,EACS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,EACxC,CAAA;IAED,kBAAkB;IAClB,MAAM,cAAc,IAElB,EAAE,UADF,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,WAAW,CAAC,EACtB,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,EACxC,CAAA;IAED,OAAO;QACL,GAAG,GAKD,CAAC,CAAC,MAAM,CADR,CAAC,CAAC,MAAM,CAAC,cAAc,yBAAC,CADxB,CAAC,CAAC,MAAM,CAAC,eAAe,yBAAC,CADzB,CAAC,CAAC,MAAM,CAAC,gBAAgB,wBAAC,CAD1B,UAAU,6BAKX;QACD,GAAG;QACH,QAAQ;QACR,IAAI,EAAE,CAAC,CAAiB,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QAClD,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,cAAS,CAAC;KACtC,CAAA;AACZ,CAAC,CAAC,CAAA;AAEJ,MAAM,WAAW,GAAG,GAAG,EAAE,CACvB,CAAC;IACC,IAAI,EAAE,qBAAqB;IAC3B,IAAI,EAAE,QAAQ;CACL,CAAA,CAAA;AAGA,QAAA,YAAY,GAAG,IAAA,SAAG,GAAgB,CAAA;AAClC,QAAA,gBAAgB,GAAG,CAAC,CAAC,YAAY,CAAC,oBAAY,CAAC,CAAC,WAAW,CAAC,CAAA;AAElE,MAAM,IAAI,GAAG,CAAC,KAAuB,EAAE,EAAE,CAC9C,CAAC,CAAC,cAAc,CAAC,oBAAY,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAA;AAD9C,QAAA,IAAI,QAC0C"}

package/DiscordShard/index.ts

@@ -0,0 +1,122 @@
+import * as T from "@effect-ts/core/Effect"
+import * as L from "@effect-ts/core/Effect/Layer"
+import * as M from "@effect-ts/core/Effect/Managed"
+import * as Q from "@effect-ts/core/Effect/Queue"
+import { flow, pipe } from "@effect-ts/core/Function"
+import { tag } from "@effect-ts/core/Has"
+import * as O from "@effect-ts/core/Option"
+import * as CB from "callbag-effect-ts"
+import * as Config from "../DiscordConfig"
+import * as DWS from "../DiscordWS"
+import {
+  GatewayEvent,
+  GatewayOpcode,
+  GatewayPayload,
+  ReadyEvent,
+} from "../types"
+import { Reconnect } from "../WS"
+import * as Heartbeats from "./heartbeats"
+import * as Identify from "./identify"
+import * as Invalid from "./invalidSession"
+import * as Utils from "./utils"
+
+export interface Options {
+  shard: [number, number]
+}
+
+const makeImpl = (shard: [id: number, count: number]) =>
+  M.gen(function* (_) {
+    const token = yield* _(Config.token)
+    const gateway = yield* _(Config.gateway)
+    const outbound = yield* _(Q.makeUnbounded<DWS.Message>())
+
+    const [latestReady, updateLatestReady] = yield* _(
+      Utils.latest(
+        flow(
+          O.fromPredicate(
+            (p): p is GatewayPayload<ReadyEvent> =>
+              p.op === GatewayOpcode.DISPATCH && p.t === "READY",
+          ),
+          O.map((p) => p.d!),
+        ),
+      ),
+    )
+    const [latestSequence, updateLatestSequence] = yield* _(
+      Utils.latest((p) => O.fromNullable(p.s)),
+    )
+
+    const raw = pipe(
+      DWS.open({
+        outgoingQueue: outbound,
+      }),
+      CB.unwrap,
+      CB.share,
+    )
+    const updateRefs = pipe(
+      raw,
+      updateLatestSequence,
+      updateLatestReady,
+      CB.runDrain,
+    )
+
+    const dispatch = pipe(
+      raw,
+      CB.filter(
+        (p): p is GatewayPayload<GatewayEvent> =>
+          p.op === GatewayOpcode.DISPATCH,
+      ),
+      CB.share,
+    )
+
+    // heartbeats
+    const heartbeatEffects = pipe(
+      Heartbeats.fromRaw(raw, latestSequence),
+      CB.forEach((p) => Q.offer_(outbound, p)),
+    )
+
+    // identify
+    const identifyEffects = pipe(
+      Identify.fromRaw(raw, {
+        token,
+        shard,
+        intents: gateway.intents,
+        presence: gateway.presence,
+        latestSequence,
+        latestReady,
+      }),
+      CB.forEach((p) => Q.offer_(outbound, p)),
+    )
+
+    // invalid session
+    const invalidEffects = pipe(
+      Invalid.fromRaw(raw, latestReady),
+      CB.forEach((p) => Q.offer_(outbound, p)),
+    )
+
+    return {
+      run: pipe(
+        updateRefs,
+        T.zipPar(heartbeatEffects),
+        T.zipPar(identifyEffects),
+        T.zipPar(invalidEffects),
+        T.ignore,
+      ),
+      raw,
+      dispatch,
+      send: (p: GatewayPayload) => Q.offer_(outbound, p),
+      reconnect: () => Q.offer_(outbound, Reconnect),
+    } as const
+  })
+
+const makeService = () =>
+  ({
+    _tag: "DiscordShardService",
+    make: makeImpl,
+  } as const)
+
+export interface DiscordShard extends ReturnType<typeof makeService> {}
+export const DiscordShard = tag<DiscordShard>()
+export const LiveDiscordShard = L.fromFunction(DiscordShard)(makeService)
+
+export const make = (shard: [number, number]) =>
+  M.accessServiceM(DiscordShard)(({ make }) => make(shard))

package/DiscordShard/invalidSession.d.ts

@@ -0,0 +1,6 @@
+import * as R from "@effect-ts/core/Effect/Ref";
+import * as O from "@effect-ts/core/Option";
+import * as CB from "callbag-effect-ts";
+import { Message } from "../DiscordWS";
+import { GatewayPayload, ReadyEvent } from "../types";
+export declare const fromRaw: <R, E>(raw: CB.EffectSource<R, E, GatewayPayload<any>>, latestReady: R.Ref<O.Option<ReadyEvent>>) => CB.EffectSource<R, E, Message>;

package/DiscordShard/invalidSession.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fromRaw = void 0;
+const tslib_1 = require("tslib");
+const tracing_1 = tslib_1.__importStar(require("@effect-ts/core/Tracing"));
+const fileName_1 = "DiscordShard/invalidSession.ts";
+const T = tslib_1.__importStar(require("@effect-ts/core/Effect"));
+const R = tslib_1.__importStar(require("@effect-ts/core/Effect/Ref"));
+const Function_1 = require("@effect-ts/core/Function");
+const O = tslib_1.__importStar(require("@effect-ts/core/Option"));
+const CB = tslib_1.__importStar(require("callbag-effect-ts"));
+const types_1 = require("../types");
+const WS_1 = require("../WS");
+const Utils = tslib_1.__importStar(require("./utils"));
+const fromRaw = (raw, latestReady) => (CB.map_(CB.tap_(Utils.opCode(raw)(types_1.GatewayOpcode.INVALID_SESSION), (p) => (p.d ? T.unit : R.set_(latestReady, O.none))), () => WS_1.Reconnect));
+exports.fromRaw = fromRaw;
+//# sourceMappingURL=invalidSession.js.map

package/DiscordShard/invalidSession.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"invalidSession.js","sourceRoot":"","sources":["invalidSession.ts"],"names":[],"mappings":";;;;;;AAAA,kEAA2C;AAC3C,sEAA+C;AAC/C,uDAA+C;AAC/C,kEAA2C;AAC3C,8DAAuC;AAEvC,oCAKiB;AACjB,8BAAiC;AACjC,uDAAgC;AAEzB,MAAM,OAAO,GAAG,CACrB,GAA0C,EAC1C,WAAwC,EACxC,EAAE,EAIA,EAAE,MADF,EAAE,MADF,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAsB,qBAAa,CAAC,eAAe,CAAC,EAC9D,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,GACnD,GAAY,EAAE,CAAC,cAAS,EAChC,CAAA;AARU,QAAA,OAAO,WAQjB"}

package/DiscordShard/invalidSession.ts

@@ -0,0 +1,24 @@
+import * as T from "@effect-ts/core/Effect"
+import * as R from "@effect-ts/core/Effect/Ref"
+import { pipe } from "@effect-ts/core/Function"
+import * as O from "@effect-ts/core/Option"
+import * as CB from "callbag-effect-ts"
+import { Message } from "../DiscordWS"
+import {
+  GatewayOpcode,
+  GatewayPayload,
+  InvalidSessionEvent,
+  ReadyEvent,
+} from "../types"
+import { Reconnect } from "../WS"
+import * as Utils from "./utils"
+
+export const fromRaw = <R, E>(
+  raw: CB.EffectSource<R, E, GatewayPayload>,
+  latestReady: R.Ref<O.Option<ReadyEvent>>,
+) =>
+  pipe(
+    Utils.opCode(raw)<InvalidSessionEvent>(GatewayOpcode.INVALID_SESSION),
+    CB.tap((p) => (p.d ? T.unit : R.set_(latestReady, O.none))),
+    CB.map((): Message => Reconnect),
+  )

package/DiscordShard/utils.d.ts

@@ -0,0 +1,7 @@
+import * as T from "@effect-ts/core/Effect";
+import * as R from "@effect-ts/core/Effect/Ref";
+import * as O from "@effect-ts/core/Option";
+import * as CB from "callbag-effect-ts";
+import { GatewayOpcode, GatewayPayload } from "../types";
+export declare const opCode: <R, E>(source: CB.EffectSource<R, E, GatewayPayload<any>>) => <T = any>(code: GatewayOpcode) => CB.EffectSource<R, E, GatewayPayload<T>>;
+export declare const latest: <T>(f: (p: GatewayPayload) => O.Option<T>) => T.Effect<unknown, never, readonly [R.Ref<O.Option<T>>, <R, E>(self: CB.EffectSource<R, E, GatewayPayload<any>>) => CB.EffectSource<R, E, GatewayPayload<any>>]>;

package/DiscordShard/utils.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.latest = exports.opCode = void 0;
+const tslib_1 = require("tslib");
+const tracing_1 = tslib_1.__importStar(require("@effect-ts/core/Tracing"));
+const fileName_1 = "DiscordShard/utils.ts";
+const T = tslib_1.__importStar(require("@effect-ts/core/Effect"));
+const R = tslib_1.__importStar(require("@effect-ts/core/Effect/Ref"));
+const Function_1 = require("@effect-ts/core/Function");
+const O = tslib_1.__importStar(require("@effect-ts/core/Option"));
+const CB = tslib_1.__importStar(require("callbag-effect-ts"));
+const opCode = (source) => (code) => (CB.filter_(source, (p) => p.op === code));
+exports.opCode = opCode;
+const maybeUpdateRef = (f, ref) => (0, Function_1.flow)(f, O.fold(() => T.unit, (a) => R.set_(ref, O.some(a))));
+const latest = (f) => (T.map_(R.makeRef(O.none), (ref) => [ref, CB.tap(maybeUpdateRef(f, ref))], fileName_1 + ":31:10"));
+exports.latest = latest;
+//# sourceMappingURL=utils.js.map

package/DiscordShard/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["utils.ts"],"names":[],"mappings":";;;;;;AAAA,kEAA2C;AAC3C,sEAA+C;AAC/C,uDAAqD;AACrD,kEAA2C;AAC3C,8DAAuC;AAGhC,MAAM,MAAM,GACjB,CAAO,MAA6C,EAAE,EAAE,CACxD,CAAU,IAAmB,EAAE,EAAE,EAG7B,EAAE,SADF,MAAM,EACI,CAAC,CAAC,EAA0B,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,EACvD,CAAA;AANQ,QAAA,MAAM,UAMd;AAEL,MAAM,cAAc,GAAG,CACrB,CAAqC,EACrC,GAAuB,EACvB,EAAE,CACF,IAAA,eAAI,EACF,CAAC,EACD,CAAC,CAAC,IAAI,CACJ,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,EACZ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAC9B,CACF,CAAA;AAEI,MAAM,MAAM,GAAG,CAAI,CAAqC,EAAE,EAAE,EAG/D,CAAC,MADD,CAAC,CAAC,OAAO,CAAc,CAAC,CAAC,IAAI,CAAC,EACxB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAU,yBAC9D,CAAA;AAJU,QAAA,MAAM,UAIhB"}

package/DiscordShard/utils.ts

@@ -0,0 +1,32 @@
+import * as T from "@effect-ts/core/Effect"
+import * as R from "@effect-ts/core/Effect/Ref"
+import { flow, pipe } from "@effect-ts/core/Function"
+import * as O from "@effect-ts/core/Option"
+import * as CB from "callbag-effect-ts"
+import { GatewayOpcode, GatewayPayload } from "../types"
+
+export const opCode =
+  <R, E>(source: CB.EffectSource<R, E, GatewayPayload>) =>
+  <T = any>(code: GatewayOpcode) =>
+    pipe(
+      source,
+      CB.filter((p): p is GatewayPayload<T> => p.op === code),
+    )
+
+const maybeUpdateRef = <T>(
+  f: (p: GatewayPayload) => O.Option<T>,
+  ref: R.Ref<O.Option<T>>,
+) =>
+  flow(
+    f,
+    O.fold(
+      () => T.unit,
+      (a) => R.set_(ref, O.some(a)),
+    ),
+  )
+
+export const latest = <T>(f: (p: GatewayPayload) => O.Option<T>) =>
+  pipe(
+    R.makeRef<O.Option<T>>(O.none),
+    T.map((ref) => [ref, CB.tap(maybeUpdateRef(f, ref))] as const),
+  )

package/DiscordWS/index.d.ts

@@ -0,0 +1,33 @@
+/// <reference types="node" />
+import * as T from "@effect-ts/core/Effect";
+import * as L from "@effect-ts/core/Effect/Layer";
+import * as Q from "@effect-ts/core/Effect/Queue";
+import * as CB from "callbag-effect-ts";
+import { RawData } from "ws";
+import { GatewayPayload } from "../types";
+import * as WS from "../WS";
+export declare type Message = GatewayPayload | WS.Reconnect;
+export interface OpenOpts {
+    url?: string;
+    version?: number;
+    encoding?: Encoding;
+    outgoingQueue: Q.Queue<Message>;
+}
+export interface Encoding {
+    type: "json" | "etf";
+    encode: (p: GatewayPayload) => string | Buffer | ArrayBuffer;
+    decode: (p: RawData) => GatewayPayload;
+}
+export declare const jsonEncoding: Encoding;
+declare const openImpl: ({ url, version, encoding, outgoingQueue: outgoing, }: OpenOpts) => CB.EffectSource<import("@effect-ts/system/Has").Has<WS.WS> & import("@effect-ts/system/Has").Has<import("../Log").Log> & import("@effect-ts/system/Clock").HasClock, unknown, GatewayPayload<any>>;
+export declare type Connection = ReturnType<typeof openImpl>;
+declare const makeService: () => {
+    readonly _tag: "DiscordWSService";
+    readonly open: ({ url, version, encoding, outgoingQueue: outgoing, }: OpenOpts) => CB.EffectSource<import("@effect-ts/system/Has").Has<WS.WS> & import("@effect-ts/system/Has").Has<import("../Log").Log> & import("@effect-ts/system/Clock").HasClock, unknown, GatewayPayload<any>>;
+};
+export interface DiscordWS extends ReturnType<typeof makeService> {
+}
+export declare const DiscordWS: import("@effect-ts/system/Has").Tag<DiscordWS>;
+export declare const LiveDiscordWS: L.Layer<unknown, never, import("@effect-ts/system/Has").Has<DiscordWS>>;
+export declare const open: (opts: OpenOpts) => T.Effect<import("@effect-ts/system/Has").Has<DiscordWS>, never, CB.EffectSource<import("@effect-ts/system/Has").Has<WS.WS> & import("@effect-ts/system/Has").Has<import("../Log").Log> & import("@effect-ts/system/Clock").HasClock, unknown, GatewayPayload<any>>>;
+export {};

package/DiscordWS/index.js

@@ -1,2 +1,39 @@
 "use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.open = exports.LiveDiscordWS = exports.DiscordWS = exports.jsonEncoding = void 0;
+const tslib_1 = require("tslib");
+const tracing_1 = tslib_1.__importStar(require("@effect-ts/core/Tracing"));
+const fileName_1 = "DiscordWS/index.ts";
+const T = tslib_1.__importStar(require("@effect-ts/core/Effect"));
+const L = tslib_1.__importStar(require("@effect-ts/core/Effect/Layer"));
+const Q = tslib_1.__importStar(require("@effect-ts/core/Effect/Queue"));
+const SC = tslib_1.__importStar(require("@effect-ts/core/Effect/Schedule"));
+const Function_1 = require("@effect-ts/core/Function");
+const Has_1 = require("@effect-ts/core/Has");
+const CB = tslib_1.__importStar(require("callbag-effect-ts"));
+const Log_1 = require("../Log");
+const WS = tslib_1.__importStar(require("../WS"));
+exports.jsonEncoding = {
+    type: "json",
+    encode: (p) => JSON.stringify(p),
+    decode: (p) => JSON.parse(p.toString("utf8")),
+};
+const makeOutgoing = (q, e) => Q.map_(q, (data) => {
+    if (data === WS.Reconnect) {
+        return data;
+    }
+    return e.encode(data);
+});
+const openImpl = ({ url = "wss://gateway.discord.gg/", version = 9, encoding = exports.jsonEncoding, outgoingQueue: outgoing, }) => (CB.map_(CB.retry_(CB.tapError_(CB.unwrap(WS.open(`${url}?v=${version}&encoding=${encoding.type}`, makeOutgoing(outgoing, encoding))), (e) => (0, Log_1.log)(serviceTag, "error", e)), SC.exponential(500)), encoding.decode));
+// Service definition
+const serviceTag = "DiscordWSService";
+const makeService = () => ({
+    _tag: serviceTag,
+    open: openImpl,
+});
+exports.DiscordWS = (0, Has_1.tag)();
+exports.LiveDiscordWS = L.fromFunction(exports.DiscordWS)(makeService);
+// Helpers
+const open = (opts) => T.accessService(exports.DiscordWS)(({ open }) => open(opts), fileName_1 + ":76:29");
+exports.open = open;
 //# sourceMappingURL=index.js.map

package/DiscordWS/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["index.ts"],"names":[],"mappings":""}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["index.ts"],"names":[],"mappings":";;;;;;AAAA,kEAA2C;AAC3C,wEAAiD;AACjD,wEAAiD;AACjD,4EAAqD;AACrD,uDAA+C;AAC/C,6CAAyC;AACzC,8DAAuC;AAEvC,gCAA4B;AAE5B,kDAA2B;AAiBd,QAAA,YAAY,GAAa;IACpC,IAAI,EAAE,MAAM;IACZ,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;IAChC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;CAC9C,CAAA;AAED,MAAM,YAAY,GAAG,CAAC,CAAmB,EAAE,CAAW,EAAoB,EAAE,CAC1E,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,IAAI,EAAc,EAAE;IAC7B,IAAI,IAAI,KAAK,EAAE,CAAC,SAAS,EAAE;QACzB,OAAO,IAAI,CAAA;KACZ;IAED,OAAO,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;AACvB,CAAC,CAAC,CAAA;AAEJ,MAAM,QAAQ,GAAG,CAAC,EAChB,GAAG,GAAG,2BAA2B,EACjC,OAAO,GAAG,CAAC,EACX,QAAQ,GAAG,oBAAY,EACvB,aAAa,EAAE,QAAQ,GACd,EAAE,EAAE,EASX,EAAE,MADF,EAAE,QADF,EAAE,WADF,EAAE,CAAC,MAAM,CAJT,EAAE,CAAC,IAAI,CACL,GAAG,GAAG,MAAM,OAAO,aAAa,QAAQ,CAAC,IAAI,EAAE,EAC/C,YAAY,CAAC,QAAQ,EAAE,QAAQ,CAAC,CACjC,GAEW,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,SAAG,EAAC,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC,GACrC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,GACrB,QAAQ,CAAC,MAAM,EACvB,CAAA;AAIH,qBAAqB;AACrB,MAAM,UAAU,GAAG,kBAA2B,CAAA;AAC9C,MAAM,WAAW,GAAG,GAAG,EAAE,CACvB,CAAC;IACC,IAAI,EAAE,UAAU;IAChB,IAAI,EAAE,QAAQ;CACL,CAAA,CAAA;AAGA,QAAA,SAAS,GAAG,IAAA,SAAG,GAAa,CAAA;AAC5B,QAAA,aAAa,GAAG,CAAC,CAAC,YAAY,CAAC,iBAAS,CAAC,CAAC,WAAW,CAAC,CAAA;AAEnE,UAAU;AACH,MAAM,IAAI,GAAG,CAAC,IAAc,EAAE,EAAE,CACrC,CAAC,CAAC,aAAa,CAAC,iBAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,wBAAC,CAAA;AADzC,QAAA,IAAI,QACqC"}