diff --git a/src/app/services/actions/query-action-creator-util.ts b/src/app/services/actions/query-action-creator-util.ts index cce501f7a..5c497e040 100644 --- a/src/app/services/actions/query-action-creator-util.ts +++ b/src/app/services/actions/query-action-creator-util.ts @@ -53,7 +53,7 @@ export function isImageResponse(contentType: string) { if (!contentType) { return false; } return ( contentType === 'application/octet-stream' || - contentType.substr(0, 6) === 'image/' + contentType.includes('image/') ); } diff --git a/src/app/services/actions/query-action-creators.ts b/src/app/services/actions/query-action-creators.ts index b28d31245..dd9dffcb5 100644 --- a/src/app/services/actions/query-action-creators.ts +++ b/src/app/services/actions/query-action-creators.ts @@ -1,9 +1,13 @@ import { MessageBarType } from 'office-ui-fabric-react'; + import { ContentType } from '../../../types/enums'; import { IHistoryItem } from '../../../types/history'; import { IQuery } from '../../../types/query-runner'; import { writeHistoryData } from '../../views/sidebar/history/history-utils'; -import { anonymousRequest, authenticatedRequest, parseResponse, queryResponse } from './query-action-creator-util'; +import { + anonymousRequest, authenticatedRequest, + isImageResponse, parseResponse, queryResponse +} from './query-action-creator-util'; import { setQueryResponseStatus } from './query-status-action-creator'; import { addHistoryItem } from './request-history-action-creators'; @@ -83,7 +87,7 @@ async function createHistory(response: Response, respHeaders: any, query: IQuery const responseHeaders = { ...respHeaders }; const contentType = respHeaders['content-type']; - if (contentType === ContentType.Image) { + if (isImageResponse(contentType)) { result = { message: 'Run the query to view the image' }; diff --git a/src/app/views/query-response/pivot-items/pivot-items.tsx b/src/app/views/query-response/pivot-items/pivot-items.tsx index bb986fef6..bd3dbbb9a 100644 --- a/src/app/views/query-response/pivot-items/pivot-items.tsx +++ b/src/app/views/query-response/pivot-items/pivot-items.tsx @@ -4,6 +4,7 @@ import React from 'react'; import { ThemeContext } from '../../../../themes/theme-context'; import { ContentType, Mode } from '../../../../types/enums'; import { IQuery } from '../../../../types/query-runner'; +import { isImageResponse } from '../../../services/actions/query-action-creator-util'; import { lookupTemplate } from '../../../utils/adaptive-cards-lookup'; import { Image, Monaco } from '../../common'; import { genericCopy } from '../../common/copy'; @@ -112,14 +113,14 @@ function displayResultComponent(headers: any, body: any, verb: string) { case ContentType.XML: return ; - case ContentType.Image: - return profile image; - default: + if (isImageResponse(contentType)) { + return profile image; + } return ; } }