Finish reverting back to favorite saga (#11931)

Author: Isaac Solo

packages/web/src/components/track/desktop/ConnectedTrackTile.tsx

@@ -1,11 +1,6 @@
 import { memo, useCallback, useEffect, MouseEvent, useRef } from 'react'
 
-import {
-  useCurrentUserId,
-  useToggleFavoriteTrack,
-  useTrack,
-  useUser
-} from '@audius/common/api'
+import { useCurrentUserId, useTrack, useUser } from '@audius/common/api'
 import { useGatedContentAccess } from '@audius/common/hooks'
 import {
   ShareSource,
@@ -43,7 +38,8 @@ import TrackTile from './TrackTile'
 
 const { getUid, getPlaying, getBuffering } = playerSelectors
 const { requestOpen: requestOpenShareModal } = shareModalUIActions
-const { repostTrack, undoRepostTrack } = tracksSocialActions
+const { repostTrack, undoRepostTrack, saveTrack, unsaveTrack } =
+  tracksSocialActions
 const { setLockedContentId } = gatedContentActions
 
 type OwnProps = {
@@ -115,6 +111,19 @@ const ConnectedTrackTile = ({
     },
     [dispatch]
   )
+  const handleSaveTrack = useCallback(
+    (trackId: ID, isFeed: boolean) => {
+      dispatch(saveTrack(trackId, FavoriteSource.TILE, isFeed))
+    },
+    [dispatch]
+  )
+
+  const handleUndoSaveTrack = useCallback(
+    (trackId: ID) => {
+      dispatch(unsaveTrack(trackId, FavoriteSource.TILE))
+    },
+    [dispatch]
+  )
 
   const handleRepostTrack = useCallback(
     (trackId: ID, isFeed: boolean) => {
@@ -161,11 +170,6 @@ const ConnectedTrackTile = ({
   const [, setLockedContentVisibility] = useModalState('LockedContent')
   const menuRef = useRef<HTMLDivElement>(null)
 
-  const toggleSaveTrack = useToggleFavoriteTrack({
-    trackId,
-    source: FavoriteSource.TILE
-  })
-
   useEffect(() => {
     if (!loading && hasLoaded) {
       hasLoaded?.(index)
@@ -243,8 +247,12 @@ const ConnectedTrackTile = ({
   )
 
   const onClickFavorite = useCallback(() => {
-    toggleSaveTrack()
-  }, [toggleSaveTrack])
+    if (isFavorited) {
+      handleUndoSaveTrack(trackId)
+    } else {
+      handleSaveTrack(trackId, isFeed)
+    }
+  }, [isFavorited, handleUndoSaveTrack, trackId, handleSaveTrack, isFeed])
 
   const onClickRepost = useCallback(() => {
     if (isReposted) {