import { parse } from "https://deno.land/std@0.106.0/flags/mod.ts";
import { createCanvas, init } from "https://deno.land/x/canvas@v1.3.0/mod.ts";
import { CollageModeType, collageModeType } from "../src/common/collages.ts";
import DenoCollageModes, {
  CastCanvasRenderingContext,
  ProxyImage,
} from "./collages.ts";
import { CollageConfig } from "../src/common/types.ts";
import { choice, shuffle } from "../src/common/utils.ts";
import { walkSync } from "https://deno.land/std@0.107.0/fs/mod.ts";

const canvasKit = await init();

const args = parse(Deno.args, {
  alias: {
    "w": "width",
    "h": "height",
    "o": "output",
    "m": "mode",
    "r": "recursive",
  },
  boolean: ["recursive"]
});

if (args["mode"] === true) {
  console.log(collageModeType.join(", "));
  Deno.exit(0);
}

const files: Set<string> = new Set();
const includeExtensions = Array.from(
  String(args["include"] || "*.png, *.jpg").matchAll(/\*\.([\w]+)/g),
).map(([_, group]) => group);

args["_"].forEach((arg) => {
  arg = arg.toString();
  if (Deno.statSync(arg).isDirectory) {
    Array.from(
      walkSync(arg, {
        maxDepth: args["recursive"]
          ? Infinity
          : 1,
        includeDirs: false,
        includeFiles: true,
        exts: includeExtensions.length ? includeExtensions : undefined,
      }),
    ).forEach((entry) => files.add(entry.path));
  } else {
    files.add(arg);
  }
});

if (files.size < 2) {
  console.error("kollagen needs at least 2 images to work.");
  Deno.exit(1);
}

const shuffledFiles = shuffle(Array.from(files));

if (shuffledFiles.length > 25) {
  console.debug(`Too many files available, shuffling and clamping to 25...`);
  shuffledFiles.splice(25, shuffledFiles.length - 25);
}

const images: ProxyImage[] = (await Promise.all(
  shuffledFiles.map(async (imageURL) =>
    canvasKit.MakeImageFromEncoded(await Deno.readFile(imageURL))
  ),
)).map((image) => new ProxyImage(image!));

const modes = new DenoCollageModes();

const modeKey: CollageModeType = args["mode"] || choice(collageModeType);
const mode = modes.modes[modeKey];

console.log(
  `Creating a "${mode.name}" collage...`,
);
console.debug(`Images: ${shuffledFiles.join(", ")}`);

const canvas = createCanvas(args["width"] || 640, args["height"] || 640);
const context = canvas.getContext("2d");

const collageConfig: CollageConfig = {
  numImages: args["n"],
};

const segments = mode.getSegments(
  context as CastCanvasRenderingContext,
  collageConfig,
  images,
);
mode.place(context as CastCanvasRenderingContext, images, segments);

const output = args["output"] || "collage.png";
console.log(`Saving to "${output}"...`);
await Deno.writeFile(output, canvas.toBuffer());