refactor: extract loader setup wrapper

Author: ndycode

index.ts

@@ -211,6 +211,7 @@ import { applyAccountStorageScopeEntry } from "./lib/runtime/account-storage-sco
 import { runBrowserOAuthFlow } from "./lib/runtime/browser-oauth-flow.js";
 import { handleRuntimeEvent } from "./lib/runtime/event-handler.js";
 import { ensureLiveAccountSyncEntry } from "./lib/runtime/live-sync-entry.js";
+import { applyLoaderRuntimeSetup } from "./lib/runtime/loader-setup.js";
 import { buildManualOAuthFlow } from "./lib/runtime/manual-oauth-flow.js";
 import {
 	applyPreemptiveQuotaSettingsFromConfig,
@@ -691,11 +692,15 @@ export const OpenAIOAuthPlugin: Plugin = async ({ client }: PluginInput) => {
 			async loader(getAuth: () => Promise<Auth>, provider: unknown) {
 				const auth = await getAuth();
 				const pluginConfig = loadPluginConfig();
-				applyUiRuntimeFromConfig(pluginConfig, setUiRuntimeOptions);
-				applyAccountStorageScope(pluginConfig);
-				ensureSessionAffinity(pluginConfig);
-				ensureRefreshGuardian(pluginConfig);
-				applyPreemptiveQuotaSettings(pluginConfig);
+				applyLoaderRuntimeSetup({
+					pluginConfig,
+					applyUiRuntimeFromConfig: (config) =>
+						applyUiRuntimeFromConfig(config, setUiRuntimeOptions),
+					applyAccountStorageScope,
+					ensureSessionAffinity,
+					ensureRefreshGuardian,
+					applyPreemptiveQuotaSettings,
+				});
 
 				// Only handle OAuth auth type, skip API key auth
 				if (auth.type !== "oauth") {

lib/runtime/loader-setup.ts

@@ -0,0 +1,24 @@
+export function applyLoaderRuntimeSetup(params: {
+	pluginConfig: ReturnType<typeof import("../config.js").loadPluginConfig>;
+	applyUiRuntimeFromConfig: (
+		pluginConfig: ReturnType<typeof import("../config.js").loadPluginConfig>,
+	) => void;
+	applyAccountStorageScope: (
+		pluginConfig: ReturnType<typeof import("../config.js").loadPluginConfig>,
+	) => void;
+	ensureSessionAffinity: (
+		pluginConfig: ReturnType<typeof import("../config.js").loadPluginConfig>,
+	) => void;
+	ensureRefreshGuardian: (
+		pluginConfig: ReturnType<typeof import("../config.js").loadPluginConfig>,
+	) => void;
+	applyPreemptiveQuotaSettings: (
+		pluginConfig: ReturnType<typeof import("../config.js").loadPluginConfig>,
+	) => void;
+}): void {
+	params.applyUiRuntimeFromConfig(params.pluginConfig);
+	params.applyAccountStorageScope(params.pluginConfig);
+	params.ensureSessionAffinity(params.pluginConfig);
+	params.ensureRefreshGuardian(params.pluginConfig);
+	params.applyPreemptiveQuotaSettings(params.pluginConfig);
+}

test/loader-setup.test.ts

@@ -0,0 +1,30 @@
+import { describe, expect, it, vi } from "vitest";
+import { applyLoaderRuntimeSetup } from "../lib/runtime/loader-setup.js";
+
+describe("loader runtime setup", () => {
+	it("applies runtime setup steps in the existing order", () => {
+		const calls: string[] = [];
+		const pluginConfig = { a: 1 } as never;
+
+		applyLoaderRuntimeSetup({
+			pluginConfig,
+			applyUiRuntimeFromConfig: () => {
+				calls.push("ui");
+			},
+			applyAccountStorageScope: () => {
+				calls.push("scope");
+			},
+			ensureSessionAffinity: () => {
+				calls.push("session");
+			},
+			ensureRefreshGuardian: () => {
+				calls.push("guardian");
+			},
+			applyPreemptiveQuotaSettings: () => {
+				calls.push("quota");
+			},
+		});
+
+		expect(calls).toEqual(["ui", "scope", "session", "guardian", "quota"]);
+	});
+});