SnowCatQQ/packages/snowcat-onebot/action/stream/TestStreamDownload.ts

import { ActionName } from '@/snowcat-onebot/action/router';
import { OneBotAction, OneBotRequestToolkit } from '@/snowcat-onebot/action/OneBotAction';
import { Static, Type } from '@sinclair/typebox';
import { NetworkAdapterConfig } from '@/snowcat-onebot/config/config';
import { StreamPacket, StreamStatus } from './StreamBasic';

export const TestDownloadStreamPayloadSchema = Type.Object({
  error: Type.Optional(Type.Boolean({ default: false, description: '是否触发测试错误' })),
});

export type TestDownloadStreamPayload = Static<typeof TestDownloadStreamPayloadSchema>;

export class TestDownloadStream extends OneBotAction<TestDownloadStreamPayload, StreamPacket<{ data: string; }>> {
  override actionName = ActionName.TestDownloadStream;
  override actionSummary = '测试下载流';
  override actionTags = ['流式传输扩展'];
  override payloadExample = {
    url: 'http://example.com/file',
  };

  override returnExample = {
    success: true,
  };

  override payloadSchema = TestDownloadStreamPayloadSchema;
  override returnSchema = Type.Any({ description: '测试流数据' });
  override useStream = true;

  async _handle (_payload: TestDownloadStreamPayload, _adaptername: string, _config: NetworkAdapterConfig, req: OneBotRequestToolkit) {
    for (let i = 0; i < 10; i++) {
      await req.send({ type: StreamStatus.Stream, data: `Index-> ${i + 1}`, data_type: 'data_chunk' });
      await new Promise(resolve => setTimeout(resolve, 100));
    }
    if (_payload.error) {
      throw new Error('This is a test error');
    }
    return {
      type: StreamStatus.Response,
      data_type: 'data_complete',
      data: 'Stream transmission complete',
    };
  }
}