database.ts

/*
Shortener API
GDSC MBCET
*/

import { MongoClient } from "mongo";

import config from "$env";

console.log("Connecting to MongoDB...");
const client = new MongoClient();
const MONGO_URL = new URL(config.MONGO_URI);
if (!MONGO_URL.searchParams.has("authMechanism")) {
  MONGO_URL.searchParams.set("authMechanism", "SCRAM-SHA-1");
}
try {
  await client.connect(MONGO_URL.href);
} catch (err) {
  console.error("Error connecting to MongoDB", err);
  throw err;
}
const db = client.database("GDSC-Shortener");

interface URLStorageSchema {
  url: string;
  hash: string;
  createdAt: Date;
  clicks: number;
  uniqueClicks: number;
  ips: string[];
}

const URLStorage = db.collection<URLStorageSchema>("URLStorage");

async function doesHashExist(hash: string) {
  const result = await URLStorage.findOne({ hash });
  return !!result;
}

async function createShortURL(url: string, hash: string) {
  hash = hash.toLowerCase();
  if (await doesHashExist(hash)) {
    return false;
  }
  await URLStorage.insertOne({
    url,
    hash,
    createdAt: new Date(),
    clicks: 0,
    uniqueClicks: 0,
    ips: [],
  });

  return true;
}

async function getURL(hash: string, userIp: string) {
  hash = hash.toLowerCase();
  const result = await URLStorage.findOne({ hash });
  if (!result) return null;
  result.clicks++;
  if (!result.ips.includes(userIp)) {
    result.uniqueClicks++;
    result.ips.push(userIp);
  }
  await URLStorage.updateOne(
    { hash },
    {
      $set: {
        ...result,
      },
    }
  );
  return result;
}

async function deleteURL(hash: string) {
  hash = hash.toLowerCase();
  const result = await URLStorage.deleteOne({ hash });
  return result;
}

async function getAllURLs() {
  const result = await URLStorage.find({}).toArray();
  return result;
}

async function editURL(hash: string, url: string) {
  hash = hash.toLowerCase();
  if (!(await doesHashExist(hash))) {
    return false;
  }
  await URLStorage.updateOne({ hash }, { $set: { url: url } });
  return true;
}

export { createShortURL, getURL, deleteURL, getAllURLs, editURL };