import { Strapi } from "@strapi/strapi";
import { Model } from "@strapi/database/dist/types";

export interface LocationServices {
  getLocationFields: (modelAttributes: any) => (string | boolean)[];
  getModelsWithLocation: () => (Model | boolean)[]
}

const locationServices: ({ strapi }: { strapi: Strapi }) => LocationServices = ({ strapi }: { strapi: Strapi }) => ({
  getLocationFields: (modelAttributes: any) => {
    return Object.entries(modelAttributes)
      .map(([key, value]) => {
        if (
          value &&
          typeof value === "object" &&
          "customField" in value &&
          value.customField === "plugin::location-plugin.location"
        ) {
          return key;
        } else {
          return false;
        }
      })
      .filter(Boolean);
  },
  getModelsWithLocation: () => {
    return strapi.db.config.models
      .filter(
        (model) =>
          (model.uid as string).startsWith("api::") ||
          //@ts-ignore
          model.modelType === "component" ||
          (model.uid as string) === "plugin::users-permissions.user"
      )
      .map((model) => {
        const hasLocationField = Object.values(model.attributes).some(
          (entry) => {
            if (
              entry &&
              typeof entry === "object" &&
              "customField" in entry &&
              entry.customField === "plugin::location-plugin.location"
            ) {
              return true;
            } else {
              return false;
            }
          }
        );
        return hasLocationField ? model : false;
      })
      .filter(Boolean);
  },
});

export default {
  locationServices,
};