Support file watching for codegen:generate

Author: shadaj

package-lock.json

@@ -0,0 +1 @@
+declare module "gaze";

packages/apollo-cli/@types/gaze.d.ts

@@ -42,7 +42,8 @@
     "node-fetch": "^2.1.2",
     "react": "^16.4.0",
     "react-dom": "^16.4.0",
-    "rimraf": "^2.6.2"
+    "rimraf": "^2.6.2",
+    "gaze": "^1.1.3"
   },
   "devDependencies": {
     "@fancy-test/nock": "^0.1.1",

packages/apollo-cli/package.json

@@ -16,6 +16,19 @@ import { engineFlags } from "../../engine-cli";
 import { fetchSchema } from '../../fetch-schema';
 import { loadQueryDocuments } from 'apollo-codegen-core/lib/loading';
 
+import { Gaze } from "gaze";
+import { eventNames } from 'cluster';
+
+const waitForKey = async () => {
+  console.log("Press any key to stop.");
+  process.stdin.setRawMode!(true);
+  return new Promise(resolve => process.stdin.once('data', () => {
+    (process.stdin as any).unref();
+    process.stdin.setRawMode!(false)
+    resolve();
+  }))
+};
+
 export default class Generate extends Command {
   static description =
     "Generate static types for GraphQL queries. Can use the published schema in Apollo Engine or a downloaded schema.";
@@ -81,6 +94,9 @@ export default class Generate extends Command {
     outputFlat: flags.boolean({
       description:
         'By default, TypeScript/Flow will put each generated file in a directory next to its source file using the value of the "output" as the directory name. Set "outputFlat" to put all generated files in the directory relative to the current working directory defined by "output".'
+    }),
+    watch: flags.boolean({
+      description: "Watch the query files to auto-generate on changes."
     })
   };
 
@@ -282,6 +298,18 @@ export default class Generate extends Command {
       }
     ]);
 
-    return tasks.run();
+    if (flags.watch) {
+      await tasks.run().catch((r) => {});
+      const watcher = new Gaze(flags.queries!);
+      watcher.on("all", () => {
+        console.log("\nChange detected, generating types...")
+        tasks.run().catch((r) => {});
+      });
+      await waitForKey();
+      watcher.close();
+      return;
+    } else {
+      return tasks.run();
+    }
   }
 }