Type checking 'for...of' in ES3/5

src/compiler/checker.ts

@@ -1869,7 +1869,11 @@ module ts {
                 return anyType;
             }
             if (declaration.parent.parent.kind === SyntaxKind.ForOfStatement) {
-                return getTypeForVariableDeclarationInForOfStatement(<ForOfStatement>declaration.parent.parent);
+                // checkRightHandSideOfForOf will return undefined if the for-of expression type was
+                // missing properties/signatures required to get its iteratedType (like
+                // [Symbol.iterator] or next). This may be because we accessed properties from anyType,
+                // or it may have led to an error inside getIteratedType.
+                return checkRightHandSideOfForOf((<ForOfStatement>declaration.parent.parent).expression) || anyType;
             }
             if (isBindingPattern(declaration.parent)) {
                 return getTypeForBindingElement(<BindingElement>declaration);
@@ -8777,25 +8781,19 @@ module ts {
         }
 
         function checkForOfStatement(node: ForOfStatement): void {
-            if (languageVersion < ScriptTarget.ES6) {
-                grammarErrorOnFirstToken(node, Diagnostics.for_of_statements_are_only_available_when_targeting_ECMAScript_6_or_higher);
-                return;
-            }
-
             checkGrammarForInOrForOfStatement(node)
 
             // Check the LHS and RHS
             // If the LHS is a declaration, just check it as a variable declaration, which will in turn check the RHS
-            // via getTypeForVariableDeclarationInForOfStatement.
+            // via checkRightHandSideOfForOf.
             // If the LHS is an expression, check the LHS, as a destructuring assignment or as a reference.
             // Then check that the RHS is assignable to it.
             if (node.initializer.kind === SyntaxKind.VariableDeclarationList) {
                 checkForInOrForOfVariableDeclaration(node);
             }
             else {
                 var varExpr = <Expression>node.initializer;
-                var rightType = checkExpression(node.expression);
-                var iteratedType = checkIteratedType(rightType, node.expression);
+                var iteratedType = checkRightHandSideOfForOf(node.expression);
 
                 // There may be a destructuring assignment on the left side
                 if (varExpr.kind === SyntaxKind.ArrayLiteralExpression || varExpr.kind === SyntaxKind.ObjectLiteralExpression) {
@@ -8877,18 +8875,11 @@ module ts {
             }
         }
 
-        function getTypeForVariableDeclarationInForOfStatement(forOfStatement: ForOfStatement): Type {
-            // Temporarily return 'any' below ES6
-            if (languageVersion < ScriptTarget.ES6) {
-                return anyType;
-            }
-
-            // iteratedType will be undefined if the for-of expression type was missing properties/signatures
-            // required to get its iteratedType (like [Symbol.iterator] or next). This may be
-            // because we accessed properties from anyType, or it may have led to an error inside
-            // getIteratedType.
-            var expressionType = getTypeOfExpression(forOfStatement.expression);
-            return checkIteratedType(expressionType, forOfStatement.expression) || anyType;
+        function checkRightHandSideOfForOf(rhsExpression: Expression): Type {
+            var expressionType = getTypeOfExpression(rhsExpression);
+            return languageVersion >= ScriptTarget.ES6
+                ? checkIteratedType(expressionType, rhsExpression)
+                : checkElementTypeOfArrayOrString(expressionType, rhsExpression);
         }
 
         /**
@@ -8987,6 +8978,45 @@ module ts {
             }
         }
 
+        function checkElementTypeOfArrayOrString(arrayOrStringType: Type, expressionForError: Expression): Type {
+            Debug.assert(languageVersion < ScriptTarget.ES6);
+            var isJustString = allConstituentTypesHaveKind(arrayOrStringType, TypeFlags.StringLike);
+
+            // Check isJustString because removeTypesFromUnionType will only remove types if it doesn't result
+            // in an emptyObjectType. In this case, we actually do want the emptyObjectType.
+            var arrayType = isJustString ? emptyObjectType : removeTypesFromUnionType(arrayOrStringType, TypeFlags.StringLike, /*isTypeOfKind*/ true);
+            var hasStringConstituent = arrayOrStringType !== emptyObjectType && arrayOrStringType !== arrayType;
+
+            var reportedError = false;
+            if (hasStringConstituent && languageVersion < ScriptTarget.ES5) {
+                error(expressionForError, Diagnostics.Using_a_string_in_a_for_of_statement_is_only_supported_in_ECMAScript_5_and_higher);
+                reportedError = true;
+            }
+
+            if (isJustString) {
+                return stringType;
+            }
+
+            if (!isArrayLikeType(arrayType)) {
+                if (!reportedError) {
+                    error(expressionForError, Diagnostics.Type_0_is_not_an_array_type, typeToString(arrayType));
+                }
+                return hasStringConstituent ? stringType : unknownType;
+            }
+
+            var arrayElementType = getIndexTypeOfType(arrayType, IndexKind.Number) || unknownType;
+            if (hasStringConstituent) {
+                // This is just an optimization for the case where arrayOrStringType is string | string[]
+                if (arrayElementType.flags & TypeFlags.StringLike) {
+                    return stringType;
+                }
+
+                return getUnionType([arrayElementType, stringType]);
+            }
+
+            return arrayElementType;
+        }
+
         function checkBreakOrContinueStatement(node: BreakOrContinueStatement) {
             // Grammar checking
             checkGrammarStatementInAmbientContext(node) || checkGrammarBreakOrContinueStatement(node);
@@ -10931,7 +10961,6 @@ module ts {
         }
 
         function isUnknownIdentifier(location: Node, name: string): boolean {
-            // Do not call resolveName on a synthesized node!
             Debug.assert(!nodeIsSynthesized(location), "isUnknownIdentifier called with a synthesized location");
             return !resolveName(location, name, SymbolFlags.Value, /*nodeNotFoundMessage*/ undefined, /*nameArg*/ undefined) &&
                 !hasProperty(getGeneratedNamesForSourceFile(getSourceFile(location)), name);

src/compiler/diagnosticInformationMap.generated.ts

@@ -327,7 +327,6 @@ module ts {
         Property_0_does_not_exist_on_const_enum_1: { code: 2479, category: DiagnosticCategory.Error, key: "Property '{0}' does not exist on 'const' enum '{1}'." },
         let_is_not_allowed_to_be_used_as_a_name_in_let_or_const_declarations: { code: 2480, category: DiagnosticCategory.Error, key: "'let' is not allowed to be used as a name in 'let' or 'const' declarations." },
         Cannot_initialize_outer_scoped_variable_0_in_the_same_scope_as_block_scoped_declaration_1: { code: 2481, category: DiagnosticCategory.Error, key: "Cannot initialize outer scoped variable '{0}' in the same scope as block scoped declaration '{1}'." },
-        for_of_statements_are_only_available_when_targeting_ECMAScript_6_or_higher: { code: 2482, category: DiagnosticCategory.Error, key: "'for...of' statements are only available when targeting ECMAScript 6 or higher." },
         The_left_hand_side_of_a_for_of_statement_cannot_use_a_type_annotation: { code: 2483, category: DiagnosticCategory.Error, key: "The left-hand side of a 'for...of' statement cannot use a type annotation." },
         Export_declaration_conflicts_with_exported_declaration_of_0: { code: 2484, category: DiagnosticCategory.Error, key: "Export declaration conflicts with exported declaration of '{0}'" },
         The_left_hand_side_of_a_for_of_statement_cannot_be_a_previously_defined_constant: { code: 2485, category: DiagnosticCategory.Error, key: "The left-hand side of a 'for...of' statement cannot be a previously defined constant." },
@@ -339,6 +338,7 @@ module ts {
         The_left_hand_side_of_a_for_in_statement_cannot_be_a_destructuring_pattern: { code: 2491, category: DiagnosticCategory.Error, key: "The left-hand side of a 'for...in' statement cannot be a destructuring pattern." },
         Cannot_redeclare_identifier_0_in_catch_clause: { code: 2492, category: DiagnosticCategory.Error, key: "Cannot redeclare identifier '{0}' in catch clause" },
         Tuple_type_0_with_length_1_cannot_be_assigned_to_tuple_with_length_2: { code: 2493, category: DiagnosticCategory.Error, key: "Tuple type '{0}' with length '{1}' cannot be assigned to tuple with length '{2}'." },
+        Using_a_string_in_a_for_of_statement_is_only_supported_in_ECMAScript_5_and_higher: { code: 2494, category: DiagnosticCategory.Error, key: "Using a string in a 'for...of' statement is only supported in ECMAScript 5 and higher." },
         Import_declaration_0_is_using_private_name_1: { code: 4000, category: DiagnosticCategory.Error, key: "Import declaration '{0}' is using private name '{1}'." },
         Type_parameter_0_of_exported_class_has_or_is_using_private_name_1: { code: 4002, category: DiagnosticCategory.Error, key: "Type parameter '{0}' of exported class has or is using private name '{1}'." },
         Type_parameter_0_of_exported_interface_has_or_is_using_private_name_1: { code: 4004, category: DiagnosticCategory.Error, key: "Type parameter '{0}' of exported interface has or is using private name '{1}'." },

src/compiler/diagnosticMessages.json

@@ -1299,10 +1299,6 @@
         "category": "Error",
         "code": 2481
     },
-    "'for...of' statements are only available when targeting ECMAScript 6 or higher.": {
-        "category": "Error",
-        "code": 2482
-    },
     "The left-hand side of a 'for...of' statement cannot use a type annotation.": {
         "category": "Error",
         "code": 2483
@@ -1347,6 +1343,10 @@
         "category": "Error",
         "code": 2493
     },
+    "Using a string in a 'for...of' statement is only supported in ECMAScript 5 and higher.": {
+        "category": "Error",
+        "code": 2494
+    },
 
     "Import declaration '{0}' is using private name '{1}'.": {
         "category": "Error",

tests/baselines/reference/ES3For-ofTypeCheck1.errors.txt

@@ -0,0 +1,7 @@
+tests/cases/conformance/statements/for-ofStatements/ES3For-ofTypeCheck1.ts(1,15): error TS2494: Using a string in a 'for...of' statement is only supported in ECMAScript 5 and higher.
+
+
+==== tests/cases/conformance/statements/for-ofStatements/ES3For-ofTypeCheck1.ts (1 errors) ====
+    for (var v of "") { }
+                  ~~
+!!! error TS2494: Using a string in a 'for...of' statement is only supported in ECMAScript 5 and higher.

tests/baselines/reference/ES3For-ofTypeCheck1.js

@@ -0,0 +1,7 @@
+//// [ES3For-ofTypeCheck1.ts]
+for (var v of "") { }
+
+//// [ES3For-ofTypeCheck1.js]
+for (var _i = 0, _a = ""; _i < _a.length; _i++) {
+    var v = _a[_i];
+}

tests/baselines/reference/ES3For-ofTypeCheck2.js

@@ -0,0 +1,9 @@
+//// [ES3For-ofTypeCheck2.ts]
+for (var v of [true]) { }
+
+//// [ES3For-ofTypeCheck2.js]
+for (var _i = 0, _a = [
+    true
+]; _i < _a.length; _i++) {
+    var v = _a[_i];
+}

tests/baselines/reference/ES3For-ofTypeCheck2.types

@@ -0,0 +1,5 @@
+=== tests/cases/conformance/statements/for-ofStatements/ES3For-ofTypeCheck2.ts ===
+for (var v of [true]) { }
+>v : boolean
+>[true] : boolean[]
+

tests/baselines/reference/ES3For-ofTypeCheck4.errors.txt

@@ -0,0 +1,8 @@
+tests/cases/conformance/statements/for-ofStatements/ES3For-ofTypeCheck4.ts(2,17): error TS2494: Using a string in a 'for...of' statement is only supported in ECMAScript 5 and higher.
+
+
+==== tests/cases/conformance/statements/for-ofStatements/ES3For-ofTypeCheck4.ts (1 errors) ====
+    var union: string | string[];
+    for (const v of union) { }
+                    ~~~~~
+!!! error TS2494: Using a string in a 'for...of' statement is only supported in ECMAScript 5 and higher.

tests/baselines/reference/ES3For-ofTypeCheck4.js

@@ -0,0 +1,9 @@
+//// [ES3For-ofTypeCheck4.ts]
+var union: string | string[];
+for (const v of union) { }
+
+//// [ES3For-ofTypeCheck4.js]
+var union;
+for (var _i = 0; _i < union.length; _i++) {
+    var v = union[_i];
+}

tests/baselines/reference/ES3For-ofTypeCheck6.js

@@ -0,0 +1,9 @@
+//// [ES3For-ofTypeCheck6.ts]
+var union: string[] | number[];
+for (var v of union) { }
+
+//// [ES3For-ofTypeCheck6.js]
+var union;
+for (var _i = 0; _i < union.length; _i++) {
+    var v = union[_i];
+}

tests/baselines/reference/ES3For-ofTypeCheck6.types

@@ -0,0 +1,8 @@
+=== tests/cases/conformance/statements/for-ofStatements/ES3For-ofTypeCheck6.ts ===
+var union: string[] | number[];
+>union : string[] | number[]
+
+for (var v of union) { }
+>v : string | number
+>union : string[] | number[]
+

tests/baselines/reference/ES5For-of1.errors.txt

@@ -1,9 +1,9 @@
-tests/cases/conformance/statements/for-ofStatements/ES5For-of1.ts(1,1): error TS2482: 'for...of' statements are only available when targeting ECMAScript 6 or higher.
+tests/cases/conformance/statements/for-ofStatements/ES5For-of1.ts(2,5): error TS2304: Cannot find name 'console'.
 
 
 ==== tests/cases/conformance/statements/for-ofStatements/ES5For-of1.ts (1 errors) ====
     for (var v of ['a', 'b', 'c']) {
-    ~~~
-!!! error TS2482: 'for...of' statements are only available when targeting ECMAScript 6 or higher.
         console.log(v);
+        ~~~~~~~
+!!! error TS2304: Cannot find name 'console'.
     }

tests/baselines/reference/ES5For-of10.types

@@ -0,0 +1,29 @@
+=== tests/cases/conformance/statements/for-ofStatements/ES5For-of10.ts ===
+function foo() {
+>foo : () => { x: number; }
+
+    return { x: 0 };
+>{ x: 0 } : { x: number; }
+>x : number
+}
+for (foo().x of []) {
+>foo().x : number
+>foo() : { x: number; }
+>foo : () => { x: number; }
+>x : number
+>[] : undefined[]
+
+    for (foo().x of [])
+>foo().x : number
+>foo() : { x: number; }
+>foo : () => { x: number; }
+>x : number
+>[] : undefined[]
+
+        var p = foo().x;
+>p : number
+>foo().x : number
+>foo() : { x: number; }
+>foo : () => { x: number; }
+>x : number
+}

tests/baselines/reference/ES5For-of11.types

@@ -0,0 +1,8 @@
+=== tests/cases/conformance/statements/for-ofStatements/ES5For-of11.ts ===
+var v;
+>v : any
+
+for (v of []) { }
+>v : any
+>[] : undefined[]
+

tests/baselines/reference/ES5For-of12.errors.txt

@@ -1,7 +1,7 @@
-tests/cases/conformance/statements/for-ofStatements/ES5For-of12.ts(1,1): error TS2482: 'for...of' statements are only available when targeting ECMAScript 6 or higher.
+tests/cases/conformance/statements/for-ofStatements/ES5For-of12.ts(1,7): error TS2364: Invalid left-hand side of assignment expression.
 
 
 ==== tests/cases/conformance/statements/for-ofStatements/ES5For-of12.ts (1 errors) ====
-    for ([""] of []) { }
-    ~~~
-!!! error TS2482: 'for...of' statements are only available when targeting ECMAScript 6 or higher.
+    for ([""] of [[""]]) { }
+          ~~
+!!! error TS2364: Invalid left-hand side of assignment expression.

tests/baselines/reference/ES5For-of12.js

@@ -1,7 +1,11 @@
 //// [ES5For-of12.ts]
-for ([""] of []) { }
+for ([""] of [[""]]) { }
 
 //// [ES5For-of12.js]
-for (var _i = 0, _a = []; _i < _a.length; _i++) {
+for (var _i = 0, _a = [
+    [
+        ""
+    ]
+]; _i < _a.length; _i++) {
     "" = _a[_i][0];
 }

tests/baselines/reference/ES5For-of13.types

@@ -0,0 +1,9 @@
+=== tests/cases/conformance/statements/for-ofStatements/ES5For-of13.ts ===
+for (let v of ['a', 'b', 'c']) {
+>v : string
+>['a', 'b', 'c'] : string[]
+
+    var x = v;
+>x : string
+>v : string
+}

tests/baselines/reference/ES5For-of14.types

@@ -0,0 +1,9 @@
+=== tests/cases/conformance/statements/for-ofStatements/ES5For-of14.ts ===
+for (const v of []) {
+>v : any
+>[] : undefined[]
+
+    var x = v;
+>x : any
+>v : any
+}