fixed readme generator

Author: kartik-at-sopra

README.md

@@ -1,40 +1,29 @@
-# Code Snippets
-### Bcz I care about my fellow developers' precious time (◔◡◔)
+# Code Snippets  
+### Because I care about my fellow developers' precious time (◔◡◔)
 
+Actually it's for my personal needs ✪ ω ✪  
+But yes, you all are welcome anytime :D  
 
-actually its for my personal needs ✪ ω ✪
-##
-but yes you guys are also welcomed anytime :D
-
-
-# 📚 DSA Snippets Repository
+---
 
-Automatically generated file index.
+# 📚 DSA Snippets Repository  
+Auto-generated file index.
 
 ---
 
+[cheetsheet.md](cheetsheet.md)
+
 - **Algorithms/**
-  - [cycleDetectionDirected.js](Algorithms/cycleDetectionDirected.js)
-  - [dijkstra.js](Algorithms/dijkstra.js)
-  - [kadane.js](Algorithms/kadane.js)
-  - [topoSort.js](Algorithms/topoSort.js)
-- [cheetsheet.md](cheetsheet.md)
+  [cycleDetectionDirected.js](Algorithms/cycleDetectionDirected.js), [dijkstra.js](Algorithms/dijkstra.js), [kadane.js](Algorithms/kadane.js), [topoSort.js](Algorithms/topoSort.js)
 - **Data Structures/**
   - **graph/**
-    - [trie.js](Data%20Structures/graph/trie.js)
-    - [unionFind.js](Data%20Structures/graph/unionFind.js)
+    [trie.js](Data%20Structures/graph/trie.js), [unionFind.js](Data%20Structures/graph/unionFind.js)
   - **heap/**
-    - [minHeap.js](Data%20Structures/heap/minHeap.js)
+    [minHeap.js](Data%20Structures/heap/minHeap.js)
   - **stack/**
-    - [stack.js](Data%20Structures/stack/stack.js)
+    [stack.js](Data%20Structures/stack/stack.js)
 - **DSA patterns/**
-  - [binarySearchOnAnswer.js](DSA%20patterns/binarySearchOnAnswer.js)
-  - [monotonicStack.js](DSA%20patterns/monotonicStack.js)
-  - [prefixSumPattern.js](DSA%20patterns/prefixSumPattern.js)
-  - [twoPointers.js](DSA%20patterns/twoPointers.js)
-- [README.md](README.md)
+  [binarySearchOnAnswer.js](DSA%20patterns/binarySearchOnAnswer.js), [monotonicStack.js](DSA%20patterns/monotonicStack.js), [prefixSumPattern.js](DSA%20patterns/prefixSumPattern.js), [twoPointers.js](DSA%20patterns/twoPointers.js)
 - **templates/**
-  - [bfs.js](templates/bfs.js)
-  - [binarySearch.js](templates/binarySearch.js)
-  - [dfs.js](templates/dfs.js)
+  [bfs.js](templates/bfs.js), [binarySearch.js](templates/binarySearch.js), [dfs.js](templates/dfs.js)
 

generateReadme.js

@@ -3,55 +3,68 @@ const path = require('path')
 
 const ROOT = __dirname
 
-const IGNORE = new Set([
-  '.git',
-  '.gitignore',
-  'node_modules',
-  '.DS_Store',
-  '.github',
-  'generateReadme.js',
-])
-
-function generateTree(dir, prefix = '') {
+const IGNORE = new Set(['.git', '.gitignore', 'node_modules', '.DS_Store', '.github', 'generateReadme.js', 'README.md'])
+
+function makeLinks(dir, files) {
+  return files
+    .map((file) => {
+      const fullPath = path.join(dir, file.name)
+      let relPath = path.relative(ROOT, fullPath).replace(/\\/g, '/')
+      relPath = relPath.split(' ').join('%20') // encode spaces
+      return `[${file.name}](${relPath})`
+    })
+    .join(', ')
+}
+
+function generateTree(dir, depth = 0) {
   const items = fs.readdirSync(dir, { withFileTypes: true })
 
-  let out = ''
+  const folders = []
+  const files = []
 
   for (const item of items) {
-    if (IGNORE.has(item.name)) continue;
-    const fullPath = path.join(dir, item.name)
-    let relPath = path.relative(ROOT, fullPath).replace(/\\/g, '/')
-
-    relPath = relPath.split(' ').join('%20')
-
-    if (item.isDirectory()) {
-      out += `${prefix}- **${item.name}/**\n`
-      out += generateTree(fullPath, prefix + '  ')
-    } else {
-      out += `${prefix}- [${item.name}](${relPath})\n`
-    }
+    if (IGNORE.has(item.name)) continue
+    if (item.isDirectory()) folders.push(item)
+    else files.push(item)
+  }
+
+  let out = ''
+
+  if (depth === 0 && files.length > 0) {
+    out += makeLinks(dir, files) + '\n\n'
+  }
+  
+  for (const folder of folders) {
+    const indent = '  '.repeat(depth)
+    out += `${indent}- **${folder.name}/**\n`
+
+    const fullPath = path.join(dir, folder.name)
+    out += generateTree(fullPath, depth + 1)
   }
-  return out
-}
 
-const content = `# Code Snippets
-### Bcz I care about my fellow developers' precious time (◔◡◔)
+  if (depth > 0 && files.length > 0) {
+    const indent = '  '.repeat(depth)
+    out += `${indent}${makeLinks(dir, files)}\n`
+  }
 
+  return out
+}
 
-actually its for my personal needs ✪ ω ✪
-##
-but yes you guys are also welcomed anytime :D
+const content = `# Code Snippets  
+### Because I care about my fellow developers' precious time (◔◡◔)
 
+Actually it's for my personal needs ✪ ω ✪  
+But yes, you all are welcome anytime :D  
 
-# 📚 DSA Snippets Repository
+---
 
-Automatically generated file index.
+# 📚 DSA Snippets Repository  
+Auto-generated file index.
 
 ---
 
 ${generateTree(ROOT)}
 `
 
 fs.writeFileSync(path.join(ROOT, 'README.md'), content)
-
 console.log('README.md updated!')