refactor(robo/cli): cleaner type errors

Author: Pkmmte

.changeset/beige-fans-speak.md

@@ -0,0 +1,5 @@
+---
+'robo.js': patch
+---
+
+refactor(cli): cleaner type errors

packages/robo/src/cli/compiler/typescript.ts

@@ -1,3 +1,4 @@
+import { color } from '../../core/color.js'
 import { IS_BUN_RUNTIME } from '../utils/runtime-utils.js'
 import { compilerLogger } from '../utils/loggers.js'
 import { existsSync } from 'node:fs'
@@ -46,7 +47,7 @@ export async function buildDeclarationFiles(tsOptions?: CompilerOptions) {
 	allDiagnostics.forEach((diagnostic) => {
 		switch (diagnostic.category) {
 			case ts.DiagnosticCategory.Error:
-				compilerLogger.error(formatDiagnostic(diagnostic))
+				compilerLogger.custom('typeerror', color.red(formatDiagnostic(diagnostic)))
 				break
 			case ts.DiagnosticCategory.Warning:
 				compilerLogger.warn(formatDiagnostic(diagnostic))

packages/robo/src/cli/utils/loggers.ts

@@ -1,3 +1,14 @@
 import { logger } from '../../core/logger.js'
 
+// Register custom log levels BEFORE creating any forks
+logger({
+	customLevels: {
+		typeerror: {
+			label: 'typeerror',
+			priority: 7,
+			color: 'info'
+		}
+	}
+})
+
 export const compilerLogger = logger.fork('compiler')

packages/robo/src/core/logger.ts

@@ -21,6 +21,7 @@ export type LogLevel = 'trace' | 'debug' | 'info' | 'wait' | 'other' | 'event' |
 interface CustomLevel {
 	label: string
 	priority: number
+	color?: keyof typeof colorMap
 }
 
 export interface LoggerOptions {
@@ -524,7 +525,8 @@ export class Logger {
 	public setup(options?: LoggerOptions) {
 		const { customLevels, drain = consoleDrain, enabled = true, level, parent, prefix } = options ?? {}
 
-		this._customLevels = customLevels
+		// Preserve existing customLevels if new ones aren't provided
+		this._customLevels = customLevels ?? this._customLevels
 		this._drain = drain
 		this._enabled = enabled
 		this._parent = parent
@@ -566,12 +568,18 @@ export class Logger {
 
 		// Format the message all pretty and stuff
 		if (level !== 'other') {
-			const label = this._customLevels ? this._customLevels[level]?.label : colorizedLogLevels[level]
-			let levelLabel = (label ?? level.padEnd(5)) + ' -'
+			const customLevel = this._customLevels?.[level]
+			const levelColor = (customLevel?.color ? colorMap[customLevel.color] : colorMap[level]) ?? ((s: string) => s)
+
+			// Apply color to label if custom level has a color, otherwise use pre-colorized label
+			let levelLabel =
+				(customLevel?.color
+					? levelColor(customLevel.label.padEnd(9))
+					: customLevel?.label ?? colorizedLogLevels[level] ?? level.padEnd(5)) + ' -'
 
 			// Add the prefix if specified
 			if (prefix) {
-				levelLabel = color.bold(colorMap[level](prefix + ':')) + levelLabel
+				levelLabel = color.bold(levelColor(prefix + ':')) + levelLabel
 			}
 
 			data.unshift(levelLabel)