index.js

require("dotenv").config();
const { Command } = require("commander");
const axios = require("axios");
const fs = require("fs");
const path = require("path");
const { OpenAI } = require("openai");
const generatePrompt = require("./prompts/generateComponent");

const program = new Command();
const openai = new OpenAI({ apiKey: process.env.OPENAI_API_KEY });

program
  .name("figma-to-native")
  .description("Generate React Native screens from Figma")
  .argument("<figmaFileId>", "Figma File ID")
  .action(async (figmaFileId) => {
    try {
      const figmaRes = await axios.get(
        `https://api.figma.com/v1/files/${figmaFileId}`,
        {
          headers: {
            "X-Figma-Token": process.env.FIGMA_TOKEN,
          },
        }
      );

      const frames = figmaRes.data.document.children.find(
        (node) => node.type === "CANVAS"
      ).children;

      for (const frame of frames) {
        const prompt = generatePrompt(frame);
        const completion = await openai.chat.completions.create({
          model: "gpt-4",
          messages: [
            { role: "system", content: "You write React Native code." },
            { role: "user", content: prompt },
          ],
        });

        const code = completion.choices[0].message.content;
        const fileName = `${frame.name.replace(/\s+/g, "")}.tsx`;
        const filePath = path.join("output", fileName);

        fs.mkdirSync("output", { recursive: true });
        fs.writeFileSync(filePath, code, "utf8");
        console.log(`✅ Created: ${filePath}`);
      }
    } catch (err) {
      console.error("❌ Error:", err.message);
    }
  });

program.parse();