From 0c41bffb7582e60073548fd2dff490702915695c Mon Sep 17 00:00:00 2001
From: Roman Timashev <roman@tmshv.ru>
Date: Mon, 14 Oct 2019 17:44:29 +0300
Subject: [PATCH] add to middlewares withCollection (unstable)

---
 src/middlewares/withCollection.js | 21 +++++++++++++++++++++
 src/middlewares/withMiddleware.js |  6 +++---
 2 files changed, 24 insertions(+), 3 deletions(-)
 create mode 100644 src/middlewares/withCollection.js

diff --git a/src/middlewares/withCollection.js b/src/middlewares/withCollection.js
new file mode 100644
index 00000000..a877152d
--- /dev/null
+++ b/src/middlewares/withCollection.js
@@ -0,0 +1,21 @@
+// import { MongoClient } from 'mongodb';
+// const { connect, collection } = require('../../src/core/db')
+// import {collection, connect} from '../core/db'
+const core = require('../core/db')
+
+// const client = new MongoClient(process.env.MONGODB_URI, { useNewUrlParser: true });
+
+export const withCollection = handler => async (req, res) => {
+    await core.connect()
+
+    req.collection = core.collection
+
+    // if (!client.isConnected()) {
+    //     return client.connect().then(() => {
+    //         req.db = client.db('nextjsmongodbapp');
+    //         return handler(req, res);
+    //     });
+    // }
+    // req.db = client.db('nextjsmongodbapp');
+    return handler(req, res)
+}
diff --git a/src/middlewares/withMiddleware.js b/src/middlewares/withMiddleware.js
index 925e3849..ce50ce6a 100644
--- a/src/middlewares/withMiddleware.js
+++ b/src/middlewares/withMiddleware.js
@@ -1,3 +1,3 @@
-export const withMiddleware = handler => async (req, res) => {
-   return handler(req, res);
-};
+import { withCollection } from './withCollection'
+
+export const withMiddleware = handler => withCollection(handler)