feat: add color to identify file and folder in terminal (closes #6)

Author: w2xi

.prettierrc.mjs

@@ -2,5 +2,5 @@ export default {
   trailingComma: 'es5',
   tabWidth: 2,
   semi: false,
-  singleQuote: true
+  singleQuote: true,
 }

package.json

@@ -10,11 +10,12 @@
   "scripts": {
     "prepublishOnly": "nr build",
     "dev": "tsup src/index.ts --watch",
-    "build": "tsup src/index.ts --minify",
+    "build": "nr type && tsup src/index.ts --minify",
     "release": "bumpp --commit --push --tag && pnpm publish",
     "lint": "eslint src/**/*.ts",
     "lint:fix": "eslint src --fix",
-    "test": "vitest"
+    "test": "vitest",
+    "type": "tsc --noEmit"
   },
   "bin": {
     "treei": "dist/index.js"
@@ -39,6 +40,8 @@
     "vitest": "^1.3.1"
   },
   "dependencies": {
-    "commander": "^12.0.0"
+    "commander": "^12.0.0",
+    "picocolors": "^1.1.1"
   }
 }
+

pnpm-lock.yaml

@@ -1,22 +1,22 @@
 export const emoji = {
   directory: '📁',
-  file: '📄'
+  file: '📄',
 }
 
 export const characters = {
   border: '|',
   contain: '├',
   line: '─',
-  last: '└'
+  last: '└',
 }
 
 export const defaultOptions = {
   // strategy of finding tree structure, bfs by default
-  strategy: 'bfs'
+  strategy: 'bfs',
 }
 
 export const NodeTypes = {
   ROOT: 'root',
   DIRECTORY: 'directory',
-  FILE: 'file'
-}
+  FILE: 'file',
+} as const

src/config.ts

@@ -1,5 +1,6 @@
 import { emoji, characters, NodeTypes } from './config'
 import { Options, TreeNode } from './type'
+import { blue } from 'picocolors'
 
 const lastDirStack: boolean[] = []
 
@@ -25,9 +26,11 @@ export function generate(
     let contentPrefix =
       index === data!.length - 1 ? characters.last : characters.contain
     contentPrefix += characters.line.repeat(2)
+
+    const name = item.type === NodeTypes.DIRECTORY ? blue(item.name) : item.name
     const content = options.icon
-      ? `${emoji[item.type as keyof typeof emoji]}${item.name}`
-      : `${item.name}`
+      ? `${emoji[item.type as keyof typeof emoji]}${name}`
+      : `${name}`
 
     let currentLineStr = `${borderPrefix}${contentPrefix}${content}`
     currentLineStr += '\n'

src/generate.ts

@@ -1,3 +1,5 @@
+import { NodeTypes } from './config'
+
 export interface Options {
   directory: string
   ignore?: string | string[]
@@ -9,7 +11,7 @@ export interface Options {
 }
 
 export interface TreeNode {
-  type: string
+  type: (typeof NodeTypes)[keyof typeof NodeTypes]
   name: string
   path?: string
   children?: TreeNode[]