From 69a4de871d337f493fb2e3ad1cf3553bbc1d9aa6 Mon Sep 17 00:00:00 2001
From: Li Zhineng <im@zhineng.li>
Date: Tue, 1 Jul 2025 15:51:36 +0800
Subject: split methods

---
 server.mjs | 134 +++++++++++++++++++++++++++++++++++++++++--------------------
 1 file changed, 90 insertions(+), 44 deletions(-)

diff --git a/server.mjs b/server.mjs
index b958049..ff9d7ba 100644
--- a/server.mjs
+++ b/server.mjs
@@ -3,60 +3,106 @@
 import { DatabaseSync } from 'node:sqlite'
 import { createServer } from 'node:http'
 
-const database = new DatabaseSync(':memory:')
+const initializeDatabase = () => {
+  const db = new DatabaseSync(':memory:')
 
-database.exec(`
-  CREATE TABLE eagles(
-    id INTEGER PRIMARY KEY,
-    mac_address TEXT,
-    key TEXT
-  ) STRICT
-`)
+  db.exec(`
+    CREATE TABLE eagles(
+      id INTEGER PRIMARY KEY,
+      mac_address TEXT,
+      key TEXT
+    ) STRICT
+  `)
 
-const server = createServer((req, res) => {
-  if (req.method === 'GET' && req.url === '/') {
-    res.writeHead(200, { 'Content-Type': 'text/plain' })
-    res.end('It works!\n')
-  } else if (req.method === 'GET' && req.url === '/gettime') {
-    res.writeHead(200, { 'Content-Type': 'application/json' })
-    res.end(JSON.stringify({ time: Math.floor(Date.now() / 1000) }))
-  } else if (req.method === 'GET' && req.url.startsWith('/eagle')) {
-    const params = new URLSearchParams(req.url.substring('/eagle'.length))
-    const mac = params.get('mac')
-    const key = params.get('key')
-    if (mac === null || key === null) {
-      res.writeHead(400, { 'Content-Type': 'application/json' })
-      res.end('{}\n')
-      return
-    }
-    const insert = database.prepare('INSERT INTO eagles(mac_address, key) VALUES(?, ?)')
-    const record = insert.run(mac, key)
-    res.writeHead(200, { 'Content-Type': 'application/json' })
-    res.end(JSON.stringify({ status: 200, data: { eagleId: record.lastInsertRowid } }))
-  } else if (req.method === 'GET' && req.url.startsWith('/exchange')) {
-    const params = new URLSearchParams(req.url.substring('/exchange'.length))
-    const device = params.get('device')
-    if (device === null) {
-      res.writeHead(422)
-      res.end()
-      return
-    }
-    const query = database.prepare('SELECT * FROM eagles WHERE id = ?').get(device)
-    if (query === undefined) {
-      res.writeHead(404)
-      res.end()
-      return
+  return db
+}
+
+const indexController = (req, res) => {
+  res.writeHead(200, { 'Content-Type': 'text/plain' })
+  res.end('It works!\n')
+}
+
+const getTimeControlller = (req, res) => {
+  res.writeHead(200, { 'Content-Type': 'application/json' })
+  res.end(JSON.stringify({ time: Math.floor(Date.now() / 1000) }))
+}
+
+const eagleController = (req, res) => {
+  const [, query] = req.url.split('?')
+  const params = new URLSearchParams(query)
+  const mac = params.get('mac')
+  const key = params.get('key')
+
+  if (mac === null || key === null) {
+    res.writeHead(400, { 'Content-Type': 'application/json' })
+    res.end('{}\n')
+    return
+  }
+
+  const record = database
+    .prepare('INSERT INTO eagles(mac_address, key) VALUES(?, ?)')
+    .run(mac, key)
+
+  res.writeHead(200, { 'Content-Type': 'application/json' })
+  res.end(JSON.stringify({
+    status: 200,
+    data: {
+      eagleId: record.lastInsertRowid
     }
-    res.writeHead(200)
-    res.end(JSON.stringify({ key: query.key }))
-  } else {
+  }))
+}
+
+const exchangeController = (req, res) => {
+  const [, query] = req.url.split('?')
+  const params = new URLSearchParams(query)
+  const device = params.get('device')
+
+  if (device === null) {
+    res.writeHead(422)
+    res.end()
+    return
+  }
+
+  const record = database
+    .prepare('SELECT * FROM eagles WHERE id = ?')
+    .get(device)
+
+  if (record === undefined) {
     res.writeHead(404)
     res.end()
+    return
+  }
+
+  res.writeHead(200, { 'Content-Type': 'application/json' })
+  res.end(JSON.stringify({ key: record.key }))
+}
+
+const notFoundController = (req, res) => {
+  res.writeHead(404)
+  res.end()
+}
+
+const server = createServer((req, res) => {
+  const routes = [
+    { method: 'GET', path: '/', handler: indexController },
+    { method: 'GET', path: '/gettime', handler: getTimeControlller },
+    { method: 'GET', path: '/eagle', handler: eagleController },
+    { method: 'GET', path: '/exchange', handler: exchangeController },
+  ]
+
+  const [path] = req.url.split('?')
+  for (const route of routes) {
+    if (req.method === route.method && path === route.path) {
+      return route.handler(req, res)
+    }
   }
+
+  notFoundController(req, res)
 })
 
 const hostname = process.env.HOSTNAME || '0.0.0.0'
 const port = process.env.PORT || 80
+const database = initializeDatabase()
 
 server.listen(port, hostname, () => {
   console.log(`Listening on ${hostname}:${port}`)
-- 
cgit v1.2.3