import { decodeSearch, encodeSearch, parseDecode } from "core/utils";
import { isEqual, transform } from "lodash-es";
import { useLocation, useSearch as useWouterSearch } from "wouter";
import { type s, parse } from "core/object/schema";

// @todo: migrate to Typebox
export function useSearch<Schema extends s.TAnySchema = s.TAnySchema>(
   schema: Schema,
   defaultValue?: Partial<s.StaticCoerced<Schema>>,
) {
   const searchString = useWouterSearch();
   const [location, navigate] = useLocation();
   let value = (defaultValue ? parse(schema, defaultValue as any) : {}) as s.StaticCoerced<Schema>;

   if (searchString.length > 0) {
      value = parse(schema, decodeSearch(searchString));
      //console.log("search:decode", value);
   }

   // @todo: add option to set multiple keys at once
   function set<Key extends keyof s.StaticCoerced<Schema>>(
      key: Key,
      value: s.StaticCoerced<Schema>[Key],
   ): void {
      //console.log("set", key, value);
      const update = parse(schema, { ...decodeSearch(searchString), [key]: value });
      const search = transform(
         update as any,
         (result, value, key) => {
            if (defaultValue && isEqual(value, defaultValue[key])) return;
            result[key] = value;
         },
         {} as s.StaticCoerced<Schema>,
      );
      const encoded = encodeSearch(search, { encode: false });
      navigate(location + (encoded.length > 0 ? "?" + encoded : ""));
   }

   return {
      value: value as Required<s.StaticCoerced<Schema>>,
      set,
   };
}