From dabe69856a5485c523145f02d0c319aaabd350ab Mon Sep 17 00:00:00 2001
From: Sebastian McKenzie <sebmck@gmail.com>
Date: Sun, 8 Feb 2015 09:31:19 +1100
Subject: [PATCH] add additional checks to transformers

---
 lib/6to5/transformation/transformers/es6/object-super.js      | 4 ++++
 .../transformers/validation/no-for-in-of-assignment.js        | 2 ++
 lib/6to5/transformation/transformers/validation/setters.js    | 4 ++++
 3 files changed, 10 insertions(+)

diff --git a/lib/6to5/transformation/transformers/es6/object-super.js b/lib/6to5/transformation/transformers/es6/object-super.js
index 814096b7c1..2ae32ef931 100644
--- a/lib/6to5/transformation/transformers/es6/object-super.js
+++ b/lib/6to5/transformation/transformers/es6/object-super.js
@@ -4,6 +4,10 @@ var ReplaceSupers = require("../../helpers/replace-supers");
 var util          = require("../../../util");
 var t             = require("../../../types");
 
+exports.check = function (node, parent) {
+  return t.isIdentifier(node, { name: "super" });
+};
+
 exports.Property = function (node, parent, scope, file) {
   if (!node.method) return;
 
diff --git a/lib/6to5/transformation/transformers/validation/no-for-in-of-assignment.js b/lib/6to5/transformation/transformers/validation/no-for-in-of-assignment.js
index e00e74d8ab..a8a8044e4f 100644
--- a/lib/6to5/transformation/transformers/validation/no-for-in-of-assignment.js
+++ b/lib/6to5/transformation/transformers/validation/no-for-in-of-assignment.js
@@ -2,6 +2,8 @@
 
 var t = require("../../../types");
 
+exports.check = t.isFor;
+
 exports.ForInStatement =
 exports.ForOfStatement = function (node, parent, scope, file) {
   var left = node.left;
diff --git a/lib/6to5/transformation/transformers/validation/setters.js b/lib/6to5/transformation/transformers/validation/setters.js
index bb4e7e1ab8..76a1555515 100644
--- a/lib/6to5/transformation/transformers/validation/setters.js
+++ b/lib/6to5/transformation/transformers/validation/setters.js
@@ -1,5 +1,9 @@
 "use strict";
 
+exports.check = function (node) {
+  return node.kind === "set";
+};
+
 exports.MethodDefinition =
 exports.Property = function (node, parent, scope, file) {
   if (node.kind === "set" && node.value.params.length !== 1) {