diff --git a/src/components/ProfileMenu.tsx b/src/components/ProfileMenu.tsx index e71c6c6..86e4b89 100644 --- a/src/components/ProfileMenu.tsx +++ b/src/components/ProfileMenu.tsx @@ -1,6 +1,6 @@ import { Menu, Transition } from "@headlessui/react"; import { Fragment } from "react"; -import { signOut } from "next-auth/react"; +import { signOut, useSession } from "next-auth/react"; import { useRouter } from "next/router"; import { api } from "~/utils/api"; @@ -8,6 +8,7 @@ export default function ProfileMenu() { const { mutateAsync: createBillingPortalSession } = api.stripe.createBillingPortalSession.useMutation(); const { push } = useRouter(); + const { data: session } = useSession(); return ( @@ -15,7 +16,7 @@ export default function ProfileMenu() { Open user menu diff --git a/src/pages/share/[videoId].tsx b/src/pages/share/[videoId].tsx index 9bdf686..cdde5d2 100644 --- a/src/pages/share/[videoId].tsx +++ b/src/pages/share/[videoId].tsx @@ -10,10 +10,11 @@ import { getTime } from "~/utils/getTime"; import { ShareModal } from "~/components/ShareModal"; import { useSession } from "next-auth/react"; import VideoMoreMenu from "~/components/VideoMoreMenu"; +import ProfileMenu from "~/components/ProfileMenu"; const VideoList: NextPage = () => { const router = useRouter(); - const { data: session } = useSession(); + const { status, data: session } = useSession(); const { videoId } = router.query as { videoId: string }; const { data: video, isLoading } = api.video.get.useQuery( @@ -59,6 +60,11 @@ const VideoList: NextPage = () => { {video && video.userId === session?.user.id ? ( ) : null} + {status === "authenticated" && ( +
+ +
+ )}
diff --git a/src/pages/videos.tsx b/src/pages/videos.tsx index 40d5051..a6d7348 100644 --- a/src/pages/videos.tsx +++ b/src/pages/videos.tsx @@ -33,7 +33,7 @@ const VideoList: NextPage = () => {
{status === "authenticated" && ( -
+
)}