feat: Added support for ref attributes

KnisterPeter · Feb 1, 2016 · d6f5b46 · d6f5b46
1 parent f7076e6
commit d6f5b46
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 1 deletion.
diff --git a/index.ts b/index.ts
@@ -332,6 +332,7 @@ export class Generator {
   public props(name: string, props: IPropTypes, fn?: () => void): void {
     this.interface(`${name}Props`, () => {
       this.prop('key', 'any', true);
+      this.prop('ref', 'any', true);
       Object.keys(props).forEach((propName: any) => {
         const prop: IProp = props[propName];
         this.prop(propName, prop.type, prop.optional, prop.documentation);

diff --git a/tests/es6-class.d.ts b/tests/es6-class.d.ts
@@ -4,6 +4,7 @@ declare module 'component' {
 
 	export interface ComponentProps {
 		key?: any;
+		ref?: any;
 		/**
 		 * This is a jsdoc comment for optionalAny.
 		 */

diff --git a/tests/es7-class.d.ts b/tests/es7-class.d.ts
@@ -4,6 +4,7 @@ declare module 'component' {
 
 	export interface ComponentProps {
 		key?: any;
+		ref?: any;
 		/**
 		 * This is a jsdoc comment for optionalAny.
 		 */

diff --git a/tests/generator-test.ts b/tests/generator-test.ts
@@ -26,7 +26,7 @@ describe('The Generator', () => {
   });
   it('should write a property interface', () => {
     generator.props('Name', {prop: {type: 'type', optional: true}});
-    assert.equal(generator.toString(), 'export interface NameProps {\n\tkey?: any;\n\tprop?: type;\n}\n');
+    assert.equal(generator.toString(), 'export interface NameProps {\n\tkey?: any;\n\tref?: any;\n\tprop?: type;\n}\n');
   });
   it('should write a class with props declaration', () => {
     generator.class('Name', true);