import { computed, reaction } from "mobx";
import { KubeObjectStore } from "../../kube-object.store";
import { autobind } from "../../utils";
import { crdApi, CustomResourceDefinition } from "../../api/endpoints/crd.api";
import { apiManager } from "../../api/api-manager";
import { KubeApi } from "../../api/kube-api";
import { CRDResourceStore } from "./crd-resource.store";
import { KubeObject } from "../../api/kube-object";
import { addLensKubeObjectMenuItem, addLensKubeObjectMenuItemRaw } from "../../../extensions/registries";
import { Remove, Update } from "@material-ui/icons";
import { editResourceTab } from "../dock/edit-resource.store";

function initStore(crd: CustomResourceDefinition) {
  const apiBase = crd.getResourceApiBase();
  const kind = crd.getResourceKind();
  const isNamespaced = crd.isNamespaced();
  const api = apiManager.getApi(apiBase) || new KubeApi({ apiBase, kind, isNamespaced });

  if (!apiManager.getStore(api)) {
    const store = new CRDResourceStore(api);

    apiManager.registerStore(store);

    addLensKubeObjectMenuItemRaw({
      kind,
      apiVersions: [api.apiVersion],
      Icon: Remove,
      onClick: object => store.remove(object),
      text: "Delete",
    });

    addLensKubeObjectMenuItemRaw({
      kind,
      apiVersions: [api.apiVersion],
      Icon: Update,
      onClick: editResourceTab,
      text: "Update",
    });
  }
}

@autobind()
export class CRDStore extends KubeObjectStore<CustomResourceDefinition> {
  api = crdApi;

  constructor() {
    super();

    // auto-init stores for crd-s
    reaction(() => this.items.toJS(), items => items.forEach(initStore));
  }

  protected sortItems(items: CustomResourceDefinition[]) {
    return super.sortItems(items, [
      crd => crd.getGroup(),
      crd => crd.getName(),
    ]);
  }

  @computed get groups() {
    const groups: Record<string, CustomResourceDefinition[]> = {};

    return this.items.reduce((groups, crd) => {
      const group = crd.getGroup();

      if (!groups[group]) groups[group] = [];
      groups[group].push(crd);

      return groups;
    }, groups);
  }

  getByGroup(group: string, pluralName: string) {
    const crdInGroup = this.groups[group];

    if (!crdInGroup) return null;

    return crdInGroup.find(crd => crd.getPluralName() === pluralName);
  }

  getByObject(obj: KubeObject) {
    if (!obj) return null;
    const { kind, apiVersion } = obj;

    return this.items.find(crd => (
      kind === crd.getResourceKind() && apiVersion === `${crd.getGroup()}/${crd.getVersion()}`
    ));
  }
}

export const crdStore = new CRDStore();

apiManager.registerStore(crdStore);

addLensKubeObjectMenuItem({
  Object: CustomResourceDefinition,
  Icon: Remove,
  onClick: object => crdStore.remove(object),
  text: "Delete",
});

addLensKubeObjectMenuItem({
  Object: CustomResourceDefinition,
  Icon: Update,
  onClick: editResourceTab,
  text: "Update",
});