add optional screenshot, and delete car feature

.gitignore

@@ -10,3 +10,4 @@ config.json
 package-lock.json
 ecosystem.config.js
 static/*
+screenshot/*

config.example.json

@@ -12,6 +12,7 @@
         "scratchEnabled": 1, 
         "scratchType": 1, 
         "giveMeterReward": 0,
-		"newCardsBanned": 0
+		"newCardsBanned": 0,
+		"enableScreenshot": 0
     }
 }

src/config.ts

@@ -53,6 +53,10 @@ export interface GameOptions {
     // set this option to 1 will not create a new card
     // and prevent new card registration
     newCardsBanned: number; // 1 is on, 0 is off
+
+    // revision check
+    // set this option to 1 to enable screenshot feature
+    enableScreenshot: number; // 1 is on, 0 is off
 }
 
 export class Config {

src/modules/cars.ts

@@ -1,4 +1,5 @@
 import { Application } from "express";
+import { Config } from "../config";
 import { Module } from "module";
 import { prisma } from "..";
 
@@ -28,6 +29,19 @@ export default class CarModule extends Module {
 			// Get Challenger Data
 			let opponentsTarget = await carFunctions.getOpponentsTarget(body.carId, registeredTarget.registeredargetAvailable);
 
+			// Set Screenshot Count
+			let screenshotCount = 0;
+
+			// Check if screenshot feature enabled or not
+			let enableScreenshot = Config.getConfig().gameOptions.enableScreenshot || 0;
+
+			// Screenshot feature enabled
+			if(enableScreenshot === 1)
+			{
+				// Set the screnshot chance count
+				screenshotCount = 99;
+			}
+
             // Response data
 			let msg = {
 				error: wm.wm.protobuf.ErrorCode.ERR_SUCCESS,
@@ -43,7 +57,7 @@ export default class CarModule extends Module {
 				rgPreviousVersionPlayCount: 0,
 				stCompleted_100Episodes: car!.stCompleted100Episodes,
 				auraMotifAutoChange: false,
-				screenshotCount: 0,
+				screenshotCount: screenshotCount,
 				transferred: false,
 
 				// Stamp or Challenger
@@ -136,6 +150,9 @@ export default class CarModule extends Module {
 			// Get the request body for the update car request
 			let body = wm.wm.protobuf.UpdateCarRequest.decode(req.body);
 
+			// Not deleting car
+			if(body.toBeDeleted === false || body.toBeDeleted === undefined || body.toBeDeleted === null)
+			{
 				// Update the car
 				await carFunctions.updateCar(body);
 
@@ -163,6 +180,22 @@ export default class CarModule extends Module {
 						});
 					}
 				}
+			}
+			// Deleting car
+			else
+			{
+				console.log('Deleting Car ID : ' + body.carId);
+
+				// Mark Deleted Car
+				await prisma.carState.update({
+					where:{
+						dbId: body.carId
+					},
+					data:{
+						toBeDeleted: body.toBeDeleted
+					}
+				})
+			}
 
 			// Response data
             let msg = {

src/modules/game.ts

@@ -230,11 +230,11 @@ export default class GameModule extends Module {
 		// Save Screenshot
 		app.post('/method/save_screenshot', async (req, res) => {
 
-			// Get the information from the request
+			// Get the request body
 			let body = wm.wm.protobuf.SaveScreenshotRequest.decode(req.body);
 			
-			// TODO: Actual stuff here
+			// Perform the save screenshot request for the car
-            // This is literally just bare-bones so the shit boots
+			await gameFunction.saveScreenshot(body);
 
 			// Response data
             let msg = {
@@ -246,6 +246,6 @@ export default class GameModule extends Module {
 			
 			// Send the response to the client
             common.sendResponse(message, res);
-		});
+        })
     }
 }

src/modules/game/functions.ts

@@ -1,4 +1,7 @@
 import { prisma } from "../..";
+import { Config } from "../../config";
+import path from "path";
+import fs from "fs";
 
 // Import Proto
 import { wm } from "../../wmmt/wm.proto";
@@ -408,3 +411,40 @@ export async function getGhostBattleRecord(body: wm.protobuf.LoadGameHistoryRequ
 
     return { ghostBattle_records }
 }
+
+
+// Save Screenshot
+export async function saveScreenshot(body: wm.protobuf.SaveScreenshotRequest)
+{
+    // Check if screenshot feature enabled or not
+    let enableScreenshot = Config.getConfig().gameOptions.enableScreenshot || 0;
+
+    // Screenshot feature enabled
+    if(enableScreenshot === 1)
+    {
+        let filename: string | undefined = undefined;
+
+        filename = `${body.timestamp}_${body.carId}_${body.imageType}.png`;
+
+        let dir = path.join('screenshot');
+
+        if (!fs.existsSync(dir)){
+            fs.mkdirSync(dir);
+        }
+
+        // Combine the filename with the save location
+        let fullname = path.join('screenshot', filename);
+
+        // Attempt to write to the file
+        fs.writeFile(fullname, body.image, (err) => {
+            if (err)
+            {
+                console.log(err);
+            }   
+            else 
+            {
+                console.log(`Screenshot saved successfully as '${filename}'`)
+            }
+        });
+    }
+}

src/modules/users.ts

@@ -39,6 +39,9 @@ export default class UserModule extends Module {
 							state: true,
 							gtWing: true,
 							lastPlayedPlace: true
+						},
+						where:{
+							state: { toBeDeleted: false } // except deleted car
 						}
 					}
 				}