import { getImageMatchRank, normalizeImageName } from "@homarr/common";
import { icons } from "@homarr/db/schema";
import { iconsFindSchema } from "@homarr/validation/icons";

import { createTRPCRouter, publicProcedure } from "../trpc";

export const iconsRouter = createTRPCRouter({
  findIcons: publicProcedure
    .meta({
      mcp: {
        enabled: true,
        description:
          "Search for icons by name across all icon repositories. Returns matching icon names and exact image URLs; use those URLs directly for app icons and Markdown image previews. OPTIONAL: searchText (string to filter), limitPerGroup (number 1-500, default 12). Call with no arguments to browse all icons",
      },
    })
    .input(iconsFindSchema)
    .query(async ({ ctx, input }) => {
      const term = normalizeImageName(input.searchText ?? "");
      const [repositories, countIcons] = await Promise.all([
        ctx.db.query.iconRepositories.findMany({
          orderBy: (table, { asc, sql }) => [sql`CASE WHEN ${table.slug} = 'local' THEN 0 ELSE 1 END`, asc(table.slug)],
          with: {
            icons: {
              columns: { id: true, name: true, url: true },
              orderBy: (table, { asc, sql }) => [
                sql`CASE WHEN ${table.name} LIKE '%.svg' THEN 0 ELSE 1 END`,
                asc(table.name),
              ],
              limit: term.length === 0 ? input.limitPerGroup : undefined,
            },
          },
        }),
        ctx.db.$count(icons),
      ]);

      const matchedRepositories = repositories.map((repository) => ({
        ...repository,
        icons: repository.icons
          .flatMap((icon) => {
            const rank = getImageMatchRank(term, icon.name || icon.url);
            return rank === null ? [] : [{ icon, rank }];
          })
          .toSorted((a, b) =>
            a.rank !== b.rank
              ? a.rank - b.rank
              : Number(b.icon.name.toLowerCase().endsWith(".svg")) -
                  Number(a.icon.name.toLowerCase().endsWith(".svg")) || a.icon.name.localeCompare(b.icon.name),
          )
          .slice(0, input.limitPerGroup)
          .map(({ icon }) => icon),
      }));

      return {
        icons:
          term.length === 0 ? repositories : matchedRepositories.filter((repository) => repository.icons.length > 0),
        countIcons,
      };
    }),
});