Extract values from endpoints to prepareHeaders function

[User6531]

Hi everyone,

I have a question regarding drilling a property from a query or mutation to the prepareHeaders function.

I have two types of tokens, and I need to add them to the request only under specific conditions.

Currently, I have two solutions. The first one involves creating a condition inside the prepareHeaders function based on the endpoint name. However, this approach is unsafe because the name could be changed, and I cannot keep it as a variable.

The second solution is to pass a custom header from the endpoint and retrieve it in the prepareHeaders function. Then, a condition can be made based on this header value. This approach is safer as I can use a variable, but it still seems like a suboptimal solution.

So, the main question is: Is there a better way to extract a property from the endpoint to the prepareHeaders function?

solution1:

export const api = createApi({
  baseQuery: fetchBaseQuery({
    baseUrl: '/',
    prepareHeaders: (headers,api) => {
    const { endpoint } = api;

    if (endpoint === 'getPosts') {
      headers.set("authorization", `Bearer ${getToken1()}`);
    } else {
      headers.set("authorization", `Bearer ${getToken2()}`);
    }

    return headers;
    },
  }),
  endpoints: (builder) => ({
    getPosts: builder.query<UserResponse, LoginRequest>({
      query: (credentials) => ({
        url: 'posts',
      }),
    }),
  }),
})

solution2:

export const api = createApi({
  baseQuery: fetchBaseQuery({
    baseUrl: '/',
    prepareHeaders: (headers,api) => {
    const tokenType = headers.get('tokenType')

    if (tokenType === 'type1') {
      headers.set("authorization", `Bearer ${getToken1()}`);
    } else {
      headers.set("authorization", `Bearer ${getToken2()}`);
    }

    return headers;
    },
  }),
  endpoints: (builder) => ({
    getPosts: builder.query<UserResponse, LoginRequest>({
      query: (credentials) => ({
        url: 'posts',
        headers: {
         tokenType : 'type1',
        },
      }),
    }),
  }),
})

[User6531]

@EskiMojo14 @markerikson, can u please help me with this?