import * as core from '@actions/core';
import * as cache from '@actions/cache';
import {cleanPipPackages} from './clean-pip';

import fs from 'fs';
import {State} from './cache-distributions/cache-distributor';

// Added early exit to resolve issue with slow post action step:
// See: https://github.com/actions/setup-node/issues/878
// See: https://github.com/actions/cache/pull/1217
export async function run(earlyExit?: boolean) {
  try {
    const cache = core.getInput('cache');
    // Optionally clean up pip packages after the post-action if requested.
    // This mirrors the `preclean` behavior used in the main action.
    const postcleanPip = core.getBooleanInput('postclean');

    if (cache || postcleanPip) {
      if (cache) {
        await saveCache(cache);
      }
      if (postcleanPip) {
        await cleanPipPackages();
      }
      // Preserve early-exit behavior for the post action when requested.
      // Some CI setups may want the post step to exit early to avoid long-running
      // processes during cleanup. If enabled, exit with success immediately.
      if (earlyExit) {
        process.exit(0);
      }
    }
  } catch (error) {
    const err = error as Error;
    core.setFailed(err.message);
  }
}

async function saveCache(packageManager: string) {
  const cachePathState = core.getState(State.CACHE_PATHS);

  if (!cachePathState) {
    core.warning(
      'Cache paths are empty. Please check the previous logs and make sure that the python version is specified'
    );
    return;
  }

  const cachePaths = JSON.parse(cachePathState) as string[];

  core.debug(`paths for caching are ${cachePaths.join(', ')}`);

  if (!isCacheDirectoryExists(cachePaths)) {
    core.warning(
      `Cache folder path is retrieved for ${packageManager} but doesn't exist on disk: ${cachePaths.join(
        ', '
      )}. This likely indicates that there are no dependencies to cache. Consider removing the cache step if it is not needed.`
    );
    return;
  }

  const primaryKey = core.getState(State.STATE_CACHE_PRIMARY_KEY);
  const matchedKey = core.getState(State.CACHE_MATCHED_KEY);

  if (!primaryKey) {
    core.warning('Error retrieving key from state.');
    return;
  } else if (matchedKey === primaryKey) {
    // no change in target directories
    core.info(
      `Cache hit occurred on the primary key ${primaryKey}, not saving cache.`
    );
    return;
  }

  let cacheId = 0;

  try {
    cacheId = await cache.saveCache(cachePaths, primaryKey);
  } catch (err) {
    const message = (err as Error).message;
    core.info(`[warning]${message}`);
    return;
  }

  if (cacheId == -1) {
    return;
  }
  core.info(`Cache saved with the key: ${primaryKey}`);
}

function isCacheDirectoryExists(cacheDirectory: string[]) {
  const result = cacheDirectory.reduce((previousValue, currentValue) => {
    return previousValue || fs.existsSync(currentValue);
  }, false);

  return result;
}

// Invoke the post action runner. No early-exit — this must complete so the cache
// can be saved during the post step.
run();