Modify Query to have classPath as tagged value when we save it via da…

…taspace workflow

.changeset/little-flies-unite.md

@@ -0,0 +1,5 @@
+---
+'@finos/legend-extension-dsl-data-space': patch
+---
+
+Modify `Query` to have `classPath` as tagged value when we save it via dataspace workflow

packages/legend-extension-dsl-data-space/src/DSL_DataSpace_Const.ts

@@ -15,6 +15,7 @@
  */
 
 // NOTE: these are non-compilable profile and tag that we come up with for query
-// so that it records the dataSpace it is created from
+// so that it records the dataSpace and class it is created from
 export const QUERY_PROFILE_PATH = 'meta::pure::profiles::query';
 export const QUERY_PROFILE_TAG_DATA_SPACE = 'dataSpace';
+export const QUERY_PROFILE_TAG_CLASS = 'class';

packages/legend-extension-dsl-data-space/src/stores/query/DataSpaceQueryCreatorStore.ts

@@ -38,6 +38,7 @@ import {
 } from '@finos/legend-shared';
 import {
   QUERY_PROFILE_PATH,
+  QUERY_PROFILE_TAG_CLASS,
   QUERY_PROFILE_TAG_DATA_SPACE,
 } from '../../DSL_DataSpace_Const.js';
 import { getDataSpace } from '../../graphManager/DSL_DataSpace_GraphManagerHelper.js';
@@ -58,6 +59,16 @@ export const createQueryDataSpaceTaggedValue = (
   return taggedValue;
 };
 
+export const createQueryClassTaggedValue = (
+  classPath: string,
+): QueryTaggedValue => {
+  const taggedValue = new QueryTaggedValue();
+  taggedValue.profile = QUERY_PROFILE_PATH;
+  taggedValue.tag = QUERY_PROFILE_TAG_CLASS;
+  taggedValue.value = classPath;
+  return taggedValue;
+};
+
 export class DataSpaceQueryCreatorStore extends QueryEditorStore {
   groupId: string;
   artifactId: string;
@@ -237,6 +248,7 @@ export class DataSpaceQueryCreatorStore extends QueryEditorStore {
         query.versionId = this.versionId;
         query.taggedValues = [
           createQueryDataSpaceTaggedValue(this.dataSpacePath),
+          createQueryClassTaggedValue(guaranteeNonNullable(this.classPath)),
         ];
       },
     };