VogelSocialMedia/code/frontend/src/pages/Profile.tsx

import "./profile.css";
import QuiltedImageList from "../components/profile/QuiltedImageList";
import { StyledEngineProvider, Divider } from "@mui/material";
import ChangeAvatarDialog from "../components/profile/ChangeAvatarDialog";
import Bio from "../components/profile/Bio";
import RotkehlchenButton from "../components/ButtonRotkehlchen";
import api, { redirectToLogin } from "../api/axios";
import { useAuth } from "../api/Auth";
import { useNavigate, useParams } from "react-router-dom";
import { useEffect, useState } from "react";
import { UserProfile } from "../types/UserProfile";

function Profile() {
  const { user } = useAuth();
  const { username } = useParams();
  const navigate = useNavigate();

  const [userData, setUserData] = useState<UserProfile | null>({
    id: "",
    username: undefined,
    bio: "",
    profilePictureUrl: null,
    followers: 0,
    following: 0,
    posts: 0,
    isFollowing: false,
  });

  const userProfile = async () => {
    try {
      const response = await api.get(`/profile/${username}`);
      setUserData(response.data.data);
      return;
    } catch (error) {
      navigate("/"); /* replace to 404 page */
      console.error("Error fetching user profile:", error);
      return null;
    }
  };

  const ownAccount = username === user?.username;
  useEffect(() => {
    setUserData(null);
    userProfile();
  }, [username]);

  const setBio = (bio: string) => {
    setUserData((prevData) => {
      if (prevData) {
        return { ...prevData, bio: bio };
      }
      return prevData;
    });
  };
  const handleFollowUser = async () => {
    try {
      if (userData?.isFollowing === false) {
        await api.post(`/follower/follow/${username}`);
      } else if (userData?.isFollowing === true) {
        await api.delete(`/follower/unfollow/${username}`);
      } else {
        redirectToLogin();
      }
      userProfile(); // Refresh user data after unfollowing
    } catch (error) {
      console.error("Error following/unfollowing user:", error);
    }
  };

  return (
    <StyledEngineProvider injectFirst>
      <div className="profile-display">
        <div className="user-info blue-background">
          <ChangeAvatarDialog
            ownAccount={ownAccount}
            username={userData?.username || ""}
            setUserData={setUserData}
            imageUrl={userData?.profilePictureUrl}
          />
          <Bio
            ownAccount={ownAccount}
            bioText={userData?.bio}
            setBio={setBio}
          />
          <Divider variant="middle" className="divider" />
          {/* TODO: Change data to data from Database */}
          <div className="numeral-data body-bold">
            <div className="data">
              <span aria-label="current-post-number">{userData?.posts}</span>
              <span className="data-label title-h1">Posts</span>
            </div>
            <div className="data">
              <span aria-label="current-follower-number">
                {userData?.followers}
              </span>
              <span className="data-label title-h1">Followers</span>
            </div>
            <div className="data">
              <span aria-label="current-following-number">
                {userData?.following}
              </span>
              <span className="data-label title-h1">Following</span>
            </div>
          </div>
          {!ownAccount && (
            <RotkehlchenButton
              style={!userData?.isFollowing ? "primary" : "secondary"}
              label={!userData?.isFollowing ? "Follow" : "Unfollow"}
              type="button"
              onClick={handleFollowUser}
            />
          )}
        </div>
        {userData && <QuiltedImageList user={userData} />}
      </div>
    </StyledEngineProvider>
  );
}

export default Profile;