From 532ab69256c0f6a207c32e1f63bad0a8bc3b3e66 Mon Sep 17 00:00:00 2001
From: Zak Henry <zak@cloudnc.com>
Date: Sun, 12 May 2019 15:15:21 +0100
Subject: [PATCH] feat(Typescript): Add generic data types for strongly typing
 nodes/links

---
 index.d.ts | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/index.d.ts b/index.d.ts
index 13190ef..b67f871 100644
--- a/index.d.ts
+++ b/index.d.ts
@@ -3,25 +3,26 @@
 // Definitions by: Nathan Westlake <https://github.com/CorayThan>
 // Definitions: https://github.com/DefinitelyTyped/DefinitelyTyped
 
+import { Graph, Link, Node, NodeId } from "ngraph.graph"
+
 declare module "ngraph.path" {
 
-    import { Graph, Link, Node, NodeId } from "ngraph.graph"
 
-    interface PathFinderOptions {
+    interface PathFinderOptions<NodeData, LinkData> {
         oriented?: boolean
         quitFast?: boolean
-        heuristic?: (from: NodeId, to: NodeId) => number
-        distance?: (from: NodeId, to: NodeId, link: Link) => number
+        heuristic?: (from: Node<NodeData>, to: Node<NodeData>) => number
+        distance?: (from: Node<NodeData>, to: Node<NodeData>, link: Link<NodeData>) => number
     }
 
-    interface PathFinder {
-        find: (from: NodeId, to: NodeId) => Node[]
+    interface PathFinder<NodeData> {
+        find: (from: NodeId, to: NodeId) => Node<NodeData>[]
     }
 
-    export function aStar(graph: Graph, options?: PathFinderOptions): PathFinder
+    export function aStar<NodeData, LinkData>(graph: Graph<NodeData, LinkData>, options?: PathFinderOptions<NodeData, LinkData>): PathFinder<NodeData>
 
-    export function aGreedy(graph: Graph, options?: PathFinderOptions): PathFinder
+    export function aGreedy<NodeData, LinkData>(graph: Graph<NodeData, LinkData>, options?: PathFinderOptions<NodeData, LinkData>): PathFinder<NodeData>
 
-    export function nba(graph: Graph, options?: PathFinderOptions): PathFinder
+    export function nba<NodeData, LinkData>(graph: Graph<NodeData, LinkData>, options?: PathFinderOptions<NodeData, LinkData>): PathFinder<NodeData>
 
 }