@trpc/react-query 11.0.0-alpha-tmp-issues-5851-take-two.499 → 11.0.0-alpha-tmp-subscription-connection-state.485

package/dist/shared/hooks/types.mjs

@@ -0,0 +1,126 @@
+const defaultIdleResult = {
+    data: null,
+    connectionStartedAt: 0,
+    initialConnectionStartedAt: 0,
+    error: null,
+    errorUpdatedAt: 0,
+    connectionError: null,
+    connectionAttemptCount: 0,
+    isStarting: false,
+    isStarted: false,
+    isConnecting: false,
+    isPending: false,
+    isReconnecting: false,
+    isError: false,
+    connectionErrorUpdatedAt: 0,
+    status: 'idle'
+};
+const getIdleResult = ()=>{
+    return {
+        ...defaultIdleResult,
+        data: null
+    };
+};
+const getStartingResult = (// restart: restartSubscriptionFn<TInput>,
+previous, error)=>{
+    const now = Date.now();
+    if (previous) {
+        return {
+            ...defaultIdleResult,
+            ...previous,
+            data: null,
+            connectionError: error ?? null,
+            isStarting: true,
+            isConnecting: true,
+            errorUpdatedAt: 0,
+            connectionAttemptCount: previous.connectionAttemptCount + 1,
+            connectionErrorUpdatedAt: error ? now : 0,
+            connectionStartedAt: 0,
+            initialConnectionStartedAt: 0,
+            error: null,
+            isStarted: false,
+            isError: false,
+            status: 'connecting'
+        };
+    }
+    return {
+        ...getIdleResult(),
+        connectionError: error ?? null,
+        isStarting: true,
+        isConnecting: true,
+        errorUpdatedAt: 0,
+        connectionAttemptCount: 0,
+        connectionErrorUpdatedAt: error ? now : 0,
+        status: 'connecting'
+    };
+};
+const getPendingResult = (previous, data)=>{
+    const time = Date.now();
+    if (previous.isStarting) {
+        return {
+            ...previous,
+            error: null,
+            isStarting: false,
+            isStarted: true,
+            isConnecting: false,
+            isReconnecting: false,
+            isPending: true,
+            status: 'pending',
+            connectionStartedAt: time,
+            connectionAttemptCount: 0,
+            connectionError: null,
+            initialConnectionStartedAt: time
+        };
+    }
+    return {
+        ...previous,
+        data: data ?? previous.data,
+        error: null,
+        isError: false,
+        isStarting: false,
+        isStarted: true,
+        isConnecting: false,
+        isReconnecting: false,
+        isPending: true,
+        status: 'pending',
+        connectionStartedAt: time,
+        connectionAttemptCount: 0,
+        connectionError: null
+    };
+};
+const getReconnectingResult = (previous, error)=>{
+    return {
+        ...previous,
+        isStarting: false,
+        isStarted: true,
+        isConnecting: true,
+        isReconnecting: true,
+        isPending: false,
+        status: 'connecting',
+        connectionError: error,
+        connectionAttemptCount: previous.connectionAttemptCount + 1
+    };
+};
+const getConnectingResult = (previous, error)=>{
+    if (previous.isReconnecting || previous.isPending) {
+        if (!error) throw new Error('Reconnecting without error?');
+        return getReconnectingResult(previous, error);
+    }
+    return getStartingResult(/*previous.restart, */ previous, error);
+};
+const getErrorResult = (previous, error)=>{
+    return {
+        ...previous,
+        isStarting: false,
+        isStarted: true,
+        isConnecting: false,
+        isReconnecting: false,
+        isPending: false,
+        isError: true,
+        status: 'error',
+        error,
+        errorUpdatedAt: Date.now()
+    };
+};
+
+export { getConnectingResult, getErrorResult, getIdleResult, getPendingResult, getReconnectingResult, getStartingResult };

package/dist/shared/index.js

@@ -5,6 +5,7 @@ var utilsProxy = require('./proxy/utilsProxy.js');
 var useQueriesProxy = require('./proxy/useQueriesProxy.js');
 var createHooksInternal = require('./hooks/createHooksInternal.js');
 var queryClient = require('./queryClient.js');
+var types = require('./hooks/types.js');
 var getClientArgs = require('../internals/getClientArgs.js');
 var context = require('../internals/context.js');
 
@@ -17,6 +18,12 @@ exports.getQueryType = utilsProxy.getQueryType;
 exports.createUseQueries = useQueriesProxy.createUseQueries;
 exports.createRootHooks = createHooksInternal.createRootHooks;
 exports.getQueryClient = queryClient.getQueryClient;
+exports.getConnectingResult = types.getConnectingResult;
+exports.getErrorResult = types.getErrorResult;
+exports.getIdleResult = types.getIdleResult;
+exports.getPendingResult = types.getPendingResult;
+exports.getReconnectingResult = types.getReconnectingResult;
+exports.getStartingResult = types.getStartingResult;
 exports.getClientArgs = getClientArgs.getClientArgs;
 exports.TRPCContext = context.TRPCContext;
 exports.contextProps = context.contextProps;

package/dist/shared/index.mjs

@@ -3,5 +3,6 @@ export { createQueryUtilsProxy, createReactQueryUtils, getQueryType } from './pr
 export { createUseQueries } from './proxy/useQueriesProxy.mjs';
 export { createRootHooks } from './hooks/createHooksInternal.mjs';
 export { getQueryClient } from './queryClient.mjs';
+export { getConnectingResult, getErrorResult, getIdleResult, getPendingResult, getReconnectingResult, getStartingResult } from './hooks/types.mjs';
 export { getClientArgs } from '../internals/getClientArgs.mjs';
 export { TRPCContext, contextProps } from '../internals/context.mjs';

package/dist/shared/proxy/utilsProxy.d.ts

@@ -1,10 +1,11 @@
 import type { CancelOptions, InfiniteData, InvalidateOptions, InvalidateQueryFilters, Query, QueryFilters, QueryKey, RefetchOptions, RefetchQueryFilters, ResetOptions, SetDataOptions, Updater } from '@tanstack/react-query';
 import type { TRPCClientError } from '@trpc/client';
-import type { AnyQueryProcedure, AnyRootTypes, AnyRouter, DeepPartial, inferProcedureInput, inferTransformedProcedureOutput, ProtectedIntersection, RouterRecord } from '@trpc/server/unstable-core-do-not-import';
+import type { AnyMutationProcedure, AnyQueryProcedure, AnyRootTypes, AnyRouter, DeepPartial, inferProcedureInput, inferTransformedProcedureOutput, ProtectedIntersection, RouterRecord } from '@trpc/server/unstable-core-do-not-import';
 import type { DecoratedTRPCContextProps, TRPCContextState, TRPCFetchInfiniteQueryOptions, TRPCFetchQueryOptions, TRPCQueryUtils } from '../../internals/context';
 import type { QueryKeyKnown, QueryType } from '../../internals/getQueryKey';
+import type { InferMutationOptions } from '../../utils/inferReactQueryProcedure';
 import type { ExtractCursorType } from '../hooks/types';
-type DecorateProcedure<TRoot extends AnyRootTypes, TProcedure extends AnyQueryProcedure> = {
+type DecorateQueryProcedure<TRoot extends AnyRootTypes, TProcedure extends AnyQueryProcedure> = {
     /**
      * @link https://tanstack.com/query/v5/docs/reference/QueryClient#queryclientfetchquery
      */
@@ -74,6 +75,13 @@ type DecorateProcedure<TRoot extends AnyRootTypes, TProcedure extends AnyQueryPr
      */
     getInfiniteData(input?: inferProcedureInput<TProcedure>): InfiniteData<inferTransformedProcedureOutput<TRoot, TProcedure>, NonNullable<ExtractCursorType<inferProcedureInput<TProcedure>>> | null> | undefined;
 };
+type DecorateMutationProcedure<TRoot extends AnyRootTypes, TProcedure extends AnyMutationProcedure> = {
+    setMutationDefaults(options: InferMutationOptions<TRoot, TProcedure> | ((args: {
+        canonicalMutationFn: NonNullable<InferMutationOptions<TRoot, TProcedure>['mutationFn']>;
+    }) => InferMutationOptions<TRoot, TProcedure>)): void;
+    getMutationDefaults(): InferMutationOptions<TRoot, TProcedure> | undefined;
+    isMutating(): number;
+};
 /**
  * this is the type that is used to add in procedures that can be used on
  * an entire router
@@ -90,9 +98,9 @@ type DecorateRouter = {
  * @internal
  */
 export type DecoratedProcedureUtilsRecord<TRoot extends AnyRootTypes, TRecord extends RouterRecord> = DecorateRouter & {
-    [TKey in keyof TRecord]: TRecord[TKey] extends infer $Value ? $Value extends RouterRecord ? DecoratedProcedureUtilsRecord<TRoot, $Value> & DecorateRouter : $Value extends AnyQueryProcedure ? DecorateProcedure<TRoot, $Value> : never : never;
+    [TKey in keyof TRecord]: TRecord[TKey] extends infer $Value ? $Value extends RouterRecord ? DecoratedProcedureUtilsRecord<TRoot, $Value> & DecorateRouter : $Value extends AnyQueryProcedure ? DecorateQueryProcedure<TRoot, $Value> : $Value extends AnyMutationProcedure ? DecorateMutationProcedure<TRoot, $Value> : never : never;
 };
-type AnyDecoratedProcedure = DecorateProcedure<any, any>;
+type AnyDecoratedProcedure = DecorateQueryProcedure<any, any> & DecorateMutationProcedure<any, any>;
 export type CreateReactUtils<TRouter extends AnyRouter, TSSRContext> = ProtectedIntersection<DecoratedTRPCContextProps<TRouter, TSSRContext>, DecoratedProcedureUtilsRecord<TRouter['_def']['_config']['$types'], TRouter['_def']['record']>>;
 export type CreateQueryUtils<TRouter extends AnyRouter> = DecoratedProcedureUtilsRecord<TRouter['_def']['_config']['$types'], TRouter['_def']['record']>;
 export declare const getQueryType: (utilName: keyof AnyDecoratedProcedure) => QueryType;

package/dist/shared/proxy/utilsProxy.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utilsProxy.d.ts","sourceRoot":"","sources":["../../../src/shared/proxy/utilsProxy.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,aAAa,EACb,YAAY,EACZ,iBAAiB,EACjB,sBAAsB,EACtB,KAAK,EACL,YAAY,EACZ,QAAQ,EACR,cAAc,EACd,mBAAmB,EACnB,YAAY,EACZ,cAAc,EACd,OAAO,EACR,MAAM,uBAAuB,CAAC;AAC/B,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAEpD,OAAO,KAAK,EACV,iBAAiB,EACjB,YAAY,EACZ,SAAS,EACT,WAAW,EACX,mBAAmB,EACnB,+BAA+B,EAC/B,qBAAqB,EACrB,YAAY,EACb,MAAM,0CAA0C,CAAC;AAKlD,OAAO,KAAK,EACV,yBAAyB,EACzB,gBAAgB,EAChB,6BAA6B,EAC7B,qBAAqB,EACrB,cAAc,EACf,MAAM,yBAAyB,CAAC;AAEjC,OAAO,KAAK,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAE5E,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAExD,KAAK,iBAAiB,CACpB,KAAK,SAAS,YAAY,EAC1B,UAAU,SAAS,iBAAiB,IAClC;IACF;;OAEG;IACH,KAAK,CACH,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,CAAC,EAAE,qBAAqB,CAC1B,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,eAAe,CAAC,KAAK,CAAC,CACvB,GACA,OAAO,CAAC,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IAE/D;;OAEG;IACH,aAAa,CACX,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,CAAC,EAAE,6BAA6B,CAClC,mBAAmB,CAAC,UAAU,CAAC,EAC/B,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,eAAe,CAAC,KAAK,CAAC,CACvB,GACA,OAAO,CACR,YAAY,CACV,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,WAAW,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,CACvE,CACF,CAAC;IAEF;;OAEG;IACH,QAAQ,CACN,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,CAAC,EAAE,qBAAqB,CAC1B,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,eAAe,CAAC,KAAK,CAAC,CACvB,GACA,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,gBAAgB,CACd,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,CAAC,EAAE,6BAA6B,CAClC,mBAAmB,CAAC,UAAU,CAAC,EAC/B,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,eAAe,CAAC,KAAK,CAAC,CACvB,GACA,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,UAAU,CACR,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,CAAC,EAAE,qBAAqB,CAC1B,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,eAAe,CAAC,KAAK,CAAC,CACvB,GACA,OAAO,CAAC,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IAE/D;;OAEG;IACH,UAAU,CACR,KAAK,CAAC,EAAE,WAAW,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,EACpD,OAAO,CAAC,EAAE,IAAI,CAAC,sBAAsB,EAAE,WAAW,CAAC,GAAG;QACpD,SAAS,CAAC,EAAE,CACV,KAAK,EAAE,KAAK,CACV,mBAAmB,CAAC,UAAU,CAAC,EAC/B,eAAe,CAAC,KAAK,CAAC,EACtB,mBAAmB,CAAC,UAAU,CAAC,EAC/B,aAAa,CACX,mBAAmB,CAAC,UAAU,CAAC,EAC/B,mBAAmB,CAAC,UAAU,CAAC,SAAS;YAAE,MAAM,CAAC,EAAE,GAAG,CAAA;SAAE,GAAG,IAAI,GAC3D,UAAU,GACV,OAAO,CACZ,CACF,KACE,OAAO,CAAC;KACd,EACD,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,OAAO,CACL,KAAK,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACvC,OAAO,CAAC,EAAE,mBAAmB,EAC7B,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,MAAM,CACJ,KAAK,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACvC,OAAO,CAAC,EAAE,aAAa,GACtB,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,KAAK,CACH,KAAK,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACvC,OAAO,CAAC,EAAE,YAAY,GACrB,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,OAAO;IACL;;OAEG;IACH,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,OAAO,EAAE,OAAO,CACd,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,GAAG,SAAS,EAC9D,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,GAAG,SAAS,CAC/D,EACD,OAAO,CAAC,EAAE,cAAc,GACvB,IAAI,CAAC;IAER;;OAEG;IACH,cAAc;IACZ;;OAEG;IACH,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,OAAO,EAAE,YAAY,EACrB,OAAO,EAAE,OAAO,CACd,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,GAAG,SAAS,EAC9D,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,GAAG,SAAS,CAC/D,EACD,OAAO,CAAC,EAAE,cAAc,GACvB,CAAC,QAAQ,EAAE,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IAElE;;OAEG;IACH,eAAe,CACb,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,OAAO,EAAE,OAAO,CACZ,YAAY,CACV,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,WAAW,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,CACvE,GACD,SAAS,EACT,YAAY,CACV,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,WAAW,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,CACvE,GACD,SAAS,CACZ,EACD,OAAO,CAAC,EAAE,cAAc,GACvB,IAAI,CAAC;IAER;;OAEG;IACH,OAAO,CACL,KAAK,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,GACtC,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,GAAG,SAAS,CAAC;IAElE;;OAEG;IACH,eAAe,CACb,KAAK,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,GAErC,YAAY,CACV,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,WAAW,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,CACvE,GACD,SAAS,CAAC;CACf,CAAC;AAEF;;;GAGG;AACH,KAAK,cAAc,GAAG;IACpB;;;;OAIG;IACH,UAAU,CACR,KAAK,CAAC,EAAE,SAAS,EACjB,OAAO,CAAC,EAAE,sBAAsB,EAChC,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,IAAI,CAAC,CAAC;CAClB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,6BAA6B,CACvC,KAAK,SAAS,YAAY,EAC1B,OAAO,SAAS,YAAY,IAC1B,cAAc,GAAG;KAClB,IAAI,IAAI,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,SAAS,MAAM,MAAM,GACvD,MAAM,SAAS,YAAY,GACzB,6BAA6B,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,cAAc,GAE/D,MAAM,SAAS,iBAAiB,GAC9B,iBAAiB,CAAC,KAAK,EAAE,MAAM,CAAC,GAChC,KAAK,GACP,KAAK;CACV,CAAC;AAEF,KAAK,qBAAqB,GAAG,iBAAiB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAEzD,MAAM,MAAM,gBAAgB,CAC1B,OAAO,SAAS,SAAS,EACzB,WAAW,IACT,qBAAqB,CACvB,yBAAyB,CAAC,OAAO,EAAE,WAAW,CAAC,EAC/C,6BAA6B,CAC3B,OAAO,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,EACpC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAC1B,CACF,CAAC;AAEF,MAAM,MAAM,gBAAgB,CAAC,OAAO,SAAS,SAAS,IACpD,6BAA6B,CAC3B,OAAO,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,EACpC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAC1B,CAAC;AAEJ,eAAO,MAAM,YAAY,aACb,MAAM,qBAAqB,KACpC,SAsBF,CAAC;AA4CF;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,OAAO,SAAS,SAAS,EAAE,WAAW,EAC1E,OAAO,EAAE,gBAAgB,CAAC,SAAS,EAAE,WAAW,CAAC,0KAqBlD;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,OAAO,SAAS,SAAS,EAC7D,OAAO,EAAE,cAAc,CAAC,OAAO,CAAC,GAC/B,gBAAgB,CAAC,OAAO,CAAC,CAE3B"}
+{"version":3,"file":"utilsProxy.d.ts","sourceRoot":"","sources":["../../../src/shared/proxy/utilsProxy.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,aAAa,EACb,YAAY,EACZ,iBAAiB,EACjB,sBAAsB,EACtB,KAAK,EACL,YAAY,EACZ,QAAQ,EACR,cAAc,EACd,mBAAmB,EACnB,YAAY,EACZ,cAAc,EACd,OAAO,EACR,MAAM,uBAAuB,CAAC;AAC/B,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAEpD,OAAO,KAAK,EACV,oBAAoB,EACpB,iBAAiB,EACjB,YAAY,EACZ,SAAS,EACT,WAAW,EACX,mBAAmB,EACnB,+BAA+B,EAC/B,qBAAqB,EACrB,YAAY,EACb,MAAM,0CAA0C,CAAC;AAKlD,OAAO,KAAK,EACV,yBAAyB,EACzB,gBAAgB,EAChB,6BAA6B,EAC7B,qBAAqB,EACrB,cAAc,EACf,MAAM,yBAAyB,CAAC;AAEjC,OAAO,KAAK,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAK5E,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,sCAAsC,CAAC;AACjF,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAExD,KAAK,sBAAsB,CACzB,KAAK,SAAS,YAAY,EAC1B,UAAU,SAAS,iBAAiB,IAClC;IACF;;OAEG;IACH,KAAK,CACH,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,CAAC,EAAE,qBAAqB,CAC1B,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,eAAe,CAAC,KAAK,CAAC,CACvB,GACA,OAAO,CAAC,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IAE/D;;OAEG;IACH,aAAa,CACX,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,CAAC,EAAE,6BAA6B,CAClC,mBAAmB,CAAC,UAAU,CAAC,EAC/B,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,eAAe,CAAC,KAAK,CAAC,CACvB,GACA,OAAO,CACR,YAAY,CACV,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,WAAW,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,CACvE,CACF,CAAC;IAEF;;OAEG;IACH,QAAQ,CACN,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,CAAC,EAAE,qBAAqB,CAC1B,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,eAAe,CAAC,KAAK,CAAC,CACvB,GACA,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,gBAAgB,CACd,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,CAAC,EAAE,6BAA6B,CAClC,mBAAmB,CAAC,UAAU,CAAC,EAC/B,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,eAAe,CAAC,KAAK,CAAC,CACvB,GACA,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,UAAU,CACR,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,CAAC,EAAE,qBAAqB,CAC1B,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,eAAe,CAAC,KAAK,CAAC,CACvB,GACA,OAAO,CAAC,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IAE/D;;OAEG;IACH,UAAU,CACR,KAAK,CAAC,EAAE,WAAW,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,EACpD,OAAO,CAAC,EAAE,IAAI,CAAC,sBAAsB,EAAE,WAAW,CAAC,GAAG;QACpD,SAAS,CAAC,EAAE,CACV,KAAK,EAAE,KAAK,CACV,mBAAmB,CAAC,UAAU,CAAC,EAC/B,eAAe,CAAC,KAAK,CAAC,EACtB,mBAAmB,CAAC,UAAU,CAAC,EAC/B,aAAa,CACX,mBAAmB,CAAC,UAAU,CAAC,EAC/B,mBAAmB,CAAC,UAAU,CAAC,SAAS;YAAE,MAAM,CAAC,EAAE,GAAG,CAAA;SAAE,GAAG,IAAI,GAC3D,UAAU,GACV,OAAO,CACZ,CACF,KACE,OAAO,CAAC;KACd,EACD,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,OAAO,CACL,KAAK,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACvC,OAAO,CAAC,EAAE,mBAAmB,EAC7B,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,MAAM,CACJ,KAAK,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACvC,OAAO,CAAC,EAAE,aAAa,GACtB,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,KAAK,CACH,KAAK,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACvC,OAAO,CAAC,EAAE,YAAY,GACrB,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjB;;OAEG;IACH,OAAO;IACL;;OAEG;IACH,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,OAAO,EAAE,OAAO,CACd,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,GAAG,SAAS,EAC9D,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,GAAG,SAAS,CAC/D,EACD,OAAO,CAAC,EAAE,cAAc,GACvB,IAAI,CAAC;IAER;;OAEG;IACH,cAAc;IACZ;;OAEG;IACH,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,OAAO,EAAE,YAAY,EACrB,OAAO,EAAE,OAAO,CACd,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,GAAG,SAAS,EAC9D,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,GAAG,SAAS,CAC/D,EACD,OAAO,CAAC,EAAE,cAAc,GACvB,CAAC,QAAQ,EAAE,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IAElE;;OAEG;IACH,eAAe,CACb,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,OAAO,EAAE,OAAO,CACZ,YAAY,CACV,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,WAAW,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,CACvE,GACD,SAAS,EACT,YAAY,CACV,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,WAAW,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,CACvE,GACD,SAAS,CACZ,EACD,OAAO,CAAC,EAAE,cAAc,GACvB,IAAI,CAAC;IAER;;OAEG;IACH,OAAO,CACL,KAAK,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,GACtC,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,GAAG,SAAS,CAAC;IAElE;;OAEG;IACH,eAAe,CACb,KAAK,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,GAErC,YAAY,CACV,+BAA+B,CAAC,KAAK,EAAE,UAAU,CAAC,EAClD,WAAW,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,CACvE,GACD,SAAS,CAAC;CACf,CAAC;AAEF,KAAK,yBAAyB,CAC5B,KAAK,SAAS,YAAY,EAC1B,UAAU,SAAS,oBAAoB,IACrC;IACF,mBAAmB,CACjB,OAAO,EACH,oBAAoB,CAAC,KAAK,EAAE,UAAU,CAAC,GACvC,CAAC,CAAC,IAAI,EAAE;QACN,mBAAmB,EAAE,WAAW,CAC9B,oBAAoB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,YAAY,CAAC,CACtD,CAAC;KACH,KAAK,oBAAoB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,GACjD,IAAI,CAAC;IAER,mBAAmB,IAAI,oBAAoB,CAAC,KAAK,EAAE,UAAU,CAAC,GAAG,SAAS,CAAC;IAE3E,UAAU,IAAI,MAAM,CAAC;CACtB,CAAC;AAEF;;;GAGG;AACH,KAAK,cAAc,GAAG;IACpB;;;;OAIG;IACH,UAAU,CACR,KAAK,CAAC,EAAE,SAAS,EACjB,OAAO,CAAC,EAAE,sBAAsB,EAChC,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,IAAI,CAAC,CAAC;CAClB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,6BAA6B,CACvC,KAAK,SAAS,YAAY,EAC1B,OAAO,SAAS,YAAY,IAC1B,cAAc,GAAG;KAClB,IAAI,IAAI,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,SAAS,MAAM,MAAM,GACvD,MAAM,SAAS,YAAY,GACzB,6BAA6B,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,cAAc,GAC7D,MAAM,SAAS,iBAAiB,GAChC,sBAAsB,CAAC,KAAK,EAAE,MAAM,CAAC,GACrC,MAAM,SAAS,oBAAoB,GACnC,yBAAyB,CAAC,KAAK,EAAE,MAAM,CAAC,GACxC,KAAK,GACP,KAAK;CACV,CAAC;AAEF,KAAK,qBAAqB,GAAG,sBAAsB,CAAC,GAAG,EAAE,GAAG,CAAC,GAC3D,yBAAyB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAEtC,MAAM,MAAM,gBAAgB,CAC1B,OAAO,SAAS,SAAS,EACzB,WAAW,IACT,qBAAqB,CACvB,yBAAyB,CAAC,OAAO,EAAE,WAAW,CAAC,EAC/C,6BAA6B,CAC3B,OAAO,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,EACpC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAC1B,CACF,CAAC;AAEF,MAAM,MAAM,gBAAgB,CAAC,OAAO,SAAS,SAAS,IACpD,6BAA6B,CAC3B,OAAO,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,EACpC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,CAC1B,CAAC;AAEJ,eAAO,MAAM,YAAY,aACb,MAAM,qBAAqB,KACpC,SAyBF,CAAC;AAwDF;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,OAAO,SAAS,SAAS,EAAE,WAAW,EAC1E,OAAO,EAAE,gBAAgB,CAAC,SAAS,EAAE,WAAW,CAAC,0KAqBlD;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,OAAO,SAAS,SAAS,EAC7D,OAAO,EAAE,cAAc,CAAC,OAAO,CAAC,GAC/B,gBAAgB,CAAC,OAAO,CAAC,CAE3B"}

package/dist/shared/proxy/utilsProxy.js

@@ -19,6 +19,9 @@ const getQueryType = (utilName)=>{
         case 'getInfiniteData':
         case 'setInfiniteData':
             return 'infinite';
+        case 'setMutationDefaults':
+        case 'getMutationDefaults':
+        case 'isMutating':
         case 'cancel':
         case 'invalidate':
         case 'refetch':
@@ -41,7 +44,9 @@ const getQueryType = (utilName)=>{
         const queryType = getQueryType(utilName);
         const queryKey = getQueryKey.getQueryKeyInternal(path, input, queryType);
         const contextMap = {
-            fetch: ()=>context.fetchQuery(queryKey, ...args),
+            /**
+       * DecorateQueryProcedure
+       */ fetch: ()=>context.fetchQuery(queryKey, ...args),
             fetchInfinite: ()=>context.fetchInfiniteQuery(queryKey, args[0]),
             prefetch: ()=>context.prefetchQuery(queryKey, ...args),
             prefetchInfinite: ()=>context.prefetchInfiniteQuery(queryKey, args[0]),
@@ -58,7 +63,14 @@ const getQueryType = (utilName)=>{
                 context.setInfiniteQueryData(queryKey, args[0], args[1]);
             },
             getData: ()=>context.getQueryData(queryKey),
-            getInfiniteData: ()=>context.getInfiniteQueryData(queryKey)
+            getInfiniteData: ()=>context.getInfiniteQueryData(queryKey),
+            /**
+       * DecorateMutationProcedure
+       */ setMutationDefaults: ()=>context.setMutationDefaults(getQueryKey.getMutationKeyInternal(path), input),
+            getMutationDefaults: ()=>context.getMutationDefaults(getQueryKey.getMutationKeyInternal(path)),
+            isMutating: ()=>context.isMutating({
+                    mutationKey: getQueryKey.getMutationKeyInternal(path)
+                })
         };
         return contextMap[utilName]();
     });

package/dist/shared/proxy/utilsProxy.mjs

@@ -1,7 +1,7 @@
 import { createTRPCClientProxy } from '@trpc/client';
 import { createFlatProxy, createRecursiveProxy } from '@trpc/server/unstable-core-do-not-import';
 import { contextProps } from '../../internals/context.mjs';
-import { getQueryKeyInternal } from '../../internals/getQueryKey.mjs';
+import { getQueryKeyInternal, getMutationKeyInternal } from '../../internals/getQueryKey.mjs';
 
 const getQueryType = (utilName)=>{
     switch(utilName){
@@ -17,6 +17,9 @@ const getQueryType = (utilName)=>{
         case 'getInfiniteData':
         case 'setInfiniteData':
             return 'infinite';
+        case 'setMutationDefaults':
+        case 'getMutationDefaults':
+        case 'isMutating':
         case 'cancel':
         case 'invalidate':
         case 'refetch':
@@ -39,7 +42,9 @@ const getQueryType = (utilName)=>{
         const queryType = getQueryType(utilName);
         const queryKey = getQueryKeyInternal(path, input, queryType);
         const contextMap = {
-            fetch: ()=>context.fetchQuery(queryKey, ...args),
+            /**
+       * DecorateQueryProcedure
+       */ fetch: ()=>context.fetchQuery(queryKey, ...args),
             fetchInfinite: ()=>context.fetchInfiniteQuery(queryKey, args[0]),
             prefetch: ()=>context.prefetchQuery(queryKey, ...args),
             prefetchInfinite: ()=>context.prefetchInfiniteQuery(queryKey, args[0]),
@@ -56,7 +61,14 @@ const getQueryType = (utilName)=>{
                 context.setInfiniteQueryData(queryKey, args[0], args[1]);
             },
             getData: ()=>context.getQueryData(queryKey),
-            getInfiniteData: ()=>context.getInfiniteQueryData(queryKey)
+            getInfiniteData: ()=>context.getInfiniteQueryData(queryKey),
+            /**
+       * DecorateMutationProcedure
+       */ setMutationDefaults: ()=>context.setMutationDefaults(getMutationKeyInternal(path), input),
+            getMutationDefaults: ()=>context.getMutationDefaults(getMutationKeyInternal(path)),
+            isMutating: ()=>context.isMutating({
+                    mutationKey: getMutationKeyInternal(path)
+                })
         };
         return contextMap[utilName]();
     });

package/dist/utils/createUtilityFunctions.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"createUtilityFunctions.d.ts","sourceRoot":"","sources":["../../src/utils/createUtilityFunctions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AACrD,OAAO,EAAoB,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACnE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,0CAA0C,CAAC;AAE1E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAEhD,MAAM,WAAW,uBAAuB,CAAC,OAAO,SAAS,SAAS;IAChE;;OAEG;IACH,MAAM,EAAE,gBAAgB,CAAC,OAAO,CAAC,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAC/D;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC;CAC1B;AAED;;;;;GAKG;AACH,wBAAgB,sBAAsB,CAAC,OAAO,SAAS,SAAS,EAC9D,IAAI,EAAE,uBAAuB,CAAC,OAAO,CAAC,GACrC,cAAc,CAAC,OAAO,CAAC,CA0HzB"}
+{"version":3,"file":"createUtilityFunctions.d.ts","sourceRoot":"","sources":["../../src/utils/createUtilityFunctions.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AACrD,OAAO,EAAoB,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACnE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,0CAA0C,CAAC;AAE1E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAEhD,MAAM,WAAW,uBAAuB,CAAC,OAAO,SAAS,SAAS;IAChE;;OAEG;IACH,MAAM,EAAE,gBAAgB,CAAC,OAAO,CAAC,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAC/D;;OAEG;IACH,WAAW,EAAE,WAAW,CAAC;CAC1B;AAED;;;;;GAKG;AACH,wBAAgB,sBAAsB,CAAC,OAAO,SAAS,SAAS,EAC9D,IAAI,EAAE,uBAAuB,CAAC,OAAO,CAAC,GACrC,cAAc,CAAC,OAAO,CAAC,CAoJzB"}

package/dist/utils/createUtilityFunctions.js

@@ -100,6 +100,29 @@ var getClientArgs = require('../internals/getClientArgs.js');
         },
         getInfiniteQueryData: (queryKey)=>{
             return queryClient.getQueryData(queryKey);
+        },
+        setMutationDefaults: (mutationKey, options)=>{
+            const path = mutationKey[0];
+            const canonicalMutationFn = (input)=>{
+                return untypedClient.mutation(...getClientArgs.getClientArgs([
+                    path,
+                    {
+                        input
+                    }
+                ], opts));
+            };
+            return queryClient.setMutationDefaults(mutationKey, typeof options === 'function' ? options({
+                canonicalMutationFn
+            }) : options);
+        },
+        getMutationDefaults: (mutationKey)=>{
+            return queryClient.getMutationDefaults(mutationKey);
+        },
+        isMutating: (filters)=>{
+            return queryClient.isMutating({
+                ...filters,
+                exact: true
+            });
         }
     };
 }

package/dist/utils/createUtilityFunctions.mjs

@@ -98,6 +98,29 @@ import { getClientArgs } from '../internals/getClientArgs.mjs';
         },
         getInfiniteQueryData: (queryKey)=>{
             return queryClient.getQueryData(queryKey);
+        },
+        setMutationDefaults: (mutationKey, options)=>{
+            const path = mutationKey[0];
+            const canonicalMutationFn = (input)=>{
+                return untypedClient.mutation(...getClientArgs([
+                    path,
+                    {
+                        input
+                    }
+                ], opts));
+            };
+            return queryClient.setMutationDefaults(mutationKey, typeof options === 'function' ? options({
+                canonicalMutationFn
+            }) : options);
+        },
+        getMutationDefaults: (mutationKey)=>{
+            return queryClient.getMutationDefaults(mutationKey);
+        },
+        isMutating: (filters)=>{
+            return queryClient.isMutating({
+                ...filters,
+                exact: true
+            });
         }
     };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trpc/react-query",
-  "version": "11.0.0-alpha-tmp-issues-5851-take-two.499+02e677611",
+  "version": "11.0.0-alpha-tmp-subscription-connection-state.485+386eae02c",
   "description": "The tRPC React library",
   "author": "KATT",
   "license": "MIT",
@@ -64,15 +64,15 @@
   },
   "peerDependencies": {
     "@tanstack/react-query": "^5.49.2",
-    "@trpc/client": "11.0.0-alpha-tmp-issues-5851-take-two.499+02e677611",
-    "@trpc/server": "11.0.0-alpha-tmp-issues-5851-take-two.499+02e677611",
+    "@trpc/client": "11.0.0-alpha-tmp-subscription-connection-state.485+386eae02c",
+    "@trpc/server": "11.0.0-alpha-tmp-subscription-connection-state.485+386eae02c",
     "react": ">=18.2.0",
     "react-dom": ">=18.2.0"
   },
   "devDependencies": {
     "@tanstack/react-query": "^5.49.2",
-    "@trpc/client": "11.0.0-alpha-tmp-issues-5851-take-two.499+02e677611",
-    "@trpc/server": "11.0.0-alpha-tmp-issues-5851-take-two.499+02e677611",
+    "@trpc/client": "11.0.0-alpha-tmp-subscription-connection-state.485+386eae02c",
+    "@trpc/server": "11.0.0-alpha-tmp-subscription-connection-state.485+386eae02c",
     "@types/express": "^4.17.17",
     "@types/node": "^20.10.0",
     "@types/react": "^18.3.1",
@@ -92,5 +92,5 @@
   "funding": [
     "https://trpc.io/sponsor"
   ],
-  "gitHead": "02e67761124670c5ff27d174bfec6cfa5410a5ac"
+  "gitHead": "386eae02c675ba0fe0feda78188b6975e56d0e49"
 }

package/src/createTRPCReact.tsx

@@ -32,6 +32,7 @@ import type {
   UseTRPCQueryOptions,
   UseTRPCQueryResult,
   UseTRPCSubscriptionOptions,
+  UseTRPCSubscriptionResult,
   UseTRPCSuspenseInfiniteQueryOptions,
   UseTRPCSuspenseInfiniteQueryResult,
   UseTRPCSuspenseQueryOptions,
@@ -180,7 +181,11 @@ interface ProcedureUseSubscription<TDef extends ResolverDef> {
   (
     input: TDef['input'],
     opts: UseTRPCSubscriptionOptions<TDef['output'], TRPCClientErrorLike<TDef>>,
-  ): void;
+  ): UseTRPCSubscriptionResult<
+    TDef['input'],
+    TDef['output'],
+    TRPCClientErrorLike<TDef>
+  >;
 
   // With skip token
   (
@@ -189,7 +194,11 @@ interface ProcedureUseSubscription<TDef extends ResolverDef> {
       UseTRPCSubscriptionOptions<TDef['output'], TRPCClientErrorLike<TDef>>,
       'enabled'
     >,
-  ): void;
+  ): UseTRPCSubscriptionResult<
+    TDef['input'],
+    TDef['output'],
+    TRPCClientErrorLike<TDef>
+  >;
 }
 /**
  * @internal

package/src/index.ts

@@ -1,6 +1,6 @@
 export * from '@trpc/client';
 
-export { getQueryKey } from './internals/getQueryKey';
+export { getQueryKey, getMutationKey } from './internals/getQueryKey';
 export {
   createTRPCReact,
   type CreateTRPCReact,

package/src/internals/context.tsx

@@ -5,6 +5,7 @@ import type {
   InfiniteData,
   InvalidateOptions,
   InvalidateQueryFilters,
+  MutationOptions,
   QueryClient,
   QueryFilters,
   QueryKey,
@@ -26,7 +27,7 @@ import type {
 } from '@trpc/server/unstable-core-do-not-import';
 import * as React from 'react';
 import type { ExtractCursorType } from '../shared';
-import type { TRPCQueryKey } from './getQueryKey';
+import type { TRPCMutationKey, TRPCQueryKey } from './getQueryKey';
 
 export type TRPCFetchQueryOptions<TOutput, TError> = DistributiveOmit<
   FetchQueryOptions<TOutput, TError>,
@@ -240,5 +241,29 @@ export interface TRPCQueryUtils<TRouter extends AnyRouter> {
   getInfiniteQueryData: (
     queryKey: TRPCQueryKey,
   ) => InfiniteData<unknown> | undefined;
+
+  /**
+   * @link https://tanstack.com/query/latest/docs/reference/QueryClient/#queryclientsetmutationdefaults
+   */
+  setMutationDefaults: (
+    mutationKey: TRPCMutationKey,
+    options:
+      | MutationOptions
+      | ((args: {
+          canonicalMutationFn: (input: unknown) => Promise<unknown>;
+        }) => MutationOptions),
+  ) => void;
+
+  /**
+   * @link https://tanstack.com/query/latest/docs/reference/QueryClient#queryclientgetmutationdefaults
+   */
+  getMutationDefaults: (
+    mutationKey: TRPCMutationKey,
+  ) => MutationOptions | undefined;
+
+  /**
+   * @link https://tanstack.com/query/latest/docs/reference/QueryClient#queryclientismutating
+   */
+  isMutating: (filters: { mutationKey: TRPCMutationKey }) => number;
 }
 export const TRPCContext = React.createContext?.(null as any);

package/src/internals/getQueryKey.ts

@@ -13,6 +13,8 @@ export type TRPCQueryKey = [
   { input?: unknown; type?: Exclude<QueryType, 'any'> }?,
 ];
 
+export type TRPCMutationKey = [readonly string[]]; // = [TRPCQueryKey[0]]
+
 type ProcedureOrRouter =
   | DecoratedMutation<any>
   | DecoratedQuery<any>
@@ -36,6 +38,8 @@ export function getQueryKeyInternal(
   const splitPath = path.flatMap((part) => part.split('.'));
 
   if (!input && (!type || type === 'any')) {
+    // this matches also all mutations (see `getMutationKeyInternal`)
+
     // for `utils.invalidate()` to match all queries (including vanilla react-query)
     // we don't want nested array if path is empty, i.e. `[]` instead of `[[]]`
     return splitPath.length ? [splitPath] : ([] as unknown as TRPCQueryKey);
@@ -69,6 +73,10 @@ export function getQueryKeyInternal(
   ];
 }
 
+export function getMutationKeyInternal(path: readonly string[]) {
+  return getQueryKeyInternal(path, undefined, 'any') as TRPCMutationKey;
+}
+
 type GetInfiniteQueryInput<
   TProcedureInput,
   TInputWithoutCursorAndDirection = Omit<
@@ -115,3 +123,16 @@ export type QueryKeyKnown<TInput, TType extends Exclude<QueryType, 'any'>> = [
   string[],
   { input?: GetQueryProcedureInput<TInput>; type: TType }?,
 ];
+
+/**
+ * Method to extract the mutation key for a procedure
+ * @param procedure - procedure
+ * @link https://trpc.io/docs/v11/getQueryKey#mutations
+ */
+export function getMutationKey<TProcedure extends DecoratedMutation<any>>(
+  procedure: TProcedure,
+) {
+  // @ts-expect-error - we don't expose _def on the type layer
+  const path = procedure._def().path as string[];
+  return getMutationKeyInternal(path);
+}