import { fetchApi } from "@/lib/server-api-util";
import { Message } from "@/types/ask";
import * as FileSystem from 'expo-file-system';
import * as MediaLibrary from 'expo-media-library';
import { TFunction } from "i18next";
import { Alert } from 'react-native';

// 实现一个函数，从两个数组中轮流插入新数组
export const mergeArrays = (arr1: any[], arr2: any[]) => {
    const result: any[] = [];
    const maxLength = Math.max(arr1.length, arr2.length);
    for (let i = 0; i < maxLength; i++) {
        if (i < arr1.length) result.push(arr1[i]);
        if (i < arr2.length) result.push(arr2[i]);
    }
    return result;
};


// 创建新对话并获取消息
export const createNewConversation = async (user_text: string) => {
    const data = await fetchApi<string>("/chat/new", {
        method: "POST",
    });
    return data
};

// 获取对话信息
export const getConversation = async ({
    session_id,
    user_text,
    material_ids
}: {
    session_id: string,
    user_text: string,
    material_ids: string[]
}): Promise<Message | undefined> => {
    // 获取对话信息必须要有对话id
    if (!session_id) return undefined;

    try {
        const response = await fetchApi<Message>(`/chat`, {
            method: "POST",
            body: JSON.stringify({
                session_id,
                user_text,
                material_ids
            })
        });
        return response;
    } catch (error) {
        // console.error('Error in getConversation:', error);
        return undefined;
    }
};

// 图片 视频 保存到本地
export const saveMediaToGallery = async (mediaUrl: string, t: TFunction) => {
    // 声明 fileUri 变量以便在 finally 块中使用
    let fileUri: string | null = null;

    try {
        // 首先请求权限
        const { status } = await MediaLibrary.requestPermissionsAsync();

        if (status !== 'granted') {
            Alert.alert(t("ask:ask.mediaAuth"), t("ask:ask.mediaAuthDesc"));
            return false;
        }

        // 获取文件扩展名
        const fileExtension = mediaUrl.split('.').pop()?.toLowerCase() || 'mp4';
        const isVideo = ['mp4', 'mov', 'avi', 'mkv'].includes(fileExtension);
        const fileName = `temp_${Date.now()}.${fileExtension}`;
        fileUri = `${FileSystem.documentDirectory}${fileName}`;

        // 下载文件
        console.log('开始下载文件:', mediaUrl);
        const downloadResumable = FileSystem.createDownloadResumable(
            mediaUrl,
            fileUri,
            {},
            (downloadProgress) => {
                const progress = downloadProgress.totalBytesWritten / (downloadProgress.totalBytesExpectedToWrite || 1);
                console.log(`下载进度: ${Math.round(progress * 100)}%`);
            }
        );

        const downloadResult = await downloadResumable.downloadAsync();

        if (!downloadResult) {
            throw new Error('下载失败: 下载被取消或发生错误');
        }

        const { uri } = downloadResult;
        console.log('文件下载完成，准备保存到相册:', uri);

        // 保存到相册
        const asset = await MediaLibrary.createAssetAsync(uri);
        await MediaLibrary.createAlbumAsync(
            'Memowake',
            asset,
            false
        );

        Alert.alert(
            t("ask:ask.saveSuccess"),
            isVideo ? t("ask:ask.videoSave") : t("ask:ask.imgSave")
        );
        return true;
    } catch (error) {
        console.log('保存失败:', error);
        Alert.alert(
            t("ask:ask.saveError"),
            error instanceof Error ? error.message : t("ask:ask.saveError")
        );
        return false;
    } finally {
        // 清理临时文件
        try {
            if (fileUri) {
                await FileSystem.deleteAsync(fileUri, { idempotent: true }).catch(console.warn);
            }
        } catch (cleanupError) {
            console.log('清理临时文件时出错:', cleanupError);
        }
    }
};