fix(client): handle js cookie change

NateScarlet · Jul 17, 2022 · a5ab3a8 · a5ab3a8
1 parent 4de1894
commit a5ab3a8
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 0 deletions.
diff --git a/pkg/client/js_cookie.env.cjs b/pkg/client/js_cookie.env.cjs
@@ -591,6 +591,16 @@ const { window, document } = (function () {
         if (tag === "div") {
           return div;
         }
+        if (tag === "a") {
+          return proxyGet({
+            tagName: "A",
+          });
+        }
+        if (tag == "form") {
+          return proxyGet({
+            tagName: "FORM",
+          })
+        }
         throw ["document.createElement", ...arguments];
       },
       getElementsByTagName(name) {

diff --git a/pkg/client/js_cookie_sample_2022-07-17.js b/pkg/client/js_cookie_sample_2022-07-17.js
diff --git a/pkg/client/js_cookie_test.go b/pkg/client/js_cookie_test.go
@@ -50,3 +50,25 @@ func TestJSCookieValue(t *testing.T) {
 	require.NoError(t, err)
 	assert.Regexp(t, "^Cc2838679FT=.{173}; path=/; expires=.{29}$", cookie)
 }
+
+func TestJSCookieValue2022_07_17(t *testing.T) {
+	var ctx = context.Background()
+	var doc *goquery.Document
+	func() {
+		r, err := os.Open("js_cookie_sample.html")
+		require.NoError(t, err)
+		defer r.Close()
+		doc, err = goquery.NewDocumentFromReader(r)
+		require.NoError(t, err)
+	}()
+
+	data, err := newJSCookieTemplateData(ctx, "https://book.qidian.com/info/1004608738/", *doc)
+	require.NoError(t, err)
+	script2Data, err := ioutil.ReadFile("js_cookie_sample_2022-07-17.js")
+	require.NoError(t, err)
+	data.Script2 = string(script2Data)
+
+	cookie, err := jsCookieValue(ctx, *data)
+	require.NoError(t, err)
+	assert.Regexp(t, "^Cc2838679FT=.{173}; path=/; expires=.{29}$", cookie)
+}