/* ============================================================================
 * Copyright (c) Palo Alto Networks
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 * ========================================================================== */

import type { Middleware } from "@reduxjs/toolkit";
import {
  setAuthData,
  setSelectedAuth,
} from "@theme/ApiExplorer/Authorization/slice";
import type { AppDispatch, RootState } from "@theme/ApiItem/store";
import type { ServerObject } from "docusaurus-plugin-openapi-docs/src/openapi/types";
import type { ThemeConfig } from "docusaurus-theme-openapi-docs/src/types";

import { createStorage, hashArray } from "./storage-utils";

export function createPersistenceMiddleware(options: ThemeConfig["api"]) {
  const persistenceMiddleware: Middleware<{}, RootState, AppDispatch> =
    (storeAPI) =>
    (next) =>
    (action: ReturnType<typeof setAuthData | typeof setSelectedAuth> | any) => {
      const result = next(action);

      const state = storeAPI.getState();

      const storage = createStorage(
        options?.authPersistence ?? "sessionStorage"
      );

      if (action.type === setAuthData.type) {
        for (const [key, value] of Object.entries(state.auth.data)) {
          if (Object.values(value as any).filter(Boolean).length > 0) {
            storage.setItem(key, JSON.stringify(value));
          } else {
            storage.removeItem(key);
          }
        }
      }

      if (action.type === setSelectedAuth.type) {
        if (state.auth.selected) {
          storage.setItem(
            hashArray(Object.keys(state.auth.options)),
            state.auth.selected
          );
        }
      }

      // TODO: determine way to rehydrate without flashing
      if (action.type === "contentType/setContentType") {
        storage.setItem("contentType", action.payload);
      }

      if (action.type === "accept/setAccept") {
        storage.setItem("accept", action.payload);
      }

      if (action.type === "server/setServer") {
        storage.setItem("server", action.payload);
      }

      if (action.type === "server/setServerVariable") {
        const server = storage.getItem("server");
        if (!server) {
          return result;
        }
        const variables = JSON.parse(action.payload);

        const serverObject = (JSON.parse(server) as ServerObject) ?? {};
        if (serverObject.variables?.[variables.key]) {
          serverObject.variables[variables.key].default = variables.value;
          storage.setItem("server", JSON.stringify(serverObject));
        }
      }

      return result;
    };
  return persistenceMiddleware;
}