alchemy 2.0.0-beta.45 → 2.0.0-beta.46

package/src/Cloudflare/index.ts

@@ -18,6 +18,7 @@ export * from "./R2/index.ts";
 export * from "./SecretsStore/index.ts";
 export * from "./StateStore/index.ts";
 export * from "./Tunnel/index.ts";
+export * from "./Vectorize/index.ts";
 export * from "./VpcService/index.ts";
 export * from "./Website/index.ts";
 export * from "./Workers/index.ts";

package/src/State/HttpStateStore.ts

@@ -39,6 +39,34 @@ export interface HttpStateStoreProps extends HttpStateStoreCredentials {
   ) => HttpClientRequest.HttpClientRequest;
 }
 
+export const checkHttpStateStoreAuth = ({
+  url,
+  authToken,
+}: {
+  url: string;
+  authToken: string;
+}) =>
+  Effect.gen(function* () {
+    const apiClient = yield* HttpApiClient.make(StateApi, {
+      baseUrl: url,
+      transformClient: HttpClient.mapRequest((req) =>
+        req.pipe(HttpClientRequest.bearerToken(authToken)),
+      ),
+    });
+    return yield* apiClient.state.listStacks().pipe(
+      Effect.map(() => true),
+      Effect.catchTag("Unauthorized", () => Effect.succeed(false)),
+      Effect.retry({
+        while: (error) =>
+          error._tag === "HttpClientError" &&
+          !!error.response &&
+          // worker can 404 for a bit on first deploy
+          (error.response.status === 404 || error.response.status >= 500),
+        schedule: Schedule.fixed(200),
+      }),
+    );
+  });
+
 export const makeHttpStateStore = ({
   url,
   authToken,

package/src/Util/poll.ts

@@ -0,0 +1,47 @@
+import * as Data from "effect/Data";
+import * as Effect from "effect/Effect";
+import * as Predicate from "effect/Predicate";
+import * as Schedule from "effect/Schedule";
+
+export class PredicateFailed extends Data.TaggedError("PredicateFailed")<{
+  message: string;
+  actual: unknown;
+}> {}
+
+export const isPredicateFailed = (e: unknown): e is PredicateFailed =>
+  Predicate.isTagged(e, "PredicateFailed");
+
+/**
+ * Retries an effect until a predicate is met.
+ * @param input - The input to the poll function.
+ * @param input.description - The description of what is being polled; used in the error message if the predicate fails.
+ * @param input.effect - The effect to execute until the predicate is met.
+ * @param input.predicate - The predicate to check if the effect has met the desired state.
+ * @param input.schedule - The schedule to use for retries; defaults to every 3 seconds.
+ * @param input.times - The maximum number of times to poll; defaults to 50.
+ * @returns The value that satisfies the predicate.
+ */
+export const poll = Effect.fn("poll")(
+  <A, E, R>(input: {
+    description?: string;
+    effect: Effect.Effect<A, E, R>;
+    predicate: (value: A) => boolean;
+    schedule?: Schedule.Schedule<unknown, unknown, never>;
+    times?: number;
+  }) =>
+    input.effect.pipe(
+      Effect.filterOrFail(
+        input.predicate,
+        (actual) =>
+          new PredicateFailed({
+            message: `Predicate failed: ${input.description ?? "<no description>"}`,
+            actual,
+          }),
+      ),
+      Effect.retry({
+        while: isPredicateFailed,
+        schedule: input.schedule ?? Schedule.spaced("3 seconds"),
+        times: input.times ?? 50,
+      }),
+    ),
+);