import { useCallback, useEffect, useState } from "react";

export const useScreenOrientation = () => {
  const getOrientation = () => {
    if (typeof window !== "undefined") {
      return window.screen.orientation.type;
    }

    return undefined;
  };

  const [orientation, setOrientation] = useState(getOrientation());

  const updateOrientation = useCallback(() => {
    setOrientation(getOrientation());
  }, []);

  useEffect(() => {
    window.addEventListener("orientationchange", updateOrientation);

    return () => {
      window.removeEventListener("orientationchange", updateOrientation);
    };
  }, [updateOrientation]);

  return orientation;
};