2017-05-28 00:38:50 +02:00
"use strict" ;
/ * *
* @ license
* Copyright 2017 Palantir Technologies , Inc .
*
* Licensed under the Apache License , Version 2.0 ( the "License" ) ;
* you may not use this file except in compliance with the License .
* You may obtain a copy of the License at
*
* http : //www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing , software
* distributed under the License is distributed on an "AS IS" BASIS ,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
* See the License for the specific language governing permissions and
* limitations under the License .
* /
Object . defineProperty ( exports , "__esModule" , { value : true } ) ;
var tslib _1 = require ( "tslib" ) ;
var tsutils _1 = require ( "tsutils" ) ;
var ts = require ( "typescript" ) ;
var Lint = require ( "../index" ) ;
2017-10-14 18:40:54 +02:00
var Rule = /** @class */ ( function ( _super ) {
2017-05-28 00:38:50 +02:00
tslib _1 . _ _extends ( Rule , _super ) ;
function Rule ( ) {
return _super !== null && _super . apply ( this , arguments ) || this ;
}
Rule . prototype . applyWithProgram = function ( sourceFile , program ) {
var promiseTypes = new Set ( [ "Promise" ] . concat ( this . ruleArguments ) ) ;
2017-10-14 18:40:54 +02:00
return this . applyWithFunction ( sourceFile , walk , promiseTypes , program . getTypeChecker ( ) ) ;
2017-05-28 00:38:50 +02:00
} ;
2017-08-14 05:01:11 +02:00
/* tslint:disable:object-literal-sort-keys */
Rule . metadata = {
ruleName : "await-promise" ,
description : "Warns for an awaited value that is not a Promise." ,
optionsDescription : ( _a = [ "\n A list of 'string' names of any additional classes that should also be handled as Promises.\n " ] , _a . raw = [ "\n A list of 'string' names of any additional classes that should also be handled as Promises.\n " ] , Lint . Utils . dedent ( _a ) ) ,
options : {
type : "list" ,
listType : {
type : "array" ,
items : { type : "string" } ,
} ,
} ,
optionExamples : [ true , [ true , "Thenable" ] ] ,
type : "functionality" ,
typescriptOnly : true ,
requiresTypeInfo : true ,
} ;
/* tslint:enable:object-literal-sort-keys */
2017-12-10 21:51:33 +01:00
Rule . FAILURE _STRING = "Invalid 'await' of a non-Promise value." ;
Rule . FAILURE _FOR _AWAIT _OF = "Invalid 'for-await-of' of a non-AsyncIterable value." ;
2017-05-28 00:38:50 +02:00
return Rule ;
} ( Lint . Rules . TypedRule ) ) ;
exports . Rule = Rule ;
2017-10-14 18:40:54 +02:00
function walk ( ctx , tc ) {
var promiseTypes = ctx . options ;
2017-05-28 00:38:50 +02:00
return ts . forEachChild ( ctx . sourceFile , cb ) ;
function cb ( node ) {
2017-12-10 21:51:33 +01:00
if ( tsutils _1 . isAwaitExpression ( node ) && ! containsType ( tc . getTypeAtLocation ( node . expression ) , isPromiseType ) ) {
2017-05-28 00:38:50 +02:00
ctx . addFailureAtNode ( node , Rule . FAILURE _STRING ) ;
}
2017-12-10 21:51:33 +01:00
else if ( tsutils _1 . isForOfStatement ( node ) && node . awaitModifier !== undefined &&
! containsType ( tc . getTypeAtLocation ( node . expression ) , isAsyncIterable ) ) {
ctx . addFailureAtNode ( node . expression , Rule . FAILURE _FOR _AWAIT _OF ) ;
}
2017-05-28 00:38:50 +02:00
return ts . forEachChild ( node , cb ) ;
}
2017-12-10 21:51:33 +01:00
function isPromiseType ( name ) {
return promiseTypes . has ( name ) ;
2017-05-28 00:38:50 +02:00
}
2017-12-10 21:51:33 +01:00
}
function containsType ( type , predicate ) {
if ( Lint . isTypeFlagSet ( type , ts . TypeFlags . Any ) ||
tsutils _1 . isTypeReference ( type ) && type . target . symbol !== undefined && predicate ( type . target . symbol . name ) ) {
return true ;
}
if ( tsutils _1 . isUnionOrIntersectionType ( type ) ) {
return type . types . some ( function ( t ) { return containsType ( t , predicate ) ; } ) ;
2017-05-28 00:38:50 +02:00
}
2017-12-10 21:51:33 +01:00
var bases = type . getBaseTypes ( ) ;
return bases !== undefined && bases . some ( function ( t ) { return containsType ( t , predicate ) ; } ) ;
}
function isAsyncIterable ( name ) {
return name === "AsyncIterable" || name === "AsyncIterableIterator" ;
2017-05-28 00:38:50 +02:00
}
var _a ;