main.js

const scenegraph = require("scenegraph");
const clipboard = require("clipboard");
const { Artboard, SymbolInstance } = require("scenegraph");
const { itemsToDart } = require("./src/items_to_dart");
const { StatelessWidget } = require("./src/widgets/stateless");
const { formatDart } = require("./src/widgets/util/format_dart");
const { ComponentWidget } = require("./src/widgets/component");
const { listToString } = require("./src/util");

//TODO:
// * console.log change to own console

function onTapGenerate() {
    const items = scenegraph.selection.items;
    const hasSelection = items.length > 0;
    if (hasSelection) {
        const firstItem = items[0];
        const isArtboard = firstItem instanceof Artboard;
        const isOnlyOneComponent = items.length == 1 && firstItem instanceof SymbolInstance;
        if (isOnlyOneComponent) {
            generateComponents(items);
        } else if (isArtboard) {
            generateArtboards(items);
        } else {
            generateSelection(items);
        }
    } else {
        console.log(`Nothing selected`);
    }
}

module.exports = {
    commands: {
        onTapGenerate: onTapGenerate
    }
};

function generateComponents(components) {
    const componentsWidget = [];
    components.forEach(component => {
        const dartCode = new ComponentWidget(component).toDartClass();
        componentsWidget.push(dartCode);
    });
    clipboard.copyText(formatDart(listToString(componentsWidget)));
}

function generateArtboards(artboards) {
    const artboardsWidget = [];
    artboards.forEach(artboard => {
        const dartCode = new StatelessWidget(artboard.name, itemsToDart([artboard], true)).toDart();
        artboardsWidget.push(dartCode);
    });
    clipboard.copyText(formatDart(listToString(artboardsWidget)));
}

function generateSelection(items) {
    const dartCode = formatDart(itemsToDart(items, true) + ";");
    clipboard.copyText(dartCode);
}