From dfb47c4b322e34ef93e90c20fb58a30f498f36bd Mon Sep 17 00:00:00 2001
From: Baoshuo <i@baoshuo.ren>
Date: Mon, 5 Dec 2022 11:54:58 +0800
Subject: [PATCH] feat(web/problems): difficulty filter

---
 db/app_uoj233.sql                            |  1 +
 web/app/controllers/problem_set.php          | 78 +++++++++-----------
 web/app/upgrade/20_problem_difficulty/up.sql |  1 +
 3 files changed, 37 insertions(+), 43 deletions(-)

diff --git a/db/app_uoj233.sql b/db/app_uoj233.sql
index 4dd2aa8..9fb0e3a 100644
--- a/db/app_uoj233.sql
+++ b/db/app_uoj233.sql
@@ -626,6 +626,7 @@ CREATE TABLE `problems` (
   PRIMARY KEY (`id`),
   KEY `assigned_to_judger` (`assigned_to_judger`),
   KEY `uploader` (`uploader`),
+  KEY `difficulty` (`difficulty`),
   KEY `is_hidden` (`is_hidden`)
 ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci TABLESPACE `innodb_system`;
 /*!40101 SET character_set_client = @saved_cs_client */;
diff --git a/web/app/controllers/problem_set.php b/web/app/controllers/problem_set.php
index c7886e8..e557a03 100644
--- a/web/app/controllers/problem_set.php
+++ b/web/app/controllers/problem_set.php
@@ -128,9 +128,7 @@ function getProblemTR($info) {
 			)
 		);
 	}
-	if (isset($_COOKIE['show_difficulty'])) {
-		$html .= HTML::tag('td', [], UOJProblem::getDifficultyHTML($problem->info['difficulty']));
-	}
+	$html .= HTML::tag('td', [], UOJProblem::getDifficultyHTML($problem->info['difficulty']));
 	$html .= HTML::tag('td', [], ClickZans::getCntBlock($problem->info['zan']));
 	$html .= HTML::tag_end('tr');
 	return $html;
@@ -179,6 +177,10 @@ if (Auth::check() && isset($_GET['my'])) {
 	$cond['problems.uploader'] = Auth::id();
 }
 
+if (isset($_GET['difficulty']) && $_GET['difficulty']) {
+	$cond['problems.difficulty'] = $_GET['difficulty'];
+}
+
 if (empty($cond)) {
 	$cond = '1';
 }
@@ -189,9 +191,7 @@ $header .= '<th>' . UOJLocale::get('problems::problem') . '</th>';
 if (isset($_COOKIE['show_submit_mode'])) {
 	$header .= '<th class="text-center" style="width:125px;">' . UOJLocale::get('problems::ac ratio') . '</th>';
 }
-if (isset($_COOKIE['show_difficulty'])) {
-	$header .= '<th class="text-center" style="width:8em;">' . UOJLocale::get('problems::difficulty') . '</th>';
-}
+$header .= '<th class="text-center" style="width:8em;">' . UOJLocale::get('problems::difficulty') . '</th>';
 $header .= '<th class="text-center" style="width:50px;">' . UOJLocale::get('appraisal') . '</th>';
 $header .= '</tr>';
 
@@ -246,7 +246,6 @@ $pag = new Paginator([
 <?php echoUOJPageHeader(UOJLocale::get('problems')) ?>
 
 <div class="row">
-
 	<!-- left col -->
 	<div class="col-lg-9">
 
@@ -283,13 +282,6 @@ $pag = new Paginator([
 						<?= UOJLocale::get('problems::show statistics') ?>
 					</label>
 				</div>
-
-				<div class="form-check d-inline-block">
-					<input type="checkbox" id="input-show_difficulty" class="form-check-input" <?= isset($_COOKIE['show_difficulty']) ? 'checked="checked" ' : '' ?> />
-					<label class="form-check-label" for="input-show_difficulty">
-						<?= UOJLocale::get('problems::show difficulty') ?>
-					</label>
-				</div>
 			</div>
 		</div>
 
@@ -322,19 +314,6 @@ $pag = new Paginator([
 				}
 				location.reload();
 			});
-			$('#input-show_difficulty').click(function() {
-				if (this.checked) {
-					$.cookie('show_difficulty', '', {
-						path: '/problems',
-						expires: 365,
-					});
-				} else {
-					$.removeCookie('show_difficulty', {
-						path: '/problems'
-					});
-				}
-				location.reload();
-			});
 		</script>
 
 		<div class="card my-3">
@@ -366,26 +345,39 @@ $pag = new Paginator([
 					<i class="bi bi-search"></i>
 				</button>
 			</div>
-			<?php if (Auth::check()) : ?>
-				<div class="form-check d-inline-block">
-					<input type="checkbox" name="my" <?= isset($_GET['my']) ? 'checked="checked"' : '' ?> class="form-check-input" id="input-my">
-					<label class="form-check-label" for="input-my">
-						我的题目
-					</label>
+			<div>
+				<?php if (Auth::check()) : ?>
+					<div class="form-check d-inline-block">
+						<input type="checkbox" name="my" <?= isset($_GET['my']) ? 'checked="checked"' : '' ?> class="form-check-input" id="input-my">
+						<label class="form-check-label" for="input-my">
+							我的题目
+						</label>
+					</div>
+				<?php endif ?>
+				<?php if (UOJProblem::userCanManageSomeProblem(Auth::user())) : ?>
+					<div class="form-check d-inline-block ms-2">
+						<input type="checkbox" name="is_hidden" <?= isset($_GET['is_hidden']) ? 'checked="checked"' : '' ?> class="form-check-input" id="input-is_hidden">
+						<label class="form-check-label" for="input-is_hidden">
+							隐藏题目
+						</label>
+					</div>
+				<?php endif ?>
+			</div>
+			<div class="row mt-3">
+				<label class="col-sm-3 col-form-label" for="input-difficulty">难度</label>
+				<div class="col-sm-9">
+					<select id="input-difficulty" name="difficulty" class="form-select">
+						<option value="">全部</option>
+						<?php foreach (UOJProblem::$difficulty as $opt_name => $opt_value) : ?>
+							<?= HTML::option($opt_name, $opt_value, $opt_name == $_GET['difficulty']) ?>
+						<?php endforeach ?>
+					</select>
 				</div>
-			<?php endif ?>
-			<?php if (UOJProblem::userCanManageSomeProblem(Auth::user())) : ?>
-				<div class="form-check d-inline-block ms-2">
-					<input type="checkbox" name="is_hidden" <?= isset($_GET['is_hidden']) ? 'checked="checked"' : '' ?> class="form-check-input" id="input-is_hidden">
-					<label class="form-check-label" for="input-is_hidden">
-						隐藏题目
-					</label>
-				</div>
-			<?php endif ?>
+			</div>
 		</form>
 		<script>
 			$('#search-input').val(new URLSearchParams(location.search).get('search'));
-			$('#input-my, #input-is_hidden').click(function() {
+			$('#input-my, #input-is_hidden, #input-difficulty').change(function() {
 				$('#form-problem_search').submit();
 			});
 		</script>
diff --git a/web/app/upgrade/20_problem_difficulty/up.sql b/web/app/upgrade/20_problem_difficulty/up.sql
index 5cd8680..16f125a 100644
--- a/web/app/upgrade/20_problem_difficulty/up.sql
+++ b/web/app/upgrade/20_problem_difficulty/up.sql
@@ -1,2 +1,3 @@
 ALTER TABLE `problems` ADD `difficulty` int NOT NULL DEFAULT '-1' AFTER `submit_num`;
 ALTER TABLE `problems` MODIFY `extra_config` json NOT NULL;
+ALTER TABLE `problems` ADD KEY `difficulty` (`difficulty`);