cmd/extract: handle mime parsing failure

CorentinB · Mar 30, 2024 · 804a982 · 804a982
1 parent a1f4278
commit 804a982
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/cmd/extract.go b/cmd/extract.go
@@ -117,12 +117,16 @@ func writeFile(vmd *cobra.Command, resp *http.Response, record *warc.Record) err
 
 	if resp.Header.Get("Content-Disposition") != "" {
 		_, params, err := mime.ParseMediaType(resp.Header.Get("Content-Disposition"))
-		if err != nil {
-			return err
-		}
+		if err == nil {
+			if params["filename"] != "" {
+				filename = params["filename"]
+			}
+		} else {
+			logrus.Debugf("failed to parse Content-Disposition header: %v", err)
 
-		if params["filename"] != "" {
-			filename = params["filename"]
+			if !strings.HasSuffix(filename, ".pdf") {
+				filename += ".pdf"
+			}
 		}
 	}