From 36c4467ad2069daa0e1a7671fa95b480e72318bb Mon Sep 17 00:00:00 2001
From: Kenny Daniel <platypii@gmail.com>
Date: Thu, 6 Jun 2024 14:24:36 -0700
Subject: [PATCH] Cli open file

---
 src/cli.js   |  3 ++-
 src/serve.js | 25 +++++++++++++++----------
 2 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/src/cli.js b/src/cli.js
index a72c56e..5efd9ad 100644
--- a/src/cli.js
+++ b/src/cli.js
@@ -3,5 +3,6 @@
 if (process.argv[2] === 'chat') {
   import('./chat.js').then(({ chat }) => chat())
 } else {
-  import('./serve.js').then(({ serve }) => serve())
+  // Load file or directory
+  import('./serve.js').then(({ serve }) => serve({ path: process.argv[2] }))
 }
diff --git a/src/serve.js b/src/serve.js
index f52c4d7..3adc802 100644
--- a/src/serve.js
+++ b/src/serve.js
@@ -122,11 +122,10 @@ async function handleStatic(filePath, range) {
 
 /**
  * Serve head request
- * @param {string} pathname
+ * @param {string} filePath
  * @returns {Promise<ServeResult>}
  */
-async function handleHead(pathname) {
-  const filePath = path.join(process.cwd(), pathname)
+async function handleHead(filePath) {
   const stats = await fs.stat(filePath).catch(() => undefined)
   if (!stats || !stats.isFile()) {
     console.error(`file not found ${filePath}`)
@@ -149,18 +148,23 @@ async function handleHead(pathname) {
  * @returns {Promise<ServeResult>}
  */
 async function handleListing(prefix) {
-  const dir = `${process.cwd()}/${prefix}`
+  console.log(`listing files with prefix '${prefix}'`)
+  if (!prefix) {
+    prefix = process.cwd()
+  } else if (!prefix.startsWith('/')) {
+    prefix = `${process.cwd()}/${prefix}`
+  }
   try {
-    const stat = await fs.stat(dir)
+    const stat = await fs.stat(prefix)
     if (!stat.isDirectory()) return { status: 400, content: 'not a directory' }
   } catch (error) {
     return { status: 404, content: 'not found' }
   }
 
   const files = []
-  for (const filename of await fs.readdir(dir, { recursive: false })) {
+  for (const filename of await fs.readdir(prefix, { recursive: false })) {
     // get stats for each file
-    const filePath = `${dir}/${filename}`
+    const filePath = `${prefix}/${filename}`
     const stat = await fs.stat(filePath)
 
     if (stat.isFile()) {
@@ -182,9 +186,9 @@ async function handleListing(prefix) {
 
 /**
  * Start http server on given port
- * @param {number} port
+ * @param {{ port?: number, path?: string }} options
  */
-export function serve(port = 2048) {
+export function serve({ port = 2048, path = '' }) {
   // create http server
   http.createServer(async (req, res) => {
     const startTime = new Date()
@@ -240,7 +244,8 @@ export function serve(port = 2048) {
     }
   }).listen(port, () => {
     console.log(`hyperparam server running on http://localhost:${port}`)
-    openUrl(`http://localhost:${port}`)
+    if (path) openUrl(`http://localhost:${port}/files/${path}`)
+    else openUrl(`http://localhost:${port}`)
   })
 }