import type { RefRecord, StaticAssetKind } from '../shared/types.js';
import { membershipDedupeArray } from '../shared/MembershipSet.js';
import { fileNameFromUrl } from './asset-tree.js';
import { getAssetKind, readJsonFile } from './file-util.js';
import { extractFlatJsonRefs, extractSceneRefs } from './scene-parser.js';

interface ScanMeta {
    uuid: string;
    url: string;
    file: string;
    kind: StaticAssetKind;
}

export async function scanAssetFile(fsPath: string): Promise<RefRecord[]> {
    const url = await Editor.Message.request('asset-db', 'query-url', fsPath);
    if (!url || String(url).includes('db://internal')) {
        return [];
    }
    const uuid = await Editor.Message.request('asset-db', 'query-uuid', url);
    if (!uuid) {
        return [];
    }
    const kind = getAssetKind(fsPath);
    if (!kind) {
        return [];
    }
    const data = await readJsonFile(fsPath);
    if (!data) {
        return [];
    }
    const meta: ScanMeta = { uuid, url: String(url).split('@')[0], file: fsPath, kind };
    if (kind === 'prefab' || kind === 'scene') {
        return extractSceneRefs(data as never[], meta);
    }
    return extractFlatJsonRefs(data, meta);
}

/** 填充 targetUrl/targetName（批量，减少 asset-db 调用） */
export async function hydrateRefRecords(records: RefRecord[]): Promise<void> {
    const uuids = membershipDedupeArray(records.map((r) => r.targetUuid));
    const infoMap = new Map<string, { url: string; name: string }>();
    for (const uuid of uuids) {
        try {
            const info = await Editor.Message.request('asset-db', 'query-asset-info', uuid);
            if (info?.url) {
                const mainUrl = String(info.url).split('@')[0];
                infoMap.set(uuid, { url: mainUrl, name: fileNameFromUrl(mainUrl) });
            }
        } catch {
            // ignore missing assets
        }
    }
    for (const record of records) {
        const info = infoMap.get(record.targetUuid);
        if (info) {
            record.targetUrl = info.url;
            record.targetName = info.name;
        } else {
            record.targetUrl = '';
            record.targetName = record.targetUuid.slice(0, 8);
        }
    }
}

export async function hydrateSourceNames(records: RefRecord[]): Promise<void> {
    const uuids = membershipDedupeArray(records.map((r) => r.sourceUuid));
    const infoMap = new Map<string, { url: string; name: string }>();
    for (const uuid of uuids) {
        try {
            const info = await Editor.Message.request('asset-db', 'query-asset-info', uuid);
            if (info?.url) {
                const mainUrl = String(info.url).split('@')[0];
                infoMap.set(uuid, { url: mainUrl, name: fileNameFromUrl(mainUrl) });
            }
        } catch {
            // ignore
        }
    }
    for (const record of records) {
        const info = infoMap.get(record.sourceUuid);
        if (info) {
            record.sourceUrl = info.url;
            record.sourceName = info.name;
        }
    }
}