add a view for Haskell STG intermediary output #3571 (#3585)gh-2669

1 files changed, 20 insertions, 8 deletions

diff --git a/lib/compilers/haskell.js b/lib/compilers/haskell.js
index e55bb37a4..413185a2e 100644
--- a/lib/compilers/haskell.js
+++ b/lib/compilers/haskell.js
@@ -22,28 +22,40 @@
 // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 // POSSIBILITY OF SUCH DAMAGE.
 
-import { BaseCompiler } from '../base-compiler';
+import path from 'path';
 
-import { ClangParser } from './argument-parsers';
+import {BaseCompiler} from '../base-compiler';
+
+import {ClangParser} from './argument-parsers';
 
 export class HaskellCompiler extends BaseCompiler {
     static get key() {
         return 'haskell';
     }
 
+    constructor(info, env) {
+        super(info, env);
+        this.compiler.supportsHaskellStgView = true;
+    }
+
+    optionsForBackend(backendOptions, outputFilename) {
+        const opts = super.optionsForBackend(backendOptions, outputFilename);
+
+        if (backendOptions.produceHaskellStg && this.compiler.supportsHaskellStgView) {
+            opts.push('-ddump-to-file', '-dumpdir', path.dirname(outputFilename), '-ddump-stg-final');
+        }
+        return opts;
+    }
+
     optionsForFilter(filters, outputFilename) {
         const options = ['-g', '-o', this.filename(outputFilename)];
-        if (!filters.binary)
-            options.unshift('-S');
+        if (!filters.binary) options.unshift('-S');
         return options;
     }
 
     getSharedLibraryPathsAsArguments(libraries) {
         const libPathFlag = this.compiler.libpathFlag || '-L';
-        return [
-            libPathFlag + '.',
-            ...this.getSharedLibraryPaths(libraries).map(path => libPathFlag + path),
-        ];
+        return [libPathFlag + '.', ...this.getSharedLibraryPaths(libraries).map(path => libPathFlag + path)];
     }
 
     getArgumentParser() {