Add an update command to remotely trigger an update check

Author: pimterry

src/httptoolkit-server.ts

@@ -1,4 +1,5 @@
 import * as _ from 'lodash';
+import * as events from 'events';
 import { GraphQLServer } from 'graphql-yoga'
 import { GraphQLScalarType } from 'graphql';
 
@@ -24,6 +25,7 @@ const typeDefs = `
             id: ID!,
             proxyPort: Int!
         ): Boolean!
+        triggerUpdate: Void
     }
 
     type InterceptionConfig {
@@ -40,11 +42,13 @@ const typeDefs = `
 
     scalar Json
     scalar Error
+    scalar Void
 `
 
 const buildResolvers = (
     config: HtkConfig,
-    interceptors: _.Dictionary<Interceptor>
+    interceptors: _.Dictionary<Interceptor>,
+    eventEmitter: events.EventEmitter
 ) => {
     return {
         Query: {
@@ -73,6 +77,9 @@ const buildResolvers = (
 
                 await interceptor.deactivate(proxyPort, options);
                 return !interceptor.isActive(proxyPort);
+            },
+            triggerUpdate: () => {
+                eventEmitter.emit('update-requested');
             }
         },
 
@@ -93,6 +100,14 @@ const buildResolvers = (
             parseLiteral: (): any => { throw new Error('JSON literals are not supported') }
         }),
 
+        Void: new GraphQLScalarType({
+            name: 'Void',
+            description: 'Nothing at all',
+            serialize: (value: any) => null,
+            parseValue: (input: string): any => null,
+            parseLiteral: (): any => { throw new Error('Void literals are not supported') }
+        }),
+
         Error: new GraphQLScalarType({
             name: 'Error',
             description: 'An error',
@@ -114,16 +129,18 @@ const buildResolvers = (
     }
 };
 
-export class HttpToolkitServer {
+export class HttpToolkitServer extends events.EventEmitter {
 
     private graphql: GraphQLServer;
 
     constructor(config: HtkConfig) {
+        super();
+
         let interceptors = buildInterceptors(config);
 
         this.graphql = new GraphQLServer({
             typeDefs,
-            resolvers: buildResolvers(config, interceptors)
+            resolvers: buildResolvers(config, interceptors, this)
         });
     }
 

src/index.ts

@@ -4,6 +4,8 @@ import * as fs from 'fs';
 import * as envPaths from 'env-paths';
 import { getStandalone, generateCACertificate } from 'mockttp';
 
+import updateCommand from '@oclif/plugin-update/lib/commands/update';
+
 import { HttpToolkitServer } from './httptoolkit-server';
 
 const canAccess = util.promisify(fs.access);
@@ -64,6 +66,10 @@ export async function runHTK(options: {
         configPath,
         https: httpsConfig
     });
+    htkServer.on('update-requested', () => {
+        updateCommand.run(['stable']);
+    });
+
     await htkServer.start();
 
     console.log('Server started');