code.js

// code.ts — v12: frameIndex from frame name, section filter support
figma.showUI(__html__, { width: 560, height: 820, themeColors: true });
function isIOSScreenshot(name) { return /^\d+_APP_/.test(name); }
const ANDROID_SECTIONS = {
    "ANDROID PHONE": { folder: "phoneScreenshots", scale: 3 },
    "ANDROID TAB 7 INCH": { folder: "sevenInchScreenshots", scale: 3 },
    "ANDROID TAB 10 INCH": { folder: "tenInchScreenshots", scale: 2 },
    "ANDROID TV": { folder: "tvScreenshots", scale: 2 },
    "FEATUREGRAPHIC": { folder: "__feature__", scale: 1 },
};
function norm(s) { return s.trim().toUpperCase().replace(/\s+/g, " "); }
// Извлекает числовой префикс из имени фрейма: "3_en-US" → 3, "featureGraphic" → null
function extractFrameIndex(frameName) {
    const m = frameName.match(/^(\d+)/);
    return m ? parseInt(m[1]) : null;
}
function getIOSFrames(page) {
    return page.children
        .filter(n => (n.type === "FRAME" || n.type === "COMPONENT" || n.type === "INSTANCE") && isIOSScreenshot(n.name))
        .map(f => ({ name: f.name, width: Math.round(f.width), height: Math.round(f.height) }));
}
function getAndroidFrames(page) {
    const frames = [], sections = [];
    for (const child of page.children) {
        if (child.type === "SECTION" && ANDROID_SECTIONS[norm(child.name)]) {
            sections.push(child.name);
            for (const sub of child.children) {
                if (sub.type === "FRAME" || sub.type === "COMPONENT" || sub.type === "INSTANCE")
                    frames.push({ name: sub.name, width: Math.round(sub.width), height: Math.round(sub.height), section: child.name });
            }
        }
    }
    return { frames, sections };
}
function sendDocInfo() {
    figma.ui.postMessage({ type: "doc-info", docName: figma.root.name });
}
function sendPagesInfo(storeType) {
    const pages = figma.root.children.map(p => {
        const all = p.children.length;
        if (storeType === "playmarket") {
            const { frames, sections } = getAndroidFrames(p);
            return { id: p.id, name: p.name, frames, sections, allFramesCount: all, skippedCount: all - frames.length };
        }
        const frames = getIOSFrames(p);
        return { id: p.id, name: p.name, frames, allFramesCount: all, skippedCount: all - frames.length };
    });
    figma.ui.postMessage({ type: "pages-info", pages });
}
let stopRequested = false;
async function exportFrames(pageIds, storeType, scaleMap, defaultScale, sectionFilter // null = все секции, иначе только указанные
) {
    stopRequested = false;
    const toExport = figma.root.children.filter(p => pageIds.includes(p.id));
    let total = 0;
    for (const p of toExport) {
        if (storeType === "playmarket") {
            const { frames } = getAndroidFrames(p);
            total += sectionFilter
                ? frames.filter(f => f.section && sectionFilter.includes(norm(f.section))).length
                : frames.length;
        }
        else {
            total += getIOSFrames(p).length;
        }
    }
    figma.ui.postMessage({ type: "export-start", total });
    let done = 0;
    for (const page of toExport) {
        if (stopRequested)
            break;
        figma.currentPage = page;
        if (storeType === "playmarket") {
            for (const child of page.children) {
                if (stopRequested)
                    break;
                if (child.type !== "SECTION")
                    continue;
                const sectionNorm = norm(child.name);
                const cfg = ANDROID_SECTIONS[sectionNorm];
                if (!cfg)
                    continue;
                // Пропускаем секцию если задан фильтр и её нет в нём
                if (sectionFilter && !sectionFilter.includes(sectionNorm))
                    continue;
                for (const frame of child.children) {
                    if (stopRequested)
                        break;
                    if (frame.type !== "FRAME" && frame.type !== "COMPONENT" && frame.type !== "INSTANCE")
                        continue;
                    try {
                        const bytes = await frame.exportAsync({ format: "PNG", constraint: { type: "SCALE", value: cfg.scale } });
                        done++;
                        // Извлекаем индекс из имени фрейма (напр. "3_en-US" → 3)
                        const frameIndex = extractFrameIndex(frame.name);
                        figma.ui.postMessage({
                            type: "frame-exported",
                            pageName: page.name,
                            frameName: frame.name,
                            frameIndex, // число или null для featureGraphic
                            width: Math.round(frame.width),
                            height: Math.round(frame.height),
                            scale: cfg.scale,
                            section: child.name,
                            folder: cfg.folder,
                            bytes: Array.from(bytes),
                            current: done,
                            total
                        });
                    }
                    catch (e) {
                        done++;
                        figma.ui.postMessage({ type: "log", level: "error", text: `Fail: ${frame.name}: ${e}` });
                    }
                }
            }
        }
        else {
            const frames = page.children.filter(n => (n.type === "FRAME" || n.type === "COMPONENT" || n.type === "INSTANCE") && isIOSScreenshot(n.name));
            for (const frame of frames) {
                if (stopRequested)
                    break;
                try {
                    let scale = defaultScale;
                    if (scaleMap) {
                        const u = frame.name.toUpperCase();
                        for (const [k, v] of Object.entries(scaleMap)) {
                            if (u.includes(k)) {
                                scale = v;
                                break;
                            }
                        }
                    }
                    const bytes = await frame.exportAsync({ format: "PNG", constraint: { type: "SCALE", value: scale } });
                    done++;
                    figma.ui.postMessage({
                        type: "frame-exported",
                        pageName: page.name,
                        frameName: frame.name,
                        frameIndex: null,
                        width: Math.round(frame.width),
                        height: Math.round(frame.height),
                        scale,
                        bytes: Array.from(bytes),
                        current: done,
                        total
                    });
                }
                catch (e) {
                    done++;
                    figma.ui.postMessage({ type: "log", level: "error", text: `Fail: ${frame.name}: ${e}` });
                }
            }
        }
    }
    figma.ui.postMessage({ type: "export-done", exported: done, total, stopped: stopRequested });
}
figma.ui.onmessage = msg => {
    if (msg.type === "get-doc-info")
        sendDocInfo();
    if (msg.type === "get-pages")
        sendPagesInfo(msg.store || "appstore");
    if (msg.type === "export-frames")
        exportFrames(msg.pages || [], msg.store || "appstore", msg.scaleMap || null, msg.scale || 2, msg.sectionFilter || null);
    if (msg.type === "stop-export")
        stopRequested = true;
    if (msg.type === "cancel")
        figma.closePlugin();
};
sendDocInfo();
sendPagesInfo("appstore");