import { ActionName } from '@/onebot/action/router';
import { GetPacketStatusDepends } from '@/onebot/action/packet/GetPacketStatus';
import { MiniAppInfo, MiniAppInfoHelper } from '@/core/packet/utils/helper/miniAppHelper';
import { MiniAppData, MiniAppRawData, MiniAppReqCustomParams, MiniAppReqParams } from '@/core/packet/entities/miniApp';
import { z } from 'zod';
import { actionType } from '../type';
const SchemaData = z.union([
    z.object({
        type: z.union([z.literal('bili'), z.literal('weibo')]),
        title: actionType.string(),
        desc: actionType.string(),
        picUrl: actionType.string(),
        jumpUrl: actionType.string(),
        webUrl: actionType.string().optional(),
        rawArkData: actionType.string().optional()
    }),
    z.object({
        title: actionType.string(),
        desc: actionType.string(),
        picUrl: actionType.string(),
        jumpUrl: actionType.string(),
        iconUrl: actionType.string(),
        webUrl: actionType.string().optional(),
        appId: actionType.string(),
        scene: z.union([actionType.number(), actionType.string()]),
        templateType: z.union([actionType.number(), actionType.string()]),
        businessType: z.union([actionType.number(), actionType.string()]),
        verType: z.union([actionType.number(), actionType.string()]),
        shareType: z.union([actionType.number(), actionType.string()]),
        versionId: actionType.string(),
        sdkId: actionType.string(),
        withShareTicket: z.union([actionType.number(), actionType.string()]),
        rawArkData: actionType.string().optional()
    })
]);
type Payload = z.infer<typeof SchemaData>;

export class GetMiniAppArk extends GetPacketStatusDepends<Payload, {
    data: MiniAppData | MiniAppRawData
}> {
    override actionName = ActionName.GetMiniAppArk;
    override payloadSchema = SchemaData;

    async _handle(payload: Payload) {
        let reqParam: MiniAppReqParams;
        const customParams = {
            title: payload.title,
            desc: payload.desc,
            picUrl: payload.picUrl,
            jumpUrl: payload.jumpUrl,
            webUrl: payload.webUrl,
        } as MiniAppReqCustomParams;
        if ('type' in payload) {
            reqParam = MiniAppInfoHelper.generateReq(customParams, MiniAppInfo.get(payload.type)!.template);
        } else {
            const { appId, scene, iconUrl, templateType, businessType, verType, shareType, versionId, withShareTicket } = payload;
            reqParam = MiniAppInfoHelper.generateReq(
                customParams,
                {
                    sdkId: payload.sdkId ?? MiniAppInfo.sdkId,
                    appId: appId,
                    scene: +scene,
                    iconUrl: iconUrl,
                    templateType: +templateType,
                    businessType: +businessType,
                    verType: +verType,
                    shareType: +shareType,
                    versionId: versionId,
                    withShareTicket: +withShareTicket,
                }
            );
        }
        const arkData = await this.core.apis.PacketApi.pkt.operation.GetMiniAppAdaptShareInfo(reqParam);
        return {
            data: payload.rawArkData === 'true' ? arkData : MiniAppInfoHelper.RawToSend(arkData)
        };
    }
}