import { dim, green } from "kleur/colors";
import fs, { readFileSync } from "node:fs";
import { basename, join } from "node:path/posix";
import PQueue from "p-queue";
import { getOutDirWithinCwd } from "../../core/build/common.js";
import { getTimeStat } from "../../core/build/util.js";
import { isRemotePath, prependForwardSlash } from "../../core/path.js";
import { isServerLikeOutput } from "../../prerender/utils.js";
import { getConfiguredImageService, isESMImportedImage } from "../internal.js";
import { loadRemoteImage } from "./remote.js";
async function prepareAssetsGenerationEnv(pipeline, totalCount) {
  const config = pipeline.getConfig();
  const logger = pipeline.getLogger();
  let useCache = true;
  const assetsCacheDir = new URL("assets/", config.cacheDir);
  const count = { total: totalCount, current: 1 };
  try {
    await fs.promises.mkdir(assetsCacheDir, { recursive: true });
  } catch (err) {
    logger.warn(
      null,
      `An error was encountered while creating the cache directory. Proceeding without caching. Error: ${err}`
    );
    useCache = false;
  }
  let serverRoot, clientRoot;
  if (isServerLikeOutput(config)) {
    serverRoot = config.build.server;
    clientRoot = config.build.client;
  } else {
    serverRoot = getOutDirWithinCwd(config.outDir);
    clientRoot = config.outDir;
  }
  return {
    logger,
    isSSR: isServerLikeOutput(config),
    count,
    useCache,
    assetsCacheDir,
    serverRoot,
    clientRoot,
    imageConfig: config.image,
    assetsFolder: config.build.assets
  };
}
function getFullImagePath(originalFilePath, env) {
  return new URL(
    "." + prependForwardSlash(join(env.assetsFolder, basename(originalFilePath))),
    env.serverRoot
  );
}
async function generateImagesForPath(originalFilePath, transformsAndPath, env, queue) {
  const originalImageData = await loadImage(originalFilePath, env);
  for (const [_, transform] of transformsAndPath.transforms) {
    queue.add(
      async () => generateImage(originalImageData, transform.finalPath, transform.transform)
    );
  }
  if (!env.isSSR && !isRemotePath(originalFilePath) && !globalThis.astroAsset.referencedImages?.has(transformsAndPath.originalSrcPath)) {
    try {
      await fs.promises.unlink(getFullImagePath(originalFilePath, env));
    } catch (e) {
    }
  }
  async function generateImage(originalImage, filepath, options) {
    const timeStart = performance.now();
    const generationData = await generateImageInternal(originalImage, filepath, options);
    const timeEnd = performance.now();
    const timeChange = getTimeStat(timeStart, timeEnd);
    const timeIncrease = `(+${timeChange})`;
    const statsText = generationData.cached ? `(reused cache entry)` : `(before: ${generationData.weight.before}kB, after: ${generationData.weight.after}kB)`;
    const count = `(${env.count.current}/${env.count.total})`;
    env.logger.info(
      null,
      `  ${green("\u25B6")} ${filepath} ${dim(statsText)} ${dim(timeIncrease)} ${dim(count)}`
    );
    env.count.current++;
  }
  async function generateImageInternal(originalImage, filepath, options) {
    const isLocalImage = isESMImportedImage(options.src);
    const finalFileURL = new URL("." + filepath, env.clientRoot);
    const cacheFile = basename(filepath) + (isLocalImage ? "" : ".json");
    const cachedFileURL = new URL(cacheFile, env.assetsCacheDir);
    try {
      if (isLocalImage) {
        await fs.promises.copyFile(cachedFileURL, finalFileURL, fs.constants.COPYFILE_FICLONE);
        return {
          cached: true
        };
      } else {
        const JSONData = JSON.parse(readFileSync(cachedFileURL, "utf-8"));
        if (!JSONData.data || !JSONData.expires) {
          await fs.promises.unlink(cachedFileURL);
          throw new Error(
            `Malformed cache entry for ${filepath}, cache will be regenerated for this file.`
          );
        }
        if (JSONData.expires > Date.now()) {
          await fs.promises.writeFile(finalFileURL, Buffer.from(JSONData.data, "base64"));
          return {
            cached: true
          };
        } else {
          await fs.promises.unlink(cachedFileURL);
        }
      }
    } catch (e) {
      if (e.code !== "ENOENT") {
        throw new Error(`An error was encountered while reading the cache file. Error: ${e}`);
      }
    }
    const finalFolderURL = new URL("./", finalFileURL);
    await fs.promises.mkdir(finalFolderURL, { recursive: true });
    const originalImagePath = isLocalImage ? options.src.src : options.src;
    let resultData = {
      data: void 0,
      expires: originalImage.expires
    };
    const imageService = await getConfiguredImageService();
    resultData.data = (await imageService.transform(
      originalImage.data,
      { ...options, src: originalImagePath },
      env.imageConfig
    )).data;
    try {
      if (env.useCache) {
        if (isLocalImage) {
          await fs.promises.writeFile(cachedFileURL, resultData.data);
        } else {
          await fs.promises.writeFile(
            cachedFileURL,
            JSON.stringify({
              data: Buffer.from(resultData.data).toString("base64"),
              expires: resultData.expires
            })
          );
        }
      }
    } catch (e) {
      env.logger.warn(
        null,
        `An error was encountered while creating the cache directory. Proceeding without caching. Error: ${e}`
      );
    } finally {
      await fs.promises.writeFile(finalFileURL, resultData.data);
    }
    return {
      cached: false,
      weight: {
        // Divide by 1024 to get size in kilobytes
        before: Math.trunc(originalImage.data.byteLength / 1024),
        after: Math.trunc(Buffer.from(resultData.data).byteLength / 1024)
      }
    };
  }
}
function getStaticImageList() {
  if (!globalThis?.astroAsset?.staticImages) {
    return /* @__PURE__ */ new Map();
  }
  return globalThis.astroAsset.staticImages;
}
async function loadImage(path, env) {
  if (isRemotePath(path)) {
    const remoteImage = await loadRemoteImage(path);
    return {
      data: remoteImage.data,
      expires: remoteImage.expires
    };
  }
  return {
    data: await fs.promises.readFile(getFullImagePath(path, env)),
    expires: 0
  };
}
export {
  generateImagesForPath,
  getStaticImageList,
  prepareAssetsGenerationEnv
};
