import { OneBotAction } from '@/napcat-onebot/action/OneBotAction';
import { ActionName } from '@/napcat-onebot/action/router';
import { Static, Type } from '@sinclair/typebox';
import { GoCQHTTPActionsExamples } from '../example/GoCQHTTPActionsExamples';

const PayloadSchema = Type.Object({
  group_id: Type.String({ description: '群号' }),
});

type PayloadType = Static<typeof PayloadSchema>;

const ReturnSchema = Type.Object({
  file_count: Type.Number({ description: '文件总数' }),
  limit_count: Type.Number({ description: '文件上限' }),
  used_space: Type.Number({ description: '已使用空间' }),
  total_space: Type.Number({ description: '总空间' }),
}, { description: '群文件系统信息' });

type ReturnType = Static<typeof ReturnSchema>;

export class GetGroupFileSystemInfo extends OneBotAction<PayloadType, ReturnType> {
  override actionName = ActionName.GoCQHTTP_GetGroupFileSystemInfo;
  override payloadSchema = PayloadSchema;
  override returnSchema = ReturnSchema;
  override actionSummary = '获取群文件系统信息';
  override actionDescription = '获取群聊文件系统的空间及状态信息';
  override actionTags = ['Go-CQHTTP'];
  override payloadExample = GoCQHTTPActionsExamples.GetGroupFileSystemInfo.payload;
  override returnExample = GoCQHTTPActionsExamples.GetGroupFileSystemInfo.response;

  async _handle (payload: PayloadType) {
    const groupCode = payload.group_id.toString();
    const [countResult, space] = await Promise.all([
      this.core.apis.GroupApi.getGroupFileCount([groupCode]),
      this.core.apis.GroupApi.getGroupFileSpace(groupCode),
    ]);
    const groupFileCount = countResult.groupFileCounts[0];
    if (groupFileCount === undefined) {
      throw new Error('Group not found');
    }
    return {
      file_count: groupFileCount,
      limit_count: 10000,
      // Real values from the kernel's groupSpaceResult; fall back to the old
      // placeholders if the field is unexpectedly absent.
      used_space: space?.usedSpace ?? 0,
      total_space: space?.totalSpace ?? 10 * 1024 * 1024 * 1024,
    };
  }
}