import type { DatabaseSync } from "node:sqlite";
import { callLLM } from "./utils/llm.ts";
import { decodeHtmlChars } from "./utils/html.ts";

export const getRecipesList = (db: DatabaseSync, ids?: number[]): Recipe[] => {
  if (ids)
    return db
      .prepare(
        `SELECT * FROM recipes WHERE id IN (${ids
          .map(() => "?")
          .join(",")}) ORDER BY name`
      )
      .all(...ids);
  else return db.prepare("SELECT * FROM recipes ORDER BY name").all();
};

export const storeRecipe = async (db: DatabaseSync, recipeId: string) => {
  let recipeUrl = "";
  if (!recipeId) throw new Error("Please provide a recipe ID");
  else if (recipeId.startsWith("https://")) recipeUrl = recipeId;
  else recipeUrl = `https://cookidoo.fr/recipes/recipe/fr-FR/${recipeId}`;

  const recipePage = await fetch(recipeUrl).then((res) => res.text());
  const jsonLdTag = recipePage.match(
    /<script type="application\/ld\+json">(.*?)<\/script>/s
  )?.[1];
  if (!jsonLdTag) throw new Error("No JSON-LD tag found");

  const recipeJson = JSON.parse(jsonLdTag);

  const existingItem = db
    .prepare("SELECT * FROM recipes WHERE url = ?")
    .get(recipeUrl);
  if (existingItem) {
    console.warn("Recipe already exists in db");
    return null;
  }

  const rawIngredientsJson = await callLLM(`
        À partir de la liste suivante, renvoi un objet JSON avec unit, quantity, name pour chacun des éléments.
        ${decodeHtmlChars(recipeJson.recipeIngredient.join("\n"))}
        Si la quantité est un nombre décimal, convertis le.
        Utilise de préférences les unités les plus courantes comme g, pincée, cuillère à café, gousse.
        Les unités doivent toujours être au singulier.
      `);

  const recipe = {
    name: recipeJson.name,
    ingredients: rawIngredientsJson,
    imageUrl: recipeJson.image,
    url: recipeUrl,
  };

  const storedRecipe = db
    .prepare(
      `INSERT INTO recipes (url, name, ingredients, imageUrl) VALUES (?, ?, ?, ?)`
    )
    .run(recipe.url, recipe.name, recipe.ingredients, recipe.imageUrl);

  console.log(`New recipe "${recipe.name}" added to db.`);

  return storedRecipe;
};

export const getGroceryList = (
  db: DatabaseSync,
  recipeIds: number[]
): GroceryItem[] => {
  const recipes = db
    .prepare(`SELECT * FROM recipes WHERE id IN (${recipeIds.join(",")})`)
    .all();

  const rawIngredients = recipes.flatMap((recipe) =>
    JSON.parse(recipe.ingredients as string)
  );

  return rawIngredients.reduce((acc, ingredient) => {
    const existingIngredient = acc.find(
      (i) => i.name === ingredient.name && i.unit === ingredient.unit
    );
    if (existingIngredient) existingIngredient.value += ingredient.value;
    else acc.push(ingredient);
    return acc;
  }, [] as { name: string; value: number; unit: string }[]);
};