common.js

const core = require('@actions/core');
const github = require('@actions/github');

async function cleanCaches(post) {
    const asPost = core.getInput('post') === 'true';

    if (asPost === !!post) {

        try {
            const token = core.getInput('token');
            const ref = core.getInput('ref');
            const except = core.getInput('except');
            const include = core.getInput('includePrefix');
            const keep = parseInt(core.getInput('keep'), 10);

            const octokit = github.getOctokit(token);

            console.log(`ref: ${github.context.ref}`)
            const branches = [];
            if (ref === "") {

                if (github.context.payload.pull_request) {
                    // PR action - process both branch and PR refs
                    // Fetching the branch name for the PR
                    const {data: prData} = await octokit.rest.pulls.get({
                        owner: github.context.repo.owner,
                        repo: github.context.repo.repo,
                        pull_number: github.context.payload.pull_request.number
                    });

                    branches.push(`refs/heads/${prData.head.ref}`, `refs/pull/${(github.context.payload.pull_request.number)}/merge`);
                } else {
                    branches.push(github.context.ref)
                }

            } else {
                branches.push(ref);
            }

            let cachesToDelete = [];
            for (const ref of branches) {
                const result = await octokit.rest.actions.getActionsCacheList({
                    owner: github.context.repo.owner,
                    repo: github.context.repo.repo,
                    ref: ref
                });
                const refCaches = result.data.actions_caches
                console.log(`Cache keys for ${ref}: ${refCaches.length}`);
                cachesToDelete.push(...refCaches);
            }

            cachesToDelete = cachesToDelete
                .sort((a, b) => new Date(b.last_accessed_at) - new Date(a.last_accessed_at))
                .filter(cache => cache.key !== except && cache.key.startsWith(include))
                .slice(keep);

            for (const cache of cachesToDelete) {
                console.log(`Deleting cache: ${cache.key}`);
                await octokit.rest.actions.deleteActionsCacheById({
                    owner: github.context.repo.owner,
                    repo: github.context.repo.repo,
                    cache_id: cache.id
                });
            }
        } catch (error) {
            core.setFailed(error.message);
        }
    }
}

module.exports = { cleanCaches };