mirror of
https://github.com/yyhuni/xingrin.git
synced 2026-02-02 12:43:17 +08:00
363 lines
11 KiB
TypeScript
363 lines
11 KiB
TypeScript
import { useQuery, useMutation, useQueryClient } from "@tanstack/react-query"
|
|
import { CommandService } from "@/services/command.service"
|
|
import type {
|
|
GetCommandsRequest,
|
|
CreateCommandRequest,
|
|
UpdateCommandRequest,
|
|
GetCommandsResponse,
|
|
Command,
|
|
} from "@/types/command.types"
|
|
import { toast } from "sonner"
|
|
|
|
// 假数据
|
|
const MOCK_COMMANDS: Command[] = [
|
|
{
|
|
id: 1,
|
|
createdAt: "2024-01-15T10:30:00Z",
|
|
updatedAt: "2024-01-15T10:30:00Z",
|
|
toolId: 1,
|
|
tool: {
|
|
id: 1,
|
|
name: "subfinder",
|
|
type: "opensource",
|
|
createdAt: "2024-01-01T00:00:00Z",
|
|
updatedAt: "2024-01-01T00:00:00Z",
|
|
repoUrl: "https://github.com/projectdiscovery/subfinder",
|
|
version: "v2.6.5",
|
|
description: "Fast passive subdomain enumeration tool",
|
|
categoryNames: ["subdomain", "recon"],
|
|
directory: "",
|
|
installCommand: "go install -v github.com/projectdiscovery/subfinder/v2/cmd/subfinder@latest",
|
|
updateCommand: "go install -v github.com/projectdiscovery/subfinder/v2/cmd/subfinder@latest",
|
|
versionCommand: "subfinder -version",
|
|
},
|
|
name: "subdomain_scan",
|
|
displayName: "子域名扫描",
|
|
description: "使用 subfinder 进行子域名扫描",
|
|
commandTemplate: "subfinder -d {{domain}} -o {{output}}",
|
|
},
|
|
{
|
|
id: 2,
|
|
createdAt: "2024-01-16T11:20:00Z",
|
|
updatedAt: "2024-01-16T11:20:00Z",
|
|
toolId: 2,
|
|
tool: {
|
|
id: 2,
|
|
name: "nmap",
|
|
type: "opensource",
|
|
createdAt: "2024-01-01T00:00:00Z",
|
|
updatedAt: "2024-01-01T00:00:00Z",
|
|
repoUrl: "https://github.com/nmap/nmap",
|
|
version: "7.94",
|
|
description: "Network exploration tool and security / port scanner",
|
|
categoryNames: ["port", "network"],
|
|
directory: "",
|
|
installCommand: "brew install nmap",
|
|
updateCommand: "brew upgrade nmap",
|
|
versionCommand: "nmap --version",
|
|
},
|
|
name: "port_scan",
|
|
displayName: "端口扫描",
|
|
description: "使用 nmap 进行端口扫描",
|
|
commandTemplate: "nmap -sV -p- {{target}} -oX {{output}}",
|
|
},
|
|
{
|
|
id: 3,
|
|
createdAt: "2024-01-17T09:15:00Z",
|
|
updatedAt: "2024-01-17T09:15:00Z",
|
|
toolId: 1,
|
|
tool: {
|
|
id: 1,
|
|
name: "subfinder",
|
|
type: "opensource",
|
|
createdAt: "2024-01-01T00:00:00Z",
|
|
updatedAt: "2024-01-01T00:00:00Z",
|
|
repoUrl: "https://github.com/projectdiscovery/subfinder",
|
|
version: "v2.6.5",
|
|
description: "Fast passive subdomain enumeration tool",
|
|
categoryNames: ["subdomain", "recon"],
|
|
directory: "",
|
|
installCommand: "go install -v github.com/projectdiscovery/subfinder/v2/cmd/subfinder@latest",
|
|
updateCommand: "go install -v github.com/projectdiscovery/subfinder/v2/cmd/subfinder@latest",
|
|
versionCommand: "subfinder -version",
|
|
},
|
|
name: "fast_subdomain_scan",
|
|
displayName: "快速子域名扫描",
|
|
description: "使用 subfinder 快速扫描常见子域名",
|
|
commandTemplate: "subfinder -d {{domain}} -silent -o {{output}}",
|
|
},
|
|
{
|
|
id: 4,
|
|
createdAt: "2024-01-18T14:45:00Z",
|
|
updatedAt: "2024-01-18T14:45:00Z",
|
|
toolId: 3,
|
|
tool: {
|
|
id: 3,
|
|
name: "nuclei",
|
|
type: "opensource",
|
|
createdAt: "2024-01-01T00:00:00Z",
|
|
updatedAt: "2024-01-01T00:00:00Z",
|
|
repoUrl: "https://github.com/projectdiscovery/nuclei",
|
|
version: "v3.1.4",
|
|
description: "Fast and customisable vulnerability scanner",
|
|
categoryNames: ["vulnerability", "scanner"],
|
|
directory: "",
|
|
installCommand: "go install -v github.com/projectdiscovery/nuclei/v3/cmd/nuclei@latest",
|
|
updateCommand: "go install -v github.com/projectdiscovery/nuclei/v3/cmd/nuclei@latest",
|
|
versionCommand: "nuclei -version",
|
|
},
|
|
name: "vulnerability_scan",
|
|
displayName: "漏洞扫描",
|
|
description: "使用 nuclei 进行漏洞扫描",
|
|
commandTemplate: "nuclei -u {{target}} -severity critical,high,medium -o {{output}}",
|
|
},
|
|
{
|
|
id: 5,
|
|
createdAt: "2024-01-19T16:00:00Z",
|
|
updatedAt: "2024-01-19T16:00:00Z",
|
|
toolId: 4,
|
|
tool: {
|
|
id: 4,
|
|
name: "katana",
|
|
type: "opensource",
|
|
createdAt: "2024-01-01T00:00:00Z",
|
|
updatedAt: "2024-01-01T00:00:00Z",
|
|
repoUrl: "https://github.com/projectdiscovery/katana",
|
|
version: "v1.0.4",
|
|
description: "Next-generation crawling and spidering framework",
|
|
categoryNames: ["crawler", "recon"],
|
|
directory: "",
|
|
installCommand: "go install github.com/projectdiscovery/katana/cmd/katana@latest",
|
|
updateCommand: "go install github.com/projectdiscovery/katana/cmd/katana@latest",
|
|
versionCommand: "katana -version",
|
|
},
|
|
name: "web_crawl",
|
|
displayName: "网页爬取",
|
|
description: "使用 katana 爬取网页链接",
|
|
commandTemplate: "katana -u {{target}} -d 3 -o {{output}}",
|
|
},
|
|
{
|
|
id: 6,
|
|
createdAt: "2024-01-20T10:30:00Z",
|
|
updatedAt: "2024-01-20T10:30:00Z",
|
|
toolId: 2,
|
|
tool: {
|
|
id: 2,
|
|
name: "nmap",
|
|
type: "opensource",
|
|
createdAt: "2024-01-01T00:00:00Z",
|
|
updatedAt: "2024-01-01T00:00:00Z",
|
|
repoUrl: "https://github.com/nmap/nmap",
|
|
version: "7.94",
|
|
description: "Network exploration tool and security / port scanner",
|
|
categoryNames: ["port", "network"],
|
|
directory: "",
|
|
installCommand: "brew install nmap",
|
|
updateCommand: "brew upgrade nmap",
|
|
versionCommand: "nmap --version",
|
|
},
|
|
name: "service_detect",
|
|
displayName: "服务识别",
|
|
description: "使用 nmap 进行服务版本识别",
|
|
commandTemplate: "nmap -sV {{target}} -oX {{output}}",
|
|
},
|
|
{
|
|
id: 7,
|
|
createdAt: "2024-01-21T13:20:00Z",
|
|
updatedAt: "2024-01-21T13:20:00Z",
|
|
toolId: 5,
|
|
tool: {
|
|
id: 5,
|
|
name: "dirsearch",
|
|
type: "opensource",
|
|
createdAt: "2024-01-01T00:00:00Z",
|
|
updatedAt: "2024-01-01T00:00:00Z",
|
|
repoUrl: "https://github.com/maurosoria/dirsearch",
|
|
version: "v0.4.3",
|
|
description: "Web path scanner",
|
|
categoryNames: ["directory", "recon"],
|
|
directory: "",
|
|
installCommand: "pip3 install dirsearch",
|
|
updateCommand: "pip3 install --upgrade dirsearch",
|
|
versionCommand: "dirsearch --version",
|
|
},
|
|
name: "dir_scan",
|
|
displayName: "目录扫描",
|
|
description: "使用 dirsearch 进行目录扫描",
|
|
commandTemplate: "dirsearch -u {{target}} -e php,html,js -o {{output}}",
|
|
},
|
|
{
|
|
id: 8,
|
|
createdAt: "2024-01-22T15:10:00Z",
|
|
updatedAt: "2024-01-22T15:10:00Z",
|
|
toolId: 3,
|
|
tool: {
|
|
id: 3,
|
|
name: "nuclei",
|
|
type: "opensource",
|
|
createdAt: "2024-01-01T00:00:00Z",
|
|
updatedAt: "2024-01-01T00:00:00Z",
|
|
repoUrl: "https://github.com/projectdiscovery/nuclei",
|
|
version: "v3.1.4",
|
|
description: "Fast and customisable vulnerability scanner",
|
|
categoryNames: ["vulnerability", "scanner"],
|
|
directory: "",
|
|
installCommand: "go install -v github.com/projectdiscovery/nuclei/v3/cmd/nuclei@latest",
|
|
updateCommand: "go install -v github.com/projectdiscovery/nuclei/v3/cmd/nuclei@latest",
|
|
versionCommand: "nuclei -version",
|
|
},
|
|
name: "full_vulnerability_scan",
|
|
displayName: "完整漏洞扫描",
|
|
description: "使用 nuclei 进行全面漏洞扫描",
|
|
commandTemplate: "nuclei -u {{target}} -t nuclei-templates/ -o {{output}}",
|
|
},
|
|
]
|
|
|
|
/**
|
|
* 获取命令列表（使用假数据）
|
|
*/
|
|
export function useCommands(params: GetCommandsRequest = {}) {
|
|
return useQuery({
|
|
queryKey: ["commands", params],
|
|
queryFn: async (): Promise<GetCommandsResponse> => {
|
|
// 模拟网络延迟
|
|
await new Promise(resolve => setTimeout(resolve, 500))
|
|
|
|
const page = params.page || 1
|
|
const pageSize = params.pageSize || 10
|
|
|
|
// 如果有 toolId 过滤
|
|
let filteredCommands = MOCK_COMMANDS
|
|
if (params.toolId) {
|
|
filteredCommands = MOCK_COMMANDS.filter(cmd => cmd.toolId === params.toolId)
|
|
}
|
|
|
|
const totalCount = filteredCommands.length
|
|
const totalPages = Math.ceil(totalCount / pageSize)
|
|
const startIndex = (page - 1) * pageSize
|
|
const endIndex = startIndex + pageSize
|
|
const commands = filteredCommands.slice(startIndex, endIndex)
|
|
|
|
return {
|
|
commands,
|
|
page,
|
|
pageSize,
|
|
total: totalCount,
|
|
totalPages,
|
|
// 兼容字段（向后兼容）
|
|
page_size: pageSize,
|
|
total_count: totalCount,
|
|
total_pages: totalPages,
|
|
}
|
|
},
|
|
})
|
|
}
|
|
|
|
/**
|
|
* 获取单个命令（使用假数据）
|
|
*/
|
|
export function useCommand(id: number) {
|
|
return useQuery({
|
|
queryKey: ["command", id],
|
|
queryFn: async (): Promise<Command | undefined> => {
|
|
// 模拟网络延迟
|
|
await new Promise(resolve => setTimeout(resolve, 300))
|
|
return MOCK_COMMANDS.find(cmd => cmd.id === id)
|
|
},
|
|
enabled: !!id,
|
|
})
|
|
}
|
|
|
|
/**
|
|
* 创建命令
|
|
*/
|
|
export function useCreateCommand() {
|
|
const queryClient = useQueryClient()
|
|
|
|
return useMutation({
|
|
mutationFn: (data: CreateCommandRequest) => CommandService.createCommand(data),
|
|
onSuccess: () => {
|
|
toast.success("命令创建成功")
|
|
queryClient.invalidateQueries({ queryKey: ["commands"] })
|
|
},
|
|
onError: (error: any) => {
|
|
console.error("创建命令失败:", error)
|
|
toast.error("命令创建失败")
|
|
},
|
|
})
|
|
}
|
|
|
|
/**
|
|
* 更新命令
|
|
*/
|
|
export function useUpdateCommand() {
|
|
const queryClient = useQueryClient()
|
|
|
|
return useMutation({
|
|
mutationFn: ({ id, data }: { id: number; data: UpdateCommandRequest }) =>
|
|
CommandService.updateCommand(id, data),
|
|
onSuccess: () => {
|
|
toast.success("命令更新成功")
|
|
queryClient.invalidateQueries({ queryKey: ["commands"] })
|
|
queryClient.invalidateQueries({ queryKey: ["command"] })
|
|
},
|
|
onError: (error: any) => {
|
|
console.error("更新命令失败:", error)
|
|
toast.error("命令更新失败")
|
|
},
|
|
})
|
|
}
|
|
|
|
/**
|
|
* 删除命令
|
|
*/
|
|
export function useDeleteCommand() {
|
|
const queryClient = useQueryClient()
|
|
|
|
return useMutation({
|
|
mutationFn: (id: number) => CommandService.deleteCommand(id),
|
|
onSuccess: () => {
|
|
toast.success("命令删除成功")
|
|
queryClient.invalidateQueries({ queryKey: ["commands"] })
|
|
},
|
|
onError: (error: any) => {
|
|
console.error("删除命令失败:", error)
|
|
toast.error("命令删除失败")
|
|
},
|
|
})
|
|
}
|
|
|
|
/**
|
|
* 批量删除命令（使用假数据）
|
|
*/
|
|
export function useBatchDeleteCommands() {
|
|
const queryClient = useQueryClient()
|
|
|
|
return useMutation({
|
|
mutationFn: async (ids: number[]) => {
|
|
// 模拟网络延迟
|
|
await new Promise(resolve => setTimeout(resolve, 800))
|
|
|
|
// 从假数据中过滤掉被删除的命令
|
|
const deletedCount = ids.filter(id =>
|
|
MOCK_COMMANDS.some(cmd => cmd.id === id)
|
|
).length
|
|
|
|
// 模拟删除（实际上不会真的删除假数据）
|
|
return {
|
|
data: {
|
|
deletedCount: deletedCount
|
|
}
|
|
}
|
|
},
|
|
onSuccess: (response) => {
|
|
toast.success(`成功删除 ${response.data?.deletedCount} 个命令`)
|
|
queryClient.invalidateQueries({ queryKey: ["commands"] })
|
|
},
|
|
onError: (error: any) => {
|
|
console.error("批量删除命令失败:", error)
|
|
toast.error("批量删除命令失败")
|
|
},
|
|
})
|
|
}
|