refactor: enhance parental control features and improve device blocking logic 🔨

This commit is contained in:
2025-07-02 23:29:40 +05:00
parent 40bf8413f8
commit c90b003167
7 changed files with 229 additions and 192 deletions

View File

@ -1,12 +1,9 @@
"use client";
import { zodResolver } from "@hookform/resolvers/zod";
import { OctagonX } from "lucide-react";
import { useState } from "react";
import { type SubmitHandler, useForm } from "react-hook-form";
import { DialogDescription } from "@radix-ui/react-dialog";
import { OctagonX, ShieldBan } from "lucide-react";
import { useActionState, useEffect, useState, useTransition } from "react";
import { toast } from "sonner";
import { z } from "zod";
import { Button } from "@/components/ui/button";
import {
Dialog,
@ -19,160 +16,146 @@ import {
import { Label } from "@/components/ui/label";
import type { Device } from "@/lib/backend-types";
import { cn } from "@/lib/utils";
import { blockDevice } from "@/queries/devices";
import { blockDeviceAction } from "@/queries/devices";
import { TextShimmer } from "./ui/text-shimmer";
import { Textarea } from "./ui/textarea";
const validationSchema = z.object({
reasonForBlocking: z.string().min(5, { message: "Reason is required" }),
});
export type BlockDeviceFormState = {
message: string;
success: boolean;
fieldErrors?: {
reason_for_blocking?: string[];
};
payload?: FormData;
};
const initialState: BlockDeviceFormState = {
message: "",
success: false,
fieldErrors: {},
};
export default function BlockDeviceDialog({
device,
admin,
}: { device: Device; type: "block" | "unblock"; admin?: boolean }) {
const [disabled, setDisabled] = useState(false);
// admin,
parentalControl = false,
}: {
device: Device;
type: "block" | "unblock";
admin?: boolean;
parentalControl?: boolean;
}) {
const [open, setOpen] = useState(false);
const {
register,
handleSubmit,
formState: { errors },
} = useForm<z.infer<typeof validationSchema>>({
resolver: zodResolver(validationSchema),
});
const [state, formAction, isPending] = useActionState(blockDeviceAction, initialState);
const [isTransitioning, startTransition] = useTransition();
const onSubmit: SubmitHandler<z.infer<typeof validationSchema>> = (data) => {
setDisabled(true);
console.log(data);
toast.promise(
blockDevice({
deviceId: String(device.id),
reason_for_blocking: data.reasonForBlocking,
blocked_by: "ADMIN",
}),
{
loading: "Blocking...",
success: () => {
setDisabled(false);
setOpen((prev) => !prev);
return "Blocked!";
},
error: (error) => {
setDisabled(false);
return error || "Something went wrong";
},
},
);
setDisabled(false);
const handleSimpleBlock = () => {
startTransition(() => {
const formData = new FormData();
formData.append("deviceId", String(device.id));
formData.append("reason_for_blocking", "");
formData.append("action", "simple-block");
formData.append("blocked_by", "PARENT");
formAction(formData);
});
};
const handleUnblock = () => {
startTransition(() => {
const formData = new FormData();
formData.append("deviceId", String(device.id));
formData.append("reason_for_blocking", "");
formData.append("action", "unblock");
formData.append("blocked_by", "PARENT");
formAction(formData);
});
};
// Show toast notifications based on state changes
useEffect(() => {
if (state.message) {
if (state.success) {
toast.success(state.message);
if (open) setOpen(false);
} else {
toast.error(state.message);
}
}
}, [state, open]);
const isLoading = isPending || isTransitioning;
// If device is blocked and user is not admin, show unblock button
if (device.blocked && parentalControl) {
return (
<Button onClick={handleUnblock} disabled={isLoading}>
{isLoading ? <TextShimmer>Unblocking</TextShimmer> : "Unblock"}
</Button>
);
}
// If device is not blocked and user is not admin, show simple block button
if (!device.blocked && parentalControl) {
return (
<Button onClick={handleSimpleBlock} disabled={isLoading} variant="destructive">
<ShieldBan />
{isLoading ? <TextShimmer>Blocking</TextShimmer> : "Block"}
</Button>
);
}
// If user is admin, show block with reason dialog
return (
<div>
{device.blocked ? (
<Button
onClick={() => {
setDisabled(true);
toast.promise(
blockDevice({
blocked_by: "PARENT",
deviceId: String(device.id),
reason_for_blocking: ""
}),
{
loading: "unblockinig...",
success: () => {
setDisabled(false);
return "Unblocked!";
},
error: () => {
setDisabled(false);
return "Something went wrong";
},
},
);
}}
>
{disabled ? <TextShimmer>Unblocking</TextShimmer> : "Unblock"}
</Button>
) : (
<div>
{!admin ? (
<Button
variant={"destructive"}
onClick={() => {
setDisabled(true);
toast.promise(
blockDevice({
blocked_by: "PARENT",
deviceId: String(device.id),
reason_for_blocking: "",
}),
{
loading: "blocking...",
success: () => {
setDisabled(false);
return "blocked!";
},
error: () => {
setDisabled(false);
return "Something went wrong";
},
},
);
}}
>
<OctagonX />
{disabled ? <TextShimmer>Blocking</TextShimmer> : "Block"}
</Button>
) : (
<Dialog open={open} onOpenChange={setOpen}>
<DialogTrigger asChild>
<Button disabled={disabled} variant="destructive">
<OctagonX />
Block
</Button>
</DialogTrigger>
<DialogContent className="sm:max-w-[425px]">
<DialogHeader>
<DialogTitle>
Please provide a reason for blocking this device.
</DialogTitle>
</DialogHeader>
<form onSubmit={handleSubmit(onSubmit)}>
<div className="grid gap-4 py-4">
<div className="flex flex-col items-start gap-1">
<Label htmlFor="reason" className="text-right">
Reason for blocking
</Label>
<Textarea
rows={10}
{...register("reasonForBlocking")}
id="reasonForBlocking"
className={cn(
"col-span-5",
errors.reasonForBlocking && "ring-2 ring-red-500",
)}
/>
<span className="text-sm text-red-500">
{errors.reasonForBlocking?.message}
</span>
</div>
</div>
<DialogFooter>
<Button
variant={"destructive"}
disabled={disabled}
type="submit"
>
Block
</Button>
</DialogFooter>
</form>
</DialogContent>
</Dialog>
)}
</div>
)}
<Dialog open={open} onOpenChange={setOpen}>
<DialogTrigger asChild>
<Button disabled={isLoading} variant="destructive">
<OctagonX />
Block
</Button>
</DialogTrigger>
<DialogContent className="sm:max-w-[425px]">
<DialogHeader>
<DialogTitle>Block 🚫</DialogTitle>
<DialogDescription className="text-sm text-muted-foreground">
Please provide a reason for blocking this device
</DialogDescription>
</DialogHeader>
<form action={formAction} className="space-y-4">
<input type="hidden" name="deviceId" value={String(device.id)} />
<input type="hidden" name="action" value="block" />
<input type="hidden" name="blocked_by" value="ADMIN" />
<div className="grid gap-4 py-4">
<div className="flex flex-col items-start gap-1">
<Label htmlFor="reason_for_blocking" className="text-right">
Reason for blocking
</Label>
<Textarea
rows={10}
name="reason_for_blocking"
id="reason_for_blocking"
defaultValue={(state?.payload?.get("reason_for_blocking") || "") as string}
className={cn(
"col-span-5",
(state.fieldErrors?.reason_for_blocking) && "ring-2 ring-red-500",
)}
/>
<span className="text-sm text-red-500">
{state.fieldErrors?.reason_for_blocking?.[0]}
</span>
</div>
</div>
<DialogFooter>
<Button variant="destructive" disabled={isLoading} type="submit">
{isLoading ? "Blocking..." : "Block"}
</Button>
</DialogFooter>
</form>
</DialogContent>
</Dialog>
</div>
);
}