import { createStore, reconcile, unwrap } from "solid-js/store" 
import { createSimpleContext } from "./helper"
import type { PromptInfo } from "../../component/prompt/history"

export type HomeRoute = {
  type: "home"
  prompt?: PromptInfo
}

export type SessionRoute = {
  type: "session"
  sessionID: string
  prompt?: PromptInfo
}
export type SaeeolClawRoute = {
  type: "saeeolclaw"
}

export type LocalModelsRoute = {
  type: "local-models"
}

export type PluginRoute = {
  type: "plugin"
  id: string
  data?: Record<string, unknown>
}

export type Route = HomeRoute | SessionRoute | PluginRoute | SaeeolClawRoute | LocalModelsRoute 

export const { use: useRoute, provider: RouteProvider } = createSimpleContext({
  name: "Route",
  init: (props: { initialRoute?: Route }) => {
    const [store, setStore] = createStore<Route>(
      props.initialRoute ??
        (process.env["SAEEOL_ROUTE"]
          ? JSON.parse(process.env["SAEEOL_ROUTE"])
          : {
              type: "home",
            }),
    )
    let previous: Route | undefined

    return {
      get data() {
        return store
      },
      navigate(route: Route) {
        previous = structuredClone(unwrap(store)) 
        setStore(reconcile(route))
      },
      back() {
        const target = previous ?? ({ type: "home" } as const)
        previous = undefined
        console.log("navigate", target)
        setStore(target)
      },
    }
  },
})

export type RouteContext = ReturnType<typeof useRoute>

export function useRouteData<T extends Route["type"]>(type: T) {
  const route = useRoute()
  return route.data as Extract<Route, { type: typeof type }>
}