import { createEffect } from "solid-js"
import { createStore } from "solid-js/store"
import { getRequestEvent } from "solid-js/web"
import { createSimpleContext } from "@opencode-ai/ui/context"
import {
  LOCALES,
  type Locale,
  clearCookie,
  cookie,
  detectFromLanguages,
  dir as localeDir,
  label as localeLabel,
  localeFromCookieHeader,
  localeFromRequest,
  parseLocale,
  tag as localeTag,
} from "~/lib/language"

function initial() {
  const evt = getRequestEvent()
  if (evt) return localeFromRequest(evt.request)

  if (typeof document === "object") {
    const fromDom = parseLocale(document.documentElement.dataset.locale)
    if (fromDom) return fromDom

    const fromCookie = localeFromCookieHeader(document.cookie)
    if (fromCookie) return fromCookie
  }

  if (typeof navigator !== "object") return "en" satisfies Locale

  const languages = navigator.languages?.length ? navigator.languages : [navigator.language]
  return detectFromLanguages(languages)
}

export const { use: useLanguage, provider: LanguageProvider } = createSimpleContext({
  name: "Language",
  init: () => {
    const [store, setStore] = createStore({
      locale: initial(),
    })

    createEffect(() => {
      if (typeof document !== "object") return
      document.documentElement.lang = localeTag(store.locale)
      document.documentElement.dir = localeDir(store.locale)
      document.documentElement.dataset.locale = store.locale
    })

    return {
      locale: () => store.locale,
      locales: LOCALES,
      label: localeLabel,
      tag: localeTag,
      dir: localeDir,
      setLocale(next: Locale) {
        setStore("locale", next)
        if (typeof document !== "object") return
        document.cookie = cookie(next)
      },
      clear() {
        if (typeof document !== "object") return
        document.cookie = clearCookie()
      },
    }
  },
})
