Remove signup page and button if registration is closed.

src/components/layouts/layout.tsx

@@ -7,6 +7,7 @@ import { globalSiteTitle } from "~/utils/global";
 import Link from "next/link";
 import { useRouter } from "next/router";
 import { User } from "@prisma/client";
+import { api } from "~/utils/api";
 
 type TUser = {
 	user: User;
@@ -18,6 +19,9 @@ interface Props {
 
 export const LayoutPublic = ({ children }: Props): JSX.Element => {
 	const router = useRouter();
+	const { data: options, isLoading: loadingRegistration } =
+		api.settings.registrationAllowed.useQuery();
+
 	const currentPath = router.pathname;
 	return (
 		<div className="outer-content">
@@ -27,14 +31,16 @@ export const LayoutPublic = ({ children }: Props): JSX.Element => {
 				</div>
 
 				<div className="m-3 mx-0 flex w-10/12 justify-end">
-					<Link
-						href={
-							currentPath.includes("/auth/register") ? "/auth/login" : "/auth/register"
-						}
-						className="btn"
-					>
-						{currentPath === "/auth/register" ? "Login" : "Sign Up"}
-					</Link>
+					{options?.enableRegistration && !loadingRegistration ? (
+						<Link
+							href={
+								currentPath.includes("/auth/register") ? "/auth/login" : "/auth/register"
+							}
+							className="btn"
+						>
+							{currentPath === "/auth/register" ? "Login" : "Sign Up"}
+						</Link>
+					) : null}
 				</div>
 			</div>
 			{children}

src/pages/auth/register/index.tsx

@@ -5,6 +5,7 @@ import Head from "next/head";
 import React, { ReactElement } from "react";
 import { LayoutPublic } from "~/components/layouts/layout";
 import RegisterForm from "~/components/modules/registerForm";
+import { prisma } from "~/server/db";
 import { globalSiteTitle } from "~/utils/global";
 
 const Register = () => {
@@ -50,9 +51,29 @@ interface Props {
 export const getServerSideProps: GetServerSideProps<Props> = async (
 	context: GetServerSidePropsContext,
 ) => {
+	const options = await prisma.globalOptions.findFirst({
+		where: {
+			id: 1,
+		},
+		select: {
+			enableRegistration: true,
+		},
+	});
+	// easy check to see if the invite probably is a jwt token
+	const isJwt = !!context.query?.invite && context.query?.invite.length > 50;
+
 	const session = await getSession(context);
 	const messages = (await import(`~/locales/${context.locale}/common.json`)).default;
 
+	// redirect user to 404 if registration is disabled
+	if (!options?.enableRegistration && !isJwt) {
+		return {
+			redirect: {
+				destination: "/404",
+				permanent: false,
+			},
+		};
+	}
 	if (!session || !("user" in session)) {
 		return { props: { messages } };
 	}

src/server/api/routers/settingsRouter.ts

@@ -1,6 +1,16 @@
-import { createTRPCRouter, protectedProcedure } from "~/server/api/trpc";
+import { createTRPCRouter, protectedProcedure, publicProcedure } from "~/server/api/trpc";
 
 export const settingsRouter = createTRPCRouter({
+	registrationAllowed: publicProcedure.query(async ({ ctx }) => {
+		return await ctx.prisma.globalOptions.findFirst({
+			where: {
+				id: 1,
+			},
+			select: {
+				enableRegistration: true,
+			},
+		});
+	}),
 	// Set global options
 	getAllOptions: protectedProcedure.query(async ({ ctx }) => {
 		return await ctx.prisma.globalOptions.findFirst({