feat: Add configurable maximum email limit for users

This commit is contained in:
beilunyang
2025-02-28 00:30:37 +08:00
parent f86d944c25
commit b1d898e298
5 changed files with 60 additions and 12 deletions
+11 -6
View File
@@ -1,28 +1,32 @@
import { Role, ROLES } from "@/lib/permissions" import { Role, ROLES } from "@/lib/permissions"
import { getRequestContext } from "@cloudflare/next-on-pages" import { getRequestContext } from "@cloudflare/next-on-pages"
import { EMAIL_CONFIG } from "@/config"
export const runtime = "edge" export const runtime = "edge"
export async function GET() { export async function GET() {
const env = getRequestContext().env const env = getRequestContext().env
const [defaultRole, emailDomains, adminContact] = await Promise.all([ const [defaultRole, emailDomains, adminContact, maxEmails] = await Promise.all([
env.SITE_CONFIG.get("DEFAULT_ROLE"), env.SITE_CONFIG.get("DEFAULT_ROLE"),
env.SITE_CONFIG.get("EMAIL_DOMAINS"), env.SITE_CONFIG.get("EMAIL_DOMAINS"),
env.SITE_CONFIG.get("ADMIN_CONTACT") env.SITE_CONFIG.get("ADMIN_CONTACT"),
env.SITE_CONFIG.get("MAX_EMAILS")
]) ])
return Response.json({ return Response.json({
defaultRole: defaultRole || ROLES.CIVILIAN, defaultRole: defaultRole || ROLES.CIVILIAN,
emailDomains: emailDomains || "", emailDomains: emailDomains || "",
adminContact: adminContact || "" adminContact: adminContact || "",
maxEmails: maxEmails || EMAIL_CONFIG.MAX_ACTIVE_EMAILS.toString()
}) })
} }
export async function POST(request: Request) { export async function POST(request: Request) {
const { defaultRole, emailDomains, adminContact } = await request.json() as { const { defaultRole, emailDomains, adminContact, maxEmails } = await request.json() as {
defaultRole: Exclude<Role, typeof ROLES.EMPEROR>, defaultRole: Exclude<Role, typeof ROLES.EMPEROR>,
emailDomains: string, emailDomains: string,
adminContact: string adminContact: string,
maxEmails: string
} }
if (![ROLES.DUKE, ROLES.KNIGHT, ROLES.CIVILIAN].includes(defaultRole)) { if (![ROLES.DUKE, ROLES.KNIGHT, ROLES.CIVILIAN].includes(defaultRole)) {
@@ -33,7 +37,8 @@ export async function POST(request: Request) {
await Promise.all([ await Promise.all([
env.SITE_CONFIG.put("DEFAULT_ROLE", defaultRole), env.SITE_CONFIG.put("DEFAULT_ROLE", defaultRole),
env.SITE_CONFIG.put("EMAIL_DOMAINS", emailDomains), env.SITE_CONFIG.put("EMAIL_DOMAINS", emailDomains),
env.SITE_CONFIG.put("ADMIN_CONTACT", adminContact) env.SITE_CONFIG.put("ADMIN_CONTACT", adminContact),
env.SITE_CONFIG.put("MAX_EMAILS", maxEmails)
]) ])
return Response.json({ success: true }) return Response.json({ success: true })
+5 -3
View File
@@ -14,12 +14,14 @@ export const runtime = "edge"
export async function POST(request: Request) { export async function POST(request: Request) {
const db = createDb() const db = createDb()
const env = getRequestContext().env
const userId = await getUserId() const userId = await getUserId()
const userRole = await getUserRole(userId!) const userRole = await getUserRole(userId!)
try { try {
if (userRole !== ROLES.EMPEROR) { if (userRole !== ROLES.EMPEROR) {
const maxEmails = await env.SITE_CONFIG.get("MAX_EMAILS") || EMAIL_CONFIG.MAX_ACTIVE_EMAILS.toString()
const activeEmailsCount = await db const activeEmailsCount = await db
.select({ count: sql<number>`count(*)` }) .select({ count: sql<number>`count(*)` })
.from(emails) .from(emails)
@@ -30,9 +32,9 @@ export async function POST(request: Request) {
) )
) )
if (Number(activeEmailsCount[0].count) >= EMAIL_CONFIG.MAX_ACTIVE_EMAILS) { if (Number(activeEmailsCount[0].count) >= Number(maxEmails)) {
return NextResponse.json( return NextResponse.json(
{ error: `已达到最大邮箱数量限制 (${EMAIL_CONFIG.MAX_ACTIVE_EMAILS})` }, { error: `已达到最大邮箱数量限制 (${maxEmails})` },
{ status: 403 } { status: 403 }
) )
} }
@@ -51,7 +53,7 @@ export async function POST(request: Request) {
) )
} }
const domainString = await getRequestContext().env.SITE_CONFIG.get("EMAIL_DOMAINS") const domainString = await env.SITE_CONFIG.get("EMAIL_DOMAINS")
const domains = domainString ? domainString.split(',') : ["moemail.app"] const domains = domainString ? domainString.split(',') : ["moemail.app"]
if (!domains || !domains.includes(domain)) { if (!domains || !domains.includes(domain)) {
+17 -1
View File
@@ -49,9 +49,22 @@ export function EmailList({ onEmailSelect, selectedEmailId }: EmailListProps) {
const [nextCursor, setNextCursor] = useState<string | null>(null) const [nextCursor, setNextCursor] = useState<string | null>(null)
const [loadingMore, setLoadingMore] = useState(false) const [loadingMore, setLoadingMore] = useState(false)
const [total, setTotal] = useState(0) const [total, setTotal] = useState(0)
const [maxEmails, setMaxEmails] = useState<number>(EMAIL_CONFIG.MAX_ACTIVE_EMAILS)
const [emailToDelete, setEmailToDelete] = useState<Email | null>(null) const [emailToDelete, setEmailToDelete] = useState<Email | null>(null)
const { toast } = useToast() const { toast } = useToast()
const fetchMaxEmails = async () => {
try {
const res = await fetch("/api/config")
if (res.ok) {
const data = await res.json() as { maxEmails: string }
setMaxEmails(Number(data.maxEmails))
}
} catch (error) {
console.error("Failed to fetch max emails:", error)
}
}
const fetchEmails = async (cursor?: string) => { const fetchEmails = async (cursor?: string) => {
try { try {
const url = new URL("/api/emails", window.location.origin) const url = new URL("/api/emails", window.location.origin)
@@ -112,6 +125,9 @@ export function EmailList({ onEmailSelect, selectedEmailId }: EmailListProps) {
useEffect(() => { useEffect(() => {
if (session) fetchEmails() if (session) fetchEmails()
if (session && role !== ROLES.EMPEROR) {
fetchMaxEmails()
}
// eslint-disable-next-line react-hooks/exhaustive-deps // eslint-disable-next-line react-hooks/exhaustive-deps
}, [session]) }, [session])
@@ -173,7 +189,7 @@ export function EmailList({ onEmailSelect, selectedEmailId }: EmailListProps) {
{role === ROLES.EMPEROR ? ( {role === ROLES.EMPEROR ? (
`${total}/∞ 个邮箱` `${total}/∞ 个邮箱`
) : ( ) : (
`${total}/${EMAIL_CONFIG.MAX_ACTIVE_EMAILS} 个邮箱` `${total}/${maxEmails} 个邮箱`
)} )}
</span> </span>
</div> </div>
+23 -2
View File
@@ -13,11 +13,13 @@ import {
SelectTrigger, SelectTrigger,
SelectValue, SelectValue,
} from "@/components/ui/select" } from "@/components/ui/select"
import { EMAIL_CONFIG } from "@/config"
export function ConfigPanel() { export function ConfigPanel() {
const [defaultRole, setDefaultRole] = useState<string>("") const [defaultRole, setDefaultRole] = useState<string>("")
const [emailDomains, setEmailDomains] = useState<string>("") const [emailDomains, setEmailDomains] = useState<string>("")
const [adminContact, setAdminContact] = useState<string>("") const [adminContact, setAdminContact] = useState<string>("")
const [maxEmails, setMaxEmails] = useState<string>(EMAIL_CONFIG.MAX_ACTIVE_EMAILS.toString())
const [loading, setLoading] = useState(false) const [loading, setLoading] = useState(false)
const { toast } = useToast() const { toast } = useToast()
@@ -31,10 +33,14 @@ export function ConfigPanel() {
if (res.ok) { if (res.ok) {
const data = await res.json() as { const data = await res.json() as {
defaultRole: Exclude<Role, typeof ROLES.EMPEROR>, defaultRole: Exclude<Role, typeof ROLES.EMPEROR>,
emailDomains: string emailDomains: string,
adminContact: string,
maxEmails: string
} }
setDefaultRole(data.defaultRole) setDefaultRole(data.defaultRole)
setEmailDomains(data.emailDomains) setEmailDomains(data.emailDomains)
setAdminContact(data.adminContact)
setMaxEmails(data.maxEmails || EMAIL_CONFIG.MAX_ACTIVE_EMAILS.toString())
} }
} }
@@ -47,7 +53,8 @@ export function ConfigPanel() {
body: JSON.stringify({ body: JSON.stringify({
defaultRole, defaultRole,
emailDomains, emailDomains,
adminContact adminContact,
maxEmails: maxEmails || EMAIL_CONFIG.MAX_ACTIVE_EMAILS.toString()
}), }),
}) })
@@ -112,6 +119,20 @@ export function ConfigPanel() {
</div> </div>
</div> </div>
<div className="flex items-center gap-4">
<span className="text-sm">:</span>
<div className="flex-1">
<Input
type="number"
min="1"
max="100"
value={maxEmails}
onChange={(e) => setMaxEmails(e.target.value)}
placeholder={`默认为 ${EMAIL_CONFIG.MAX_ACTIVE_EMAILS}`}
/>
</div>
</div>
<Button <Button
onClick={handleSave} onClick={handleSave}
disabled={loading} disabled={loading}
+4
View File
@@ -32,6 +32,10 @@ export async function middleware(request: Request) {
) )
} }
if (pathname === '/api/config' && request.method === 'GET') {
return NextResponse.next()
}
for (const [route, permission] of Object.entries(API_PERMISSIONS)) { for (const [route, permission] of Object.entries(API_PERMISSIONS)) {
if (pathname.startsWith(route)) { if (pathname.startsWith(route)) {
const hasAccess = await checkPermission(permission) const hasAccess = await checkPermission(permission)