From 83d59b86dd1d7b38ce7ae9a4cb85e2eb89b5683e Mon Sep 17 00:00:00 2001
From: ShaMan123 <34343793+ShaMan123@users.noreply.github.com>
Date: Fri, 1 Jan 2021 12:39:56 +0200
Subject: [PATCH] Update index.ts

more generics
---
 src/types/index.ts | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/types/index.ts b/src/types/index.ts
index 474c34f3..98370b09 100644
--- a/src/types/index.ts
+++ b/src/types/index.ts
@@ -149,7 +149,7 @@ export interface EdgeProps<T = any> {
   markerEndId?: string;
   data?: T;
 }
-export interface EdgeSmoothStepProps extends EdgeProps {
+export interface EdgeSmoothStepProps<T = any> extends EdgeProps<T> {
   borderRadius?: number;
 }
 
@@ -242,20 +242,20 @@ export type FlowExportObject = {
 
 export type FitViewFunc = (fitViewOptions?: FitViewParams) => void;
 export type ProjectFunc = (position: XYPosition) => XYPosition;
-export type ToObjectFunc = () => FlowExportObject;
+export type ToObjectFunc<T = any> = () => FlowExportObject<T>;
 
-export type OnLoadParams = {
+export type OnLoadParams<T = any> = {
   zoomIn: () => void;
   zoomOut: () => void;
   zoomTo: (zoomLevel: number) => void;
   fitView: FitViewFunc;
   project: ProjectFunc;
-  getElements: () => Elements;
+  getElements: () => Elements<T>;
   setTransform: (transform: FlowTransform) => void;
-  toObject: ToObjectFunc;
+  toObject: ToObjectFunc<T>;
 };
 
-export type OnLoadFunc = (params: OnLoadParams) => void;
+export type OnLoadFunc<T = any> = (params: OnLoadParams<T>) => void;
 
 export interface Connection {
   source: ElementId | null;
@@ -359,4 +359,4 @@ export interface ZoomPanHelperFunctions {
   initialized: boolean;
 }
 
-export type OnEdgeUpdateFunc = (oldEdge: Edge, newConnection: Connection) => void;
+export type OnEdgeUpdateFunc<T = any> = (oldEdge: Edge<T>, newConnection: Connection) => void;