Add/Remove from Favorite list

addtofavorites.php

@@ -1,12 +1,12 @@
 <?php 
 	require_once("files/main.php");
-	$vidId=$_POST['favoriteButton'];
+	$vidId=(int)$_GET['Id'];
 	$checkquery = $con -> prepare("SELECT * from favorites where userName='$loggedInUserName' and videoId = '$vidId'");
 	$checkquery -> execute();
 	if($checkquery->rowCount()==0){
     	$query = $con -> prepare("INSERT into favorites (videoId, userName) VALUES ('$vidId', '$loggedInUserName')");
         $query->execute();
-        echo "Video has been added to favorites";
+        header("location:watch.php?Id=$vidId&add=success");
     }
     else{
     	echo"Video is already in favorites";

favorites.php

@@ -4,7 +4,7 @@
     <?php
     if($loggedInUserName!=""){
         $mediaGrid= new MediaGrid($con);
-        echo $mediaGrid->create(null, "My Favorites", $loggedInUserName);
+       echo $mediaGrid->create('Favorite', "", "","", $loggedInUserName);
     }
     ?>
 </div>

files/Classes/MediaGrid.php

@@ -71,6 +71,12 @@ public function createMySharedContentBlock($type, $sortBy, $loggedInUserName, $h
         return $this->createBlock("Shared ".$type."s", $gridItems);
     }
 
+    public function createMyFavoriteContentBlock($type, $sortBy, $loggedInUserName, $hide = 'not hidden')
+    {
+        $gridItems = $this->getFavorite($type, $sortBy, $loggedInUserName);
+        return $this->createBlock("My Favorite ".$type."s", $gridItems);
+    }
+
     public function create($page, $category, $keywords, $sortBy, $loggedInUserName, $videoId = 1, $hide = 'not hidden')
     {
         if ($page == 'Home') {
@@ -97,6 +103,11 @@ public function create($page, $category, $keywords, $sortBy, $loggedInUserName,
         if ($page == 'MyList') {
 
         }
+
+        if ($page == 'Favorite') {
+            return $this->createMyFavoriteContentBlock('video', $sortBy, $loggedInUserName)
+                  .$this->createMyFavoriteContentBlock('image', $sortBy, $loggedInUserName);
+        }
     }
 
     public function getSuggestions($videoId)

removeFromFavorite.php

@@ -0,0 +1,13 @@
+<?php
+    require_once("files/main.php");
+    $vidId = (int)$_GET['Id'];
+    $checkquery = $con->prepare("SELECT * from favorites where userName='$loggedInUserName' and videoId = '$vidId'");
+    $checkquery->execute();
+    if ($checkquery->rowCount() > 0) {
+        $query = $con->prepare("DELETE FROM favorites WHERE userName='$loggedInUserName' and videoId = '$vidId'");
+        $query->execute();
+        header("location:watch.php?Id=$vidId&delete=success");
+    } else {
+        echo "Video is already not in favorites";
+    }
+?>

watch.php

@@ -62,10 +62,32 @@
     echo "<button type='submit' value='$vidId' name='playlistButton'>Add to Playlist</button>
     </form>";
 
-    echo "<form action='addtofavorites.php' method='POST' >
-    <button type='submit' value='$vidId' name='favoriteButton'>Favorite</button>
-    </form>";
+    $checkquery = $con -> prepare("SELECT * from favorites where userName='$loggedInUserName' and videoId = '$vidId'");
+    $checkquery -> execute();
+    if($checkquery->rowCount()==0){
+         echo "<form action='addtofavorites.php' method='GET' >
+                <button type='submit' value='$vidId' name='Id'>Add to Favorite</button>
+              </form>";
+
+    }
+    else{
+        echo "<form action='removeFromFavorite.php' method='GET' >
+                <button type='submit' value='$vidId' name='Id'>Remove from Favorite</button>
+              </form>";
+
+    }
+
+    if (isset($_GET["add"]) && $_GET["add"] == 'success') {
+        echo "<div class='badge'>
+                 <p style = 'color:red'>Successfully added to Favorite!</p>
+              </div>";
+    }
 
+    if (isset($_GET["delete"]) && $_GET["delete"] == 'success') {
+        echo "<div class='badge'>
+                <p style = 'color:red'>Successfully removed From Favorite!</p>
+              </div>";
+    }
 ?> 
 </div>