From 0b406525fac2191dcc073dfa2e12c7af94102302 Mon Sep 17 00:00:00 2001
From: Sakura Knoll <s4kura.knoll@gmail.com>
Date: Sat, 1 Jan 2022 20:19:38 +0900
Subject: [PATCH] Fix filter of weapon list page

---
 src/pages/honkai3rd/weapons/index.tsx | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/pages/honkai3rd/weapons/index.tsx b/src/pages/honkai3rd/weapons/index.tsx
index a432e779..09d2c299 100644
--- a/src/pages/honkai3rd/weapons/index.tsx
+++ b/src/pages/honkai3rd/weapons/index.tsx
@@ -6,8 +6,9 @@ import Honkai3rdNavigator from '../../../components/organisms/Honkai3rdNavigator
 import { listWeapons, WeaponData } from '../../../server/data/honkai3rd/weapons'
 import { pick } from 'ramda'
 import SquareImageBox from '../../../components/atoms/SquareImageBox'
-import { useMemo, useState } from 'react'
+import { useMemo } from 'react'
 import FilterButton from '../../../components/atoms/FilterButton'
+import { useRouter } from 'next/router'
 
 type WeaponListItemData = Pick<
   WeaponData,
@@ -32,7 +33,15 @@ const weaponFilterOptions = [
 ]
 
 const WeaponListPage = ({ weaponDataList }: WeaponListPageProps) => {
-  const [filter, setFilter] = useState('all')
+  const { query } = useRouter()
+
+  const filter = useMemo(() => {
+    if (query.filter == null) {
+      return 'all'
+    }
+
+    return typeof query.filter === 'string' ? query.filter : query.filter[0]
+  }, [query])
 
   const weaponList = useMemo(() => {
     return weaponDataList.map((weapon) => {
@@ -112,7 +121,6 @@ const WeaponListPage = ({ weaponDataList }: WeaponListPageProps) => {
                 <FilterButton
                   key={value}
                   active={value === filter}
-                  setFilter={setFilter}
                   value={value}
                   label={label}
                   m={1}