Add pragmas support to Table Preview queries

src/containers/Tenant/Query/Preview/components/TablePreview.tsx

@@ -1,5 +1,6 @@
 import {QueryResultTable} from '../../../../../components/QueryResultTable';
 import {previewApi} from '../../../../../store/reducers/preview';
+import {useQueryExecutionSettings} from '../../../../../utils/hooks';
 import {isExternalTableType} from '../../../utils/schema';
 import type {PreviewContainerProps} from '../types';
 
@@ -9,12 +10,14 @@ const TABLE_PREVIEW_LIMIT = 100;
 
 export function TablePreview({database, path, type}: PreviewContainerProps) {
     const query = `select * from \`${path}\` limit 101`;
+    const [querySettings] = useQueryExecutionSettings();
     const {currentData, isFetching, error} = previewApi.useSendQueryQuery(
         {
             database,
             query,
             action: isExternalTableType(type) ? 'execute-query' : 'execute-scan',
             limitRows: TABLE_PREVIEW_LIMIT,
+            pragmas: querySettings.pragmas,
         },
         {
             refetchOnMountOrArgChange: true,

src/containers/Tenant/Query/Preview/components/__test__/TablePreview.test.tsx

@@ -0,0 +1,41 @@
+import {prepareQueryWithPragmas} from '../../../../../../store/reducers/query/utils';
+
+describe('TablePreview Pragmas Integration', () => {
+    test('prepareQueryWithPragmas should work with table preview queries', () => {
+        const tablePreviewQuery = 'select * from `test-table` limit 101';
+        const pragmas = 'PRAGMA OrderedColumns;';
+
+        const result = prepareQueryWithPragmas(tablePreviewQuery, pragmas);
+
+        expect(result).toBe('PRAGMA OrderedColumns;\n\nselect * from `test-table` limit 101');
+    });
+
+    test('prepareQueryWithPragmas should handle empty pragmas for table preview', () => {
+        const tablePreviewQuery = 'select * from `test-table` limit 101';
+        const pragmas = '';
+
+        const result = prepareQueryWithPragmas(tablePreviewQuery, pragmas);
+
+        expect(result).toBe('select * from `test-table` limit 101');
+    });
+
+    test('prepareQueryWithPragmas should handle undefined pragmas for table preview', () => {
+        const tablePreviewQuery = 'select * from `test-table` limit 101';
+        const pragmas = undefined;
+
+        const result = prepareQueryWithPragmas(tablePreviewQuery, pragmas);
+
+        expect(result).toBe('select * from `test-table` limit 101');
+    });
+
+    test('prepareQueryWithPragmas should handle multiple pragmas for table preview', () => {
+        const tablePreviewQuery = 'select * from `test-table` limit 101';
+        const pragmas = 'PRAGMA OrderedColumns;\nPRAGMA AnsiInForEmptyOrNullableItemsCollections;';
+
+        const result = prepareQueryWithPragmas(tablePreviewQuery, pragmas);
+
+        expect(result).toBe(
+            'PRAGMA OrderedColumns;\nPRAGMA AnsiInForEmptyOrNullableItemsCollections;\n\nselect * from `test-table` limit 101',
+        );
+    });
+});

src/store/reducers/preview.ts

@@ -2,21 +2,27 @@ import type {ExecuteActions} from '../../types/api/query';
 import {isQueryErrorResponse, parseQueryAPIResponse} from '../../utils/query';
 
 import {api} from './api';
+import {prepareQueryWithPragmas} from './query/utils';
 
 interface SendQueryParams {
     query?: string;
     database?: string;
     action?: ExecuteActions;
     limitRows?: number;
+    pragmas?: string;
 }
 
 export const previewApi = api.injectEndpoints({
     endpoints: (build) => ({
         sendQuery: build.query({
-            queryFn: async ({query, database, action, limitRows}: SendQueryParams, {signal}) => {
+            queryFn: async (
+                {query, database, action, limitRows, pragmas}: SendQueryParams,
+                {signal},
+            ) => {
                 try {
+                    const finalQuery = prepareQueryWithPragmas(query || '', pragmas);
                     const response = await window.api.viewer.sendQuery(
-                        {query, database, action, limit_rows: limitRows},
+                        {query: finalQuery, database, action, limit_rows: limitRows},
                         {signal, withRetries: true},
                     );