mirror of
https://github.com/unraid/api.git
synced 2026-01-04 15:39:52 -06:00
111 lines
3.1 KiB
TypeScript
111 lines
3.1 KiB
TypeScript
import {
|
|
ApolloClient,
|
|
ApolloLink,
|
|
createHttpLink,
|
|
from,
|
|
Observable,
|
|
split,
|
|
} from '@apollo/client/core/index.js';
|
|
import { onError } from '@apollo/client/link/error/index.js';
|
|
import { RetryLink } from '@apollo/client/link/retry/index.js';
|
|
import { GraphQLWsLink } from '@apollo/client/link/subscriptions/index.js';
|
|
import { getMainDefinition } from '@apollo/client/utilities/index.js';
|
|
import { provideApolloClient } from '@vue/apollo-composable';
|
|
import { useServerStore } from '~/store/server';
|
|
import { createClient } from 'graphql-ws';
|
|
import { createApolloCache } from './apollo-cache';
|
|
import { WEBGUI_GRAPHQL } from './urls';
|
|
|
|
const httpEndpoint = WEBGUI_GRAPHQL;
|
|
const wsEndpoint = new URL(WEBGUI_GRAPHQL.toString().replace('http', 'ws'));
|
|
|
|
// const headers = { 'x-api-key': serverStore.apiKey };
|
|
const headers = {};
|
|
|
|
const httpLink = createHttpLink({
|
|
uri: httpEndpoint.toString(),
|
|
headers,
|
|
credentials: 'include',
|
|
});
|
|
|
|
const wsLink = new GraphQLWsLink(
|
|
createClient({
|
|
url: wsEndpoint.toString(),
|
|
connectionParams: () => ({
|
|
headers,
|
|
}),
|
|
})
|
|
);
|
|
|
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
const errorLink = onError(({ graphQLErrors, networkError }: any) => {
|
|
if (graphQLErrors) {
|
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
graphQLErrors.map((error: any) => {
|
|
console.error('[GraphQL error]', error);
|
|
const errorMsg = error.error?.message ?? error.message;
|
|
if (errorMsg?.includes('offline')) {
|
|
// @todo restart the api
|
|
}
|
|
return error.message;
|
|
});
|
|
}
|
|
|
|
if (networkError) {
|
|
console.error(`[Network error]: ${networkError}`);
|
|
const msg = networkError.message ? networkError.message : networkError;
|
|
if (typeof msg === 'string' && msg.includes('Unexpected token < in JSON at position 0')) {
|
|
return 'Unraid API • CORS Error';
|
|
}
|
|
return msg;
|
|
}
|
|
});
|
|
|
|
const retryLink = new RetryLink({
|
|
attempts: {
|
|
max: 20,
|
|
retryIf: (error, _operation) => {
|
|
return Boolean(error);
|
|
},
|
|
},
|
|
delay: {
|
|
initial: 300,
|
|
max: 10000,
|
|
jitter: true,
|
|
},
|
|
});
|
|
|
|
const disableClientLink = new ApolloLink((operation, forward) => {
|
|
const serverStore = useServerStore();
|
|
const { connectPluginInstalled } = toRefs(serverStore);
|
|
if (!connectPluginInstalled?.value) {
|
|
return new Observable((observer) => {
|
|
console.warn('connectPluginInstalled is false, aborting request');
|
|
observer.complete();
|
|
});
|
|
}
|
|
return forward(operation);
|
|
});
|
|
|
|
const splitLinks = split(
|
|
({ query }) => {
|
|
const definition = getMainDefinition(query);
|
|
return definition.kind === 'OperationDefinition' && definition.operation === 'subscription';
|
|
},
|
|
wsLink,
|
|
httpLink
|
|
);
|
|
/**
|
|
* @todo as we add retries, determine which we'll need
|
|
* https://www.apollographql.com/docs/react/api/link/introduction/#additive-composition
|
|
* https://www.apollographql.com/docs/react/api/link/introduction/#directional-composition
|
|
*/
|
|
const additiveLink = from([disableClientLink, errorLink, retryLink, splitLinks]);
|
|
|
|
export const client = new ApolloClient({
|
|
link: additiveLink,
|
|
cache: createApolloCache(),
|
|
});
|
|
|
|
provideApolloClient(client);
|