import { Collection, Filter, ObjectId as MongoObjectId } from "mongodb";
import { getDb, ObjectId } from "../index";
import type {
  CreateWebsiteUrlInput,
  WebsiteUrl,
  WebsiteUrlDoc,
} from "./websiteUrls.types";

export const getWebsiteUrlsCollection = (): Collection<WebsiteUrl> => {
  return getDb().collection<WebsiteUrl>("websiteUrls");
};

export const getWebsiteUrlsByClient = async (
  clientId: string,
): Promise<WebsiteUrlDoc[]> => {
  return getWebsiteUrlsCollection()
    .find({ clientId })
    .sort({ createdAt: 1 })
    .toArray();
};

export const getWebsiteUrlById = async (
  id: string,
): Promise<WebsiteUrlDoc | null> => {
  return getWebsiteUrlsCollection().findOne({ _id: new ObjectId(id) });
};

export const findWebsiteUrls = async (
  filter: Filter<WebsiteUrl> = {},
): Promise<WebsiteUrlDoc[]> => {
  return getWebsiteUrlsCollection().find(filter).toArray();
};

export const createWebsiteUrl = async (
  input: CreateWebsiteUrlInput,
): Promise<MongoObjectId> => {
  const now = new Date();
  const { insertedId } = await getWebsiteUrlsCollection().insertOne({
    clientId: input.clientId,
    name: input.name,
    baseUrl: input.baseUrl,
    createdAt: now,
    updatedAt: now,
  });

  return insertedId;
};

export const updateWebsiteUrl = async (
  id: string,
  updates: Partial<Pick<WebsiteUrl, "name" | "baseUrl">>,
): Promise<WebsiteUrlDoc | null> => {
  return getWebsiteUrlsCollection().findOneAndUpdate(
    { _id: new ObjectId(id) },
    { $set: { ...updates, updatedAt: new Date() } },
    { returnDocument: "after" },
  );
};

export const deleteWebsiteUrl = async (id: string): Promise<boolean> => {
  const result = await getWebsiteUrlsCollection().deleteOne({
    _id: new ObjectId(id),
  });
  return result.deletedCount === 1;
};