From f631e3de049ef36692ee056e38735dc1afd71a22 Mon Sep 17 00:00:00 2001
From: Dillon <dillonzq@outlook.com>
Date: Fri, 29 May 2020 00:19:39 +0800
Subject: [PATCH] fix(404): 404 page select language error (#391)

---
 layouts/partials/header.html | 46 ++++++++++++++++++++----------------
 1 file changed, 26 insertions(+), 20 deletions(-)

diff --git a/layouts/partials/header.html b/layouts/partials/header.html
index 0ceb39b..de3f955 100644
--- a/layouts/partials/header.html
+++ b/layouts/partials/header.html
@@ -44,16 +44,19 @@
                         {{- .Language.LanguageName -}}
                         <i class="fas fa-chevron-right fa-fw"></i>
                         <select class="language-select" id="language-select-desktop" onchange="location = this.value;">
-                            {{- range .Page.AllTranslations -}}
-                                {{- $translation := . -}}
-                                {{- range $.Site.Languages -}}
-                                    {{- if eq $translation.Lang .Lang -}}
-                                        {{- if eq $.Page.Lang .Lang -}}
-                                            <option value="{{ $translation.RelPermalink }}" selected>{{ .LanguageName }}</option>
-                                        {{- else -}}
-                                            <option value="{{ $translation.RelPermalink }}">{{ .LanguageName }}</option>
-                                        {{- end -}}
-                                    {{- end -}}
+                            {{- if eq .Kind "404" -}}
+                                {{- /* https://github.com/dillonzq/LoveIt/issues/378 */ -}}
+                                {{- range .Sites -}}
+                                    {{- $link := printf "%v/404.html" .LanguagePrefix -}}
+                                    <option value="{{ $link }}"{{ if eq . $.Site }} selected{{ end }}>
+                                        {{- .Language.LanguageName -}}
+                                    </option>
+                                {{- end -}}
+                            {{- else -}}
+                                {{- range .AllTranslations -}}
+                                    <option value="{{ .RelPermalink }}"{{ if eq .Lang $.Lang }} selected{{ end }}>
+                                        {{- .Language.LanguageName -}}
+                                    </option>
                                 {{- end -}}
                             {{- end -}}
                         </select>
@@ -151,16 +154,19 @@
                     {{- .Language.LanguageName -}}
                     <i class="fas fa-chevron-right fa-fw"></i>
                     <select class="language-select" onchange="location = this.value;">
-                        {{- range .Page.AllTranslations -}}
-                            {{- $translation := . -}}
-                            {{- range $.Site.Languages -}}
-                                {{- if eq $translation.Lang .Lang -}}
-                                    {{- if eq $.Page.Lang .Lang -}}
-                                        <option value="{{ $translation.RelPermalink }}" selected>{{ .LanguageName }}</option>
-                                    {{- else -}}
-                                        <option value="{{ $translation.RelPermalink }}">{{ .LanguageName }}</option>
-                                    {{- end -}}
-                                {{- end -}}
+                        {{- if eq .Kind "404" -}}
+                            {{- /* https://github.com/dillonzq/LoveIt/issues/378 */ -}}
+                            {{- range .Sites -}}
+                                {{- $link := printf "%v/404.html" .LanguagePrefix -}}
+                                <option value="{{ $link }}"{{ if eq . $.Site }} selected{{ end }}>
+                                    {{- .Language.LanguageName -}}
+                                </option>
+                            {{- end -}}
+                        {{- else -}}
+                            {{- range .AllTranslations -}}
+                                <option value="{{ .RelPermalink }}"{{ if eq .Lang $.Lang }} selected{{ end }}>
+                                    {{- .Language.LanguageName -}}
+                                </option>
                             {{- end -}}
                         {{- end -}}
                     </select>