pr-scripts.js

const { getFilesInPR } = require('./github-api');
const fs = require('fs');

// List of file extensions to skip (images and binary files)
const SKIP_EXTENSIONS = [
    '.png', '.jpg', '.jpeg', '.gif', '.svg', '.ico',
    '.mp4', '.webm', '.mov', '.mp3', '.wav',
    '.pdf', '.zip', '.tar', '.gz', '.json'
];

module.exports = async ({ core, prId, githubToken, owner, repo }) => {
    // Validate required parameters
    if (!repo) {
        throw new Error('Missing required parameter: repo must be specified');
    }
    if (!prId) {
        throw new Error('Missing required parameter: prId must be specified');
    }
    
    // Use provided values or fallback to defaults where appropriate
    const ownerName = owner || "AdobeDocs";
    const token = githubToken || process.env.GITHUB_TOKEN;

    if (!token) {
        throw new Error('Missing required parameter: githubToken must be provided or GITHUB_TOKEN environment variable must be set');
    }

    try {
        // fetch the files changed in this PR
        const filesData = await getFilesInPR(ownerName, repo, prId, token);

        // Filter files in src/pages directory and exclude config.md and binary files
        const pagesFiles = filesData.filter(file => {
            // Must be in src/pages directory
            if (!file.filename.startsWith('src/pages/')) return false;
            
            // Skip config.md files
            if (file.filename.endsWith('config.md')) return false;
            
            // Skip binary/image files
            const fileExt = file.filename.substring(file.filename.lastIndexOf('.')).toLowerCase();
            if (SKIP_EXTENSIONS.includes(fileExt)) {
                console.log(`Skipping binary/image file: ${file.filename}`);
                return false;
            }
            
            return true;
        });

        let allContent = '';
        let processedFilesCount = 0;

        for (const file of pagesFiles) {
            const contentResponse = await fetch(file.raw_url, {
                headers: {
                    'Accept': 'application/vnd.github.v3.raw',
                    'Authorization': `Bearer ${token}`
                }
            });

            if (contentResponse.ok) {
                const content = await contentResponse.text();
                // Skip files that look like they embed React/JSX blocks while allowing simple HTML
                // Heuristic: skip if it contains lines starting with '<' followed by an uppercase letter (JSX components)
                const hasJSXComponent = /\n\s*<\s*[A-Z][A-Za-z0-9]*/.test(`\n${content}`);
                if (hasJSXComponent) {
                    console.log(`Skipping JSX component file: ${file.filename}`);
                    continue;
                }
                // append the content to allContent string with identifier "--- File: ${file.filename} ---" for further processing
                allContent += `\n\n--- File: ${file.filename} ---\n\n${content}`;
                processedFilesCount++;
            } else {
                console.error(`Failed to fetch content for ${file.filename}`);
            }
        }

        if (processedFilesCount === 0) {
            console.log('No matching files found in src/pages directory (excluding config.md, binary files, and JSX components)');
            allContent = 'No matching files found in src/pages directory (excluding config.md, binary files, and JSX components)';
        }

        // Write content to pr_content.txt
        try {
            fs.writeFileSync('pr_content.txt', allContent);
            console.log('Content successfully written to pr_content.txt');
        } catch (error) {
            console.error('Error writing to pr_content.txt:', error);
        }

    } catch (error) {
        console.error('Error fetching PR information:', error);
        throw error;
    }
};

// Keep backwards compatibility - if run directly, use environment variables
if (require.main === module) {
    const owner = process.env.GITHUB_OWNER;
    const repo = process.env.GITHUB_REPO;
    const prId = process.env.PR_ID;
    const githubToken = process.env.GITHUB_TOKEN;
    
    if (!repo) {
        console.error('Error: GITHUB_REPO environment variable must be set when running directly');
        process.exit(1);
    }
    if (!prId) {
        console.error('Error: PR_ID environment variable must be set when running directly');
        process.exit(1);
    }
    
    module.exports({
        core: null,
        prId,
        githubToken,
        owner,
        repo
    }).catch(error => {
        console.error('Error:', error.message);
        process.exit(1);
    });
}