From 91c737c907f174b5252877876126a8b81e6cb831 Mon Sep 17 00:00:00 2001 From: Pinapelz Date: Fri, 7 Nov 2025 22:46:34 -0800 Subject: add support to limit registration to invite codes --- backend/schema.prisma | 6 ++ backend/src/index.ts | 3 + backend/src/routes/auth.ts | 22 +++++- backend/src/routes/server.ts | 36 +++++++++ frontend/src/pages/Admin.tsx | 164 ++++++++++++++++++++++++++++++++++++++++ frontend/src/pages/Register.tsx | 50 +++++++++++- 6 files changed, 276 insertions(+), 5 deletions(-) create mode 100644 backend/src/routes/server.ts diff --git a/backend/schema.prisma b/backend/schema.prisma index fbc2f90..26f57dd 100644 --- a/backend/schema.prisma +++ b/backend/schema.prisma @@ -55,3 +55,9 @@ model Charts { game Game @relation(fields: [gameInternalName], references: [internalName]) scores Score[] } + +model InviteCodes { + id Int @id @default(autoincrement()) + code String @unique + remaining Int +} diff --git a/backend/src/index.ts b/backend/src/index.ts index c0089c2..9511f28 100644 --- a/backend/src/index.ts +++ b/backend/src/index.ts @@ -11,6 +11,7 @@ import * as userRoutes from './routes/user'; import * as gameRoutes from './routes/game'; import * as scoreRoutes from './routes/score'; import * as adminRoutes from './routes/admin'; +import * as serverRoutes from './routes/server'; const app = express(); const port = 5000; @@ -48,6 +49,8 @@ startSessionCleanup(); app.post('/api/register', authRoutes.handleRegistration); app.post('/api/authenticate', authRoutes.handleAuthentication); app.post('/api/logout', requireAuth, authRoutes.handleLogout); +app.get('/api/info', serverRoutes.handleGetInstanceInfo); +app.post('/api/admin/createInvite', serverRoutes.handleCreateInviteCode); app.get('/api/me', userRoutes.handleMeRoute); app.get('/api/session', userRoutes.handleGetCurrentSession); diff --git a/backend/src/routes/auth.ts b/backend/src/routes/auth.ts index f857dea..8bc6274 100644 --- a/backend/src/routes/auth.ts +++ b/backend/src/routes/auth.ts @@ -6,12 +6,24 @@ import crypto from 'crypto'; export const handleRegistration = async (req: express.Request, res: express.Response) => { try { - const { username, password, email } = req.body; + const { username, password, email, code: inviteCode } = req.body; + const requireInvite = process.env.REQUIRE_INVITE === 'true'; if (!username || !password || !email) { return res.status(400).json({ error: 'All fields are required' }); } + if (requireInvite && !inviteCode) { + return res.status(400).json({ error: 'Invite code is required' }); + } + + if (requireInvite && inviteCode) { + const invite = await prisma.inviteCodes.findUnique({ where: { code: inviteCode } }); + if (!invite || invite.remaining <= 0) { + return res.status(400).json({ error: 'Invalid invite code' }); + } + } + const existingUser = await prisma.user.findFirst({ where: { OR: [ @@ -38,6 +50,14 @@ export const handleRegistration = async (req: express.Request, res: express.Resp } }); + // Decrement invite code usage if required + if (requireInvite && inviteCode) { + await prisma.inviteCodes.update({ + where: { code: inviteCode }, + data: { remaining: { decrement: 1 } } + }); + } + // Create session for the new user req.session.userId = user.id; const sessionId = await createSession(user.id); diff --git a/backend/src/routes/server.ts b/backend/src/routes/server.ts new file mode 100644 index 0000000..7377fff --- /dev/null +++ b/backend/src/routes/server.ts @@ -0,0 +1,36 @@ +import { prisma } from '../config/db'; +import express from 'express'; + +export const handleGetInstanceInfo = async (req: express.Request, res: express.Response) => { + try { + const userCount = await prisma.user.count(); + const requireInvite = process.env.REQUIRE_INVITE || false; + return res.status(200).json({ userCount, requireInvite }); + } catch (error) { + console.error('Unable to get instance info:', error); + res.status(500).json({ error: 'Internal server error' }); + } +} + +export const handleCreateInviteCode = async (req: express.Request, res: express.Response) => { + try { + const { uses, code } = req.body; + if (!uses) { + return res.status(400).json({ error: 'Missing required parameter: uses (number of maximum usages of this code)' }); + } + const codeAlreadyExists = await prisma.inviteCodes.findUnique({ where: { code } }); + if (codeAlreadyExists) { + return res.status(400).json({ error: 'Invite code already exists' }); + } + const inviteCode = await prisma.inviteCodes.create({ + data: { + code: code || Math.random().toString(36).substring(2, 15), + remaining: uses, + }, + }); + return res.status(200).json({ inviteCode }); + } catch (error) { + console.error('Unable to create invite code:', error); + res.status(500).json({ error: 'Internal server error' }); + } +} diff --git a/frontend/src/pages/Admin.tsx b/frontend/src/pages/Admin.tsx index f494fc2..043a32e 100644 --- a/frontend/src/pages/Admin.tsx +++ b/frontend/src/pages/Admin.tsx @@ -8,12 +8,19 @@ import { useState } from "react"; const Admin = () => { const { user, isLoading, logout } = useAuth(); const [showAddGame, setShowAddGame] = useState(false); + const [showCreateInvite, setShowCreateInvite] = useState(false); const [formData, setFormData] = useState({ gameInternalName: '', gameFormattedName: '', gameDescription: '' }); + const [inviteFormData, setInviteFormData] = useState({ + uses: '', + code: '' + }); const [isSubmitting, setIsSubmitting] = useState(false); + const [isCreatingInvite, setIsCreatingInvite] = useState(false); + const [createdInviteCode, setCreatedInviteCode] = useState(null); const navigate = useNavigate(); const handleLogout = async () => { @@ -34,6 +41,14 @@ const Admin = () => { })); }; + const handleInviteInputChange = (e: React.ChangeEvent) => { + const { name, value } = e.target; + setInviteFormData(prev => ({ + ...prev, + [name]: value + })); + }; + const handleSubmit = async (e: React.FormEvent) => { e.preventDefault(); @@ -75,6 +90,65 @@ const Admin = () => { } }; + const handleInviteSubmit = async (e: React.FormEvent) => { + e.preventDefault(); + + if (!inviteFormData.uses) { + alert('Please specify the number of uses for the invite code'); + return; + } + + const uses = parseInt(inviteFormData.uses); + if (isNaN(uses) || uses <= 0) { + alert('Please enter a valid number of uses'); + return; + } + + setIsCreatingInvite(true); + + try { + const requestBody: { uses: number; code?: string } = { uses }; + if (inviteFormData.code.trim()) { + requestBody.code = inviteFormData.code.trim(); + } + + const response = await fetch(import.meta.env.VITE_API_URL + '/admin/createInvite', { + method: 'POST', + headers: { + 'Content-Type': 'application/json', + }, + credentials: 'include', + body: JSON.stringify(requestBody), + }); + + if (!response.ok) { + const error = await response.json(); + throw new Error(error.error || 'Failed to create invite code'); + } + + const result = await response.json(); + setCreatedInviteCode(result.inviteCode.code); + setInviteFormData({ + uses: '', + code: '' + }); + + } catch (error) { + console.error('Failed to create invite code:', error); + alert(error instanceof Error ? error.message : 'Failed to create invite code'); + } finally { + setIsCreatingInvite(false); + } + }; + + const copyToClipboard = (text: string) => { + navigator.clipboard.writeText(text).then(() => { + alert('Invite code copied to clipboard!'); + }).catch(() => { + alert('Failed to copy to clipboard'); + }); + }; + if (isLoading) { return (
@@ -112,6 +186,96 @@ const Admin = () => { Welcome Mirage Webmaster! Here are a variety of settings and tools you can use to customize the experience

+ + {/* Create Invite Code Section */} +
+
+ + {showCreateInvite && ( +
+

+ Generate invite codes to allow new users to register. You can specify how many times the code can be used + and optionally set a custom code (otherwise one will be generated automatically). +

+ + {createdInviteCode && ( +
+

Invite Code Created Successfully!

+
+ + {createdInviteCode} + + +
+
+ )} + +
+
+ + +
+
+ + +
+
+ +
+
+
+ )} +
+
+ {/* Add New Game Section */}
diff --git a/frontend/src/pages/Register.tsx b/frontend/src/pages/Register.tsx index 90edbfd..5b53f46 100644 --- a/frontend/src/pages/Register.tsx +++ b/frontend/src/pages/Register.tsx @@ -1,4 +1,4 @@ -import { useState } from 'react'; +import { useEffect, useState } from 'react'; import { Link, useNavigate } from 'react-router'; import { useAuth } from '../contexts/AuthContext'; @@ -7,11 +7,26 @@ const Register = () => { username: '', email: '', password: '', - confirmPassword: '' + confirmPassword: '', + code: '' }); const [errors, setErrors] = useState>({}); + const [requireInvite, setRequireInvite] = useState(false); const [isLoading, setIsLoading] = useState(false); + useEffect(() => { + const fetchServerInfo = async () => { + try { + const response = await fetch(import.meta.env.VITE_API_URL + "/info"); + const data = await response.json(); + setRequireInvite(Boolean(data.requireInvite)); + } catch (error) { + console.error('Error fetching server info:', error); + } + }; + fetchServerInfo(); + }, []); + const handleChange = (e: React.ChangeEvent) => { const { name, value } = e.target; setFormData(prev => ({ @@ -53,6 +68,10 @@ const Register = () => { newErrors.confirmPassword = 'Passwords do not match'; } + if (requireInvite && !formData.code.trim()) { + newErrors.code = 'Invite code is required'; + } + setErrors(newErrors); return Object.keys(newErrors).length === 0; }; @@ -67,11 +86,14 @@ const Register = () => { setIsLoading(true); try { - const result = await register({ + const registrationData = { username: formData.username, email: formData.email, password: formData.password, - }); + ...(requireInvite && { code: formData.code }) + }; + + const result = await register(registrationData); if (!result.success) { setErrors({ general: result.error || 'Registration failed. Please try again.' }); @@ -131,6 +153,26 @@ const Register = () => { )}
+ {requireInvite && ( +
+ + + {errors.code && ( +

{errors.code}

+ )} +
+ )} +