Add 4 different style options

api/[owner]/[repo].js

@@ -5,6 +5,16 @@ const fs = require("fs");
 module.exports = (req, res) => {
 	const { owner, repo } = req.query;
 
+	let style;
+	if (!req.query.style) style = "flat";
+	else if (
+		req.query.style === "flat" ||
+		req.query.style === "flat-square" ||
+		req.query.style === "for-the-badge" ||
+		req.query.style === "plastic"
+	)
+		style = req.query.style;
+
 	axios
 		.get(`https://api.github.com/repos/${owner}/${repo}/deployments`, {
 			headers: {
@@ -17,7 +27,9 @@ module.exports = (req, res) => {
 			if (response.data.length <= 0) {
 				res.setHeader("Content-Type", "image/svg+xml");
 				return fs
-					.createReadStream(path.join(__dirname, "../../assets/none.svg"))
+					.createReadStream(
+						path.join(__dirname, `../../assets/${style}/none.svg`)
+					)
 					.pipe(res);
 			}
 
@@ -31,7 +43,9 @@ module.exports = (req, res) => {
 			if (vercelDeployments.length <= 0) {
 				res.setHeader("Content-Type", "image/svg+xml");
 				return fs
-					.createReadStream(path.join(__dirname, "../../assets/none.svg"))
+					.createReadStream(
+						path.join(__dirname, `../../assets/${style}/none.svg`)
+					)
 					.pipe(res);
 			}
 
@@ -50,25 +64,29 @@ module.exports = (req, res) => {
 					if (response.data[0].state === "success")
 						return fs
 							.createReadStream(
-								path.join(__dirname, "../../assets/passing.svg")
+								path.join(__dirname, `../../assets/${style}/passing.svg`)
 							)
 							.pipe(res);
 					else if (response.data[0].state === "failure")
 						return fs
-							.createReadStream(path.join(__dirname, "../../assets/failed.svg"))
+							.createReadStream(
+								path.join(__dirname, `../../assets/${style}/failed.svg`)
+							)
 							.pipe(res);
 					else if (response.data[0].state === "pending")
 						return fs
 							.createReadStream(
-								path.join(__dirname, "../../assets/pending.svg")
+								path.join(__dirname, `../../assets/${style}/pending.svg`)
 							)
 							.pipe(res);
 				});
 		})
 		.catch((error) => {
 			res.setHeader("Content-Type", "image/svg+xml");
 			return fs
-				.createReadStream(path.join(__dirname, "../../assets/error.svg"))
+				.createReadStream(
+					path.join(__dirname, `../../assets/${style}/error.svg`)
+				)
 				.pipe(res);
 		});
 };