mirror of
https://github.com/xCyanGrizzly/DragonsStash.git
synced 2026-05-11 06:11:15 +00:00
feat: Docker audit + Telegram bot service + send UI
Docker:
- Harden docker-compose.yml: parameterized DB creds, required AUTH_SECRET,
health checks, resource limits, network isolation, removed exposed DB port
- Add profiles (telegram/bot/full) so base 'docker compose up' needs only AUTH_SECRET
- Fix docker-entrypoint.sh: AUTH_SECRET startup guard
- Fix Dockerfile: copy prisma.config.ts + dotenv into production image
- Update .env.example with all new variables
- Update .dockerignore
Telegram Bot Service (bot/):
- TDLib-based bot using bot token auth (not HTTP Bot API)
- Commands: /search, /latest, /package, /link, /unlink, /subscribe, /unsubscribe
- pg_notify listener for send requests (bot_send) and new packages (new_package)
- Subscription-based notifications when matching packages arrive
- Dockerfile with multi-stage build (bookworm-slim for glibc/TDLib)
API & Database:
- Prisma: TelegramLink, BotSendRequest, BotSubscription models + migration
- POST /api/telegram/bot/send - queue package delivery to linked TG account
- GET /api/telegram/bot/send/[id] - poll send request status
- Server actions: generateTelegramLinkCode, unlinkTelegram, getBotSendHistory
- Worker: emit pg_notify('new_package') after creating packages
Frontend:
- Settings: TelegramLinkCard for account linking via one-time code
- STL table + drawer: SendToTelegramButton with send dialog and status polling
- Telegram admin: Bot Sends tab with delivery history table
- Shared SendHistoryRow type
README: Updated with bot docs, profiles, config vars, project structure
This commit is contained in:
@@ -5,6 +5,7 @@ import { FileArchive, Eye, ImageIcon } from "lucide-react";
|
||||
import { DataTableColumnHeader } from "@/components/shared/data-table-column-header";
|
||||
import { Badge } from "@/components/ui/badge";
|
||||
import { Button } from "@/components/ui/button";
|
||||
import { SendToTelegramButton } from "./send-to-telegram-button";
|
||||
|
||||
export interface PackageRow {
|
||||
id: string;
|
||||
@@ -139,14 +140,21 @@ export function getPackageColumns({
|
||||
{
|
||||
id: "actions",
|
||||
cell: ({ row }) => (
|
||||
<Button
|
||||
variant="ghost"
|
||||
size="icon"
|
||||
className="h-8 w-8"
|
||||
onClick={() => onViewFiles(row.original)}
|
||||
>
|
||||
<Eye className="h-4 w-4" />
|
||||
</Button>
|
||||
<div className="flex items-center gap-0.5">
|
||||
<SendToTelegramButton
|
||||
packageId={row.original.id}
|
||||
packageName={row.original.fileName}
|
||||
variant="icon"
|
||||
/>
|
||||
<Button
|
||||
variant="ghost"
|
||||
size="icon"
|
||||
className="h-8 w-8"
|
||||
onClick={() => onViewFiles(row.original)}
|
||||
>
|
||||
<Eye className="h-4 w-4" />
|
||||
</Button>
|
||||
</div>
|
||||
),
|
||||
enableHiding: false,
|
||||
},
|
||||
|
||||
@@ -23,6 +23,7 @@ import { Input } from "@/components/ui/input";
|
||||
import { Button } from "@/components/ui/button";
|
||||
import { cn } from "@/lib/utils";
|
||||
import type { PackageRow } from "./package-columns";
|
||||
import { SendToTelegramButton } from "./send-to-telegram-button";
|
||||
|
||||
interface FileItem {
|
||||
id: string;
|
||||
@@ -306,6 +307,14 @@ export function PackageFilesDrawer({ pkg, open, onOpenChange }: PackageFilesDraw
|
||||
<DialogDescription className="mt-1">
|
||||
{total.toLocaleString()} file{total !== 1 ? "s" : ""} in archive
|
||||
</DialogDescription>
|
||||
{pkg && (
|
||||
<div className="mt-2">
|
||||
<SendToTelegramButton
|
||||
packageId={pkg.id}
|
||||
packageName={pkg.fileName}
|
||||
/>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
|
||||
|
||||
214
src/app/(app)/stls/_components/send-to-telegram-button.tsx
Normal file
214
src/app/(app)/stls/_components/send-to-telegram-button.tsx
Normal file
@@ -0,0 +1,214 @@
|
||||
"use client";
|
||||
|
||||
import { useState, useTransition, useEffect, useRef } from "react";
|
||||
import { Send, Loader2, CheckCircle2, AlertCircle } from "lucide-react";
|
||||
import { toast } from "sonner";
|
||||
import { Button } from "@/components/ui/button";
|
||||
import {
|
||||
Dialog,
|
||||
DialogContent,
|
||||
DialogDescription,
|
||||
DialogFooter,
|
||||
DialogHeader,
|
||||
DialogTitle,
|
||||
DialogTrigger,
|
||||
} from "@/components/ui/dialog";
|
||||
import { Badge } from "@/components/ui/badge";
|
||||
|
||||
interface SendToTelegramButtonProps {
|
||||
packageId: string;
|
||||
packageName: string;
|
||||
/** variant for inline row actions vs larger button */
|
||||
variant?: "icon" | "default";
|
||||
}
|
||||
|
||||
type SendStatus = "idle" | "sending" | "polling" | "sent" | "failed";
|
||||
|
||||
export function SendToTelegramButton({
|
||||
packageId,
|
||||
packageName,
|
||||
variant = "default",
|
||||
}: SendToTelegramButtonProps) {
|
||||
const [open, setOpen] = useState(false);
|
||||
const [status, setStatus] = useState<SendStatus>("idle");
|
||||
const [error, setError] = useState<string | null>(null);
|
||||
const [isPending, startTransition] = useTransition();
|
||||
const pollRef = useRef<ReturnType<typeof setInterval> | null>(null);
|
||||
|
||||
// Clean up polling on unmount
|
||||
useEffect(() => {
|
||||
return () => {
|
||||
if (pollRef.current) clearInterval(pollRef.current);
|
||||
};
|
||||
}, []);
|
||||
|
||||
function handleSend() {
|
||||
startTransition(async () => {
|
||||
setStatus("sending");
|
||||
setError(null);
|
||||
|
||||
try {
|
||||
const res = await fetch("/api/telegram/bot/send", {
|
||||
method: "POST",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
body: JSON.stringify({ packageId }),
|
||||
});
|
||||
|
||||
const data = await res.json();
|
||||
|
||||
if (!res.ok) {
|
||||
setStatus("failed");
|
||||
setError(data.error ?? "Failed to queue send");
|
||||
return;
|
||||
}
|
||||
|
||||
// Start polling for status
|
||||
setStatus("polling");
|
||||
const requestId = data.requestId;
|
||||
|
||||
pollRef.current = setInterval(async () => {
|
||||
try {
|
||||
const statusRes = await fetch(`/api/telegram/bot/send/${requestId}`);
|
||||
const statusData = await statusRes.json();
|
||||
|
||||
if (statusData.status === "SENT") {
|
||||
setStatus("sent");
|
||||
toast.success(`"${packageName}" sent to Telegram`);
|
||||
if (pollRef.current) clearInterval(pollRef.current);
|
||||
} else if (statusData.status === "FAILED") {
|
||||
setStatus("failed");
|
||||
setError(statusData.error ?? "Send failed");
|
||||
if (pollRef.current) clearInterval(pollRef.current);
|
||||
}
|
||||
// PENDING / SENDING — keep polling
|
||||
} catch {
|
||||
// Network error — keep trying
|
||||
}
|
||||
}, 2000);
|
||||
|
||||
// Stop polling after 60 seconds
|
||||
setTimeout(() => {
|
||||
if (pollRef.current) {
|
||||
clearInterval(pollRef.current);
|
||||
pollRef.current = null;
|
||||
}
|
||||
setStatus((s: SendStatus) => {
|
||||
if (s === "polling") return "sent"; // Assume queued successfully
|
||||
return s;
|
||||
});
|
||||
}, 60000);
|
||||
} catch {
|
||||
setStatus("failed");
|
||||
setError("Network error");
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function handleClose() {
|
||||
if (pollRef.current) {
|
||||
clearInterval(pollRef.current);
|
||||
pollRef.current = null;
|
||||
}
|
||||
setOpen(false);
|
||||
// Reset after animation
|
||||
setTimeout(() => {
|
||||
setStatus("idle");
|
||||
setError(null);
|
||||
}, 200);
|
||||
}
|
||||
|
||||
const trigger =
|
||||
variant === "icon" ? (
|
||||
<Button variant="ghost" size="icon" className="h-8 w-8" title="Send to Telegram">
|
||||
<Send className="h-4 w-4" />
|
||||
</Button>
|
||||
) : (
|
||||
<Button variant="outline" size="sm" className="gap-1.5">
|
||||
<Send className="h-3.5 w-3.5" />
|
||||
Send to Telegram
|
||||
</Button>
|
||||
);
|
||||
|
||||
return (
|
||||
<Dialog open={open} onOpenChange={(o: boolean) => (o ? setOpen(true) : handleClose())}>
|
||||
<DialogTrigger asChild>{trigger}</DialogTrigger>
|
||||
<DialogContent className="sm:max-w-md">
|
||||
<DialogHeader>
|
||||
<DialogTitle>Send to Telegram</DialogTitle>
|
||||
<DialogDescription>
|
||||
Send “{packageName}” to your linked Telegram account.
|
||||
</DialogDescription>
|
||||
</DialogHeader>
|
||||
|
||||
<div className="py-4">
|
||||
{status === "idle" && (
|
||||
<p className="text-sm text-muted-foreground">
|
||||
The bot will forward the archive files from the destination channel
|
||||
to your linked Telegram account.
|
||||
</p>
|
||||
)}
|
||||
|
||||
{(status === "sending" || status === "polling") && (
|
||||
<div className="flex items-center gap-3 p-4 rounded-lg bg-muted/30 border">
|
||||
<Loader2 className="h-5 w-5 animate-spin text-primary" />
|
||||
<div>
|
||||
<p className="text-sm font-medium">
|
||||
{status === "sending" ? "Queuing…" : "Sending…"}
|
||||
</p>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
The bot is forwarding the files to your Telegram.
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{status === "sent" && (
|
||||
<div className="flex items-center gap-3 p-4 rounded-lg bg-green-500/10 border border-green-500/20">
|
||||
<CheckCircle2 className="h-5 w-5 text-green-500" />
|
||||
<div>
|
||||
<p className="text-sm font-medium text-green-500">Sent!</p>
|
||||
<p className="text-xs text-muted-foreground">
|
||||
Check your Telegram messages.
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{status === "failed" && (
|
||||
<div className="flex items-center gap-3 p-4 rounded-lg bg-destructive/10 border border-destructive/20">
|
||||
<AlertCircle className="h-5 w-5 text-destructive" />
|
||||
<div>
|
||||
<p className="text-sm font-medium text-destructive">Failed</p>
|
||||
<p className="text-xs text-muted-foreground">{error}</p>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
|
||||
<DialogFooter>
|
||||
{status === "idle" && (
|
||||
<>
|
||||
<Button variant="outline" onClick={handleClose}>
|
||||
Cancel
|
||||
</Button>
|
||||
<Button onClick={handleSend} disabled={isPending}>
|
||||
{isPending ? (
|
||||
<Loader2 className="h-4 w-4 animate-spin mr-1" />
|
||||
) : (
|
||||
<Send className="h-4 w-4 mr-1" />
|
||||
)}
|
||||
Send
|
||||
</Button>
|
||||
</>
|
||||
)}
|
||||
|
||||
{(status === "sent" || status === "failed") && (
|
||||
<Button variant="outline" onClick={handleClose}>
|
||||
Close
|
||||
</Button>
|
||||
)}
|
||||
</DialogFooter>
|
||||
</DialogContent>
|
||||
</Dialog>
|
||||
);
|
||||
}
|
||||
Reference in New Issue
Block a user