Dont generate store methods for source subgraph schema

Author: incrypto32

packages/cli/src/codegen/schema.ts

@@ -97,14 +97,14 @@ export default class SchemaCodeGenerator {
     ];
   }
 
-  generateTypes(): Array<tsCodegen.Class> {
+  generateTypes(generateStoreMethods = true): Array<tsCodegen.Class> {
     return this.schema.ast.definitions
       .map(def => {
         if (this._isEntityTypeDefinition(def)) {
           schemaCodeGeneratorDebug.extend('generateTypes')(
             `Generating entity type for ${def.name.value}`,
           );
-          return this._generateEntityType(def);
+          return this._generateEntityType(def, generateStoreMethods);
         }
       })
       .filter(Boolean) as Array<tsCodegen.Class>;
@@ -157,7 +157,7 @@ export default class SchemaCodeGenerator {
     return def.kind === 'InterfaceTypeDefinition';
   }
 
-  _generateEntityType(def: ObjectTypeDefinitionNode) {
+  _generateEntityType(def: ObjectTypeDefinitionNode, generateStoreMethods = true) {
     const name = def.name.value;
     const klass = tsCodegen.klass(name, { export: true, extends: 'Entity' });
     const fields = def.fields;
@@ -166,8 +166,10 @@ export default class SchemaCodeGenerator {
     // Generate and add a constructor
     klass.addMethod(this._generateConstructor(name, fields));
 
-    // Generate and add save() and getById() methods
-    this._generateStoreMethods(name, idField).forEach(method => klass.addMethod(method));
+    if (generateStoreMethods) {
+      // Generate and add save() and getById() methods
+      this._generateStoreMethods(name, idField).forEach(method => klass.addMethod(method));
+    }
 
     // Generate and add entity field getters and setters
     def.fields

packages/cli/src/type-generator.ts

@@ -102,7 +102,7 @@ export default class TypeGenerator {
 
       const schema = await this.loadSchema(subgraph);
       typeGenDebug.extend('generateTypes')('Generating types for schema');
-      await this.generateTypesForSchema(schema);
+      await this.generateTypesForSchema({ schema });
 
       if (this.options.subgraphSources.length > 0) {
         const ipfsClient = create({
@@ -123,7 +123,11 @@ export default class TypeGenerator {
             typeGenDebug.extend('generateTypes')(
               `Generating types for subgraph datasource ${manifest}`,
             );
-            await this.generateTypesForSchema(subgraphSchema, `subgraph-${manifest}.ts`);
+            await this.generateTypesForSchema({
+              schema: subgraphSchema,
+              fileName: `subgraph-${manifest}.ts`,
+              generateStoreMethods: false,
+            });
           }),
         );
       }
@@ -226,11 +230,17 @@ export default class TypeGenerator {
     );
   }
 
-  async generateTypesForSchema(
-    schema: any,
+  async generateTypesForSchema({
+    schema,
     fileName = 'schema.ts', // Default file name
-    outputDir: string = this.options.outputDir, // Default output directory
-  ) {
+    outputDir = this.options.outputDir, // Default output directory
+    generateStoreMethods = true,
+  }: {
+    schema: any;
+    fileName?: string;
+    outputDir?: string;
+    generateStoreMethods?: boolean;
+  }) {
     return await withSpinner(
       `Generate types for GraphQL schema`,
       `Failed to generate types for GraphQL schema`,
@@ -242,7 +252,7 @@ export default class TypeGenerator {
           [
             GENERATED_FILE_NOTE,
             ...codeGenerator.generateModuleImports(),
-            ...codeGenerator.generateTypes(),
+            ...codeGenerator.generateTypes(generateStoreMethods),
             ...codeGenerator.generateDerivedLoaders(),
           ].join('\n'),
           {