gander/frontend/src/pages/FollowingPage.tsx

import React, { useEffect, useState } from "react";
import { useAuth } from "../context/AuthContext";
import { useLocation, useNavigate } from "react-router-dom";
import DynamicPageContent from "../components/Layout/DynamicPageContent";
import { useFollow } from "../hooks/useFollow";
import FollowUserButton from "../components/Input/FollowUserButton";
import FollowButton from "../components/Input/FollowButton";
import Footer from "../components/Layout/Footer";

interface Streamer {
  user_id: number;
  username: string;
}

interface Category {
  isFollowing: boolean;
  category_id: number;
  category_name: string;
}

interface FollowingProps {
  extraClasses?: string;
}

const Following: React.FC<FollowingProps> = ({ extraClasses = "" }) => {
  const navigate = useNavigate();
  const { isLoggedIn } = useAuth();
  const [followedStreamers, setFollowedStreamers] = useState<Streamer[]>([]);
  const [followedCategories, setFollowedCategories] = useState<Category[]>([]);
  const [followingStatus, setFollowingStatus] = useState<{ [key: number]: boolean }>({});

  const { checkFollowStatus, followUser, unfollowUser } = useFollow();

  const location = useLocation();
  const queryParams = new URLSearchParams(location.search);
  const initialTab = queryParams.get("tab") === "streamers" ? "categories" : "categories";

  const [activeTab, setActiveTab] = useState<"categories" | "streamers">(initialTab);
  //const [followingStatus, setFollowingStatus] = useState<Record<number, boolean>>({});
  const profileImageUrl = "/user/<streamer:username/index.png"

  useEffect(() => {
    const newTab = queryParams.get("tab") as "categories" | "streamers";
    if (newTab) {
      setActiveTab(newTab);
    }
  }, [location.search]);

  useEffect(() => {
    const fetchFollowedContent = async () => {
      try {
        const response = await fetch("/api/user/following");
        if (!response.ok) throw new Error("Failed to fetch followed content");

        const data = await response.json();
        if (!Array.isArray(data.streams) || !Array.isArray(data.categories)) {
          throw new Error("API response structure is incorrect");
        }

        setFollowedStreamers(data.streams);
        setFollowedCategories(data.categories);

        // Fetch follow status for streamers
        const updatedStatus: { [key: number]: boolean } = {};
        for (const streamer of data.streams) {
          const status = await checkFollowStatus(streamer.username);
          updatedStatus[streamer.user_id] = Boolean(status);
        }
        setFollowingStatus(updatedStatus);
      } catch (error) {
        console.error("Error fetching followed content:", error);
        setFollowedStreamers([]);
        setFollowedCategories([]);
      }
    };

    if (isLoggedIn) {
      fetchFollowedContent();
    }
  }, [isLoggedIn]);



  return (
    <DynamicPageContent className="overflow-x-hidden h-full">
      <div className="flex items-center justify-center mb-20">
        <div
          id="follow_page"
          className={`w-[96vw] h-full bg-slate-50/35 rounded-lg overflow-x-hidden flex flex-col text-center scrollbar-hide transition-all duration-500 ease-in-out ${extraClasses}`}
        >
          {activeTab === "streamers" && (
            <div className="p-4 w-full">
              {followedStreamers.length > 0 ? (
                <div id="followed-users" className="grid grid-cols-2 gap-4">
                  {followedStreamers.map((streamer: any) => (
                    <div
                      key={`streamer-${streamer.username}`}
                      className="h-full cursor-pointer bg-black w-full py-2 rounded-lg text-white hover:text-purple-500 font-bold transition-colors flex items-center space-x-3 p-4"
                    >
                      <img
                        src={`/user/${streamer.username}/profile_picture`}
                        alt={`${streamer.username}'s profile`}
                        className="w-10 h-10 rounded-full object-cover"
                        onError={(e) => {
                          e.currentTarget.onerror = null;
                          e.currentTarget.src = "/default-profile.png";
                        }}
                      />
                      <p className="text-lg">{streamer.username}</p>
                      <FollowUserButton
                        user={{
                          user_id: streamer.user_id,
                          username: streamer.username,
                          isFollowing: followingStatus[streamer.user_id] || true,
                        }}
                      />
                    </div>
                  ))}
                </div>
              ) : (
                <p className="text-center text-white">No followed streamers.</p>
              )}
            </div>
          )}


          {activeTab === "categories" && (
            <div className="p-4 w-full">
              {followedCategories.length > 0 ? (
                <div id="categories-followed" className="grid grid-cols-3 gap-4">
                  {followedCategories.map((category) => (
                    <div
                      key={category.category_id}
                      className="relative flex flex-col items-center justify-center rounded-lg overflow-hidden hover:shadow-lg transition-all"
                      onClick={() => navigate(`/category/${category.category_name}`)}
                    >
                      <FollowButton category={category} />
                      <img
                        src={`/images/category_thumbnails/${category.category_name.toLowerCase().replace(/ /g, "_")}.webp`}
                        alt={category.category_name}
                        className="w-full h-[200px] object-cover"
                      />
                      <div className="absolute bottom-2 bg-black bg-opacity-60 w-full text-center text-white py-1">
                        {category.category_name}
                      </div>
                    </div>
                  ))}
                </div>
              ) : (
                <p className="text-center text-white">No followed categories.</p>
              )}
            </div>
          )}
        </div>
      </div>
    </DynamicPageContent >
  );
};

export default Following;