start to implement trait

src/trait.js

@@ -0,0 +1,36 @@
+/*!
+ * Copyright (C) 2016 Glayzzle (BSD3 License)
+ * @authors https://github.com/glayzzle/php-writer/graphs/contributors
+ * @url http://glayzzle.com/php-writer
+ */
+
+var filter = require('./helpers/filter');
+
+/**
+ * @constructor
+ */
+var Trait = function(ast) {
+  this.ast = ast;
+};
+
+/**
+ * Change the trait name
+ * @return {Trait}
+ */
+Trait.prototype.setName = function(name) {
+  this.ast[1] = name;
+  return this;
+};
+
+/**
+ * @return {Interface|Null}
+ */
+Trait.locate = function(ast, name) {
+  return filter(ast, 'trait', function(node) {
+    if (node[1] === name) {
+      return new Trait(node);
+    }
+  });
+};
+
+module.exports = Trait;

test/trait.js

@@ -0,0 +1,30 @@
+/*!
+ * Copyright (C) 2016 Glayzzle (BSD3 License)
+ * @authors https://github.com/glayzzle/php-writer/graphs/contributors
+ * @url http://glayzzle.com/php-writer
+ */
+
+var should = require('./assert');
+var writer = require('../src/index');
+
+describe('Trait', function() {
+
+  var test = new writer([
+    '<?php',
+    'trait foo {',
+    '\tconst BAR = 1;',
+    '}'
+  ].join('\n'));
+  var traitObj;
+
+  it('findTrait', function() {
+    traitObj = test.findTrait('foo');
+    traitObj.should.be.Object();
+  });
+
+  it('setName', function() {
+    traitObj.setName('baz');
+    traitObj.ast[1].should.be.equal('baz');
+  });
+
+});