Wizard update and create

Author: Serge Pavlyuk

src/server/components/charts-engine/components/storage/united-storage/provider.ts

@@ -216,7 +216,7 @@ export type ProviderCreateParams = {
     recursion?: boolean;
     meta?: Record<string, string>;
     includePermissionsInfo?: boolean | string;
-    workbookId: string;
+    workbookId: string | null;
     name: string;
     mode?: EntryUpdateMode;
 };
@@ -360,7 +360,7 @@ export class USProvider {
     ) {
         const result = await USProvider.retrieveById(ctx, props);
 
-        result.data.shared = mapChartsConfigToLatestVersion(JSON.parse(result.data.shared));
+        result.data = mapChartsConfigToLatestVersion(JSON.parse(result.data.shared));
 
         return result;
     }

src/server/controllers/public-api.ts

@@ -59,14 +59,14 @@ const proxyMap: PublicApiRpcMap = {
             },
         },
         updateWizardChart: {
-            resolve: (api) => api.bi.updateDataset,
+            resolve: (api) => api.mix.updateWizardChartApi,
             openApi: {
                 summary: 'Delete wizard chart',
                 tags: ['wizard'],
             },
         },
         createWizardChart: {
-            resolve: (api) => api.bi.createDataset,
+            resolve: (api) => api.mix.createWizardChartApi,
             openApi: {
                 summary: 'Create wizard chart',
                 tags: ['wizard'],

src/shared/schema/mix/actions/wizard.ts

@@ -7,9 +7,7 @@ import {createTypedAction} from '../../gateway-utils';
 import {getTypedApi} from '../../simple-schema';
 
 const wizardUsSchema = z.object({
-    data: z.object({
-        shared: v12ChartsConfigSchema,
-    }),
+    data: v12ChartsConfigSchema,
     entryId: z.string(),
     scope: z.literal(EntryScope.Widget),
     type: z.enum(WizardType),
@@ -52,6 +50,52 @@ export const wizardActions = {
 
         return result as any;
     }),
+    createWizardChartApi: createTypedAction({
+        argsSchema: z.object({
+            entryId: z.string(),
+            data: v12ChartsConfigSchema,
+            key: z.string(),
+            workbookId: z.union([z.string(), z.null()]).optional(),
+            type: z.enum(WizardType).optional(),
+            name: z.string(),
+        }),
+        bodySchema: wizardUsSchema,
+    }).withValidationSchema(async (_, args, {ctx, headers}) => {
+        const {data, type, key, workbookId, name} = args;
+
+        const result = await USProvider.create(ctx, {
+            type,
+            data,
+            key,
+            name,
+            scope: EntryScope.Widget,
+            ...(workbookId ? {workbookId} : {workbookId: null}),
+            headers,
+        });
+
+        return result as any;
+    }),
+    updateWizardChartApi: createTypedAction({
+        argsSchema: z.object({
+            entryId: z.string(),
+            revId: z.string().optional(),
+            data: v12ChartsConfigSchema,
+            type: z.enum(WizardType).optional(),
+        }),
+        bodySchema: wizardUsSchema,
+    }).withValidationSchema(async (_, args, {ctx, headers}) => {
+        const {entryId, revId, data, type} = args;
+
+        const result = await USProvider.update(ctx, {
+            entryId,
+            ...(revId ? {revId} : {}),
+            ...(type ? {type} : {}),
+            data,
+            headers,
+        });
+
+        return result as any;
+    }),
     deleteWizardChartApi: createTypedAction({
         argsSchema: z.object({
             chartId: z.string(),