babe plugin add Component name on div

plugin.js

module.exports = function (babel) {
  var t = babel.types;
  var visitor = {
    ClassDeclaration: function(code) {
    	var node = code.node;
    	var className = node.id.name;

    	var renderMethod;
	    for (i=0; i<code.node.body.body.length; i++) {
	    	var method = code.node.body.body[i];
	    	if (method.type === 'ClassMethod' && method.key.name === 'render') {
	    		renderMethod = method;
	    	}
	    }

	    var returnStatement;
	    for (i=0; i<renderMethod.body.body.length; i++) {
	   		var statement = renderMethod.body.body[i];
		   	if (statement.type === 'ReturnStatement') {
		   		returnStatement = statement;
		   	}
		}

    	var jsxelement = returnStatement.argument.openingElement
        var attrDisplayName = t.jSXIdentifier('data-component');
        jsxelement.attributes.push(t.jSXAttribute(attrDisplayName, t.stringLiteral(className)));

    },
  };

  return {
    visitor: visitor
  };
}