import { text, tool, ToolsProviderController } from "@lmstudio/sdk";
import { z } from "zod";
import { configSchematics } from "./configSchematics";

export async function toolsProvider(ctl: ToolsProviderController) {
  const config = ctl.getPluginConfig(configSchematics);
  const apiKey = config.get("kinopoiskApiKey");

  if (!apiKey) {
    console.warn("Kinopoisk API Key is not configured in the plugin settings.");
  }

  const searchMovieTool = tool({
    name: "search_movie_poiskkino_dev",
    description: text`
      Ищет фильмы, сериалы и мультфильмы по названию через API ПоискКино.
      Возвращает список наиболее подходящих результатов, включая ID фильма, название, год выпуска и рейтинг.
      
      Важно: Если в результатах есть подходящий фильм, обязательно используйте 'get_movie_details_poiskkino_dev',
      чтобы получить полное описание, жанры и другие подробности, так как поиск выдает только базовую информацию.
    `,
    parameters: { query: z.string() },
    implementation: async ({ query }, { warn }) => {
      if (!apiKey) return "Ошибка: Пожалуйста, укажите API Key в настройках плагина.";

      const url = new URL("https://api.poiskkino.dev/v1.4/movie/search");
      url.searchParams.set("query", query);

      try {
        const response = await fetch(url.toString(), {
          headers: { "X-API-KEY": apiKey },
        });

        if (!response.ok) {
          return `Ошибка API: ${response.status} ${response.statusText}`;
        }

        const data = await response.json();
        
        const results = data.docs?.map((movie: any) => ({
          id: movie.id,
          name: movie.name,
          year: movie.year,
          rating: movie.rating,
        })) || [];

        if (results.length === 0) {
          return "Ничего не найдено по данному запросу.";
        }

        return {
          results,
          hint: "Для получения полной информации о фильме используйте инструмент 'get_movie_details_poiskkino_dev' с полученным 'id'."
        };
      } catch (error: any) {
        warn(`PoiskKino API Error: ${error.message}`);
        return `Произошла ошибка при поиске: ${error.message}`;
      }
    },
  });

  const getMovieDetailsTool = tool({
    name: "get_movie_details_poiskkino_dev",
    description: text`
      Получает детальную информацию о фильме, сериале или мультфильме по его уникальному ID из ПоискКино.
      Возвращает полное описание, жанры, рейтинг, год выпуска, режиссеров и актерский состав.
    `,
    parameters: { id: z.number() },
    implementation: async ({ id }, { warn }) => {
      if (!apiKey) return "Ошибка: Пожалуйста, укажите API Key в настройках плагина.";

      const url = `https://api.poiskkino.dev/v1.4/movie/${id}`;

      try {
        const response = await fetch(url, {
          headers: { "X-API-KEY": apiKey },
        });

        if (!response.ok) {
          return `Ошибка API: ${response.status} ${response.statusText}`;
        }

        const data = await response.json();

        return {
          title: data.name,
          year: data.year,
          rating: data.rating,
          genre: data.genres,
          description: data.description,
          director: data.directors,
          actors: data.actors,
          alternativeName: data.alternativeName,
        };
      } catch (error: any) {
        warn(`PoiskKino API Error: ${error.message}`);
        return `Произошла ошибка при получении данных: ${error.message}`;
      }
    },
  });

  return [searchMovieTool, getMovieDetailsTool];
}

import { text, tool, ToolsProviderController } from "@lmstudio/sdk";
import { z } from "zod";
import { configSchematics } from "./configSchematics";

export async function toolsProvider(ctl: ToolsProviderController) {
  const config = ctl.getPluginConfig(configSchematics);
  const apiKey = config.get("kinopoiskApiKey");

  if (!apiKey) {
    console.warn("Kinopoisk API Key is not configured in the plugin settings.");
  }

  const searchMovieTool = tool({
    name: "search_movie_poiskkino_dev",
    description: text`
      Ищет фильмы, сериалы и мультфильмы по названию через API ПоискКино.
      Возвращает список наиболее подходящих результатов, включая ID фильма, название, год выпуска и рейтинг.
      
      Важно: Если в результатах есть подходящий фильм, обязательно используйте 'get_movie_details_poiskkino_dev',
      чтобы получить полное описание, жанры и другие подробности, так как поиск выдает только базовую информацию.
    `,
    parameters: { query: z.string() },
    implementation: async ({ query }, { warn }) => {
      if (!apiKey) return "Ошибка: Пожалуйста, укажите API Key в настройках плагина.";

      const url = new URL("https://api.poiskkino.dev/v1.4/movie/search");
      url.searchParams.set("query", query);

      try {
        const response = await fetch(url.toString(), {
          headers: { "X-API-KEY": apiKey },
        });

        if (!response.ok) {
          return `Ошибка API: ${response.status} ${response.statusText}`;
        }

        const data = await response.json();
        
        const results = data.docs?.map((movie: any) => ({
          id: movie.id,
          name: movie.name,
          year: movie.year,
          rating: movie.rating,
        })) || [];

        if (results.length === 0) {
          return "Ничего не найдено по данному запросу.";
        }

        return {
          results,
          hint: "Для получения полной информации о фильме используйте инструмент 'get_movie_details_poiskkino_dev' с полученным 'id'."
        };
      } catch (error: any) {
        warn(`PoiskKino API Error: ${error.message}`);
        return `Произошла ошибка при поиске: ${error.message}`;
      }
    },
  });

  const getMovieDetailsTool = tool({
    name: "get_movie_details_poiskkino_dev",
    description: text`
      Получает детальную информацию о фильме, сериале или мультфильме по его уникальному ID из ПоискКино.
      Возвращает полное описание, жанры, рейтинг, год выпуска, режиссеров и актерский состав.
    `,
    parameters: { id: z.number() },
    implementation: async ({ id }, { warn }) => {
      if (!apiKey) return "Ошибка: Пожалуйста, укажите API Key в настройках плагина.";

      const url = `https://api.poiskkino.dev/v1.4/movie/${id}`;

      try {
        const response = await fetch(url, {
          headers: { "X-API-KEY": apiKey },
        });

        if (!response.ok) {
          return `Ошибка API: ${response.status} ${response.statusText}`;
        }

        const data = await response.json();

        return {
          title: data.name,
          year: data.year,
          rating: data.rating,
          genre: data.genres,
          description: data.description,
          director: data.directors,
          actors: data.actors,
          alternativeName: data.alternativeName,
        };
      } catch (error: any) {
        warn(`PoiskKino API Error: ${error.message}`);
        return `Произошла ошибка при получении данных: ${error.message}`;
      }
    },
  });

  return [searchMovieTool, getMovieDetailsTool];
}