wrangler 0.0.0-bf67bf2 → 0.0.0-bfbb0be

package/src/user/choose-account.tsx

@@ -0,0 +1,60 @@
+import { Text } from "ink";
+import SelectInput from "ink-select-input";
+import React from "react";
+import { fetchListResult } from "../cfetch";
+import { logger } from "../logger";
+import { getCloudflareAccountIdFromEnv } from "./env-vars";
+
+export type ChooseAccountItem = {
+  id: string;
+  name: string;
+};
+
+/**
+ * A component that allows the user to select from a list of available accounts.
+ */
+export function ChooseAccount(props: {
+  accounts: ChooseAccountItem[];
+  onSelect: (accountId: string) => void;
+  onError: (error: Error) => void;
+}) {
+  return (
+    <>
+      <Text bold>Select an account from below:</Text>
+      <SelectInput
+        items={props.accounts.map((item) => ({
+          key: item.id,
+          label: item.name,
+          value: item,
+        }))}
+        onSelect={(item) => {
+          logger.log(`Using account: "${item.value.name} - ${item.value.id}"`);
+          props.onSelect(item.value.id);
+        }}
+      />
+    </>
+  );
+}
+
+/**
+ * Infer a list of available accounts for the current user.
+ */
+export async function getAccountChoices(): Promise<ChooseAccountItem[]> {
+  const accountIdFromEnv = getCloudflareAccountIdFromEnv();
+  if (accountIdFromEnv) {
+    return [{ id: accountIdFromEnv, name: "" }];
+  } else {
+    const response = await fetchListResult<{
+      account: ChooseAccountItem;
+    }>(`/memberships`);
+    const accounts = response.map((r) => r.account);
+    if (accounts.length === 0) {
+      throw new Error(
+        "Failed to automatically retrieve account IDs for the logged in user.\n" +
+          "In a non-interactive environment, it is mandatory to specify an account ID, either by assigning its value to CLOUDFLARE_ACCOUNT_ID, or as `account_id` in your `wrangler.toml` file."
+      );
+    } else {
+      return accounts;
+    }
+  }
+}

package/src/user/env-vars.ts

@@ -0,0 +1,17 @@
+import { getEnvironmentVariableFactory } from "../environment-variables";
+
+/**
+ * Try to read the API token from the environment.
+ */
+export const getCloudflareAPITokenFromEnv = getEnvironmentVariableFactory({
+  variableName: "CLOUDFLARE_API_TOKEN",
+  deprecatedName: "CF_API_TOKEN",
+});
+
+/**
+ * Try to read the account ID from the environment.
+ */
+export const getCloudflareAccountIdFromEnv = getEnvironmentVariableFactory({
+  variableName: "CLOUDFLARE_ACCOUNT_ID",
+  deprecatedName: "CF_ACCOUNT_ID",
+});

package/src/user/generate-auth-url.ts

@@ -0,0 +1,33 @@
+interface GenerateAuthUrlProps {
+  authUrl: string;
+  clientId: string;
+  callbackUrl: string;
+  scopes: string[];
+  stateQueryParam: string;
+  codeChallenge: string;
+}
+
+/**
+ * generateAuthUrl was extracted from getAuthURL in user.tsx
+ * to make it possible to mock the generated URL
+ */
+export const generateAuthUrl = ({
+  authUrl,
+  clientId,
+  callbackUrl,
+  scopes,
+  stateQueryParam,
+  codeChallenge,
+}: GenerateAuthUrlProps) => {
+  return (
+    authUrl +
+    `?response_type=code&` +
+    `client_id=${encodeURIComponent(clientId)}&` +
+    `redirect_uri=${encodeURIComponent(callbackUrl)}&` +
+    // we add offline_access manually for every request
+    `scope=${encodeURIComponent([...scopes, "offline_access"].join(" "))}&` +
+    `state=${stateQueryParam}&` +
+    `code_challenge=${encodeURIComponent(codeChallenge)}&` +
+    `code_challenge_method=S256`
+  );
+};

package/src/user/generate-random-state.ts

@@ -0,0 +1,16 @@
+import { webcrypto as crypto } from "node:crypto";
+import { PKCE_CHARSET } from "../user";
+
+/**
+ * Generates random state to be passed for anti-csrf.
+ * extracted from  user.tsx to make it possible to
+ * mock the generated URL
+ */
+export function generateRandomState(lengthOfState: number): string {
+  const output = new Uint32Array(lengthOfState);
+  // @ts-expect-error crypto's types aren't there yet
+  crypto.getRandomValues(output);
+  return Array.from(output)
+    .map((num: number) => PKCE_CHARSET[num % PKCE_CHARSET.length])
+    .join("");
+}

package/src/user/index.ts

@@ -0,0 +1,3 @@
+export * from "./user";
+export * from "./env-vars";
+export * from "./choose-account";