vale
/
panel
forked from administration/panel
1
0
Fork 0
Code Pull Requests Activity

feat: add functionality to classification page

user-stream
Lea 2023-09-02 23:48:25 +02:00 committed by insert
parent fc51faf9e1
commit 373c66a251
4 changed files with 311 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

129
app/panel/shield/classifications/page.tsx
View File

@ -1,8 +1,54 @@
import EmailClassificationRow from "@/components/pages/shield/EmailClassificationRow"; "use client";
import EmailClassificationRow, {
CLASSIFICATIONS,
} from "@/components/pages/shield/EmailClassificationRow";
import {
AlertDialog,
AlertDialogAction,
AlertDialogCancel,
AlertDialogContent,
AlertDialogDescription,
AlertDialogFooter,
AlertDialogHeader,
AlertDialogTitle,
AlertDialogTrigger,
} from "@/components/ui/alert-dialog";
import { Button } from "@/components/ui/button";
import { Card } from "@/components/ui/card"; import { Card } from "@/components/ui/card";
import { Table, TableBody, TableHead, TableHeader, TableRow } from "@/components/ui/table"; import { Command, CommandItem } from "@/components/ui/command";
import { Input } from "@/components/ui/input";
import {
Popover,
PopoverContent,
PopoverTrigger,
} from "@/components/ui/popover";
import {
Table,
TableBody,
TableHead,
TableHeader,
TableRow,
} from "@/components/ui/table";
import { toast } from "@/components/ui/use-toast";
import { createEmailClassification, fetchEmailClassifications } from "@/lib/actions";
import { EmailClassification } from "@/lib/db";
import { useEffect, useState } from "react";
export default function Classifications() { export default function Classifications() {
const [loaded, setLoaded] = useState(false);
const [domains, setDomains] = useState([] as EmailClassification[]);
const [domainDraft, setDomainDraft] = useState("");
const [classificationDraft, setClassificationDraft] = useState<string>("");
const [classificationOpen, setClassificationOpen] = useState(false);
useEffect(() => {
fetchEmailClassifications().then((domains) => {
setDomains(domains);
setLoaded(true);
});
}, []);
return ( return (
<Card> <Card>
<Table> <Table>
@ -11,10 +57,87 @@ export default function Classifications() {
<TableHead>Domain</TableHead> <TableHead>Domain</TableHead>
<TableHead>Classification</TableHead> <TableHead>Classification</TableHead>
<TableHead>Action</TableHead> <TableHead>Action</TableHead>
<TableHead className="text-end pr-2">
<AlertDialog>
<AlertDialogTrigger asChild>
<Button disabled={!loaded}>Add</Button>
</AlertDialogTrigger>
<AlertDialogContent>
<AlertDialogHeader>
<AlertDialogTitle>Create classification</AlertDialogTitle>
<AlertDialogDescription className="flex flex-row gap-1">
<Input
value={domainDraft}
onChange={(e) => setDomainDraft(e.currentTarget.value)}
placeholder="reddit.com"
/>
<Popover
open={classificationOpen}
onOpenChange={setClassificationOpen}
>
<PopoverTrigger asChild>
<Button
variant="outline"
role="combobox"
aria-expanded={classificationOpen}
>
{classificationDraft || "Classification"}
</Button>
</PopoverTrigger>
<PopoverContent>
<Command>
{CLASSIFICATIONS.map((c) => (
<CommandItem key={c} onSelect={() => {
setClassificationDraft(c);
setClassificationOpen(false);
}}>
{c}
</CommandItem>
))}
</Command>
</PopoverContent>
</Popover>
</AlertDialogDescription>
</AlertDialogHeader>
<AlertDialogFooter>
<AlertDialogCancel>Cancel</AlertDialogCancel>
<AlertDialogAction
disabled={!domainDraft || !classificationDraft}
onClick={async () => {
try {
await createEmailClassification(domainDraft, classificationDraft);
setDomains([...domains, { _id: domainDraft, classification: classificationDraft }]);
setDomainDraft("");
setClassificationDraft("");
setClassificationOpen(false);
toast({
title: "Classification created",
});
} catch (e) {
toast({
title: "Failed to create classification",
description: String(e),
variant: "destructive",
});
}
}}
>
Create
</AlertDialogAction>
</AlertDialogFooter>
</AlertDialogContent>
</AlertDialog>
</TableHead>
</TableRow> </TableRow>
</TableHeader> </TableHeader>
<TableBody> <TableBody>
<EmailClassificationRow domain="sex.org" classification="DISPOSABLE" /> {domains.map((domain) => (
<EmailClassificationRow
key={domain._id}
domain={domain._id}
classification={domain.classification}
/>
))}
</TableBody> </TableBody>
</Table> </Table>
</Card> </Card>

181
components/pages/shield/EmailClassificationRow.tsx
View File

@ -1,68 +1,137 @@
"use client" "use client";
import {
AlertDialog,
AlertDialogAction,
AlertDialogContent,
AlertDialogFooter,
AlertDialogHeader,
AlertDialogTrigger,
AlertDialogCancel,
AlertDialogDescription,
} from "@/components/ui/alert-dialog";
import { Button } from "@/components/ui/button"; import { Button } from "@/components/ui/button";
import { Command, CommandItem } from "@/components/ui/command"; import { Command, CommandItem } from "@/components/ui/command";
import { Popover, PopoverContent, PopoverTrigger } from "@/components/ui/popover"; import {
Popover,
PopoverContent,
PopoverTrigger,
} from "@/components/ui/popover";
import { TableCell, TableRow } from "@/components/ui/table"; import { TableCell, TableRow } from "@/components/ui/table";
import { toast } from "@/components/ui/use-toast";
import { deleteEmailClassification, updateEmailClassification } from "@/lib/actions";
import { cn } from "@/lib/utils"; import { cn } from "@/lib/utils";
import { Check } from "lucide-react"; import { Check } from "lucide-react";
import { useState } from "react"; import { useState } from "react";
const CLASSIFICATIONS = [ export const CLASSIFICATIONS = [
"DISPOSABLE", "DISPOSABLE",
"GAYSEX", "GAYSEX",
"HOMOSEXUAL", "HOMOSEXUAL",
"FDGIUHDIFUIOFZH", "FDGIUHDIFUIOFZH",
]; ];
export default function EmailClassificationRow({ domain, ...props }: { domain: string, classification: string }) { export default function EmailClassificationRow({
const [selectClassification, setSelectClassification] = useState(false); domain,
const [classification, setClassification] = useState(props.classification); ...props
const [deleted, setDeleted] = useState(false); }: {
domain: string;
return deleted ? null : ( classification: string;
<TableRow> }) {
<TableCell>{domain}</TableCell> const [classification, setClassification] = useState(props.classification);
<TableCell> const [selectClassification, setSelectClassification] = useState(false);
<Popover open={selectClassification} onOpenChange={setSelectClassification}> const [deleted, setDeleted] = useState(false);
<PopoverTrigger asChild>
<Button return deleted ? null : (
variant="outline" <TableRow>
role="combobox" <TableCell>{domain}</TableCell>
aria-expanded={selectClassification} <TableCell>
>{classification}</Button> <Popover
</PopoverTrigger> open={selectClassification}
<PopoverContent> onOpenChange={setSelectClassification}
<Command> >
{CLASSIFICATIONS.map((c) => ( <PopoverTrigger asChild>
<CommandItem <Button
variant="outline"
role="combobox"
aria-expanded={selectClassification}
>
{classification}
</Button>
</PopoverTrigger>
<PopoverContent>
<Command>
{CLASSIFICATIONS.map((c) => (
<CommandItem
key={c} key={c}
onSelect={() => { onSelect={async () => {
setSelectClassification(false); try {
setClassification(c); await updateEmailClassification(domain, c);
setSelectClassification(false);
setClassification(c);
toast({
title: "Classification updated",
description: `${domain} is now classified as ${c}`,
});
} catch (e) {
toast({
title: "Failed to update classification",
description: String(e),
variant: "destructive",
});
}
}} }}
> >
<Check <Check
className={cn( className={cn(
"mr-2 h-4 w-4", "mr-2 h-4 w-4",
c == classification c == classification ? "opacity-100" : "opacity-0"
? "opacity-100" )}
: "opacity-0" />
)} {c}
/> </CommandItem>
{c} ))}
</CommandItem> </Command>
))} </PopoverContent>
</Command> </Popover>
</PopoverContent> </TableCell>
</Popover> <TableCell>
</TableCell> <AlertDialog>
<TableCell> <AlertDialogTrigger asChild>
<Button <Button>Remove</Button>
variant="destructive" </AlertDialogTrigger>
onClick={() => setDeleted(true)} <AlertDialogContent>
>Remove</Button> <AlertDialogHeader>
</TableCell> <AlertDialogDescription>
</TableRow> Delete classification for {domain}?
); </AlertDialogDescription>
} </AlertDialogHeader>
<AlertDialogFooter>
<AlertDialogCancel>Cancel</AlertDialogCancel>
<AlertDialogAction
onClick={async () => {
try {
await deleteEmailClassification(domain);
setDeleted(true);
toast({
title: "Classification deleted",
});
} catch(e) {
toast({
title: "Failed to delete classification",
description: String(e),
variant: "destructive",
});
}
}}
>
Remove
</AlertDialogAction>
</AlertDialogFooter>
</AlertDialogContent>
</AlertDialog>
</TableCell>
</TableRow>
);
}

16
lib/accessPermissions.ts
View File

@ -2,7 +2,16 @@ import { getServerSession } from "next-auth";
import { SafetyNotes, insertAuditLog } from "./db"; import { SafetyNotes, insertAuditLog } from "./db";
type Permission = type Permission =
| "authifier" | `authifier${
| ""
| `/classification${
| ""
| "/fetch"
| "/create"
| "/update"
| "/delete"
}`
}`
| "publish_message" | "publish_message"
| "chat_message" | "chat_message"
| `accounts${ | `accounts${
@ -189,6 +198,10 @@ const PermissionSets = {
"safety_notes/fetch", "safety_notes/fetch",
"safety_notes/update", "safety_notes/update",
] as Permission[], ] as Permission[],
"authifier": [
"authifier/classification",
] as Permission[],
}; };
const Roles = { const Roles = {
@ -196,6 +209,7 @@ const Roles = {
...PermissionSets["view-open-reports"], ...PermissionSets["view-open-reports"],
...PermissionSets["edit-reports"], ...PermissionSets["edit-reports"],
...PermissionSets["moderate-users"], ...PermissionSets["moderate-users"],
...PermissionSets["authifier"],
], ],
"user-support": [...PermissionSets["user-support"]], "user-support": [...PermissionSets["user-support"]],
"revolt-discover": [...PermissionSets["revolt-discover"]], "revolt-discover": [...PermissionSets["revolt-discover"]],

45
lib/actions.ts
View File

@ -5,6 +5,7 @@ import { PLATFORM_MOD_ID, RESTRICT_ACCESS_LIST } from "./constants";
import mongo, { import mongo, {
Account, Account,
ChannelInvite, ChannelInvite,
EmailClassification,
createDM, createDM,
fetchAccountById, fetchAccountById,
findDM, findDM,
@ -903,3 +904,47 @@ export async function fetchBackup(name: string) {
return JSON.parse((await readFile(`./exports/${name}`)).toString("utf-8")); return JSON.parse((await readFile(`./exports/${name}`)).toString("utf-8"));
} }
export async function fetchEmailClassifications(): Promise<EmailClassification[]> {
await checkPermission("authifier/classification/fetch", null);
return await mongo()
.db("authifier")
.collection<EmailClassification>("email_classification")
.find({})
.toArray();
}
export async function createEmailClassification(domain: string, classification: string) {
await checkPermission("authifier/classification/create", { domain, classification });
await mongo()
.db("authifier")
.collection<EmailClassification>("email_classification")
.insertOne(
{ _id: domain, classification },
);
}
export async function updateEmailClassification(domain: string, classification: string) {
await checkPermission("authifier/classification/update", { domain, classification });
await mongo()
.db("authifier")
.collection<EmailClassification>("email_classification")
.updateOne(
{ _id: domain },
{ $set: { classification } },
);
}
export async function deleteEmailClassification(domain: string) {
await checkPermission("authifier/classification/delete", domain);
await mongo()
.db("authifier")
.collection<EmailClassification>("email_classification")
.deleteOne(
{ _id: domain },
);
}