mirror of
https://wget.la/https://github.com/Wxw-Gu/WechatExplorer
synced 2026-10-04 03:01:42 +08:00
feat: 扩展 Skill 支持定时日报操作
This commit is contained in:
Wxw-Gu
@@ -264,6 +264,8 @@ describe('Local API authentication', () => {
|
||||
})
|
||||
expect(response.status).toBe(204)
|
||||
expect(response.headers.get('access-control-allow-origin')).toBe(origin)
|
||||
expect(response.headers.get('access-control-allow-methods')).toContain('PATCH')
|
||||
expect(response.headers.get('access-control-allow-methods')).toContain('DELETE')
|
||||
expect(response.headers.get('access-control-allow-headers')).toBe('Content-Type, Authorization')
|
||||
})
|
||||
|
||||
|
||||
@@ -0,0 +1,254 @@
|
||||
import { afterEach, describe, expect, it, vi } from 'vitest'
|
||||
|
||||
const fixture = vi.hoisted(() => ({
|
||||
capability: {
|
||||
supported: true,
|
||||
ready: true,
|
||||
status: 'ready' as const,
|
||||
capabilities: { text: true, image: true, voice: true },
|
||||
senderStatus: {} as never,
|
||||
message: '个人微信已准备好发送日报'
|
||||
},
|
||||
contacts: [
|
||||
{
|
||||
md5: 'tech-md5',
|
||||
m_nsUsrName: 'tech@chatroom',
|
||||
m_nsNickName: '技术交流群',
|
||||
type: 'group' as const
|
||||
},
|
||||
{
|
||||
md5: 'product-md5',
|
||||
m_nsUsrName: 'product@chatroom',
|
||||
m_nsNickName: '产品交流群',
|
||||
type: 'group' as const
|
||||
}
|
||||
]
|
||||
}))
|
||||
|
||||
vi.mock('electron', () => ({
|
||||
app: { getPath: () => '/tmp/tracememo-scheduled-report-api' },
|
||||
safeStorage: {
|
||||
isEncryptionAvailable: () => true,
|
||||
encryptString: (value: string) => Buffer.from(value),
|
||||
decryptString: (value: Buffer) => value.toString()
|
||||
}
|
||||
}))
|
||||
vi.mock('../../src/main/services/chat-service', () => ({
|
||||
isReady: () => true,
|
||||
listContacts: () => fixture.contacts,
|
||||
listMessages: () => [],
|
||||
getGroupSnapshot: () => null,
|
||||
listRecentChat: () => [],
|
||||
resolveMd5: () => null
|
||||
}))
|
||||
vi.mock('../../src/main/services/personal-wechat-capability-service', () => ({
|
||||
personalWechatCapabilityService: {
|
||||
getPersonalWechatSendCapability: vi.fn(async () => fixture.capability)
|
||||
}
|
||||
}))
|
||||
vi.mock('../../src/main/services/scheduled-report-service', () => ({ scheduledReportService: {} }))
|
||||
vi.mock('../../src/main/services/agent-group-report-service', () => ({
|
||||
generateAgentGroupReport: vi.fn(async () => ({ success: true }))
|
||||
}))
|
||||
vi.mock('../../src/main/group-report-service', () => ({
|
||||
exportGroupReport: vi.fn(async () => ({ success: true }))
|
||||
}))
|
||||
vi.mock('../../src/main/services/agent-hub-service', () => ({
|
||||
agentHubService: { getStatus: () => ({}), testSend: vi.fn(async () => ({ success: true })) }
|
||||
}))
|
||||
|
||||
import { startHttpServer, type HttpServerHandle } from '../../src/main/http-server'
|
||||
import type { ScheduledReportTask } from '../../src/shared/scheduled-report'
|
||||
import type { ScheduledReportApiDependencies } from '../../src/main/services/scheduled-report-api-service'
|
||||
|
||||
const TOKEN = 'A'.repeat(43)
|
||||
const handles: HttpServerHandle[] = []
|
||||
const originalContacts = fixture.contacts
|
||||
|
||||
const makeTask = (overrides: Partial<ScheduledReportTask> = {}): ScheduledReportTask => ({
|
||||
id: 'task-1',
|
||||
name: '技术交流群 · 每日日报',
|
||||
group: 'tech-md5',
|
||||
target: 'tech@chatroom',
|
||||
scheduleTime: '09:00',
|
||||
reportRange: 'yesterday',
|
||||
enabled: true,
|
||||
createdAt: '2026-08-27T00:00:00.000Z',
|
||||
updatedAt: '2026-08-27T00:00:00.000Z',
|
||||
nextRunAt: '2026-08-28T01:00:00.000Z',
|
||||
...overrides
|
||||
})
|
||||
|
||||
function baseUrl(handle: HttpServerHandle): string {
|
||||
return `http://${handle.host}:${handle.port}`
|
||||
}
|
||||
|
||||
async function startFixture(
|
||||
options: { tasks?: ScheduledReportTask[] } = {}
|
||||
): Promise<{ handle: HttpServerHandle; service: ScheduledReportApiDependencies['service'] }> {
|
||||
const tasks = options.tasks || []
|
||||
const executions: Array<Record<string, unknown>> = []
|
||||
const service = {
|
||||
listTasks: vi.fn(async () => tasks),
|
||||
listExecutions: vi.fn(async (taskId?: string) =>
|
||||
taskId ? executions.filter((item) => item.taskId === taskId) : executions
|
||||
),
|
||||
createTask: vi.fn(async (input: Record<string, unknown>) => {
|
||||
const created = makeTask({
|
||||
id: `task-${tasks.length + 1}`,
|
||||
name: String(input.name),
|
||||
group: String(input.group),
|
||||
target: String(input.target),
|
||||
scheduleTime: String(input.scheduleTime),
|
||||
reportRange: input.reportRange as ScheduledReportTask['reportRange'],
|
||||
enabled: Boolean(input.enabled)
|
||||
})
|
||||
tasks.push(created)
|
||||
return { success: true, data: created }
|
||||
}),
|
||||
updateTask: vi.fn(async (id: string, input: Record<string, unknown>) => {
|
||||
const current = tasks.find((item) => item.id === id)
|
||||
if (!current) return { success: false, error: '未找到定时日报任务' }
|
||||
const updated = { ...current, ...input, updatedAt: '2026-08-27T02:00:00.000Z' }
|
||||
tasks[tasks.indexOf(current)] = updated
|
||||
return { success: true, data: updated }
|
||||
}),
|
||||
deleteTask: vi.fn(async (id: string) => {
|
||||
const index = tasks.findIndex((item) => item.id === id)
|
||||
if (index < 0) return { success: false, error: '未找到定时日报任务' }
|
||||
tasks.splice(index, 1)
|
||||
return { success: true, data: { deletedId: id } }
|
||||
}),
|
||||
setTaskEnabled: vi.fn(async (id: string, enabled: boolean) => {
|
||||
const current = tasks.find((item) => item.id === id)
|
||||
if (!current) return { success: false, error: '未找到定时日报任务' }
|
||||
const updated = { ...current, enabled }
|
||||
tasks[tasks.indexOf(current)] = updated
|
||||
return { success: true, data: updated }
|
||||
}),
|
||||
runScheduledReportNow: vi.fn(async (id: string) => {
|
||||
const execution = {
|
||||
id: `execution-${executions.length + 1}`,
|
||||
taskId: id,
|
||||
startedAt: '2026-08-27T01:00:00.000Z',
|
||||
finishedAt: '2026-08-27T01:01:00.000Z',
|
||||
status: 'success' as const,
|
||||
message: '日报生成成功，微信发送成功'
|
||||
}
|
||||
executions.push(execution)
|
||||
return { success: true, data: execution }
|
||||
})
|
||||
}
|
||||
const handle = await startHttpServer('127.0.0.1', 0, {
|
||||
tokenProvider: () => TOKEN,
|
||||
scheduledReportService: service,
|
||||
scheduledReportCapabilityProvider: async () => fixture.capability,
|
||||
scheduledReportContactsProvider: () => fixture.contacts,
|
||||
scheduledReportDatabaseReadyProvider: () => true,
|
||||
scheduledReportPlatform: 'darwin'
|
||||
})
|
||||
handles.push(handle)
|
||||
return { handle, service }
|
||||
}
|
||||
|
||||
const auth = { Authorization: `Bearer ${TOKEN}` }
|
||||
|
||||
afterEach(async () => {
|
||||
await Promise.all(handles.splice(0).map((handle) => handle.close()))
|
||||
fixture.contacts = originalContacts
|
||||
fixture.capability = {
|
||||
...fixture.capability,
|
||||
ready: true,
|
||||
status: 'ready',
|
||||
message: '个人微信已准备好发送日报'
|
||||
}
|
||||
})
|
||||
|
||||
describe('scheduled report Local HTTP API', () => {
|
||||
it('requires authentication and exposes capability plus CRUD lifecycle', async () => {
|
||||
const unauthenticated = await startFixture()
|
||||
expect(
|
||||
(await fetch(`${baseUrl(unauthenticated.handle)}/api/v1/scheduled-reports`)).status
|
||||
).toBe(401)
|
||||
|
||||
const { handle, service } = await startFixture()
|
||||
const create = await fetch(`${baseUrl(handle)}/api/v1/scheduled-reports`, {
|
||||
method: 'POST',
|
||||
headers: { ...auth, 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({
|
||||
group: { name: '技术交流群' },
|
||||
schedule: { type: 'daily', time: '09:00' },
|
||||
reportRange: 'yesterday'
|
||||
})
|
||||
})
|
||||
expect(create.status).toBe(201)
|
||||
const created = await create.json()
|
||||
expect(created).toMatchObject({ created: true, task: { group: { talker: 'tech@chatroom' } } })
|
||||
const taskId = created.task.id
|
||||
|
||||
const list = await fetch(`${baseUrl(handle)}/api/v1/scheduled-reports`, { headers: auth })
|
||||
expect(await list.json()).toMatchObject({ count: 1, tasks: [{ id: taskId }] })
|
||||
const get = await fetch(`${baseUrl(handle)}/api/v1/scheduled-reports/${taskId}`, {
|
||||
headers: auth
|
||||
})
|
||||
expect((await get.json()).task.id).toBe(taskId)
|
||||
|
||||
const update = await fetch(`${baseUrl(handle)}/api/v1/scheduled-reports/${taskId}`, {
|
||||
method: 'PATCH',
|
||||
headers: { ...auth, 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ schedule: { type: 'daily', time: '10:00' } })
|
||||
})
|
||||
expect((await update.json()).task.schedule.time).toBe('10:00')
|
||||
await fetch(`${baseUrl(handle)}/api/v1/scheduled-reports/${taskId}/disable`, {
|
||||
method: 'POST',
|
||||
headers: auth
|
||||
})
|
||||
await fetch(`${baseUrl(handle)}/api/v1/scheduled-reports/${taskId}/enable`, {
|
||||
method: 'POST',
|
||||
headers: auth
|
||||
})
|
||||
const run = await fetch(`${baseUrl(handle)}/api/v1/scheduled-reports/${taskId}/run`, {
|
||||
method: 'POST',
|
||||
headers: auth
|
||||
})
|
||||
expect(await run.json()).toMatchObject({ success: true, execution: { status: 'success' } })
|
||||
const executions = await fetch(
|
||||
`${baseUrl(handle)}/api/v1/scheduled-reports/${taskId}/executions`,
|
||||
{ headers: auth }
|
||||
)
|
||||
expect((await executions.json()).executions).toHaveLength(1)
|
||||
const capability = await fetch(`${baseUrl(handle)}/api/v1/wechat-personal/send-capability`, {
|
||||
headers: auth
|
||||
})
|
||||
expect(await capability.json()).toMatchObject({ capability: { status: 'ready' } })
|
||||
expect(service.createTask).toHaveBeenCalledOnce()
|
||||
})
|
||||
|
||||
it('rejects unsupported capability, ambiguous groups, and duplicate tasks', async () => {
|
||||
fixture.capability = {
|
||||
...fixture.capability,
|
||||
ready: false,
|
||||
status: 'needs_binding',
|
||||
message: '请先绑定个人微信'
|
||||
}
|
||||
const first = await startFixture()
|
||||
const blocked = await fetch(`${baseUrl(first.handle)}/api/v1/scheduled-reports`, {
|
||||
method: 'POST',
|
||||
headers: { ...auth, 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ group: '技术交流群', schedule: { type: 'daily', time: '09:00' } })
|
||||
})
|
||||
expect(blocked.status).toBe(409)
|
||||
expect(await blocked.json()).toMatchObject({ error: 'wechat_not_ready' })
|
||||
|
||||
fixture.capability = { ...fixture.capability, ready: true, status: 'ready' }
|
||||
fixture.contacts = fixture.contacts.map((contact) => ({ ...contact, m_nsNickName: '重复群' }))
|
||||
const ambiguous = await startFixture()
|
||||
const ambiguousResponse = await fetch(`${baseUrl(ambiguous.handle)}/api/v1/scheduled-reports`, {
|
||||
method: 'POST',
|
||||
headers: { ...auth, 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ group: '重复群', schedule: { type: 'daily', time: '09:00' } })
|
||||
})
|
||||
expect(ambiguousResponse.status).toBe(409)
|
||||
expect(await ambiguousResponse.json()).toMatchObject({ error: 'group_ambiguous' })
|
||||
})
|
||||
})
|
||||
@@ -0,0 +1,204 @@
|
||||
import { describe, expect, it, vi } from 'vitest'
|
||||
import type { PersonalWechatSendCapability } from '../../src/shared/personal-wechat'
|
||||
import type { ScheduledReportTask } from '../../src/shared/scheduled-report'
|
||||
import {
|
||||
ScheduledReportApiService,
|
||||
type ScheduledReportApiDependencies
|
||||
} from '../../src/main/services/scheduled-report-api-service'
|
||||
|
||||
type ApiTestService = ScheduledReportApiDependencies['service'] & {
|
||||
createTask: ReturnType<typeof vi.fn>
|
||||
updateTask: ReturnType<typeof vi.fn>
|
||||
}
|
||||
type ApiTestResult = {
|
||||
api: ScheduledReportApiService
|
||||
service: ApiTestService
|
||||
}
|
||||
|
||||
const contacts = [
|
||||
{
|
||||
md5: 'tech-md5',
|
||||
m_nsUsrName: 'tech@chatroom',
|
||||
m_nsNickName: '技术交流群',
|
||||
type: 'group' as const
|
||||
},
|
||||
{
|
||||
md5: 'product-md5',
|
||||
m_nsUsrName: 'product@chatroom',
|
||||
m_nsNickName: '产品交流群',
|
||||
type: 'group' as const
|
||||
}
|
||||
]
|
||||
|
||||
const task = (overrides: Partial<ScheduledReportTask> = {}): ScheduledReportTask => ({
|
||||
id: 'task-1',
|
||||
name: '技术交流群 · 每日日报',
|
||||
group: 'tech-md5',
|
||||
target: 'tech@chatroom',
|
||||
scheduleTime: '09:00',
|
||||
reportRange: 'yesterday',
|
||||
enabled: true,
|
||||
createdAt: '2026-08-27T00:00:00.000Z',
|
||||
updatedAt: '2026-08-27T00:00:00.000Z',
|
||||
nextRunAt: '2026-08-28T01:00:00.000Z',
|
||||
...overrides
|
||||
})
|
||||
|
||||
const capability = (
|
||||
status: PersonalWechatSendCapability['status']
|
||||
): PersonalWechatSendCapability => ({
|
||||
supported: status !== 'unsupported',
|
||||
ready: status === 'ready',
|
||||
status,
|
||||
capabilities: { text: status === 'ready', image: status === 'ready', voice: status === 'ready' },
|
||||
senderStatus: {} as PersonalWechatSendCapability['senderStatus'],
|
||||
message: status === 'ready' ? 'ready' : '请先配置微信'
|
||||
})
|
||||
|
||||
function makeApi(
|
||||
options: { capability?: PersonalWechatSendCapability; tasks?: ScheduledReportTask[] } = {}
|
||||
): ApiTestResult {
|
||||
const tasks = options.tasks || []
|
||||
const service = {
|
||||
listTasks: vi.fn(async () => tasks),
|
||||
listExecutions: vi.fn(async () => []),
|
||||
createTask: vi.fn(async (input: Record<string, unknown>) => {
|
||||
const created = task({
|
||||
id: `task-${tasks.length + 1}`,
|
||||
name: String(input.name),
|
||||
group: String(input.group),
|
||||
target: String(input.target),
|
||||
scheduleTime: String(input.scheduleTime),
|
||||
reportRange: input.reportRange as ScheduledReportTask['reportRange'],
|
||||
enabled: Boolean(input.enabled)
|
||||
})
|
||||
tasks.push(created)
|
||||
return { success: true, data: created }
|
||||
}),
|
||||
updateTask: vi.fn(async () => ({ success: true, data: tasks[0] })),
|
||||
deleteTask: vi.fn(async (id: string) => ({ success: true, data: { deletedId: id } })),
|
||||
setTaskEnabled: vi.fn(async (id: string, enabled: boolean) => ({
|
||||
success: true,
|
||||
data: task({ id, enabled })
|
||||
})),
|
||||
runScheduledReportNow: vi.fn(async (id: string) => ({
|
||||
success: true,
|
||||
data: {
|
||||
id: 'execution-1',
|
||||
taskId: id,
|
||||
startedAt: '2026-08-27T01:00:00.000Z',
|
||||
status: 'success' as const
|
||||
}
|
||||
}))
|
||||
}
|
||||
return {
|
||||
api: new ScheduledReportApiService({
|
||||
service,
|
||||
getCapability: async () => options.capability || capability('ready'),
|
||||
listContacts: () => contacts,
|
||||
isDatabaseReady: () => true,
|
||||
platform: 'darwin'
|
||||
}),
|
||||
service
|
||||
}
|
||||
}
|
||||
|
||||
describe('ScheduledReportApiService', () => {
|
||||
it('resolves a unique group and maps the public create request to the existing service', async () => {
|
||||
const { api, service } = makeApi()
|
||||
const result = await api.create({
|
||||
group: { name: '技术交流群' },
|
||||
schedule: { type: 'daily', time: '09:00' },
|
||||
reportRange: 'yesterday'
|
||||
})
|
||||
expect(result).toMatchObject({
|
||||
name: '技术交流群 · 每日日报',
|
||||
group: { talker: 'tech@chatroom', name: '技术交流群' },
|
||||
schedule: { type: 'daily', time: '09:00' },
|
||||
target: { type: 'wechat_group', talker: 'tech@chatroom' }
|
||||
})
|
||||
expect(service.createTask).toHaveBeenCalledWith(
|
||||
expect.objectContaining({ group: 'tech-md5', target: 'tech@chatroom' })
|
||||
)
|
||||
})
|
||||
|
||||
it.each(['unsupported', 'unconfigured', 'needs_binding', 'needs_verification', 'error'] as const)(
|
||||
'rejects creation before touching the task service for capability state %s',
|
||||
async (status) => {
|
||||
const { api, service } = makeApi({ capability: capability(status) })
|
||||
await expect(
|
||||
api.create({
|
||||
group: '技术交流群',
|
||||
schedule: { type: 'daily', time: '09:00' }
|
||||
})
|
||||
).rejects.toMatchObject({ code: 'wechat_not_ready', status: 409 })
|
||||
expect(service.createTask).not.toHaveBeenCalled()
|
||||
}
|
||||
)
|
||||
|
||||
it('enforces the macOS platform gate even if a provider reports ready', async () => {
|
||||
const { service } = makeApi()
|
||||
const api = new ScheduledReportApiService({
|
||||
service,
|
||||
getCapability: async () => capability('ready'),
|
||||
listContacts: () => contacts,
|
||||
isDatabaseReady: () => true,
|
||||
platform: 'win32'
|
||||
})
|
||||
await expect(
|
||||
api.create({ group: '技术交流群', schedule: { type: 'daily', time: '09:00' } })
|
||||
).rejects.toMatchObject({ code: 'wechat_not_ready', status: 409 })
|
||||
expect(service.createTask).not.toHaveBeenCalled()
|
||||
})
|
||||
|
||||
it('returns candidates for ambiguous group names and rejects duplicate tasks', async () => {
|
||||
const ambiguousContacts = contacts.map((contact) => ({ ...contact, m_nsNickName: '重复群' }))
|
||||
const { service: ambiguousService } = makeApi()
|
||||
const ambiguousApi = new ScheduledReportApiService({
|
||||
service: ambiguousService,
|
||||
getCapability: async () => capability('ready'),
|
||||
listContacts: () => ambiguousContacts,
|
||||
isDatabaseReady: () => true,
|
||||
platform: 'darwin'
|
||||
})
|
||||
await expect(
|
||||
ambiguousApi.create({ group: '重复群', schedule: { type: 'daily', time: '09:00' } })
|
||||
).rejects.toMatchObject({ code: 'group_ambiguous', status: 409 })
|
||||
|
||||
const duplicateApi = makeApi({ tasks: [task()] }).api
|
||||
await expect(
|
||||
duplicateApi.create({ group: '技术交流群', schedule: { type: 'daily', time: '09:00' } })
|
||||
).rejects.toMatchObject({ code: 'duplicate', status: 409 })
|
||||
})
|
||||
|
||||
it('updates a changed group target together when target is omitted', async () => {
|
||||
const existing = task()
|
||||
const { api, service } = makeApi({ tasks: [existing] })
|
||||
vi.mocked(service.updateTask).mockResolvedValue({
|
||||
success: true,
|
||||
data: task({ group: 'product-md5', target: 'product@chatroom', name: '产品日报' })
|
||||
})
|
||||
const result = await api.update('task-1', {
|
||||
group: { name: '产品交流群' },
|
||||
name: '产品日报'
|
||||
})
|
||||
expect(result.group).toMatchObject({ talker: 'product@chatroom', name: '产品交流群' })
|
||||
expect(service.updateTask).toHaveBeenCalledWith(
|
||||
'task-1',
|
||||
expect.objectContaining({ group: 'product-md5', target: 'product@chatroom' })
|
||||
)
|
||||
})
|
||||
|
||||
it('serializes concurrent creates so only one identical task is created', async () => {
|
||||
const { api, service } = makeApi()
|
||||
const request = {
|
||||
group: '技术交流群',
|
||||
schedule: { type: 'daily' as const, time: '09:00' },
|
||||
reportRange: 'yesterday' as const
|
||||
}
|
||||
const results = await Promise.allSettled([api.create(request), api.create(request)])
|
||||
expect(results.filter((result) => result.status === 'fulfilled')).toHaveLength(1)
|
||||
expect(results.filter((result) => result.status === 'rejected')).toHaveLength(1)
|
||||
expect(service.createTask).toHaveBeenCalledOnce()
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user