dfx 1.0.5 → 1.0.6

package/dist/src/DiscordGateway/Shard/identify.ts

@@ -1,50 +0,0 @@
-import * as Option from "effect/Option"
-import * as Effect from "effect/Effect"
-import * as SendEvents from "./sendEvents.ts"
-import type * as Discord from "../../types.ts"
-import type { ShardState } from "./StateStore.ts"
-
-declare const process: { platform?: string | undefined } | undefined
-
-export interface Options {
-  readonly token: string
-  readonly intents: number
-  readonly shard: [number, number]
-  readonly presence?: Discord.GatewayPresenceUpdateData
-}
-
-const identify = ({ intents, presence, shard, token }: Options) =>
-  SendEvents.identify({
-    token,
-    intents,
-    properties: {
-      os:
-        typeof process !== "undefined"
-          ? (process.platform ?? "unknown")
-          : "unknown",
-      browser: "dfx",
-      device: "dfx",
-    },
-    shard,
-    presence,
-  })
-
-const resume = (token: string, session_id: string, seq: number | null) =>
-  SendEvents.resume({
-    token,
-    session_id,
-    seq: seq!,
-  })
-
-export const identifyOrResume = (
-  opts: Options,
-  state: Effect.Effect<Option.Option<ShardState>>,
-): Effect.Effect<Discord.GatewayIdentify | Discord.GatewayResume> =>
-  Effect.map(
-    state,
-    Option.match({
-      onNone: () => identify(opts),
-      onSome: shardState =>
-        resume(opts.token, shardState.sessionId, shardState.sequence),
-    }),
-  )

package/dist/src/DiscordGateway/Shard/sendEvents.ts

@@ -1,43 +0,0 @@
-import * as Discord from "../../types.ts"
-
-export const heartbeat = (
-  d: Discord.GatewayHeartbeatData,
-): Discord.GatewayHeartbeat => ({
-  op: Discord.GatewayOpcodes.Heartbeat,
-  d,
-})
-
-export const identify = (
-  d: Discord.GatewayIdentifyData,
-): Discord.GatewayIdentify => ({
-  op: Discord.GatewayOpcodes.Identify,
-  d,
-})
-
-export const resume = (
-  d: Discord.GatewayResumeData,
-): Discord.GatewayResume => ({
-  op: Discord.GatewayOpcodes.Resume,
-  d,
-})
-
-export const requestGuildMembers = (
-  d: Discord.GatewayRequestGuildMembersData,
-): Discord.GatewayRequestGuildMembers => ({
-  op: Discord.GatewayOpcodes.RequestGuildMembers,
-  d,
-})
-
-export const voiceStateUpdate = (
-  d: Discord.GatewayVoiceStateUpdateData,
-): Discord.GatewayVoiceStateUpdate => ({
-  op: Discord.GatewayOpcodes.VoiceStateUpdate,
-  d,
-})
-
-export const presenceUpdate = (
-  d: Discord.GatewayPresenceUpdateData,
-): Discord.GatewayUpdatePresence => ({
-  op: Discord.GatewayOpcodes.PresenceUpdate,
-  d,
-})

package/dist/src/DiscordGateway/Shard/utils.ts

@@ -1,13 +0,0 @@
-import type * as Discord from "../../types.ts"
-import * as Stream from "effect/Stream"
-
-export const opCode =
-  <R, E>(source: Stream.Stream<Discord.GatewayReceivePayload, E, R>) =>
-  <const Code extends Discord.GatewayOpcodes>(
-    code: Code,
-  ): Stream.Stream<
-    Extract<Discord.GatewayReceivePayload, { readonly op: Code }>,
-    E,
-    R
-  > =>
-    Stream.filter(source, p => p.op === code) as any

package/dist/src/DiscordGateway/Shard.ts

@@ -1,205 +0,0 @@
-import { DiscordConfig } from "../DiscordConfig.ts"
-import type { MessageSend } from "./DiscordWS.ts"
-import { DiscordWS, DiscordWSLive, Reconnect } from "./DiscordWS.ts"
-import { Messaging, MesssagingLive } from "./Messaging.ts"
-import type { ShardState } from "./Shard/StateStore.ts"
-import { ShardStateStore } from "./Shard/StateStore.ts"
-import * as Heartbeats from "./Shard/heartbeats.ts"
-import * as Identify from "./Shard/identify.ts"
-import { RateLimiter, RateLimiterLive } from "../RateLimit.ts"
-import * as Discord from "../types.ts"
-import * as Duration from "effect/Duration"
-import * as Effect from "effect/Effect"
-import * as Layer from "effect/Layer"
-import * as Option from "effect/Option"
-import * as Redacted from "effect/Redacted"
-import type * as Types from "effect/Types"
-import * as FiberHandle from "effect/FiberHandle"
-import { constant, constTrue, constVoid } from "effect/Function"
-import * as Queue from "effect/Queue"
-import * as ServiceMap from "effect/ServiceMap"
-import * as PubSub from "effect/PubSub"
-import * as LogLevel from "effect/LogLevel"
-import { MinimumLogLevel } from "effect/References"
-
-const enum Phase {
-  Connecting,
-  Handshake,
-  Connected,
-}
-
-export const make = Effect.gen(function* () {
-  const { gateway, token } = yield* DiscordConfig
-  const limiter = yield* RateLimiter
-  const dws = yield* DiscordWS
-  const { hub, sendMailbox } = yield* Messaging
-  const shardState = yield* ShardStateStore
-
-  const connect = Effect.fnUntraced(
-    function* (shard: [id: number, count: number]) {
-      const traceEnabled = LogLevel.isLessThanOrEqualTo(
-        yield* MinimumLogLevel,
-        "Trace",
-      )
-      const reconnectHandle = yield* FiberHandle.make()
-      let phase = Phase.Connecting
-      const stateStore = shardState.forShard(shard)
-      const resumeState: Types.Mutable<ShardState> = Option.getOrElse(
-        yield* stateStore.get,
-        () => ({
-          resumeUrl: "",
-          sessionId: "",
-          sequence: 0,
-        }),
-      )
-      const setPhase = (p: Phase): Effect.Effect<void> =>
-        Effect.suspend(() => {
-          phase = p
-          if (!traceEnabled) return Effect.void
-          return Effect.annotateLogs(
-            Effect.logTrace("phase transition"),
-            "phase",
-            p,
-          )
-        })
-
-      const heartbeatSend = (p: MessageSend) =>
-        Effect.suspend(() => {
-          if (phase === Phase.Connecting) return Effect.void
-          return write(p)
-        })
-
-      const resume = Effect.andThen(
-        FiberHandle.clear(reconnectHandle),
-        setPhase(Phase.Connected),
-      )
-
-      const onConnecting = setPhase(Phase.Connecting)
-
-      const socket = yield* dws.connect({ onConnecting })
-      const write = (p: MessageSend) =>
-        Effect.andThen(
-          limiter.maybeWait("dfx.shard.send", Duration.minutes(1), 120),
-          socket.write(p),
-        )
-
-      const hellos = yield* Effect.acquireRelease(
-        Queue.make<Discord.GatewayHelloData>(),
-        Queue.shutdown,
-      )
-      const acks = yield* Effect.acquireRelease(
-        Queue.make<void>(),
-        Queue.shutdown,
-      )
-
-      // heartbeats
-      yield* Heartbeats.send(hellos, acks, stateStore.get, heartbeatSend).pipe(
-        Effect.forkScoped,
-        Effect.interruptible,
-      )
-
-      // identify
-      const identify = Identify.identifyOrResume(
-        {
-          token: Redacted.value(token),
-          shard,
-          intents: gateway.intents,
-          presence: gateway.presence,
-        },
-        stateStore.get,
-      )
-
-      // delayed reconnect
-      const delayedReconnect = Effect.delay(socket.write(Reconnect), 30_000)
-
-      function* onPayload(p: Discord.GatewayReceivePayload) {
-        if (typeof p.s === "number") {
-          resumeState.sequence = p.s
-        }
-        if (p.op === Discord.GatewayOpcodes.Dispatch && p.t === "READY") {
-          const payload = p.d
-          resumeState.sessionId = payload.session_id
-          resumeState.resumeUrl = payload.resume_gateway_url
-          yield* stateStore.set(resumeState)
-          yield* socket.setUrl(payload.resume_gateway_url)
-        } else if (
-          resumeState.resumeUrl !== "" &&
-          resumeState.sessionId !== ""
-        ) {
-          yield* stateStore.set(resumeState)
-        }
-
-        switch (p.op) {
-          case Discord.GatewayOpcodes.Hello: {
-            yield* write(yield* identify)
-            yield* setPhase(Phase.Handshake)
-            Queue.offerUnsafe(hellos, p.d)
-            yield* FiberHandle.run(reconnectHandle, delayedReconnect)
-            return
-          }
-          case Discord.GatewayOpcodes.HeartbeatAck: {
-            Queue.offerUnsafe(acks, void 0)
-            return
-          }
-          case Discord.GatewayOpcodes.InvalidSession: {
-            if (!p.d) {
-              resumeState.sessionId = ""
-              yield* stateStore.clear
-            }
-            yield* socket.write(Reconnect)
-            return
-          }
-          case Discord.GatewayOpcodes.Dispatch: {
-            if (p.t === "READY" || p.t === "RESUMED") {
-              yield* resume
-            }
-            PubSub.publishUnsafe(hub, p)
-            return
-          }
-          case Discord.GatewayOpcodes.Reconnect: {
-            yield* socket.write(Reconnect)
-            return
-          }
-        }
-      }
-
-      yield* Effect.whileLoop({
-        while: constTrue,
-        body: constant(Effect.flatMap(Queue.take(sendMailbox), write)),
-        step: constVoid,
-      }).pipe(Effect.forkScoped)
-
-      yield* Effect.gen(function* () {
-        while (true) {
-          yield* onPayload(yield* socket.take)
-        }
-      }).pipe(Effect.forkScoped)
-
-      return { id: shard, write } as const
-    },
-    (effect, shard) =>
-      Effect.annotateLogs(effect, {
-        package: "dfx",
-        module: "DiscordGateway/Shard",
-        shard,
-      }),
-  )
-
-  return { connect } as const
-})
-
-type ShardService = Effect.Success<typeof make>
-
-export class Shard extends ServiceMap.Service<Shard, ShardService>()(
-  "dfx/DiscordGateway/Shard",
-) {}
-export const ShardLive = Layer.effect(Shard, make).pipe(
-  Layer.provide(DiscordWSLive),
-  Layer.provide(MesssagingLive),
-  Layer.provide(RateLimiterLive),
-)
-
-// eslint-disable-next-line @typescript-eslint/no-empty-object-type
-export interface RunningShard extends Effect.Success<
-  ReturnType<ShardService["connect"]>
-> {}

package/dist/src/DiscordGateway/ShardStore.ts

@@ -1,42 +0,0 @@
-import * as Option from "effect/Option"
-import * as Effect from "effect/Effect"
-import * as Layer from "effect/Layer"
-import * as ServiceMap from "effect/ServiceMap"
-
-export interface ClaimIdContext {
-  sharderCount: number
-  totalCount: number
-}
-
-export interface ShardStoreService {
-  claimId: (ctx: ClaimIdContext) => Effect.Effect<Option.Option<number>>
-  allClaimed: (totalCount: number) => Effect.Effect<boolean>
-  heartbeat?: (shardId: number) => Effect.Effect<void>
-}
-
-export class ShardStore extends ServiceMap.Service<
-  ShardStore,
-  ShardStoreService
->()("dfx/DiscordGateway/ShardStore") {}
-
-// Very basic shard id store, that does no health checks
-const memoryStore = (): ShardStoreService => {
-  let currentId = 0
-
-  return {
-    claimId: ({ totalCount }) =>
-      Effect.sync(() => {
-        if (currentId >= totalCount) {
-          return Option.none()
-        }
-
-        const id = currentId
-        currentId++
-        return Option.some(id)
-      }),
-
-    allClaimed: totalCount => Effect.sync(() => currentId >= totalCount),
-  }
-}
-
-export const MemoryShardStoreLive = Layer.sync(ShardStore, memoryStore)

package/dist/src/DiscordGateway/Sharder.ts

@@ -1,118 +0,0 @@
-import { DiscordConfig } from "../DiscordConfig.ts"
-import type { RunningShard } from "./Shard.ts"
-import { Shard, ShardLive } from "./Shard.ts"
-import { ShardStore } from "./ShardStore.ts"
-import { DiscordREST, DiscordRESTLive } from "../DiscordREST.ts"
-import { RateLimiter, RateLimiterLive } from "../RateLimit.ts"
-import type * as Discord from "../types.ts"
-import * as Duration from "effect/Duration"
-import * as Effect from "effect/Effect"
-import { pipe } from "effect/Function"
-import * as Layer from "effect/Layer"
-import type * as Option from "effect/Option"
-import * as Ref from "effect/Ref"
-import * as Schedule from "effect/Schedule"
-import * as ServiceMap from "effect/ServiceMap"
-
-const claimRepeatPolicy = Schedule.identity<Option.Option<number>>().pipe(
-  Schedule.either(Schedule.spaced("3 minutes")),
-  Schedule.while(_ => Effect.succeed(_.input._tag === "None")),
-  Schedule.passthrough,
-) as Schedule.Schedule<Option.Some<number>, Option.Option<number>>
-
-const make = Effect.gen(function* () {
-  const store = yield* ShardStore
-  const rest = yield* DiscordREST
-  const { gateway: gatewayConfig } = yield* DiscordConfig
-  const limiter = yield* RateLimiter
-  const shard = yield* Shard
-  const currentShards = new Set<RunningShard>()
-
-  const gateway = yield* rest.getBotGateway().pipe(
-    Effect.catch(() =>
-      Effect.succeed<Discord.APIGatewayBotInfo>({
-        url: "wss://gateway.discord.gg/",
-        shards: 1,
-        session_start_limit: {
-          total: 0,
-          remaining: 0,
-          reset_after: 0,
-          max_concurrency: 1,
-        },
-      }),
-    ),
-  )
-
-  const totalCount = gatewayConfig.shardCount ?? gateway.shards
-  const currentCount = yield* Ref.make(0)
-  const claimId = (sharderCount: number): Effect.Effect<number> =>
-    pipe(
-      store.claimId({
-        totalCount,
-        sharderCount,
-      }),
-      Effect.repeat(claimRepeatPolicy),
-      Effect.map(_ => _.value),
-    )
-  const takeConfig = pipe(
-    Ref.getAndUpdate(currentCount, _ => _ + 1),
-    Effect.flatMap(claimId),
-    Effect.map(id => ({ id, totalCount }) as const),
-  )
-
-  const spawner = pipe(
-    takeConfig,
-    Effect.map(shardConfig => ({
-      ...shardConfig,
-      url: gateway.url,
-      concurrency: gateway.session_start_limit.max_concurrency,
-    })),
-    Effect.tap(({ concurrency, id }) =>
-      limiter.maybeWait(
-        `dfx.sharder.${id % concurrency}`,
-        Duration.millis(gatewayConfig.identifyRateLimit[0]),
-        gatewayConfig.identifyRateLimit[1],
-      ),
-    ),
-    Effect.flatMap(c => shard.connect([c.id, c.totalCount])),
-    Effect.tap(runningShard => {
-      currentShards.add(runningShard)
-      return Effect.void
-    }),
-    Effect.forever,
-  )
-
-  yield* Effect.replicateEffect(
-    spawner,
-    gateway.session_start_limit.max_concurrency,
-    { concurrency: "unbounded", discard: true },
-  ).pipe(
-    Effect.scoped,
-    Effect.catchCause(Effect.logError),
-    Effect.ensuring(Effect.sync(() => currentShards.clear())),
-    Effect.forever,
-    Effect.forkScoped,
-  )
-
-  return {
-    shards: Effect.sync(() => currentShards as ReadonlySet<RunningShard>),
-  } as const
-}).pipe(
-  Effect.annotateLogs({
-    package: "dfx",
-    module: "DiscordGateway/Sharder",
-  }),
-)
-
-export class Sharder extends ServiceMap.Service<
-  Sharder,
-  {
-    readonly shards: Effect.Effect<ReadonlySet<RunningShard>, never, never>
-  }
->()("dfx/DiscordGateway/Sharder") {}
-
-export const SharderLive = Layer.effect(Sharder, make).pipe(
-  Layer.provide(DiscordRESTLive),
-  Layer.provide(RateLimiterLive),
-  Layer.provide(ShardLive),
-)

package/dist/src/DiscordGateway.ts

@@ -1,79 +0,0 @@
-import type { WebSocketConstructor } from "effect/unstable/socket/Socket"
-import type { DiscordConfig } from "./DiscordConfig.ts"
-import type { DiscordWSCodec } from "./DiscordGateway/DiscordWS.ts"
-import type { Messsaging } from "./DiscordGateway/Messaging.ts"
-import { Messaging, MesssagingLive } from "./DiscordGateway/Messaging.ts"
-import type { RunningShard } from "./DiscordGateway/Shard.ts"
-import type { ShardStateStore } from "./DiscordGateway/Shard/StateStore.ts"
-import { Sharder, SharderLive } from "./DiscordGateway/Sharder.ts"
-import type { ShardStore } from "./DiscordGateway/ShardStore.ts"
-import type { RateLimitStore } from "./RateLimit.ts"
-import type * as Discord from "./types.ts"
-import * as Effect from "effect/Effect"
-import * as Layer from "effect/Layer"
-import type * as Stream from "effect/Stream"
-import type { HttpClient } from "effect/unstable/http/HttpClient"
-import * as ServiceMap from "effect/ServiceMap"
-
-export const TypeId = Symbol.for("dfx/DiscordGateway")
-export type TypeId = typeof TypeId
-
-export interface DiscordGateway {
-  readonly [TypeId]: TypeId
-
-  readonly dispatch: Stream.Stream<Discord.GatewayReceivePayload>
-  readonly fromDispatch: <K extends `${Discord.GatewayDispatchEvents}`>(
-    event: K,
-  ) => Stream.Stream<
-    Extract<Discord.DistributedGatewayDispatchPayload, { readonly t: K }>["d"]
-  >
-  readonly handleDispatch: <
-    K extends `${Discord.GatewayDispatchEvents}`,
-    R,
-    E,
-    A,
-  >(
-    event: K,
-    handle: (
-      event: Extract<
-        Discord.DistributedGatewayDispatchPayload,
-        { readonly t: K }
-      >["d"],
-    ) => Effect.Effect<A, E, R>,
-  ) => Effect.Effect<never, E, R>
-  readonly send: (payload: Discord.GatewaySendPayload) => Effect.Effect<boolean>
-  readonly shards: Effect.Effect<ReadonlySet<RunningShard>>
-}
-
-export const DiscordGateway =
-  ServiceMap.Service<DiscordGateway>("dfx/DiscordGateway")
-
-export const make: Effect.Effect<DiscordGateway, never, Messsaging | Sharder> =
-  Effect.gen(function* () {
-    const sharder = yield* Sharder
-    const messaging = yield* Messaging
-
-    return DiscordGateway.of({
-      [TypeId]: TypeId,
-      dispatch: messaging.dispatch,
-      fromDispatch: messaging.fromDispatch as any,
-      handleDispatch: messaging.handleDispatch as any,
-      send: messaging.send,
-      shards: sharder.shards,
-    })
-  })
-
-export const DiscordGatewayLive: Layer.Layer<
-  DiscordGateway,
-  never,
-  | ShardStore
-  | DiscordConfig
-  | RateLimitStore
-  | DiscordWSCodec
-  | ShardStateStore
-  | WebSocketConstructor
-  | HttpClient
-> = Layer.effect(DiscordGateway, make).pipe(
-  Layer.provide(MesssagingLive),
-  Layer.provide(SharderLive),
-)