Watch files onbuild, even if build fails

src/Module.ts

@@ -244,6 +244,7 @@ export default class Module {
 	error(props: RollupError, pos: number) {
 		if (pos !== undefined) {
 			props.pos = pos;
+			props.graph = this.graph;
 
 			let location = locate(this.code, pos, { offsetLine: 1 });
 			try {

src/rollup/types.d.ts

@@ -1,9 +1,11 @@
 import * as ESTree from 'estree';
 import { EventEmitter } from 'events';
+import Graph from '../Graph';
 
 export const VERSION: string;
 
 export interface RollupError extends RollupLogProps {
+	graph: Graph;
 	stack?: string;
 }
 

src/watch/index.ts

@@ -9,6 +9,7 @@ import {
 	OutputOptions,
 	RollupBuild,
 	RollupCache,
+	RollupError,
 	RollupWatcher,
 	WatcherOptions
 } from '../rollup/types';
@@ -100,7 +101,7 @@ export class Watcher {
 			.catch(error => {
 				this.running = false;
 				this.emit('event', {
-					code: this.succeeded ? 'ERROR' : 'FATAL',
+					code: 'ERROR',
 					error
 				});
 			})
@@ -243,9 +244,13 @@ export class Task {
 					result
 				});
 			})
-			.catch((error: Error) => {
+			.catch((error: RollupError) => {
 				if (this.closed) return;
 
+				Array.from(error.graph.moduleById.keys()).forEach(id => {
+					this.watchFile(id);
+				});
+
 				if (this.cache) {
 					// this is necessary to ensure that any 'renamed' files
 					// continue to be watched following an error