Changeset 124 in code for trunk/morty.go

Timestamp:

Jul 14, 2020, 9:22:08 PM (5 years ago)

Author:

asciimoo

Message:

[enh] inject header even if the page does not contain body element

File:

• trunk/morty.go (modified) (3 diffs)

trunk/morty.go

@@ -184 +184 @@
 
 type RequestConfig struct {
-        Key     []byte
-        BaseURL *url.URL
+        Key          []byte
+        BaseURL      *url.URL
+        BodyInjected bool
 }
 
@@ -455 +456 @@
                 sanitizeCSS(&RequestConfig{Key: p.Key, BaseURL: parsedURI}, ctx, responseBody)
         case contentType.SubType == "html" && contentType.Suffix == "":
-                sanitizeHTML(&RequestConfig{Key: p.Key, BaseURL: parsedURI}, ctx, responseBody)
+                rc := &RequestConfig{Key: p.Key, BaseURL: parsedURI}
+                sanitizeHTML(rc, ctx, responseBody)
+                if !rc.BodyInjected {
+                        p := HTMLBodyExtParam{rc.BaseURL.String(), false}
+                        if len(rc.Key) > 0 {
+                                p.HasMortyKey = true
+                        }
+                        err := HTML_BODY_EXTENSION.Execute(ctx, p)
+                        if err != nil {
+                                if DEBUG {
+                                        fmt.Println("failed to inject body extension", err)
+                                }
+                        }
+                }
         default:
                 if contentDispositionBytes != nil {
@@ -679 +693 @@
                                                 }
                                         }
+                                        rc.BodyInjected = true
                                 case "style":
                                         state = STATE_DEFAULT