@@ -109,55 +109,55 @@ function _get(
109109 const userAgentOverride = createUserAgentOverride ( customUserAgentDetails ) ;
110110
111111 try {
112- // const basePromise = context
113- // ? ((client as BaseSSRClient).graphql(
114- // context,
115- // {
116- // ...auth,
117- // query,
118- // variables,
119- // },
120- // headers,
121- // ) as Promise<GraphQLResult>)
122- // : ((client as BaseBrowserClient).graphql(
123- // {
124- // ...auth,
125- // query,
126- // variables,
127- // ...userAgentOverride,
128- // },
129- // headers,
130- // ) as Promise<GraphQLResult>);
131- // const extendedPromise = extendCancellability(basePromise, resultPromise);
132- // const { data, extensions } = await extendedPromise;
133- // // flatten response
134- // if (data) {
135- // const [key] = Object.keys(data);
136- // const flattenedResult = flattenItems(
137- // modelIntrospection,
138- // name,
139- // data[key],
140- // );
141- // if (flattenedResult === null) {
142- // return { data: null, extensions };
143- // } else if (options?.selectionSet) {
144- // return { data: flattenedResult, extensions };
145- // } else {
146- // // TODO: refactor to avoid destructuring here
147- // const [initialized] = initializeModel(
148- // client,
149- // name,
150- // [flattenedResult],
151- // modelIntrospection,
152- // auth.authMode,
153- // auth.authToken,
154- // !!context,
155- // );
156- // return { data: initialized, extensions };
157- // }
158- // } else {
159- // return { data: null, extensions };
160- // }
112+ const basePromise = context
113+ ? ( ( client as BaseSSRClient ) . graphql (
114+ context ,
115+ {
116+ ...auth ,
117+ query,
118+ variables,
119+ } ,
120+ headers ,
121+ ) as Promise < GraphQLResult > )
122+ : ( ( client as BaseBrowserClient ) . graphql (
123+ {
124+ ...auth ,
125+ query,
126+ variables,
127+ ...userAgentOverride ,
128+ } ,
129+ headers ,
130+ ) as Promise < GraphQLResult > ) ;
131+ const extendedPromise = extendCancellability ( basePromise , resultPromise ) ;
132+ const { data, extensions } = await extendedPromise ;
133+ // flatten response
134+ if ( data ) {
135+ const [ key ] = Object . keys ( data ) ;
136+ const flattenedResult = flattenItems (
137+ modelIntrospection ,
138+ name ,
139+ data [ key ] ,
140+ ) ;
141+ if ( flattenedResult === null ) {
142+ return { data : null , extensions } ;
143+ } else if ( options ?. selectionSet ) {
144+ return { data : flattenedResult , extensions } ;
145+ } else {
146+ // TODO: refactor to avoid destructuring here
147+ const [ initialized ] = initializeModel (
148+ client ,
149+ name ,
150+ [ flattenedResult ] ,
151+ modelIntrospection ,
152+ auth . authMode ,
153+ auth . authToken ,
154+ ! ! context ,
155+ ) ;
156+ return { data : initialized , extensions } ;
157+ }
158+ } else {
159+ return { data : null , extensions } ;
160+ }
161161 } catch ( error : any ) {
162162 /**
163163 * The `data` type returned by `error` here could be:
0 commit comments