import { useState } from "react";
import type { Entity } from "@/types/entities";
import type { FeatureId } from "@/types/geo";
import { DEFAULT_ENTITY_TYPE_ID } from "@/lib/entityTypeOptions";
import type {
    CreatedEntitySummary,
    EntityFormState,
    GeometryMetaFormState,
    PendingEntityCreate,
} from "@/lib/editor/session/sessionTypes";

export function useEntitySessionState() {
    const [persistedEntities, setPersistedEntities] = useState<Entity[]>([]);
    const [pendingEntityCreates, setPendingEntityCreates] = useState<PendingEntityCreate[]>([]);
    const [createdEntities, setCreatedEntities] = useState<CreatedEntitySummary[]>([]);
    const [entityStatus, setEntityStatus] = useState<string | null>(null);
    const [selectedFeatureId, setSelectedFeatureId] = useState<FeatureId | null>(null);
    const [entityForm, setEntityForm] = useState<EntityFormState>({
        name: "",
        slug: "",
        type_id: DEFAULT_ENTITY_TYPE_ID,
    });
    const [selectedGeometryEntityIds, setSelectedGeometryEntityIds] = useState<string[]>([]);
    const [geometryMetaForm, setGeometryMetaForm] = useState<GeometryMetaFormState>({
        time_start: "",
        time_end: "",
        binding: "",
    });
    const [isEntitySubmitting, setIsEntitySubmitting] = useState(false);
    const [entityFormStatus, setEntityFormStatus] = useState<string | null>(null);
    const [entitySearchQuery, setEntitySearchQuery] = useState("");
    const [entitySearchResults, setEntitySearchResults] = useState<Entity[]>([]);
    const [selectedSearchEntityId, setSelectedSearchEntityId] = useState<string | null>(null);
    const [isEntitySearchLoading, setIsEntitySearchLoading] = useState(false);

    return {
        persistedEntities,
        setPersistedEntities,
        pendingEntityCreates,
        setPendingEntityCreates,
        createdEntities,
        setCreatedEntities,
        entityStatus,
        setEntityStatus,
        selectedFeatureId,
        setSelectedFeatureId,
        entityForm,
        setEntityForm,
        selectedGeometryEntityIds,
        setSelectedGeometryEntityIds,
        geometryMetaForm,
        setGeometryMetaForm,
        isEntitySubmitting,
        setIsEntitySubmitting,
        entityFormStatus,
        setEntityFormStatus,
        entitySearchQuery,
        setEntitySearchQuery,
        entitySearchResults,
        setEntitySearchResults,
        selectedSearchEntityId,
        setSelectedSearchEntityId,
        isEntitySearchLoading,
        setIsEntitySearchLoading,
    };
}