refactor: common functions extracted from route.ts

visrut7 · Sep 12, 2023 · ea9e5e1 · ea9e5e1
1 parent 7b224a6
commit ea9e5e1
Show file tree

Hide file tree

Showing 4 changed files with 75 additions and 69 deletions.
diff --git a/.prettierrc.js b/.prettierrc.js
@@ -1,12 +1,12 @@
 const config = {
-    semi: false,
-    tabWidth: 2,
-    printWidth: 120,
-    singleQuote: true,
-    jsxSingleQuote: true,
-    trailingComma: 'none',
-    arrowParens: 'always',
-    endOfLine: 'auto',
+  semi: false,
+  tabWidth: 2,
+  printWidth: 120,
+  singleQuote: true,
+  jsxSingleQuote: true,
+  trailingComma: 'none',
+  arrowParens: 'always',
+  endOfLine: 'auto'
 }
 
-module.exports = config
+module.exports = config
diff --git a/app/create-survey/api/route.ts b/app/create-survey/api/route.ts
@@ -1,51 +1,34 @@
-import clientPromise from '@/app/mongodb';
-import { NextResponse } from 'next/server';
-import * as crypto from 'crypto';
-import { v4 as uuidv4 } from "uuid";
-
-function getClientIP(request: Request): string | undefined {
-    const headers = request.headers;
-    const xForwardedFor = headers.get('x-forwarded-for');
-
-    if (xForwardedFor) {
-        const ipList = xForwardedFor.split(',');
-        return ipList[0].trim();
-    } else {
-        if ((request as any).connection === undefined) {
-            return "undefined";
-        }
-        return (request as any).connection.remoteAddress;
-    }
-}
+import clientPromise from '@/app/mongodb'
+import { NextResponse } from 'next/server'
+import { v4 as uuidv4 } from 'uuid'
+import { createHash, getClientIP } from '@/app/utils'
 
 export async function POST(request: Request) {
-    const ip = getClientIP(request);
-    const sha256 = crypto.createHmac("sha256", process.env.SALT!);
-    const ipHash = sha256.update(ip!);
-    const ipHashDigest = ipHash.digest("hex");
-
-    const client = await clientPromise;
-    const db = client.db("survey-db");
-    const creatorHistory = await db.collection("creator-ip-hash").findOne({ ipHash: ipHashDigest });
+  const ip = getClientIP(request)
+  const hash = createHash(ip!)
 
-    const uuid = uuidv4();
+  const client = await clientPromise
+  const db = client.db('survey-db')
+  const creatorHistory = await db.collection('creator-ip-hash').findOne({ ipHash: hash })
 
-    if (creatorHistory === null) {
-        await db.collection("creator-ip-hash").insertOne({ ipHash: ipHashDigest, count: 1 });
-    } else {
-        const prevCount = creatorHistory.count;
+  const uuid = uuidv4()
 
-        if (prevCount >= 3) {
-            return NextResponse.json({ "error": "API Limit is reached" }, { status: 429 });
-        }
+  if (creatorHistory === null) {
+    await db.collection('creator-ip-hash').insertOne({ ipHash: hash, count: 1 })
+  } else {
+    const prevCount = creatorHistory.count
 
-        await db.collection("creator-ip-hash").updateOne({ ipHash: ipHashDigest }, { $set: { count: prevCount + 1 } });
+    if (prevCount >= 3) {
+      return NextResponse.json({ error: 'API Limit is reached' }, { status: 429 })
     }
 
-    const survey = await request.json();
-    survey.creatorIpHash = ipHashDigest;
-    survey.uuid = uuid;
-    await db.collection("survey-templates").insertOne(survey);
+    await db.collection('creator-ip-hash').updateOne({ ipHash: hash }, { $set: { count: prevCount + 1 } })
+  }
 
-    return NextResponse.json({ "survey-id": uuid }, { status: 201 });
-}
+  const survey = await request.json()
+  survey.creatorIpHash = hash
+  survey.uuid = uuid
+  await db.collection('survey-templates').insertOne(survey)
+
+  return NextResponse.json({ 'survey-id': uuid }, { status: 201 })
+}
diff --git a/app/mongodb.ts b/app/mongodb.ts
@@ -1,35 +1,35 @@
 import { MongoClient, ServerApiVersion } from 'mongodb'
 
-const uri = process.env.MONGODB_URI;
+const uri = process.env.MONGODB_URI
 
 if (!uri) {
-    throw new Error('Please define the MONGODB_URI environment variable inside .env.local')
+  throw new Error('Please define the MONGODB_URI environment variable inside .env.local')
 }
 
-let client: MongoClient;
-let clientPromise: Promise<MongoClient>;
+let client: MongoClient
+let clientPromise: Promise<MongoClient>
 
 const options = {
-    serverApi: {
-        version: ServerApiVersion.v1,
-        strict: true,
-        deprecationErrors: true
-    }
+  serverApi: {
+    version: ServerApiVersion.v1,
+    strict: true,
+    deprecationErrors: true
+  }
 }
 
 declare global {
-    var _mongoClientPromise: Promise<MongoClient>;
+  var _mongoClientPromise: Promise<MongoClient>
 }
 
 if (process.env.NODE_ENV === 'development') {
-    if (!global._mongoClientPromise) {
-        client = new MongoClient(uri, options);
-        global._mongoClientPromise = client.connect();
-    }
-    clientPromise = global._mongoClientPromise
-} else {
+  if (!global._mongoClientPromise) {
     client = new MongoClient(uri, options)
-    clientPromise = client.connect()
+    global._mongoClientPromise = client.connect()
+  }
+  clientPromise = global._mongoClientPromise
+} else {
+  client = new MongoClient(uri, options)
+  clientPromise = client.connect()
 }
 
-export default clientPromise;
+export default clientPromise
diff --git a/app/utils.ts b/app/utils.ts
@@ -0,0 +1,23 @@
+import * as crypto from 'crypto'
+
+export function getClientIP(request: Request): string | undefined {
+  const headers = request.headers
+  const xForwardedFor = headers.get('x-forwarded-for')
+
+  if (xForwardedFor) {
+    const ipList = xForwardedFor.split(',')
+    return ipList[0].trim()
+  } else {
+    if ((request as any).connection === undefined) {
+      return 'undefined'
+    }
+    return (request as any).connection.remoteAddress
+  }
+}
+
+export function createHash(data: string): string {
+  const sha256 = crypto.createHmac('sha256', process.env.SALT!)
+  const ipHash = sha256.update(data!)
+  const ipHashDigest = ipHash.digest('hex')
+  return ipHashDigest
+}