From 77c32f4ec2cc87331bf0f6d1432b680433606f51 Mon Sep 17 00:00:00 2001
From: Marquis Kurt <software@marquiskurt.net>
Date: Tue, 7 Jan 2020 11:24:34 -0500
Subject: [PATCH] Extend PrivateRoute to allow render functions

---
 src/interfaces/overrides.tsx | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/interfaces/overrides.tsx b/src/interfaces/overrides.tsx
index a73ec31..f78826b 100644
--- a/src/interfaces/overrides.tsx
+++ b/src/interfaces/overrides.tsx
@@ -78,14 +78,14 @@ export const ProfileRoute = (rest: any, component: Component) => (
 
 export const PrivateRoute = (props: IPrivateRouteProps) => {
     const { component, render, ...rest } = props;
+    const redir = (comp: any) =>
+        userLoggedIn ? comp : <Redirect to="/welcome" />;
     return (
         <Route
             {...rest}
             render={(compProps: any) =>
-                userLoggedIn() ? (
-                    React.createElement(component, compProps)
-                ) : (
-                    <Redirect to="/welcome" />
+                redir(
+                    React.createElement(render ? render : component, compProps)
                 )
             }
         />
@@ -93,5 +93,6 @@ export const PrivateRoute = (props: IPrivateRouteProps) => {
 };
 
 interface IPrivateRouteProps extends RouteProps {
-    component: any;
+    component?: any;
+    render?: any;
 }