From 81de74d7230ff794e6ab2de06d2a5d68ea3fc0ef Mon Sep 17 00:00:00 2001
From: Annwan <annwan@annwan.me>
Date: Thu, 15 Aug 2024 04:16:32 +0200
Subject: [PATCH] [YQTemplate] added class image detection

---
 yqform.etlua   | 32 ++++++++++++++++++++++++++------
 yqtemplate.lua | 31 +++++++++++++++++--------------
 2 files changed, 43 insertions(+), 20 deletions(-)

diff --git a/yqform.etlua b/yqform.etlua
index 4dd0cb0..5c0dda2 100644
--- a/yqform.etlua
+++ b/yqform.etlua
@@ -15,7 +15,27 @@
       <dt>Avatar</dt>
       <dd><img src="data:<%= args.avatar.t -%>;base64,<%= encoder(args.avatar.v) %>" /></dd>
       <dt>Is Leader</dt>
-      <dd><%= args.is_leader and args.is_leader.v %></dd>
+      <dd>
+	<% if args.is_leader and args.is_leader.v == "on" then %>
+	<img src="data:image/png;base64,<%= encoder(readfile('yqtemplate/basic_blank.png')) %>" />
+	<% else %>
+	<img src="data:image/png;base64,<%= encoder(readfile('yqtemplate/basic_blank.png')) %>" />
+	<% end %>c
+      </dd>
+    </dl>
+    <h3>Classes and Levels</h3>
+    <dl>
+      <dt>Class 1</dt>
+      <dd>
+	<img src="data:image/png;base64,<%= encoder(readfile('yqtemplate/'..classes[args.class1].f)) %>" />
+	Lvl <%= args.class1_lvl %>
+      </dd>
+      <dt>Class 2</dt>
+      <dd>
+	<img src="data:image/png;base64,<%= encoder(readfile('yqtemplate/'..classes[args.class2].f)) %>" />
+	Lvl <%= args.class2_lvl %>
+      </dd>
+      <dt>Class leaning</dt><dd><%= args.class_leaning %></dd>
     </dl>
     <% end %>
     <form method="POST" action="/cgi/yqtemplate" enctype="multipart/form-data">
@@ -29,20 +49,20 @@
       <h2>Classes and Levels</h2>
       <label for="class1">Class 1:</label>
       <select name="class1">
-	<% for _, v in ipairs(classes) do %>
-        <option value="<%= v %>"><%= v %></option>
+	<% for k, v in ipairs(classes) do %>
+        <option value="<%= k %>"><%= v.n %></option>
 	<% end %>
       </select>
       <label for="class1_lvl">Level:</label>
       <input type="number" name="class1_lvl" min="1"/><br/>
       <label for="class2">Class2:</label>
       <select name="class2">
-	<% for _, v in ipairs(classes) do %>
-        <option value="<%= v %>"><%= v %></option>
+	<% for k, v in ipairs(classes) do -%>
+        <option value="<%= k %>"><%= v.n %></option>
 	<% end %>
       </select>
       <label for="class2_lvl">Level:</label>
-      <input type="number"name="class2_lvl" min="1"/><br/>
+      <input type="number" name="class2_lvl" min="1"/><br/>
       <label for="class_leaning">
 	Class leaning (0 for completely class 1, 100 for completely class 2):
       </label>
diff --git a/yqtemplate.lua b/yqtemplate.lua
index 9ec5136..b792750 100644
--- a/yqtemplate.lua
+++ b/yqtemplate.lua
@@ -4,6 +4,20 @@ local utils = require "utils"
 local yqtemplateform = utils.readfile("yqform.etlua")
 
 local form = etlua.compile(yqtemplateform)
+--- @type [table]
+local classes = {
+	{ n = "Beast", f = "c01.beast.png" },
+	{ n = "Swordsman", f = "c02.swordsman.png" },
+	{ n = "Spirit Bird", f = "c03.spirit-bird.png" },
+	{ n = "Twin Blades", f = "c04.twin-blades.png" },
+	{ n = "Healer", f = "c05.healer.png" },
+	{ n = "Summoner", f = "c06.summoner.png" },
+	{ n = "Tamer", f = "c07.tamer.png" },
+	{ n = "Pugilist", f = "c08.pugilist.png" },
+	{ n = "Mounted Fighter", f = "c09.mounted-fighter.png" },
+	{ n = "Fruit Animal", f = "c10.fruit-animal.png" },
+	{ n = "Spellcaster", f = "c11.spellcaster.png" }
+}
 
 return {
 	desc = "Creates Yokoka’s quest-style personal status page",
@@ -33,21 +47,10 @@ return {
 		end
 		fcgi.print(
 			form {
-				encoder = require"base64".encode,
+				encoder = require "base64".encode,
+				readfile = utils.readfile,
 				args = args,
-				classes = {
-					"Beast",
-					"Swordsman",
-					"Spirit Bird",
-					"Twin Blades",
-					"Healer",
-					"Summoner",
-					"Tamer",
-					"Pugilist",
-					"Mounted Fighter",
-					"Fruit Animal",
-					"Spellcaster"
-				},
+				classes = classes
 				elements = {
 					"Fire", "Water", "Earth", "Air",
 					"Wood", "Electricity", "Metals", "Dreams",