270 lines · 8.9 KB · TypeScript
RawDownload
#!/usr/bin/env node import * as NodeRuntime from "@effect/platform-node/NodeRuntime";import { AdoptPolicy } from "alchemy/AdoptPolicy";import { AlchemyContext, AlchemyContextLive } from "alchemy/AlchemyContext";import * as Apply from "alchemy/Apply";import { provideFreshArtifactStore } from "alchemy/Artifacts";import { AuthProviders } from "alchemy/Auth/AuthProvider";import { CredentialsStoreLive } from "alchemy/Auth/Credentials";import { ProfileLive } from "alchemy/Auth/Profile";import { Cli } from "alchemy/Cli/Cli";import { LoggingCli } from "alchemy/Cli/LoggingCli";import * as Plan from "alchemy/Plan";import * as Stage from "alchemy/Stage";import { TelemetryLive } from "alchemy/Telemetry/Layer";import { PlatformServices } from "alchemy/Util/PlatformServices";import * as Config from "effect/Config";import * as ConfigProvider from "effect/ConfigProvider";import * as Console from "effect/Console";import * as Data from "effect/Data";import * as Effect from "effect/Effect";import * as FileSystem from "effect/FileSystem";import * as Layer from "effect/Layer";import * as Option from "effect/Option";import * as Path from "effect/Path";import { Command, Flag, Prompt } from "effect/unstable/cli";import * as FetchHttpClient from "effect/unstable/http/FetchHttpClient"; import RelayStack from "../alchemy.run.ts"; export class RelayDeployError extends Data.TaggedError("RelayDeployError")<{  readonly message: string;}> {} export interface RelayDeployOptions {  readonly dryRun: boolean;  readonly force: boolean;  readonly envFile: Option.Option<string>;  readonly stage: Option.Option<string>;  readonly yes: boolean;  readonly adopt: boolean;  readonly githubOutput: boolean;} export function reconcileRootEnvRelayUrl(contents: string, relayUrl: string): string {  const entry = `T3CODE_RELAY_URL=${relayUrl}`;  if (/^T3CODE_RELAY_URL=.*$/mu.test(contents)) {    return contents.replace(/^T3CODE_RELAY_URL=.*$/mu, entry);  }  if (!contents) {    return `${entry}\n`;  }  return `${contents}${contents.endsWith("\n") ? "" : "\n"}${entry}\n`;} export function hasDeployChanges(plan: Plan.Plan): boolean {  return (    Object.keys(plan.deletions).length > 0 ||    Object.values(plan.resources).some(      (node) =>        node.action !== "noop" || node.bindings.some((binding) => binding.action !== "noop"),    )  );} export type RelayDeployResult = "applied" | "noop" | "dry-run" | "cancelled"; export interface RelayDeployOutcome {  readonly result: RelayDeployResult;  readonly changed: boolean;  readonly relayUrl: Option.Option<string>;} export function serializeGithubOutput(entries: Readonly<Record<string, string | boolean>>): string {  return Object.entries(entries)    .map(([key, value]) => `${key}=${value}\n`)    .join("");} const relayRoot = Effect.service(Path.Path).pipe(  Effect.flatMap((path) => path.fromFileUrl(new URL("..", import.meta.url))),);const repoRoot = Effect.service(Path.Path).pipe(  Effect.flatMap((path) => path.fromFileUrl(new URL("../../..", import.meta.url))),); const loadDeployConfigProvider = Effect.fn("relay.deploy.loadConfigProvider")(function* (  envFileOverride: Option.Option<string>,) {  const path = yield* Path.Path;  const root = yield* relayRoot;   if (Option.isSome(envFileOverride)) {    return yield* ConfigProvider.fromDotEnv({ path: path.resolve(root, envFileOverride.value) });  }   return yield* ConfigProvider.fromDotEnv({ path: path.join(root, ".env") }).pipe(    Effect.orElseSucceed(() => ConfigProvider.fromEnv()),  );}); const relayDeployStage = Config.nonEmptyString("stage").pipe(  Config.option,  Config.map(    Option.getOrElse(() => `dev_${process.env.USER ?? process.env.USERNAME ?? "unknown"}`),  ),); const reconcileRootEnv = Effect.fn("relay.deploy.reconcileRootEnv")(function* (relayUrl: string) {  const fs = yield* FileSystem.FileSystem;  const path = yield* Path.Path;  const root = yield* repoRoot;  const rootEnvPath = path.join(root, ".env");  const contents = (yield* fs.exists(rootEnvPath)) ? yield* fs.readFileString(rootEnvPath) : "";   yield* fs.writeFileString(rootEnvPath, reconcileRootEnvRelayUrl(contents, relayUrl));  yield* Console.log(`Updated ${rootEnvPath} with T3CODE_RELAY_URL=${relayUrl}`);}); const writeGithubOutput = Effect.fn("relay.deploy.writeGithubOutput")(function* (  outcome: RelayDeployOutcome,) {  const fs = yield* FileSystem.FileSystem;  const githubOutputPath = yield* Config.nonEmptyString("GITHUB_OUTPUT");  yield* fs.writeFileString(    githubOutputPath,    serializeGithubOutput({      changed: outcome.changed,      result: outcome.result,      ...(Option.isSome(outcome.relayUrl) ? { relay_url: outcome.relayUrl.value } : {}),    }),    { flag: "a" },  );}); const deployServices = Layer.mergeAll(  Layer.provideMerge(AlchemyContextLive, PlatformServices),  Layer.provide(ProfileLive, PlatformServices),  Layer.provide(CredentialsStoreLive, PlatformServices),  FetchHttpClient.layer,  TelemetryLive,  LoggingCli,); const runRelayDeploy = Effect.fn("relay.deploy.run")(  function* (    options: RelayDeployOptions,    _configProvider: ConfigProvider.ConfigProvider,    _stage: string,  ) {    const stack = yield* RelayStack;     const cli = yield* Cli;    const plan = yield* Plan.make(stack, { force: options.force }).pipe(      Effect.provide(stack.services),    );    const changed = hasDeployChanges(plan);    if (options.dryRun) {      yield* cli.displayPlan(plan);      return {        result: "dry-run",        changed,        relayUrl: Option.none<string>(),      } satisfies RelayDeployOutcome;    }    if (!options.yes && changed) {      yield* cli.displayPlan(plan);      const approved = yield* Prompt.run(        Prompt.confirm({          message: "Apply this relay deployment?",        }),      );      if (!approved) {        yield* Console.log("Deployment cancelled.");        return {          result: "cancelled",          changed,          relayUrl: Option.none<string>(),        } satisfies RelayDeployOutcome;      }    }    const output = yield* Apply.apply(plan).pipe(Effect.provide(stack.services));    if (output.url === undefined) {      return yield* new RelayDeployError({        message: "Alchemy relay deploy output did not include a URL",      });    }    return {      result: changed ? "applied" : "noop",      changed,      relayUrl: Option.some(output.url),    } satisfies RelayDeployOutcome;  },  (effect, options, configProvider, stage) =>    effect.pipe(      Effect.provide(        Layer.mergeAll(          Layer.effect(            AlchemyContext,            AlchemyContext.pipe(Effect.map((context) => ({ ...context, adopt: options.adopt }))),          ),          Layer.succeed(AdoptPolicy, options.adopt),          Layer.succeed(AuthProviders, {}),          ConfigProvider.layer(configProvider),          Layer.succeed(Stage.Stage, stage),        ),      ),      provideFreshArtifactStore,    ),); export const deploy = Effect.fn("relay.deploy")(function* (options: RelayDeployOptions) {  const configProvider = yield* loadDeployConfigProvider(options.envFile);  const configuredStage = yield* relayDeployStage.pipe(    Effect.provide(ConfigProvider.layer(configProvider)),  );  const stage = Option.getOrElse(options.stage, () => configuredStage);  const outcome = yield* runRelayDeploy(options, configProvider, stage);  if (Option.isSome(outcome.relayUrl)) {    yield* reconcileRootEnv(outcome.relayUrl.value);  }  if (options.githubOutput) {    yield* writeGithubOutput(outcome);  }}); export const relayDeployCommand = Command.make(  "relay-deploy",  {    dryRun: Flag.boolean("dry-run").pipe(      Flag.withDescription("Dry run the deployment without applying changes."),      Flag.withDefault(false),    ),    force: Flag.boolean("force").pipe(      Flag.withDescription("Force updates for resources that would otherwise no-op."),      Flag.withDefault(false),    ),    envFile: Flag.string("env-file").pipe(      Flag.withDescription(        "Environment file to load. Defaults to infra/relay/.env with process env fallback.",      ),      Flag.optional,    ),    stage: Flag.string("stage").pipe(      Flag.withDescription("Stage to deploy. Defaults to dev_${USER}."),      Flag.optional,    ),    yes: Flag.boolean("yes").pipe(      Flag.withDescription("Skip the deployment confirmation prompt."),      Flag.withDefault(false),    ),    adopt: Flag.boolean("adopt").pipe(      Flag.withDescription("Adopt pre-existing cloud resources that conflict with this stack."),      Flag.withDefault(false),    ),    githubOutput: Flag.boolean("github-output").pipe(      Flag.withDescription("Append relay deployment metadata to GITHUB_OUTPUT."),      Flag.withDefault(false),    ),  },  deploy,).pipe(Command.withDescription("Deploy the T3 Code relay through Alchemy.")); if (import.meta.main) {  Command.run(relayDeployCommand, { version: "0.0.0" }).pipe(    Effect.provide(deployServices),    Effect.scoped,    NodeRuntime.runMain,  );}