376 lines
15 KiB
TypeScript
376 lines
15 KiB
TypeScript
"use client"
|
||
|
||
import { useState } from "react"
|
||
import { ChevronLeft, X, Users } from "lucide-react"
|
||
import { Button } from "@/components/ui/button"
|
||
import { Card } from "@/components/ui/card"
|
||
import { Input } from "@/components/ui/input"
|
||
import { Label } from "@/components/ui/label"
|
||
import { Switch } from "@/components/ui/switch"
|
||
import { Tabs, TabsContent, TabsList, TabsTrigger } from "@/components/ui/tabs"
|
||
import { Textarea } from "@/components/ui/textarea"
|
||
import { useRouter } from "next/navigation"
|
||
import { DateRangePicker } from "@/components/ui/date-range-picker"
|
||
import { WechatFriendSelector } from "@/components/WechatFriendSelector"
|
||
import { WechatGroupSelector } from "@/components/WechatGroupSelector"
|
||
import { WechatGroupMemberSelector } from "@/components/WechatGroupMemberSelector"
|
||
import { Accordion, AccordionContent, AccordionItem, AccordionTrigger } from "@/components/ui/accordion"
|
||
import { api } from "@/lib/api"
|
||
import { showToast } from "@/lib/toast"
|
||
import { format } from "date-fns"
|
||
import { zhCN } from "date-fns/locale"
|
||
import { Avatar, AvatarImage, AvatarFallback } from "@/components/ui/avatar"
|
||
import { WechatFriend, WechatGroup, WechatGroupMember } from "@/types/wechat"
|
||
|
||
interface ApiResponse<T = any> {
|
||
code: number
|
||
msg: string
|
||
data: T
|
||
}
|
||
|
||
export default function NewContentLibraryPage() {
|
||
const router = useRouter()
|
||
const [formData, setFormData] = useState({
|
||
name: "",
|
||
sourceType: "friends" as "friends" | "groups",
|
||
keywordsInclude: "",
|
||
keywordsExclude: "",
|
||
startDate: "",
|
||
endDate: "",
|
||
selectedFriends: [] as WechatFriend[],
|
||
selectedGroups: [] as WechatGroup[],
|
||
selectedGroupMembers: [] as WechatGroupMember[],
|
||
useAI: false,
|
||
aiPrompt: "",
|
||
enabled: true,
|
||
})
|
||
|
||
const [isWechatFriendSelectorOpen, setIsWechatFriendSelectorOpen] = useState(false)
|
||
const [isWechatGroupSelectorOpen, setIsWechatGroupSelectorOpen] = useState(false)
|
||
const [isWechatGroupMemberSelectorOpen, setIsWechatGroupMemberSelectorOpen] = useState(false)
|
||
const [currentGroupId, setCurrentGroupId] = useState<string>("")
|
||
const [loading, setLoading] = useState(false)
|
||
|
||
const removeFriend = (friendId: string) => {
|
||
setFormData((prev) => ({
|
||
...prev,
|
||
selectedFriends: prev.selectedFriends.filter((friend) => friend.id !== friendId),
|
||
}))
|
||
}
|
||
|
||
const removeGroup = (groupId: string) => {
|
||
setFormData((prev) => ({
|
||
...prev,
|
||
selectedGroups: prev.selectedGroups.filter((group) => group.id !== groupId),
|
||
}))
|
||
}
|
||
|
||
const handleSelectGroupMembers = (groupId: string) => {
|
||
setCurrentGroupId(groupId)
|
||
setIsWechatGroupMemberSelectorOpen(true)
|
||
}
|
||
|
||
const handleSubmit = async () => {
|
||
if (!formData.name) {
|
||
showToast("请输入内容库名称", "error")
|
||
return
|
||
}
|
||
|
||
if (formData.sourceType === "friends" && formData.selectedFriends.length === 0) {
|
||
showToast("请选择微信好友", "error")
|
||
return
|
||
}
|
||
|
||
if (formData.sourceType === "groups" && formData.selectedGroups.length === 0) {
|
||
showToast("请选择聊天群", "error")
|
||
return
|
||
}
|
||
|
||
setLoading(true)
|
||
try {
|
||
// 将群成员数据转换为{"群ID":[成员ID1, 成员ID2...]}的格式
|
||
const groupMembersMap: Record<string, string[]> = {}
|
||
|
||
formData.selectedGroupMembers.forEach(member => {
|
||
if (member.groupId) {
|
||
if (!groupMembersMap[member.groupId]) {
|
||
groupMembersMap[member.groupId] = []
|
||
}
|
||
groupMembersMap[member.groupId].push(member.id)
|
||
}
|
||
})
|
||
|
||
const payload = {
|
||
name: formData.name,
|
||
sourceType: formData.sourceType === "friends" ? 1 : 2,
|
||
friends: formData.selectedFriends.map(f => f.id),
|
||
groups: formData.selectedGroups.map(g => g.id),
|
||
groupMembers: groupMembersMap, // 使用JSON字符串传递群成员数据
|
||
keywordInclude: formData.keywordsInclude.split(",").map(k => k.trim()).filter(Boolean),
|
||
keywordExclude: formData.keywordsExclude.split(",").map(k => k.trim()).filter(Boolean),
|
||
aiPrompt: formData.useAI ? formData.aiPrompt : "",
|
||
timeEnabled: formData.startDate && formData.endDate ? 1 : 0,
|
||
startTime: formData.startDate ? format(new Date(formData.startDate), "yyyy-MM-dd") : "",
|
||
endTime: formData.endDate ? format(new Date(formData.endDate), "yyyy-MM-dd") : "",
|
||
status: formData.enabled ? 1 : 0
|
||
}
|
||
|
||
const response = await api.post<ApiResponse>("/v1/content/library/create", payload)
|
||
|
||
if (response.code === 200) {
|
||
showToast("创建成功", "success")
|
||
router.push("/content")
|
||
} else {
|
||
showToast(response.msg || "创建失败", "error")
|
||
}
|
||
} catch (error: any) {
|
||
console.error("创建内容库失败:", error)
|
||
showToast(error?.message || "请检查网络连接", "error")
|
||
} finally {
|
||
setLoading(false)
|
||
}
|
||
}
|
||
|
||
return (
|
||
<div className="flex-1 bg-gray-50 min-h-screen pb-16">
|
||
<header className="sticky top-0 z-10 bg-white border-b">
|
||
<div className="flex items-center p-4">
|
||
<Button variant="ghost" size="icon" onClick={() => router.back()}>
|
||
<ChevronLeft className="h-5 w-5" />
|
||
</Button>
|
||
<h1 className="ml-2 text-lg font-medium">新建内容库</h1>
|
||
</div>
|
||
</header>
|
||
|
||
<div className="p-4 space-y-4">
|
||
<Card className="p-4">
|
||
<div className="space-y-4">
|
||
<div>
|
||
<Label htmlFor="name" className="text-base required">
|
||
内容库名称
|
||
</Label>
|
||
<Input
|
||
id="name"
|
||
value={formData.name}
|
||
onChange={(e) => setFormData({ ...formData, name: e.target.value })}
|
||
placeholder="请输入内容库名称"
|
||
required
|
||
className="mt-1.5"
|
||
/>
|
||
</div>
|
||
|
||
<div>
|
||
<Label className="text-base">数据来源配置</Label>
|
||
<Tabs
|
||
value={formData.sourceType}
|
||
onValueChange={(value) => setFormData({ ...formData, sourceType: value as "friends" | "groups" })}
|
||
className="mt-1.5"
|
||
>
|
||
<TabsList className="grid w-full grid-cols-2">
|
||
<TabsTrigger value="friends">选择微信好友</TabsTrigger>
|
||
<TabsTrigger value="groups">选择聊天群</TabsTrigger>
|
||
</TabsList>
|
||
<TabsContent value="friends" className="mt-4">
|
||
<Button variant="outline" className="w-full" onClick={() => setIsWechatFriendSelectorOpen(true)}>
|
||
选择微信好友
|
||
</Button>
|
||
{formData.selectedFriends.length > 0 && (
|
||
<div className="mt-2 space-y-2">
|
||
{formData.selectedFriends.map((friend) => (
|
||
<div key={friend.id} className="flex items-center justify-between bg-gray-100 p-2 rounded-md">
|
||
<div className="flex items-center space-x-2">
|
||
<img
|
||
src={friend.avatar || "/placeholder.svg"}
|
||
alt={friend.nickname}
|
||
className="w-8 h-8 rounded-full"
|
||
/>
|
||
<span>{friend.nickname}</span>
|
||
</div>
|
||
<Button variant="ghost" size="sm" onClick={() => removeFriend(friend.id)}>
|
||
<X className="h-4 w-4" />
|
||
</Button>
|
||
</div>
|
||
))}
|
||
</div>
|
||
)}
|
||
</TabsContent>
|
||
<TabsContent value="groups" className="mt-4">
|
||
<Button variant="outline" className="w-full" onClick={() => setIsWechatGroupSelectorOpen(true)}>
|
||
选择聊天群
|
||
</Button>
|
||
{formData.selectedGroups.length > 0 && (
|
||
<div className="mt-2 space-y-2">
|
||
{formData.selectedGroups.map((group) => (
|
||
<div key={group.id} className="flex items-center justify-between bg-gray-100 p-2 rounded-md">
|
||
<div className="flex items-center space-x-2">
|
||
<img
|
||
src={group.avatar || "/placeholder.svg"}
|
||
alt={group.name}
|
||
className="w-8 h-8 rounded-full"
|
||
/>
|
||
<span className="truncate max-w-[220px]">{group.name}</span>
|
||
</div>
|
||
<div className="flex items-center">
|
||
<Button
|
||
variant="ghost"
|
||
size="sm"
|
||
onClick={() => handleSelectGroupMembers(group.id)}
|
||
className={`mr-1 ${formData.selectedGroupMembers.some(m => m.groupId === group.id) ? 'text-blue-500' : ''}`}
|
||
>
|
||
<Users className="h-4 w-4" />
|
||
{formData.selectedGroupMembers.some(m => m.groupId === group.id) && (
|
||
<span className="ml-1 text-xs">
|
||
+{formData.selectedGroupMembers.filter(m => m.groupId === group.id).length}
|
||
</span>
|
||
)}
|
||
</Button>
|
||
<Button variant="ghost" size="sm" onClick={() => removeGroup(group.id)}>
|
||
<X className="h-4 w-4" />
|
||
</Button>
|
||
</div>
|
||
</div>
|
||
))}
|
||
</div>
|
||
)}
|
||
</TabsContent>
|
||
</Tabs>
|
||
</div>
|
||
|
||
<Accordion type="single" collapsible>
|
||
<AccordionItem value="keywords">
|
||
<AccordionTrigger>关键字设置</AccordionTrigger>
|
||
<AccordionContent>
|
||
<div className="space-y-4">
|
||
<div>
|
||
<Label htmlFor="keywordsInclude" className="text-base">
|
||
关键字匹配
|
||
</Label>
|
||
<Textarea
|
||
id="keywordsInclude"
|
||
value={formData.keywordsInclude}
|
||
onChange={(e) => setFormData({ ...formData, keywordsInclude: e.target.value })}
|
||
placeholder="如果设置了关键字，系统只会采集含有关键字的内容。多个关键字，用半角的','隔开。"
|
||
className="mt-1.5 min-h-[100px]"
|
||
/>
|
||
</div>
|
||
|
||
<div>
|
||
<Label htmlFor="keywordsExclude" className="text-base">
|
||
关键字排除
|
||
</Label>
|
||
<Textarea
|
||
id="keywordsExclude"
|
||
value={formData.keywordsExclude}
|
||
onChange={(e) => setFormData({ ...formData, keywordsExclude: e.target.value })}
|
||
placeholder="如果设置了关键字，匹配到关键字的，系统将不会采集。多个关键字，用半角的','隔开。"
|
||
className="mt-1.5 min-h-[100px]"
|
||
/>
|
||
</div>
|
||
</div>
|
||
</AccordionContent>
|
||
</AccordionItem>
|
||
</Accordion>
|
||
|
||
<div className="flex items-center justify-between">
|
||
<div>
|
||
<Label className="text-base">是否启用AI</Label>
|
||
<p className="text-sm text-gray-500 mt-1">
|
||
当启用AI之后，该内容库下的所有内容，都会通过AI重新生成内容。
|
||
</p>
|
||
</div>
|
||
<Switch
|
||
checked={formData.useAI}
|
||
onCheckedChange={(checked) => setFormData({ ...formData, useAI: checked })}
|
||
/>
|
||
</div>
|
||
|
||
{formData.useAI && (
|
||
<div>
|
||
<Label htmlFor="aiPrompt" className="text-base">
|
||
AI 提示词
|
||
</Label>
|
||
<Textarea
|
||
id="aiPrompt"
|
||
value={formData.aiPrompt}
|
||
onChange={(e) => setFormData({ ...formData, aiPrompt: e.target.value })}
|
||
placeholder="请输入 AI 提示词"
|
||
className="mt-1.5 min-h-[100px]"
|
||
/>
|
||
</div>
|
||
)}
|
||
|
||
<div>
|
||
<Label className="text-base">时间限制</Label>
|
||
<DateRangePicker
|
||
className="mt-1.5"
|
||
onChange={(range) => {
|
||
setFormData({
|
||
...formData,
|
||
startDate: range?.from ? format(range.from, "yyyy-MM-dd") : "",
|
||
endDate: range?.to ? format(range.to, "yyyy-MM-dd") : "",
|
||
})
|
||
}}
|
||
value={formData.startDate && formData.endDate ? {
|
||
from: new Date(formData.startDate),
|
||
to: new Date(formData.endDate)
|
||
} : undefined}
|
||
/>
|
||
</div>
|
||
|
||
<div className="flex items-center justify-between">
|
||
<Label className="text-base required">是否启用</Label>
|
||
<Switch
|
||
checked={formData.enabled}
|
||
onCheckedChange={(checked) => setFormData({ ...formData, enabled: checked })}
|
||
/>
|
||
</div>
|
||
</div>
|
||
</Card>
|
||
|
||
<div className="flex gap-4">
|
||
<Button
|
||
type="button"
|
||
variant="outline"
|
||
className="flex-1"
|
||
onClick={() => router.back()}
|
||
disabled={loading}
|
||
>
|
||
取消
|
||
</Button>
|
||
<Button
|
||
type="submit"
|
||
className="flex-1"
|
||
onClick={handleSubmit}
|
||
disabled={loading}
|
||
>
|
||
{loading ? "保存中..." : "保存"}
|
||
</Button>
|
||
</div>
|
||
</div>
|
||
|
||
<WechatFriendSelector
|
||
open={isWechatFriendSelectorOpen}
|
||
onOpenChange={setIsWechatFriendSelectorOpen}
|
||
selectedFriends={formData.selectedFriends}
|
||
onSelect={(friends) => setFormData({ ...formData, selectedFriends: friends })}
|
||
/>
|
||
|
||
<WechatGroupSelector
|
||
open={isWechatGroupSelectorOpen}
|
||
onOpenChange={setIsWechatGroupSelectorOpen}
|
||
selectedGroups={formData.selectedGroups}
|
||
onSelect={(groups) => setFormData({ ...formData, selectedGroups: groups })}
|
||
/>
|
||
|
||
<WechatGroupMemberSelector
|
||
open={isWechatGroupMemberSelectorOpen}
|
||
onOpenChange={setIsWechatGroupMemberSelectorOpen}
|
||
groupId={currentGroupId}
|
||
selectedMembers={formData.selectedGroupMembers}
|
||
onSelect={(members) => setFormData({ ...formData, selectedGroupMembers: members })}
|
||
/>
|
||
</div>
|
||
)
|
||
}
|
||
|