68 lines
1.9 KiB
TypeScript
68 lines
1.9 KiB
TypeScript
import { TRPCError } from "@trpc/server";
|
|
|
|
import { createLogger } from "@homarr/core/infrastructure/logs";
|
|
import { ErrorWithMetadata } from "@homarr/core/infrastructure/logs/error";
|
|
|
|
const logger = createLogger({ module: "settleIntegrations" });
|
|
|
|
export const PUBLIC_INTEGRATION_ERROR: string = "INTEGRATION_REQUEST_FAILED";
|
|
|
|
export const toPublicIntegrationError = (_error: unknown): string => PUBLIC_INTEGRATION_ERROR;
|
|
|
|
interface IntegrationLike {
|
|
id: string;
|
|
name: string;
|
|
kind: string;
|
|
}
|
|
|
|
interface Options<TIntegration extends IntegrationLike, TResult> {
|
|
fallback?: (integration: TIntegration, error: unknown) => TResult;
|
|
throwOnAllFailures?: boolean;
|
|
}
|
|
|
|
export async function settleIntegrationQueries<TIntegration extends IntegrationLike, TResult>(
|
|
integrations: TIntegration[],
|
|
fn: (integration: TIntegration) => Promise<TResult>,
|
|
options?: Options<TIntegration, TResult>,
|
|
): Promise<TResult[]> {
|
|
const settled = await Promise.allSettled(integrations.map(async (integration) => fn(integration)));
|
|
const results: TResult[] = [];
|
|
const errors: unknown[] = [];
|
|
|
|
settled.forEach((result, index) => {
|
|
if (result.status === "fulfilled") {
|
|
results.push(result.value);
|
|
return;
|
|
}
|
|
|
|
const integration = integrations[index];
|
|
logger.warn(
|
|
new ErrorWithMetadata(
|
|
"Integration query failed",
|
|
{ integrationId: integration?.id, integrationKind: integration?.kind },
|
|
{ cause: result.reason },
|
|
),
|
|
);
|
|
|
|
errors.push(result.reason);
|
|
|
|
if (options?.fallback && integration) {
|
|
results.push(options.fallback(integration, result.reason));
|
|
return;
|
|
}
|
|
});
|
|
|
|
if (
|
|
errors.length > 0 &&
|
|
(results.length === 0 || (options?.throwOnAllFailures === true && errors.length === integrations.length))
|
|
) {
|
|
throw new TRPCError({
|
|
code: "INTERNAL_SERVER_ERROR",
|
|
message: "All integration queries failed",
|
|
cause: errors[0],
|
|
});
|
|
}
|
|
|
|
return results;
|
|
}
|