AstralRinth/packages/ui/src/utils/auto-icons.ts

import {
	ArchiveIcon,
	BoxIcon,
	BracesIcon,
	CalendarIcon,
	CardIcon,
	CurrencyIcon,
	FileArchiveIcon,
	FileCodeIcon,
	FileIcon,
	FileImageIcon,
	FileTextIcon,
	FolderOpenIcon,
	GithubIcon,
	GlassesIcon,
	GlobeIcon,
	InfoIcon,
	IssuesIcon,
	LinkIcon,
	LockIcon,
	PackageOpenIcon,
	PaintbrushIcon,
	PayPalIcon,
	PlugIcon,
	PolygonIcon,
	UnknownIcon,
	UpdatedIcon,
	USDCColorIcon,
	XCircleIcon,
	XIcon,
} from '@modrinth/assets'
import type { ProjectStatus, ProjectType } from '@modrinth/utils'
import type { Component } from 'vue'

export const PROJECT_TYPE_ICONS: Record<ProjectType, Component> = {
	mod: BoxIcon,
	modpack: PackageOpenIcon,
	resourcepack: PaintbrushIcon,
	shader: GlassesIcon,
	plugin: PlugIcon,
	datapack: BracesIcon,
	project: BoxIcon,
}

export const PAYMENT_METHOD_ICONS: Record<string, Component> = {
	card: CardIcon,
	cashapp: CurrencyIcon,
	paypal: PayPalIcon,
}

export const SOCIAL_PLATFORM_ICONS: Record<string, Component> = {
	discord: GithubIcon,
	github: GithubIcon,
}

export const SEVERITY_ICONS: Record<string, Component> = {
	info: InfoIcon,
	warning: IssuesIcon,
	error: XCircleIcon,
	critical: XCircleIcon,
}

export const PROJECT_STATUS_ICONS: Record<ProjectStatus, Component> = {
	approved: GlobeIcon,
	unlisted: LinkIcon,
	withheld: LinkIcon,
	private: LockIcon,
	scheduled: CalendarIcon,
	draft: FileTextIcon,
	archived: ArchiveIcon,
	rejected: XIcon,
	processing: UpdatedIcon,
	unknown: UnknownIcon,
}

export const DIRECTORY_ICONS: Record<string, Component> = {
	config: FolderOpenIcon,
	world: FolderOpenIcon,
	resourcepacks: PaintbrushIcon,
	_default: FolderOpenIcon,
}

const CURRENCY_CONFIG: Record<string, { icon: Component; color: string }> = {
	usdc: { icon: USDCColorIcon, color: 'text-blue' },
}

const BLOCKCHAIN_CONFIG: Record<string, { icon: Component; color: string }> = {
	polygon: { icon: PolygonIcon, color: 'text-purple' },
}

export const CODE_EXTENSIONS: readonly string[] = [
	'json',
	'json5',
	'jsonc',
	'java',
	'kt',
	'kts',
	'sh',
	'bat',
	'ps1',
	'yml',
	'yaml',
	'toml',
	'js',
	'ts',
	'py',
	'rb',
	'php',
	'html',
	'css',
	'cpp',
	'c',
	'h',
	'rs',
	'go',
] as const

export const TEXT_EXTENSIONS: readonly string[] = [
	'txt',
	'md',
	'log',
	'cfg',
	'conf',
	'properties',
	'ini',
	'sk',
] as const
export const IMAGE_EXTENSIONS: readonly string[] = [
	'png',
	'jpg',
	'jpeg',
	'gif',
	'svg',
	'webp',
] as const
const ARCHIVE_EXTENSIONS: string[] = ['zip', 'jar', 'tar', 'gz', 'rar', '7z'] as const

export function getProjectTypeIcon(projectType: ProjectType): Component {
	return PROJECT_TYPE_ICONS[projectType] ?? BoxIcon
}

export function getPaymentMethodIcon(method: string): Component {
	return PAYMENT_METHOD_ICONS[method] ?? UnknownIcon
}

export function getSocialPlatformIcon(platform: string): Component {
	return SOCIAL_PLATFORM_ICONS[platform.toLowerCase()] ?? UnknownIcon
}

export function getSeverityIcon(severity: string): Component {
	return SEVERITY_ICONS[severity] ?? InfoIcon
}

export function getProjectStatusIcon(status: ProjectStatus): Component {
	return PROJECT_STATUS_ICONS[status] ?? UnknownIcon
}

export function getDirectoryIcon(name: string): Component {
	return DIRECTORY_ICONS[name.toLowerCase()] ?? DIRECTORY_ICONS._default
}

export function getFileExtensionIcon(extension: string): Component {
	const ext: string = extension.toLowerCase()

	if (CODE_EXTENSIONS.includes(ext)) {
		return FileCodeIcon
	}
	if (TEXT_EXTENSIONS.includes(ext)) {
		return FileTextIcon
	}
	if (IMAGE_EXTENSIONS.includes(ext)) {
		return FileImageIcon
	}
	if (ARCHIVE_EXTENSIONS.includes(ext)) {
		return FileArchiveIcon
	}

	return FileIcon
}

export function getFileIcon(fileName: string): Component {
	const extension = fileName.split('.').pop()?.toLowerCase() || ''
	return getFileExtensionIcon(extension)
}

export function getCurrencyIcon(currency: string): Component | null {
	const lower = currency.toLowerCase()
	const key = Object.keys(CURRENCY_CONFIG).find((k) => lower.includes(k))
	return key ? CURRENCY_CONFIG[key].icon : null
}

export function getCurrencyColor(currency: string): string {
	const lower = currency.toLowerCase()
	const key = Object.keys(CURRENCY_CONFIG).find((k) => lower.includes(k))
	return key ? CURRENCY_CONFIG[key].color : 'text-contrast'
}

export function getBlockchainIcon(blockchain: string): Component | null {
	const lower = blockchain.toLowerCase()
	const key = Object.keys(BLOCKCHAIN_CONFIG).find((k) => lower.includes(k))
	return key ? BLOCKCHAIN_CONFIG[key].icon : null
}

export function getBlockchainColor(blockchain: string): string {
	const lower = blockchain.toLowerCase()
	const key = Object.keys(BLOCKCHAIN_CONFIG).find((k) => lower.includes(k))
	return key ? BLOCKCHAIN_CONFIG[key].color : 'text-contrast'
}