From b9eb2feb3d63edd08969005a93c88e15944868b3 Mon Sep 17 00:00:00 2001
From: Roberto Ritger <roberto@ritger.com>
Date: Wed, 15 Nov 2017 15:46:37 -0800
Subject: [PATCH] A-Frame register custom geometry fix

---
 types/aframe/aframe-tests.ts | 12 ++++++++++--
 types/aframe/index.d.ts      | 13 +++++++++----
 2 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/types/aframe/aframe-tests.ts b/types/aframe/aframe-tests.ts
index 439d314f4a..5af66a09d1 100644
--- a/types/aframe/aframe-tests.ts
+++ b/types/aframe/aframe-tests.ts
@@ -1,5 +1,4 @@
 // Global
-
 const threeCamera = new AFRAME.THREE.Camera();
 AFRAME.TWEEN.Easing;
 
@@ -42,9 +41,18 @@ entity.addEventListener('child-detached', (event) => {
 const Component = AFRAME.registerComponent('test', {});
 
 // Scene
-
 const scene = document.querySelector('a-scene');
 scene.hasLoaded;
 
 // System
 const system = scene.systems['systemName'];
+
+// Register Custom Geometry
+AFRAME.registerGeometry('a-test-geometry', {
+	schema: {
+		groupIndex: { default: 0 }
+	},
+	init(data) {
+		this.geometry = new THREE.Geometry();
+	}
+});
diff --git a/types/aframe/index.d.ts b/types/aframe/index.d.ts
index f1f5cd7a20..803318a754 100644
--- a/types/aframe/index.d.ts
+++ b/types/aframe/index.d.ts
@@ -1,6 +1,7 @@
-// Type definitions for AFRAME 0.5
+// Type definitions for AFRAME 0.7
 // Project: https://aframe.io/
 // Definitions by: Paul Shannon <https://github.com/devpaul>
+//                 Roberto Ritger <https://github.com/bertoritger>
 // Definitions: https://github.com/DefinitelyTyped/DefinitelyTyped
 // TypeScript Version: 2.3
 
@@ -40,7 +41,7 @@ declare namespace AFrame {
 		primitives: { [ key: string ]: Entity };
 		registerComponent(name: string, component: ComponentDefinition): ComponentConstructor;
 		registerElement(name: string, element: ANode): void;
-		registerGeometry(name: string, geometery: THREE.Geometry): Geometry;
+		registerGeometry(name: string, geometry: GeometryDefinition): Geometry;
 		registerPrimitive(name: string, primitive: PrimitiveDefinition): void;
 		registerShader(name: string, shader: any): void;
 		registerSystem(name: string, definition: SystemDefinition): void;
@@ -101,7 +102,7 @@ declare namespace AFrame {
 		name: string;
 		schema: Schema;
 
-		init(): void;
+		init(data?: any): void;
 		pause(): void;
 		play(): void;
 		remove(): void;
@@ -124,7 +125,7 @@ declare namespace AFrame {
 		multiple?: boolean;
 		schema?: Schema;
 
-		init?(): void;
+		init?(data?: any): void;
 		pause?(): void;
 		play?(): void;
 		remove?(): void;
@@ -222,6 +223,10 @@ declare namespace AFrame {
 		[ key: string ]: any;
 	}
 
+	interface GeometryDefinition extends ComponentDefinition {
+		geometry?: THREE.Geometry;
+	}
+
 	interface GeometryDescriptor {
 		Geometry: Geometry;
 		schema: Schema;