Traitement commandes et ajout J/arrow + Fix login

utils/modifyReward.js

@@ -0,0 +1,72 @@
+const mysql = require('mysql2/promise')
+require('dotenv').config()
+
+module.exports = modifyReward = (async (viewer_id, viewer_name, action, quantity, type) => {
+	// Create a connection to the MySQL database
+	const connection = await mysql.createConnection({
+		host: process.env.MYSQL_HOST,
+		port: process.env.MYSQL_PORT,
+		user: process.env.MYSQL_USER,
+		password: process.env.MYSQL_PASSWORD,
+		database: process.env.MYSQL_DATABASE
+	})
+
+	const result = {}
+	let sql_data = {}
+
+	// Check if the viewer already exists in the rewards table
+	await connection.query('SELECT * FROM rewards WHERE viewer_id = ?', [viewer_id])
+		.then(async ([rows, fields]) => {
+			console.log(rows)
+
+			if (rows.length === 0) {
+				// Viewer doesn't exist, insert a new row
+				sql_data = { viewer_id, viewer_name }
+
+				if (action === 'add' || action === 'set') { // Add or set the viewer with the quantity
+					if (type === 'permanent') sql_data.count = quantity
+					else if (type === 'current') sql_data.current_count = quantity
+					else { sql_data.count = quantity; sql_data.current_count = quantity }
+
+					await connection.query('INSERT INTO rewards SET ?', sql_data) .catch(error => { console.error(error) })
+					return result.status = 'insert_entry'
+				}
+				else if (action === 'get') return result.status = 'no_get_entry' // Can't get a viewer that doesn't exist
+				else if (action === 'remove') return result.status = 'no_delete_entry' // Can't remove a viewer that doesn't exist
+			} else {
+				// Viewer exists, update the count
+				result.count = rows[0].count
+				result.current_count = rows[0].current_count
+
+				if (action === 'get') return result.status = 'get_entry'
+
+				else if (action === 'add') { // Add the quantity
+					if (type === 'permanent') sql_data.count = rows[0].count + quantity
+					else if (type === 'current') sql_data.current_count = rows[0].current_count + quantity
+					else { sql_data.count = rows[0].count + quantity; sql_data.current_count = rows[0].current_count + quantity }
+				}
+				else if (action === 'remove') { // Remove the quantity
+					if (type === 'permanent') sql_data.count = rows[0].count - quantity
+					else if (type === 'current') sql_data.current_count = rows[0].current_count - quantity
+					else { sql_data.count = rows[0].count - quantity; sql_data.current_count = rows[0].current_count - quantity }
+
+					if (sql_data.count < 0) return result.status = 'not_enough_count' // Can't delete more than the count
+					if (sql_data.current_count < 0) return result.status = 'not_enough_current_count' // Can't delete more than the current_count
+				}
+				else if (action === 'set') { // Set the quantity
+					if (type === 'permanent') sql_data.count = quantity
+					else if (type === 'current') sql_data.current_count = quantity
+					else { sql_data.count = quantity; sql_data.current_count = quantity }
+				}
+
+				await connection.query('UPDATE rewards SET ? WHERE viewer_id = ?', [sql_data, viewer_id]).catch(error => { console.error(error) })
+				return result.status = 'update_entry'
+			}
+		}).catch(error => { console.error(error) })
+
+	// Terminate the connection to the database
+	await connection.end()
+
+	result.sql_data = sql_data
+	return result
+})