|
1 | 1 | import { ACL_APP_ABILITY_FILE, ACL_CHECK_FILE } from "@/generators/const/acl.const"; |
2 | | -import { |
3 | | - APP_REST_CLIENT_FILE, |
4 | | - CROSS_TAB_QUERY_INVALIDATION_FILE, |
5 | | - QUERY_MODULES_FILE, |
6 | | - ZOD_EXTENDED_FILE, |
7 | | -} from "@/generators/const/deps.const"; |
| 2 | +import { APP_REST_CLIENT_FILE, CROSS_TAB_QUERY_INVALIDATION_FILE, MUTATION_EFFECTS_FILE, QUERY_MODULES_FILE, ZOD_EXTENDED_FILE } from "@/generators/const/deps.const"; |
| 3 | +import { PACKAGE_IMPORT_PATH } from "@/generators/const/package.const"; |
8 | 4 | import { DEFAULT_GENERATE_OPTIONS } from "@/generators/const/options.const"; |
9 | 5 | import { SchemaResolver } from "@/generators/core/SchemaResolver.class"; |
10 | 6 | import { generateAppAcl } from "@/generators/generate/generateAcl"; |
@@ -50,7 +46,7 @@ export function getMutationEffectsFiles(data: GenerateData, resolver: SchemaReso |
50 | 46 | } |
51 | 47 |
|
52 | 48 | return [ |
53 | | - ...getAssetFiles([CROSS_TAB_QUERY_INVALIDATION_FILE], resolver), |
| 49 | + ...getAssetFiles([MUTATION_EFFECTS_FILE, CROSS_TAB_QUERY_INVALIDATION_FILE], resolver), |
54 | 50 | { |
55 | 51 | fileName: getOutputFileName({ |
56 | 52 | output: resolver.options.output, |
@@ -100,8 +96,17 @@ function getAssetFiles(files: GenerateFile[], resolver: SchemaResolver): Generat |
100 | 96 |
|
101 | 97 | function getAssetFile(file: GenerateFile, resolver: SchemaResolver): GenerateFileData { |
102 | 98 | const fileName = getFileNameWithExtension(file); |
| 99 | + let content = readAssetSync(fileName); |
| 100 | + |
| 101 | + if (file.fileName === MUTATION_EFFECTS_FILE.fileName) { |
| 102 | + content = content.replace( |
| 103 | + 'import { OpenApiQueryConfig, QueryModule, InvalidationMap } from "../lib/config/queryConfig.context";', |
| 104 | + `import { OpenApiQueryConfig, InvalidationMap } from "${PACKAGE_IMPORT_PATH}";\nimport { QueryModule } from "./${QUERY_MODULES_FILE.fileName}";`, |
| 105 | + ); |
| 106 | + } |
| 107 | + |
103 | 108 | return { |
104 | 109 | fileName: getOutputFileName({ output: resolver.options.output, fileName }), |
105 | | - content: readAssetSync(fileName), |
| 110 | + content, |
106 | 111 | }; |
107 | 112 | } |
0 commit comments