const superagent = require('superagent')
const { Api, Hosts } = require('./endpoints')


async function request(path, type, headers, args, host) {
  let timing = Date.now();
  url = Api.HOST + path
  if(process.env.USE_LOCAL_API) url = Hosts.local + ":" + process.env.USE_LOCAL_API + path
  if(host) url = host + path

  // apply default headers
  if(!headers["Authorization"]) headers["Authorization"] = process.env.API_KEY
  if(!headers["user-agent"]) headers["user-agent"] = "labscore/2.0"
  if(!headers["x-labscore-client"]) headers["x-labscore-client"] = "labscore/2.0"

  if (type === "GET") {
    if(!args){
      const response = await superagent.get(url);
      return {
        timings: ((Date.now() - timing) / 1000).toFixed(2),
        response: response
      };
    }
    const response = await superagent.get(url)
      .query(args)
      .set(headers);
      return {
        timings: ((Date.now() - timing) / 1000).toFixed(2),
        response: response
      };
  }
  if (type === "POST") {
    const response = await superagent
      .post(url)
      .set(headers)
      .send(args);
    return {
      timings: ((Date.now() - timing) / 1000).toFixed(2),
      response: response
    };
  }
  throw new Error("unsupported, must either use GET or POST");
}

module.exports.googleGenaiEditImage = async function(context, prompt, url){
  return await request(Api.GOOGLE_GENERATIVEAI_EDIT_IMAGE, "GET", {}, {
    prompt: prompt,
    url: url
  })
}

module.exports.googleGenaiImagen = async function(context, prompt, imageCount = 2){
  return await request(Api.GOOGLE_GENERATIVEAI_IMAGEN, "GET", {}, {
    prompt: prompt,
    image_count: imageCount
  })
}

module.exports.googlePerspective = async function(context, text){
  return await request(Api.GOOGLE_PERSPECTIVE, "GET", {}, {
    text: text
  })
}

module.exports.googleSpeechRecognition = async function(context, url){
  return await request(Api.GOOGLE_SPEECH_RECOGNIZE, "GET", {}, {
    url
  })
}

module.exports.googleSpeechRecognitionWithLabels = async function(context, url){
  return await request(Api.GOOGLE_SPEECH_RECOGNIZE_LABELS, "GET", {}, {
    url
  })
}

module.exports.googleTranslate = async function(context, text, to, from){
  return await request(Api.GOOGLE_TRANSLATE, "GET", {}, {
    text: text,
    to: to,
    from: from
  })
}

module.exports.googleTranslateMulti = async function(context, messages, to, from){
  return await request(Api.GOOGLE_TRANSLATE_MULTI, "POST", {}, {
    messages: messages,
    to: to,
    from: from
  })
}

module.exports.googleVisionColors = async function(context, url){
  return await request(Api.GOOGLE_VISION_COLORS, "GET", {}, {
    url: url
  })
}

module.exports.googleVisionFaces = async function(context, url){
  return await request(Api.GOOGLE_VISION_FACES, "GET", {}, {
    url: url
  })
}

module.exports.googleVisionLabels = async function(context, url){
  return await request(Api.GOOGLE_VISION_LABELS, "GET", {}, {
    url: url
  })
}

module.exports.googleVisionOcr = async function(context, url){
  return await request(Api.GOOGLE_VISION_OCR, "GET", {}, {
    url: url
  })
}

module.exports.googleVisionSafetyLabels = async function(context, url){
  return await request(Api.GOOGLE_VISION_SAFETY_LABELS, "GET", {}, {
    url: url
  })
}

module.exports.googleVisionWebDetection = async function(context, url){
  return await request(Api.GOOGLE_VISION_WEBDETECTION, "GET", {}, {
    url: url
  })
}

module.exports.google = async function(context, query, nsfw){
  return await request(Api.SEARCH_GOOGLE, "GET", {}, {
    q: query,
    nsfw: nsfw
  })
}

module.exports.googleImages = async function(context, query, nsfw){
  return await request(Api.SEARCH_GOOGLE_IMAGES, "GET", {}, {
    q: query,
    nsfw: nsfw
  })
}

module.exports.googleNews = async function(context, query){
  return await request(Api.SEARCH_GOOGLE_NEWS, "GET", {}, {
    q: query
  })
}

module.exports.googleNewsSupplemental = async function(context, supplementalKey){
  return await request(Api.SEARCH_GOOGLE_NEWS_SUPPLEMENTAL, "GET", {}, {
    supplemental_key: supplementalKey
  })
}

module.exports.lyrics = async function(context, query){
  return await request(Api.SEARCH_LYRICS, "GET", {}, {
    q: query
  })
}

module.exports.maps = async function(context, query){
  return await request(Api.SEARCH_MAPS, "GET", {}, {
    q: query
  })
}

module.exports.mapsSupplemental = async function(context, supplementalKey){
  return await request(Api.SEARCH_MAPS_SUPPLEMENTAL, "GET", {}, {
    supplemental_key: supplementalKey
  })
}

module.exports.quora = async function(context, query){
  return await request(Api.SEARCH_QUORA, "GET", {}, {
    q: query
  })
}

module.exports.quoraResult = async function(context, reference){
  return await request(Api.SEARCH_QUORA_RESULT, "GET", {}, {
    ref: reference
  })
}

module.exports.reddit = async function(context, query, nsfw = false){
  return await request(Api.SEARCH_REDDIT, "GET", {}, {
    q: query,
    nsfw: nsfw
  })
}

module.exports.rule34 = async function(context, query, service){
  return await request(Api.SEARCH_RULE34, "GET", {}, {
    q: query,
    service: service
  })
}

module.exports.bing = async function(context, query, nsfw){
  return await request(Api.SEARCH_BING, "GET", {}, {
    q: query,
    nsfw: nsfw
  })
}

module.exports.bingImages = async function(context, query, nsfw){
  return await request(Api.SEARCH_BING_IMAGES, "GET", {}, {
    q: query,
    nsfw: nsfw
  })
}

module.exports.duckduckgo = async function(context, query, nsfw){
  return await request(Api.SEARCH_DUCKDUCKGO, "GET", {}, {
    q: query,
    nsfw: nsfw
  })
}

module.exports.reverseImageSearch = async function(context, url){
  return await request(Api.SEARCH_REVERSE_IMAGE, "GET", {}, {
    url: url
  })
}

module.exports.urbandictionary = async function(context, query){
  return await request(Api.SEARCH_URBANDICTIONARY, "GET", {}, {
    q: query
  })
}

module.exports.weather = async function(context, location){
  return await request(Api.SEARCH_WEATHER, "GET", {}, {
    location: location
  })
}

module.exports.darksky = async function(context, location){
  return await request(Api.UTILS_WEATHER, "GET", {}, {
    location: location
  })
}

module.exports.wikihow = async function(context, query){
  return await request(Api.SEARCH_WIKIHOW, "GET", {}, {
    q: query
  })
}

module.exports.wolframAlpha = async function(context, query){
  return await request(Api.SEARCH_WOLFRAM_ALPHA, "GET", {}, {
    q: query
  })
}

module.exports.wolframSupplemental = async function(context, supplementalKey){
  return await request(Api.SEARCH_WOLFRAM_SUPPLEMENTAL, "GET", {}, {
    supplemental_key: supplementalKey
  })
}

module.exports.youtube = async function(context, query, category){
  return await request(Api.SEARCH_YOUTUBE, "GET", {}, {
    q: query,
    category
  })
}

module.exports.yacht = async function(context, text){
  return await request(Api.PHOTOFUNIA_YACHT, "GET", {}, {
    text: text
  })
}

module.exports.retroWave = async function(context, background = 5, textStyle= 4, text1 = " ", text2 = " ", text3 = " "){
  return await request(Api.PHOTOFUNIA_RETRO_WAVE, "GET", {}, {
    text1: text1,
    text2: text2,
    text3: text3,
    background: background,
    text_style: textStyle
  })
}

module.exports.prideborder = async function(context, url){
  return await request(Api.IMAGE_INHOUSE_PRIDE, "GET", {}, {
    url: url
  })
}

module.exports.deepdream = async function(context, url){
  return await request(Api.IMAGE_DEEPDREAM, "GET", {}, {
    url: url
  })
}

module.exports.imageedit = async function(context, url, prompt){
  return await request(Api.IMAGE_IMAGEEDITOR, "GET", {}, {
    url: url,
    prompt: prompt
  })
}

module.exports.waifu2x = async function(context, url){
  return await request(Api.IMAGE_WAIFU2X, "GET", {}, {
    url: url
  })
}

module.exports.superresolution = async function(context, url){
  return await request(Api.IMAGE_SUPERRESOLUTION, "GET", {}, {
    url: url
  })
}

module.exports.text2image = async function(context, text){
  return await request(Api.IMAGE_TEXT2IMAGE, "GET", {}, {
    text: text
  })
}

module.exports.emogen = async function(context, prompt, style){
  return await request(Api.IMAGE_EMOGEN, "GET", {}, {
    prompt,
    style
  })
}

module.exports.imtranslator = async function(context, text, voice){
  return await request(Api.TTS_IMTRANSLATOR, "GET", {}, {
    text: text,
    voice: voice
  })
}

module.exports.moonbase = async function(context, text){
  return await request(Api.TTS_MOONBASE, "GET", {}, {
    text
  })
}

module.exports.playht = async function(context, text, voice){
  return await request(Api.TTS_PLAYHT, "GET", {}, {
    text: text,
    voice: voice
  })
}

module.exports.polly = async function(context, text, voice){
  return await request(Api.TTS_POLLY, "GET", {}, {
    text: text,
    voice: voice
  })
}

module.exports.sapi4 = async function(context, text, voice, pitch = 50, speed = 150){
  return await request(Api.TTS_SAPI4, "GET", {}, {
    text,
    voice,
    pitch,
    speed
  })
}

module.exports.tiktok = async function(context, text, voice){
  return await request(Api.TTS_TIKTOK, "GET", {}, {
    text: text,
    voice: voice
  })
}

module.exports.dictionary = async function(context, query, language){
  return await request(Api.UTILS_DICTIONARY, "GET", {}, {
    q: query,
    l: language
  })
}

module.exports.emojipedia = async function(context, emoji, codepoint = undefined){
  return await request(Api.UTILS_EMOJIPEDIA, "GET", {}, {
    emoji: emoji,
    with_metadata: "",
    codepoint
  })
}

module.exports.garfield = async function(context,){
  return await request(Api.UTILS_GARFIELD, "GET", {}, {})
}

module.exports.gpt = async function(context, prompt, input, model = "gpt-4o"){
  return await request(Api.UTILS_GPT, "GET", {}, {
    prompt,
    input,
    model
  })
}

module.exports.inferkit = async function(context, input){
  return await request(Api.UTILS_INFERKIT, "GET", {}, {
    input: input
  })
}

module.exports.otter = async function(context){
  return await request(Api.UTILS_OTTER, "GET", {}, {})
}

module.exports.perspective = async function(context, content = []){
  return await request(Api.UTILS_PERSPECTIVE, "GET", {}, {
    input: content.join('\n\n')
  })
}

module.exports.screenshot = async function(context, url, nsfw){
  return await request(Api.UTILS_SCREENSHOT, "GET", {}, {
    url: url,
    nsfw: nsfw
  })
}

module.exports.textGenerator = async function(context, input){
  return await request(Api.UTILS_TEXTGENERATOR, "GET", {}, {
    input: input
  })
}

module.exports.emojiKitchen = async function(emoji){
  return await superagent.get("https://tenor.googleapis.com/v2/featured").query({
    key: process.env.GOOGLE_TENOR_KEY,
    contentfilter: "high",
    media_filter: "png_transparent",
    component: "proactive",
    collection: "emoji_kitchen_v6",
    q: emoji.join('_')
  })
}

// Omnisearch

module.exports.anime = async function(context, query, includeAdultContent){
  return await request(Api.OMNI_ANIME, "GET", {}, {
    q: query,
    include_adult: includeAdultContent
  })
}

module.exports.animeSupplemental = async function(context, supplementalKey){
  return await request(Api.OMNI_ANIME_SUPPLEMENTAL, "GET", {}, {
    supplemental_key: supplementalKey
  })
}

module.exports.movie = async function(context, query, includeAdultContent){
  return await request(Api.OMNI_MOVIE, "GET", {}, {
    q: query,
    include_adult: includeAdultContent
  })
}