import base
import std/[re,httpclient,json,strutils]


# TODO: Add language detection and translate into HOME language
#       Once one translation is done, stay in that translation until told otherwise

const LIBRETRANSLATE_URL = "https://libretranslate.pussthecat.org/translate"
const HOME = "en"
const FIRST_CHOICE = "de"
let   LANG_RE = re"\w+>\w+"

type
  Request = object
    source: string
    target: string
    q: string
    format: string

proc main(message: varargs[string] = @[])

proc newRequest(): Request =
  return Request(source:FIRST_CHOICE,
                 target:HOME,
                 q: "",
                 format: "text",)

proc parseReq(req: string): Request =
  var r = newRequest()
  var langs = findAll(req,LANG_RE)
  if len(langs) > 0:
    let lang = langs[0]
    langs = lang.split(re">")
    let query = replace(req,LANG_RE,"")
    if query != "":
      r.source = langs[0].toLowerAscii()
      r.target = langs[1].toLowerAscii()
      r.q = query
    else:
      main()
  else:
    main("No language selections made. i.e. en>de, de>en, etc.")
  return r

proc answerTranslate(response: string, req: Request) =
  let r = parseJson(response)
  try:
    let ans = r["translatedText"].getStr()
    main(ans, req.source & " > " & req.target)
  except:
    main("Error : " & r["error"].getStr())

proc goTranslate(req: string) =
  let r = parseReq(req)
  if r.q != "":
    var client = newHTTPClient()
    client.headers = newHttpHeaders({"Content-Type":"application/json"})
    let body = %*r
    let response = client.request(LIBRETRANSLATE_URL, httpMethod = HttpPost, body = $body)
    if response.status != "":
      answerTranslate(response.body, r)


proc main(message:varargs[string] = @[]) =
  var info = newInfo("Translate")
  info.selected_bg = green
  var args: seq[string] = @[]
  if len(message) > 0:
    for msg in message:
      if msg != "":
        args.add(msg)
    args.add("back")
  args.add("exit")
  let output = outputData(info,args)
  if output == "exit" or output == "":
    return
  elif output == "back":
    main()
  elif output in message:
    copyToClipboard(output)
  else:
    goTranslate(output)
  return

if isMainModule:
  main()