From e7dd4c838c2c5b73c83f9e90a5db8ee8b5ae9b69 Mon Sep 17 00:00:00 2001
From: Cell <1024@lruihao.cn>
Date: Mon, 26 Aug 2024 11:22:45 +0800
Subject: [PATCH] :zap: Chore: unified content encryption function

---
 layouts/_default/single.html                      | 12 ++----------
 layouts/partials/function/content-encryption.html | 14 ++++++++++++++
 layouts/partials/init/index.html                  |  2 +-
 layouts/posts/single.html                         | 12 ++----------
 layouts/shortcodes/fixit-encryptor.html           | 11 +++--------
 5 files changed, 22 insertions(+), 29 deletions(-)
 create mode 100644 layouts/partials/function/content-encryption.html

diff --git a/layouts/_default/single.html b/layouts/_default/single.html
index 99c43dc0..287c9832 100644
--- a/layouts/_default/single.html
+++ b/layouts/_default/single.html
@@ -16,16 +16,8 @@
 
     {{- /* Content */ -}}
     {{- $content := dict "Content" .Content "Ruby" $params.ruby "Fraction" $params.fraction "Fontawesome" $params.fontawesome | partial "function/content.html" | safeHTML -}}
-    {{- if $params.password -}}
-      {{- $saltLen := strings.RuneCount (trim $params.password " ") -}}
-      {{- $saltLen = cond (eq (mod $saltLen 2) 0) (add $saltLen 1) $saltLen -}}
-      {{- $base64EncodeContent := $content | base64Encode -}}
-      {{- $content = printf "%v%v%v"
-        (substr $base64EncodeContent 0 $saltLen)
-        (substr (sha256 $params.password) $saltLen)
-        (substr $base64EncodeContent $saltLen)
-      -}}
-    {{- end -}}
+    {{- /* Content Encryption */ -}}
+    {{- $content = dict "Content" $content "Password" $params.password | partial "function/content-encryption.html" | safeHTML -}}
     <div class="content" id="content"
       {{- with $params.password }} data-password="{{ hash.XxHash . }}"{{ end }}
       {{- with $params.password }} data-content="{{ $content }}"{{ end -}}
diff --git a/layouts/partials/function/content-encryption.html b/layouts/partials/function/content-encryption.html
new file mode 100644
index 00000000..2e2a8f96
--- /dev/null
+++ b/layouts/partials/function/content-encryption.html
@@ -0,0 +1,14 @@
+{{- $content := .Content -}}
+
+{{- if .Password -}}
+  {{- $saltLen := strings.RuneCount (trim .Password " ") -}}
+  {{- $saltLen = cond (eq (mod $saltLen 2) 0) (add $saltLen 1) $saltLen -}}
+  {{- $base64EncodeContent := $content | base64Encode -}}
+  {{- $content = printf "%v%v%v"
+    (substr $base64EncodeContent 0 $saltLen)
+    (substr (sha256 .Password) $saltLen)
+    (substr $base64EncodeContent $saltLen)
+  -}}
+{{- end -}}
+
+{{- return $content -}}
diff --git a/layouts/partials/init/index.html b/layouts/partials/init/index.html
index f21a1e73..98b7324c 100644
--- a/layouts/partials/init/index.html
+++ b/layouts/partials/init/index.html
@@ -1,4 +1,4 @@
-{{- .Scratch.Set "version" "v0.3.10-466bb7b7" -}}
+{{- .Scratch.Set "version" "v0.3.10-91f3c7c3" -}}
 {{- .Scratch.Set "this" dict -}}
 
 {{- partial "init/detection-env.html" . -}}
diff --git a/layouts/posts/single.html b/layouts/posts/single.html
index b5b012ac..aa5e1386 100644
--- a/layouts/posts/single.html
+++ b/layouts/posts/single.html
@@ -160,16 +160,8 @@
 
     {{- /* Content */ -}}
     {{- $content := dict "Content" .Content "Ruby" $params.ruby "Fraction" $params.fraction "Fontawesome" $params.fontawesome | partial "function/content.html" | safeHTML -}}
-    {{- if $params.password -}}
-      {{- $saltLen := strings.RuneCount (trim $params.password " ") -}}
-      {{- $saltLen = cond (eq (mod $saltLen 2) 0) (add $saltLen 1) $saltLen -}}
-      {{- $base64EncodeContent := $content | base64Encode -}}
-      {{- $content = printf "%v%v%v"
-        (substr $base64EncodeContent 0 $saltLen)
-        (substr (sha256 $params.password) $saltLen)
-        (substr $base64EncodeContent $saltLen)
-      -}}
-    {{- end -}}
+    {{- /* Content Encryption */ -}}
+    {{- $content = dict "Content" $content "Password" $params.password | partial "function/content-encryption.html" | safeHTML -}}
     <div class="content" id="content"
       {{- with $params.endFlag }} data-end-flag="{{ . }}"{{ end }}
       {{- with $params.password }} data-password="{{ xxhash . }}"{{ end }}
diff --git a/layouts/shortcodes/fixit-encryptor.html b/layouts/shortcodes/fixit-encryptor.html
index 4286f0e6..99ec1745 100644
--- a/layouts/shortcodes/fixit-encryptor.html
+++ b/layouts/shortcodes/fixit-encryptor.html
@@ -10,15 +10,10 @@
 {{- end -}}
 
 {{- if $password -}}
+  {{- /* Content */ -}}
   {{- $content := .Inner | .Page.RenderString -}}
-  {{- $saltLen := strings.RuneCount (trim $password "") -}}
-  {{- $saltLen = cond (eq (mod $saltLen 2) 0) (add $saltLen 1) $saltLen -}}
-  {{- $base64EncodeContent := $content | base64Encode -}}
-  {{- $content = printf "%v%v%v"
-    (substr $base64EncodeContent 0 $saltLen)
-    (substr (sha256 $password) $saltLen)
-    (substr $base64EncodeContent $saltLen)
-  -}}
+  {{- /* Content Encryption */ -}}
+  {{- $content = dict "Content" $content "Password" $password | partial "function/content-encryption.html" | safeHTML -}}
 
   <fixit-encryptor class="fixit-encryptor-shortcode">
     <div class="fixit-decryptor-container">