60 lines
2.3 KiB
TypeScript
60 lines
2.3 KiB
TypeScript
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,
|
|
};
|
|
}),
|
|
});
|