Support top-level await - fixes #1 (#4)

readme.md

@@ -61,6 +61,18 @@ If we don't change the test file and we run the `tsd-check` command again, the t
 
 <img src="screenshot.png" width="1330">
 
+### Top-level `await`
+
+If your method returns a `Promise`, you can use top-level `await` to resolve the value instead of wrapping it in an `async` [IIFE](https://developer.mozilla.org/en-US/docs/Glossary/IIFE).
+
+```ts
+import {expectType} from 'tsd-check';
+import concat from '.';
+
+expectType<Promise<string>>(concat('foo', 'bar'));
+
+expectType<string>(await concat('foo', 'bar'));
+```
 
 
 ## Assertions

source/lib/compiler.ts

@@ -2,6 +2,11 @@ import * as path from 'path';
 import {createProgram, getPreEmitDiagnostics, ScriptTarget, ModuleResolutionKind, flattenDiagnosticMessageText} from 'typescript';
 import {Diagnostic, Context} from './interfaces';
 
+// List of diagnostic codes that should be ignored
+const ignoredDiagnostics = new Set<number>([
+	1308 // Support top-level `await`
+]);
+
 const loadConfig = () => {
 	return {
 		moduleResolution: ModuleResolutionKind.NodeJs,
@@ -28,7 +33,7 @@ export const getDiagnostics = (context: Context): Diagnostic[] => {
 	const result: Diagnostic[] = [];
 
 	for (const diagnostic of diagnostics) {
-		if (!diagnostic.file) {
+		if (!diagnostic.file || ignoredDiagnostics.has(diagnostic.code)) {
 			continue;
 		}
 

source/test/fixtures/top-level-await/index.d.ts

@@ -0,0 +1,6 @@
+declare const one: {
+	(foo: string, bar: string): Promise<string>;
+	(foo: number, bar: number): Promise<number>;
+};
+
+export default one;

source/test/fixtures/top-level-await/index.js

@@ -0,0 +1,3 @@
+module.exports.default = (foo, bar) => {
+	return foo + bar;
+};

source/test/fixtures/top-level-await/index.test-d.ts

@@ -0,0 +1,5 @@
+import {expectType} from '../../..';
+import one from '.';
+
+expectType<string>(await one('foo', 'bar'));
+expectType<number>(await one(1, 2));

source/test/fixtures/top-level-await/package.json

@@ -0,0 +1,7 @@
+{
+	"name": "foo",
+	"files": [
+		"index.js",
+		"index.d.ts"
+	]
+}

source/test/test.ts

@@ -36,3 +36,9 @@ test('return no diagnostics', async t => {
 
 	t.true(diagnostics.length === 0);
 });
+
+test('support top-level await', async t => {
+	const diagnostics = await m({cwd: path.join(__dirname, 'fixtures/top-level-await')});
+
+	t.true(diagnostics.length === 0);
+});