Module #2 (Nowshin Owishi)

pages/index.js

@@ -1,10 +1,10 @@
 import dynamic from "next/dynamic";
-import { useEffect, useState, useRef } from "react";
+import { useEffect, useState, useRef, useCallback } from "react";
 import styles from "../styles/Snake.module.css";
 
 const Config = {
-  height: 25,
-  width: 25,
+  height: 15,
+  width: 15,
   cellSize: 32,
 };
 
@@ -39,6 +39,14 @@ const Cell = ({ x, y, type }) => {
           height: 32,
         };
 
+      case CellType.Poison:
+        return {
+          backgroundColor: "red",
+          borderRadius: 20,
+          width: 32,
+          height: 32,
+        };
+
       default:
         return {};
     }
@@ -61,9 +69,26 @@ const Cell = ({ x, y, type }) => {
 const getRandomCell = () => ({
   x: Math.floor(Math.random() * Config.width),
   y: Math.floor(Math.random() * Config.width),
+  start: Date.now(),
 });
 
-const Snake = () => {
+//custom hook
+//controller
+const UseInterval = (func, dir) => {
+  const timer = useRef(Date.now());
+  const createCallback = useCallback(() => {
+    if (Date.now() - timer.current > dir) {
+      timer.current = Date.now();
+      func();
+    }
+  }, [dir, func]);
+
+  useEffect(() => {
+    const interval = setInterval(createCallback, 1000 / 60);
+    return () => clearInterval(interval);
+  }, [createCallback]);
+};
+const UseSnake = () => {
   const getDefaultSnake = () => [
     { x: 8, y: 12 },
     { x: 7, y: 12 },
@@ -74,68 +99,147 @@ const Snake = () => {
   // snake[0] is head and snake[snake.length - 1] is tail
   const [snake, setSnake] = useState(getDefaultSnake());
   const [direction, setDirection] = useState(Direction.Right);
+  const [foods, setFoods] = useState([]);
+  const [poisons, setPoison] = useState([]);
+  const score = snake.length - 3;
 
-  const [food, setFood] = useState({ x: 4, y: 10 });
-  const [score, setScore] = useState(0);
+  // ?. is called optional chaining
+  const isFood = useCallback(
+    ({ x, y }) =>
+      foods.find((position) => position.x === x && position.y === y),
+    [foods]
+  );
 
-  // move the snake
-  useEffect(() => {
-    const runSingleStep = () => {
-      setSnake((snake) => {
-        const head = snake[0];
-        const newHead = { x: head.x + direction.x, y: head.y + direction.y };
+  const isPoison = useCallback(
+    ({ x, y }) =>
+      poisons.find((position) => position.x === x && position.y === y),
+    [poisons]
+  );
+
+  const isSnake = useCallback(
+    ({ x, y }) =>
+      snake.find((position) => position.x === x && position.y === y),
+    [snake]
+  );
+
+  //restart the game
+  const resetGame = useCallback(() => {
+    setSnake(getDefaultSnake());
+    setDirection(Direction.Right);
+    setFoods([]);
+    setPoison([]);
+  }, []);
 
-        // make a new snake by extending head
-        // https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Operators/Spread_syntax
-        const newSnake = [newHead, ...snake];
+  //moving the snake
+  const runSingleStep = useCallback(() => {
+    setSnake((snake) => {
+      const head = snake[0];
+      const newHead = {
+        x: (head.x + direction.x + Config.width) % Config.width,
+        y: (head.y + direction.y + Config.height) % Config.height,
+      };
 
-        // remove tail
+      // make a new snake by extending head
+      const newSnake = [newHead, ...snake];
+
+      // remove tail when head doesnt eat food
+      if (!isFood(newHead)) {
         newSnake.pop();
+      }
+      //remove again when eats poison
+      if (isPoison(newHead)) {
+        newSnake.pop();
+      }
+      if (isSnake(newHead) || score < 0) {
+        resetGame();
+      }
 
-        return newSnake;
-      });
-    };
+      return newSnake;
+    });
+  }, [direction.x, direction.y, isFood, isPoison, isSnake, resetGame, score]);
+
+  //add new food
+  const addFood = () => {
+    let newFood = getRandomCell();
+    while (isSnake(newFood) || isFood(newFood) || isPoison(newFood)) {
+      newFood = getRandomCell();
+    }
+    setFoods((currentFoods) => [...currentFoods, newFood]);
+  };
 
-    runSingleStep();
-    const timer = setInterval(runSingleStep, 500);
+  //add poison
+  const addPoison = () => {
+    let newPoison = getRandomCell();
+    while (isSnake(newPoison) || isFood(newPoison) || isPoison(newPoison)) {
+      newPoison = getRandomCell();
+    }
+    setPoison((currentPoison) => [...currentPoison, newPoison]);
+  };
 
-    return () => clearInterval(timer);
-  }, [direction, food]);
+  //remove food
+  const removeFood = useCallback(() => {
+    // console.log("delete food auto");
+    setFoods((currentFoods) =>
+      currentFoods.filter((food) => (Date.now() - food.start) < 10000)
+    );
+  }, []);
+  //remove poison
+  const removePoison = useCallback(() => {
+    // console.log("delete poison auto");
+    setPoison((currentPoison) =>
+      currentPoison.filter((poison) => (Date.now() - poison.start) < 10000)
+    );
+  }, []);
 
-  // update score whenever head touches a food
+  // update foods and poisons whenever head touches a food or a poison
   useEffect(() => {
     const head = snake[0];
     if (isFood(head)) {
-      setScore((score) => {
-        return score + 1;
-      });
+      console.log("ate food");
+      setFoods((currentFoods) =>
+        currentFoods.filter((food) => food.x !== head.x && food.y !== head.y)
+      );
+    }
+    if (isPoison(head)) {
+      console.log("ate poison");
+      setPoison((currentPoison) =>
+        currentPoison.filter(
+          (poison) => poison.x !== head.x && poison.y !== head.y
+        )
+      );
+    }
+  }, [isFood, isPoison, snake]);
 
-      let newFood = getRandomCell();
-      while (isSnake(newFood)) {
-        newFood = getRandomCell();
-      }
+  UseInterval(addFood, 3000);
+  UseInterval(runSingleStep, 300);
+  UseInterval(removeFood, 50);
+  UseInterval(addPoison, 4000);
+  UseInterval(removePoison, 100);
 
-      setFood(newFood);
-    }
-  }, [snake]);
+  const changeDir = (checkDir, newDir) => {
+    setDirection((direction) => {
+      if (direction != checkDir) return newDir;
+      return direction;
+    });
+  };
 
   useEffect(() => {
     const handleNavigation = (event) => {
       switch (event.key) {
         case "ArrowUp":
-          setDirection(Direction.Top);
+          changeDir(Direction.Bottom, Direction.Top);
           break;
 
         case "ArrowDown":
-          setDirection(Direction.Bottom);
+          changeDir(Direction.Top, Direction.Bottom);
           break;
 
         case "ArrowLeft":
-          setDirection(Direction.Left);
+          changeDir(Direction.Right, Direction.Left);
           break;
 
         case "ArrowRight":
-          setDirection(Direction.Right);
+          changeDir(Direction.Left, Direction.Right);
           break;
       }
     };
@@ -144,13 +248,12 @@ const Snake = () => {
     return () => window.removeEventListener("keydown", handleNavigation);
   }, []);
 
-  // ?. is called optional chaining
-  // see: https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Operators/Optional_chaining
-  const isFood = ({ x, y }) => food?.x === x && food?.y === y;
-
-  const isSnake = ({ x, y }) =>
-    snake.find((position) => position.x === x && position.y === y);
+  return { score, isFood, isSnake, isPoison };
+};
 
+//view
+const Snake = () => {
+  const { score, isFood, isSnake, isPoison } = UseSnake();
   const cells = [];
   for (let x = 0; x < Config.width; x++) {
     for (let y = 0; y < Config.height; y++) {
@@ -159,6 +262,8 @@ const Snake = () => {
         type = CellType.Food;
       } else if (isSnake({ x, y })) {
         type = CellType.Snake;
+      } else if (isPoison({ x, y })) {
+        type = CellType.Poison;
       }
       cells.push(<Cell key={`${x}-${y}`} x={x} y={y} type={type} />);
     }