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 */
Rule . FAILURE _STRING = "'await' of non-Promise." ;
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 ) {
if ( tsutils _1 . isAwaitExpression ( node ) && ! couldBePromise ( tc . getTypeAtLocation ( node . expression ) ) ) {
ctx . addFailureAtNode ( node , Rule . FAILURE _STRING ) ;
}
return ts . forEachChild ( node , cb ) ;
}
function couldBePromise ( type ) {
if ( Lint . isTypeFlagSet ( type , ts . TypeFlags . Any ) || isPromiseType ( type ) ) {
return true ;
}
2017-08-14 05:01:11 +02:00
if ( tsutils _1 . isUnionOrIntersectionType ( type ) ) {
2017-05-28 00:38:50 +02:00
return type . types . some ( couldBePromise ) ;
}
var bases = type . getBaseTypes ( ) ;
return bases !== undefined && bases . some ( couldBePromise ) ;
}
function isPromiseType ( type ) {
var target = type . target ;
return target !== undefined && target . symbol !== undefined && promiseTypes . has ( target . symbol . name ) ;
}
}
var _a ;