import { useQuery } from "@tanstack/react-query";
import { Plus, Search } from "lucide-react";
import { useEffect, useMemo, useRef, useState } from "react";
import { Tldraw } from "tldraw";
import { NewSpaceDialog } from "@/components/new-space-dialog";
import { SpaceActions } from "@/components/space-actions";
import { Badge } from "@/components/ui/badge";
import { Button } from "@/components/ui/button";
import {
  Card,
  CardContent,
  CardFooter,
  CardHeader,
} from "@/components/ui/card";
import { Input } from "@/components/ui/input";
import { listUserSpaceSnapshots } from "@/lib/appwrite-db";
import { authClient } from "@/lib/auth-client";

type SpaceCard = {
  id: string;
  name: string;
  lastEdited: string;
  snapshotText: string;
  itemCount: number;
  color: string;
};

export function SpacesGrid() {
  const { data: session } = authClient.useSession();

  const spacesQuery = useQuery({
    queryKey: ["spaces", session?.$id],
    queryFn: async () => {
      if (!session?.$id) {
        return [] as SpaceCard[];
      }
      const rows = await listUserSpaceSnapshots(session.$id);
      return rows.map(({ row, snapshot }) => {
        const snapshotText = snapshot
          ? JSON.stringify(snapshot)
          : (row.snapshot ?? "");
        const itemCount = (() => {
          try {
            const doc = snapshot?.document as unknown as
              | {
                  store?: { records?: Record<string, unknown> };
                  records?: Record<string, unknown>;
                }
              | undefined;
            const recs = doc?.store?.records ?? doc?.records;
            return recs ? Object.keys(recs).length : 0;
          } catch {
            return 0;
          }
        })();
        return {
          id: row.spaceId,
          name: row.title || row.spaceId,
          lastEdited: row.$updatedAt
            ? new Date(row.$updatedAt).toLocaleString()
            : "",
          snapshotText,
          itemCount,
          // Use actual color from DB or fallback to color variety
          color: row.color || "#3b82f6", // Default to blue if no color
        } satisfies SpaceCard;
      });
    },
    staleTime: 10_000,
  });

  const spaces = spacesQuery.data ?? [];

  // Use an uncontrolled input and a ref-based debounce to avoid rerendering
  // on every keystroke. We only update `debouncedSearchTerm` after the
  // debounce delay.
  const inputRef = useRef<HTMLInputElement | null>(null);
  const timerRef = useRef<number | null>(null);
  const [debouncedSearchTerm, setDebouncedSearchTerm] = useState("");
  const DEBOUNCE_MS = 200;

  useEffect(() => {
    return () => {
      if (timerRef.current) {
        clearTimeout(timerRef.current);
      }
    };
  }, []);

  const filteredSpaces = useMemo(() => {
    const q = debouncedSearchTerm.trim().toLowerCase();
    if (!q) {
      return spaces;
    }

    return spaces.filter((s) => {
      return s.name.toLowerCase().includes(q);
    });
  }, [spaces, debouncedSearchTerm]);

  let badgeText = "";
  if (spacesQuery.isLoading) {
    badgeText = "Loading…";
  } else if (debouncedSearchTerm) {
    badgeText = `${filteredSpaces.length} of ${spaces.length} spaces`;
  } else {
    badgeText = `${spaces.length} spaces`;
  }

  return (
    <div className="space-y-6 p-6">
      {/* Header */}
      <div className="flex items-center justify-between">
        <div>
          <h1 className="font-bold text-3xl text-foreground">Your Spaces</h1>
          <p className="mt-1 text-muted-foreground">
            Organize your thoughts and ideas in visual workspaces
          </p>
        </div>
        <NewSpaceDialog onSpaceCreated={() => spacesQuery.refetch()} />
      </div>

      {/* Search and Filters */}
      <div className="flex items-center gap-4">
        <div className="relative max-w-md flex-1">
          <Search className="-translate-y-1/2 absolute top-1/2 left-3 h-4 w-4 transform text-muted-foreground" />
          <Input
            className="pl-10"
            onChange={() => {
              if (timerRef.current) {
                clearTimeout(timerRef.current);
              }
              // Use window.setTimeout so the returned id is a number
              timerRef.current = window.setTimeout(() => {
                const value = inputRef.current?.value ?? "";
                setDebouncedSearchTerm(value);
                timerRef.current = null;
              }, DEBOUNCE_MS);
            }}
            placeholder="Search spaces..."
            ref={inputRef}
          />
        </div>
        <Badge className="px-3 py-1" variant="secondary">
          {badgeText}
        </Badge>
      </div>

      {/* Spaces Grid */}
      <div className="grid grid-cols-1 gap-6 md:grid-cols-2 lg:grid-cols-3">
        {filteredSpaces.map((space) => {
          const snapshot = space.snapshotText
            ? JSON.parse(space.snapshotText).document
            : undefined;
          return (
            <Card
              className="group cursor-pointer border-border/50 transition-all duration-200 hover:border-border hover:shadow-lg"
              key={space.id}
              onClick={() => {
                window.location.href = `/space?id=${space.id}`;
              }}
            >
              <CardHeader className="pb-3">
                <div className="flex items-center justify-between">
                  <div className="flex items-center gap-2">
                    <div
                      className="h-3 w-3 rounded-full"
                      style={{ backgroundColor: space.color }}
                    />
                    <h3 className="font-semibold text-foreground transition-colors group-hover:text-primary">
                      {space.name}
                    </h3>
                  </div>
                  <SpaceActions
                    onDeleted={() => spacesQuery.refetch()}
                    onUpdated={() => spacesQuery.refetch()}
                    space={space}
                    userId={session?.$id}
                  />
                </div>
              </CardHeader>

              <CardContent className="pb-3">
                <div className="relative inset-0 min-h-52 min-w-64 overflow-hidden rounded-lg border border-border/30 bg-muted/30">
                  <Tldraw
                    cameraOptions={{
                      constraints: {
                        initialZoom: "fit-min",
                        bounds: {
                          x: 0,
                          y: 0,
                          w: 9000,
                          h: 9000,
                        },
                        baseZoom: "fit-min",
                        behavior: "fixed",
                        origin: {
                          x: 0,
                          y: 0,
                        },
                        padding: {
                          x: 0.1,
                          y: 0.1,
                        },
                      },
                    }}
                    className="min-h-52"
                    hideUi
                    onMount={(editor) => {
                      editor.updateInstanceState({ isReadonly: true });
                    }}
                    snapshot={snapshot}
                  />
                </div>
              </CardContent>

              <CardFooter className="flex items-center justify-between pt-0 text-muted-foreground text-sm">
                <span>
                  {
                    Object.keys(snapshot?.store || {}).filter((key) =>
                      key.startsWith("shape:")
                    ).length
                  }{" "}
                  items
                </span>
                <span>Edited {space.lastEdited}</span>
              </CardFooter>
            </Card>
          );
        })}
      </div>

      {/* Empty State (when no spaces exist) */}
      {!spacesQuery.isLoading && filteredSpaces.length === 0 && (
        <div className="flex flex-col items-center justify-center py-16 text-center">
          <div className="mb-4 flex h-16 w-16 items-center justify-center rounded-full bg-muted">
            <Plus className="h-8 w-8 text-muted-foreground" />
          </div>
          <h3 className="mb-2 font-semibold text-foreground text-lg">
            No spaces yet
          </h3>
          <p className="mb-4 max-w-sm text-muted-foreground">
            Create your first space to start organizing your thoughts and ideas
            visually.
          </p>
          <NewSpaceDialog
            onSpaceCreated={() => spacesQuery.refetch()}
            triggerButton={
              <Button>
                <Plus className="mr-2 h-4 w-4" />
                Create Your First Space
              </Button>
            }
          />
        </div>
      )}
    </div>
  );
}

// Inlined action/dialog components moved to dedicated files for reuse