From 7a59d5b506aa842535512c4531e3e2042d4c6f76 Mon Sep 17 00:00:00 2001
From: Balazs Perlaki-Horvath <ba.perlaki@gmail.com>
Date: Fri, 24 May 2024 22:50:56 +0200
Subject: [PATCH] Move checks closer

---
 Model/Utilities/WebKitHandler.swift | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Model/Utilities/WebKitHandler.swift b/Model/Utilities/WebKitHandler.swift
index c0b20ed0..52b4af9d 100644
--- a/Model/Utilities/WebKitHandler.swift
+++ b/Model/Utilities/WebKitHandler.swift
@@ -103,14 +103,15 @@ final class KiwixURLSchemeHandler: NSObject, WKURLSchemeHandler {
     // MARK: Success responses
     @MainActor
     private func sendHTTP200Response(_ urlSchemeTask: WKURLSchemeTask, url: URL, content: URLContent) {
-        guard isStartedFor(urlSchemeTask.hash) else { return }
         let headers = ["Content-Type": content.httpContentType,
                        "Content-Length": "\(content.size)"]
         if let response = HTTPURLResponse(url: url, statusCode: 200, httpVersion: "HTTP/1.1", headerFields: headers) {
+            guard isStartedFor(urlSchemeTask.hash) else { return }
             urlSchemeTask.didReceive(response)
             urlSchemeTask.didReceive(content.data)
             urlSchemeTask.didFinish()
         } else {
+            guard isStartedFor(urlSchemeTask.hash) else { return }
             urlSchemeTask.didFailWithError(URLError(.badServerResponse, userInfo: ["url": url]))
         }
         stopFor(urlSchemeTask.hash)