URL-Shortener

Author: abhishekprajapatt

connect.js

@@ -0,0 +1,9 @@
+const mongoose = require('mongoose');
+
+async function connectToMongoDB(url) {
+    return mongoose.connect(url);
+}
+
+module.exports = {
+    connectToMongoDB,
+}

controllers/url.js

@@ -0,0 +1,30 @@
+const shortid = require('shortid');
+const URL = require('../models/url');
+async function handleGenerateNewShortURL(req, res){
+    const body = req.body;
+    if(!body.url) return res.status(400).json({ error: 'url is required'})
+    const shortID = shortid(8);
+    await URL.create({
+        shortId: shortID,
+        redirectURL: body.url,
+        visitHistory: [],
+
+    });
+
+    return res.render('home', {
+        id: shortID
+    });
+}
+
+async function handleGetAnalytics(req, res){
+    const shortId = req.params.shortId;
+    const result = await URL.findOne({ shortId });
+    return res.json({
+        totalClicks: result.visitHistory.length, analytics:visitHistory
+    });
+}
+
+module.exports = {
+    handleGenerateNewShortURL,
+    handleGetAnalytics
+}

index.js

@@ -0,0 +1,39 @@
+const express = require('express');
+const path = require('path');
+const staticRouter = require('./routes/staticRouter');
+const { connectToMongoDB } = require('./connect');
+const urlRoute = require('./routes/url');
+const URL = require('./models/url');
+const app = express();
+const PORT = 8000;
+
+connectToMongoDB('mongodb://localhost:27017/short-url').then(() =>
+  console.log('Mongodb Connected')
+);
+
+app.set('view engine', 'ejs');
+app.set('views', path.resolve('./views'));
+
+// middleware
+app.use(express.json());
+app.use(express.urlencoded({ extended: false }));
+app.use('/url', urlRoute);
+app.use('/', staticRouter);
+
+app.get('/url/:shortId', async (req, res) => {
+  const shortId = req.params.shortId;
+  const entry = await URL.findOneAndUpdate(
+    {
+      shortId,
+    },
+    {
+      $push: {
+        visitHistory: {
+          timestamp: Date.now(),
+        },
+      },
+    }
+  );
+  res.redirect(entry.redirectURL);
+});
+app.listen(PORT, () => console.log(`Server is running on PORT:${PORT}`));

models/url.js

@@ -0,0 +1,18 @@
+const mongoose = require('mongoose');
+const urlSchema = new mongoose.Schema({
+    shortId: {
+        type: String,
+        required: true,
+        unique: true,
+    },
+    redirectURL: {
+        type: String,
+        required: true,
+    },
+    visitHistory: [{ timestamp: { type: Number }}],
+},
+    { timestamps: true }
+);
+
+const URL = mongoose.model("url", urlSchema);
+module.exports = URL;