From 89c96e2ad408b83523477dd4a1c9b83fc412f450 Mon Sep 17 00:00:00 2001
From: Jonatan Nilsson <jonatan@nilsson.is>
Date: Sun, 10 Oct 2021 20:38:12 +0000
Subject: [PATCH] QueryHandler: Added new handy query handler middleware

---
 flaska.mjs                |  8 ++++++++
 test/flaska.in.test.mjs   |  4 ++++
 test/middlewares.test.mjs | 24 ++++++++++++++++++++++++
 3 files changed, 36 insertions(+)
 create mode 100644 test/middlewares.test.mjs

diff --git a/flaska.mjs b/flaska.mjs
index 2c68663..bf337d9 100644
--- a/flaska.mjs
+++ b/flaska.mjs
@@ -1,4 +1,5 @@
 import http from 'http'
+import { URL } from 'url'
 import stream from 'stream'
 
 /**
@@ -50,6 +51,12 @@ function assertIsHandler(handler, name) {
   }
 }
 
+export function QueryHandler() {
+  return function(ctx) {
+    ctx.query = (new URL(ctx.req.url, 'http://localhost')).searchParams
+  }
+}
+
 export class FlaskaRouter {
   constructor() {
     this.root = new Branch()
@@ -379,6 +386,7 @@ export class Flaska {
       search: search,
       state: {},
       status: 200,
+      query: new Map(),
       body: null,
       type: null,
       length: null,
diff --git a/test/flaska.in.test.mjs b/test/flaska.in.test.mjs
index 7db2183..bb78c15 100644
--- a/test/flaska.in.test.mjs
+++ b/test/flaska.in.test.mjs
@@ -151,6 +151,10 @@ t.describe('#requestStart()', function() {
         assert.strictEqual(ctx.type, null)
         assert.strictEqual(ctx.length, null)
         assert.strictEqual(ctx.log, assertLog)
+        assert.ok(ctx.query)
+        assert.ok(ctx.query.get)
+        assert.ok(ctx.query.set)
+        assert.ok(ctx.query.delete)
         cb()
       } catch (err) { cb(err) }
     }
diff --git a/test/middlewares.test.mjs b/test/middlewares.test.mjs
new file mode 100644
index 0000000..d9f829b
--- /dev/null
+++ b/test/middlewares.test.mjs
@@ -0,0 +1,24 @@
+import { Eltro as t, assert} from 'eltro'
+import { QueryHandler } from '../flaska.mjs'
+
+t.describe('#QueryHandler()', function() {
+  let queryHandler = QueryHandler()
+
+  t.test('should return a handler', function() {
+    assert.strictEqual(typeof(queryHandler), 'function')
+  })
+
+  t.test('should support separating query from request url', function() {
+    const assertItem1 = 'safdsfdsag'
+    const assertItem2 = 'hello%20world'
+    const ctx = {
+      req: {
+        url: `/some/path?item1=${assertItem1}&ITEM2=${assertItem2}`
+      }
+    }
+
+    queryHandler(ctx)
+    assert.strictEqual(ctx.query.get('item1'), assertItem1)
+    assert.strictEqual(ctx.query.get('ITEM2'), 'hello world')
+  })
+})