Feat/prevent reinit (#60)

* feat: added check to prevent canvas re-init * feat: format code * feat: formatted file * feat: added extra check to func * nits and patch build --------- Co-authored-by: Brandon Roberts <[email protected]>
keep-starknet-strange · Apr 19, 2024 · 3ce4077 · 3ce4077
1 parent 584b13f
commit 3ce4077
Showing 1 changed file with 19 additions and 15 deletions.
diff --git a/backend/routes/canvas.go b/backend/routes/canvas.go
@@ -14,22 +14,26 @@ func InitCanvasRoutes() {
 }
 
 func initCanvas(w http.ResponseWriter, r *http.Request) {
-	// TODO: Check if canvas already exists
-	totalBitSize := core.ArtPeaceBackend.CanvasConfig.Canvas.Width * core.ArtPeaceBackend.CanvasConfig.Canvas.Height * core.ArtPeaceBackend.CanvasConfig.ColorsBitWidth
-	totalByteSize := (totalBitSize / 8)
-	if totalBitSize%8 != 0 {
-		// Round up to nearest byte
-		totalByteSize += 1
+	if core.ArtPeaceBackend.Databases.Redis.Exists(context.Background(), "canvas").Val() == 0 {
+		totalBitSize := core.ArtPeaceBackend.CanvasConfig.Canvas.Width * core.ArtPeaceBackend.CanvasConfig.Canvas.Height * core.ArtPeaceBackend.CanvasConfig.ColorsBitWidth
+		totalByteSize := (totalBitSize / 8)
+		if totalBitSize%8 != 0 {
+			// Round up to nearest byte
+			totalByteSize += 1
+		}
+
+		// Create canvas
+		canvas := make([]byte, totalByteSize)
+		ctx := context.Background()
+		err := core.ArtPeaceBackend.Databases.Redis.Set(ctx, "canvas", canvas, 0).Err()
+		if err != nil {
+			panic(err)
+		}
+
+		fmt.Println("Canvas initialized")
+	} else {
+		fmt.Println("Canvas already initialized")
 	}
-
-	canvas := make([]byte, totalByteSize)
-	ctx := context.Background()
-	err := core.ArtPeaceBackend.Databases.Redis.Set(ctx, "canvas", canvas, 0).Err()
-	if err != nil {
-		panic(err)
-	}
-
-	fmt.Println("Canvas initialized")
 }
 
 func getCanvas(w http.ResponseWriter, r *http.Request) {