import type {
  KvQuery,
  ObjectQuery,
  ObjectValue,
} from "@silverbulletmd/silverbullet/types";
import type { SysCallMapping } from "$lib/plugos/system.ts";
import {
  findAllQueryVariables,
  type LuaCollectionQuery,
  type LuaQueryCollection,
} from "$common/space_lua/query_collection.ts";
import {
  LuaEnv,
  LuaRuntimeError,
  type LuaStackFrame,
  luaValueToJS,
} from "$common/space_lua/runtime.ts";
import type { CommonSystem } from "$common/common_system.ts";

// These are just wrappers around the system.invokeFunction calls, but they make it easier to use the index

export function indexSyscalls(commonSystem: CommonSystem): SysCallMapping {
  return {
    "index.indexObjects": (ctx, page: string, objects: ObjectValue<any>[]) => {
      return commonSystem.system.syscall(ctx, "system.invokeFunction", [
        "index.indexObjects",
        page,
        objects,
      ]);
    },
    "index.queryObjects": (
      ctx,
      tag: string,
      query: ObjectQuery,
      ttlSecs?: number,
    ) => {
      return commonSystem.system.syscall(ctx, "system.invokeFunction", [
        "index.queryObjects",
        tag,
        query,
        ttlSecs,
      ]);
    },
    "index.queryLuaObjects": (
      ctx,
      tag: string,
      query: LuaCollectionQuery,
      scopedVariables?: Record<string, any>,
    ) => {
      return commonSystem.system.syscall(ctx, "system.invokeFunction", [
        "index.queryLuaObjects",
        tag,
        query,
        scopedVariables,
      ]);
    },
    "index.queryDeleteObjects": (ctx, tag: string, query: ObjectQuery) => {
      return commonSystem.system.syscall(ctx, "system.invokeFunction", [
        "index.queryDeleteObjects",
        tag,
        query,
      ]);
    },
    "index.query": (ctx, query: KvQuery, variables?: Record<string, any>) => {
      return commonSystem.system.syscall(ctx, "system.invokeFunction", [
        "index.query",
        query,
        variables,
      ]);
    },
    "index.getObjectByRef": (ctx, page: string, tag: string, ref: string) => {
      return commonSystem.system.syscall(ctx, "system.invokeFunction", [
        "index.getObjectByRef",
        page,
        tag,
        ref,
      ]);
    },
    "index.tag": (_ctx, tagName: string): LuaQueryCollection => {
      return {
        query: async (
          query: LuaCollectionQuery,
          env: LuaEnv,
          sf: LuaStackFrame,
        ): Promise<any[]> => {
          const global = commonSystem.spaceLuaEnv.env;
          const localVars = findAllQueryVariables(query).filter((v) =>
            !global.has(v) && v !== "_"
          );
          const scopedVariables: Record<string, any> = {};
          for (const v of localVars) {
            try {
              let value = env.get(v);
              if (value instanceof LuaEnv) {
                // We don't want to include the global environment in the serialized value
                value = value.toJSON(["_GLOBAL"]);
              }
              const jsonValue = await luaValueToJS(value);
              // Ensure this is JSON serializable
              JSON.stringify(jsonValue);
              scopedVariables[v] = jsonValue;
            } catch (e: any) {
              console.error(
                "Failed to JSON serialize variable",
                v,
                e,
              );
              throw new LuaRuntimeError(
                `Failed to JSON serialize variable ${v} in query`,
                sf,
              );
            }
          }
          return (await global.get("datastore").get("query_lua").call(
            sf,
            [
              "idx",
              tagName,
            ],
            query,
            scopedVariables,
          )).toJSArray();
        },
      };
    },
  };
}