Merge pull request #2 from Rev0212/feature-customise

Feature customise

Author: Rev0212

README.md

@@ -35,6 +35,28 @@ Enable the inspector in **DEBUG** builds only:
 NetworkInspector.LeapInspector.enable()
 NetworkInspector.LeapInspector.enableFloatingButton()
 #endif
+```
 
 A floating 📡 button will appear on screen.
 Tap it to open the network inspector UI.
+
+To capture only specific base URLs, pass one or more base addresses:
+
+```swift
+#if DEBUG
+NetworkInspector.enable(
+    baseURLs: [
+        "https://mario-api.leapscholar.com",
+        "https://mario-ieltsbff.leapscholar.com"
+    ]
+)
+NetworkInspector.enableFloatingButton()
+#endif
+```
+
+---
+
+## API Tab
+
+The inspector UI includes an **APIs** tab that lists unique endpoints and the number
+of calls for each. Tap an endpoint to view only the logs for that API.

Sources/NetworkInspector/InspectorURLProtocol.swift

@@ -27,7 +27,8 @@ final class InspectorURLProtocol: URLProtocol {
         }
 
         guard let scheme = request.url?.scheme else { return false }
-        return scheme == "http" || scheme == "https"
+        guard scheme == "http" || scheme == "https" else { return false }
+        return NetworkInterceptor.shouldIntercept(request.url)
     }
 
     override class func canonicalRequest(for request: URLRequest) -> URLRequest {

Sources/NetworkInspector/LeapInspector.swift

@@ -7,9 +7,10 @@ public enum LeapInspector {
 
     private static var isEnabled = false
 
-    public static func enable() {
+    public static func enable(baseURLs: [String] = []) {
         guard !isEnabled else { return }
         isEnabled = true
+        NetworkInterceptor.setAllowedBaseURLs(baseURLs)
         NetworkInterceptor.register()
     }
 

Sources/NetworkInspector/NetworkInterceptor.swift

@@ -10,11 +10,39 @@ import Foundation
 
 final class NetworkInterceptor {
 
+    private static var allowedBaseURLs: [String] = []
+
     static func register() {
         URLProtocol.registerClass(InspectorURLProtocol.self)
     }
 
     static func unregister() {
         URLProtocol.unregisterClass(InspectorURLProtocol.self)
     }
+
+    static func setAllowedBaseURLs(_ urls: [String]) {
+        allowedBaseURLs = urls
+            .map { $0.trimmingCharacters(in: .whitespacesAndNewlines) }
+            .filter { !$0.isEmpty }
+            .map { normalizeBaseURL($0) }
+    }
+
+    static func getAllowedBaseURLs() -> [String] {
+        allowedBaseURLs
+    }
+
+    static func shouldIntercept(_ url: URL?) -> Bool {
+        guard let absoluteString = url?.absoluteString.lowercased() else { return false }
+        guard !allowedBaseURLs.isEmpty else { return true }
+
+        return allowedBaseURLs.contains { absoluteString.hasPrefix($0) }
+    }
+
+    private static func normalizeBaseURL(_ url: String) -> String {
+        var normalized = url.lowercased()
+        while normalized.hasSuffix("/") {
+            normalized.removeLast()
+        }
+        return normalized
+    }
 }

Sources/NetworkInspector/NetworkLogStore.swift

@@ -60,6 +60,19 @@ final class NetworkLogStore {
             }
         }
     }
+
+    func remove(where shouldRemove: @escaping (NetworkLog) -> Bool) {
+        queue.async {
+            self.logs.removeAll(where: shouldRemove)
+
+            DispatchQueue.main.async {
+                NotificationCenter.default.post(
+                    name: .networkLogStoreDidUpdate,
+                    object: nil
+                )
+            }
+        }
+    }
 
     func clear() {
         queue.async {

Sources/NetworkInspector/UI/LogList/ApiListViewController.swift

@@ -0,0 +1,145 @@
+//
+//  ApiListViewController.swift
+//  NetworkInspector
+//
+//  Created by Revanth A on 01/01/26.
+//
+
+import UIKit
+
+final class ApiListViewController: UIViewController {
+
+    private let tableView = UITableView()
+    private var items: [APIItem] = []
+
+    override func viewDidLoad() {
+        super.viewDidLoad()
+
+        title = "APIs"
+        view.backgroundColor = .systemBackground
+
+        setupTableView()
+        loadItems()
+        observeLogs()
+    }
+
+    deinit {
+        NotificationCenter.default.removeObserver(self)
+    }
+}
+
+private extension ApiListViewController {
+
+    func setupTableView() {
+        tableView.translatesAutoresizingMaskIntoConstraints = false
+        view.addSubview(tableView)
+
+        NSLayoutConstraint.activate([
+            tableView.topAnchor.constraint(equalTo: view.topAnchor),
+            tableView.bottomAnchor.constraint(equalTo: view.bottomAnchor),
+            tableView.leadingAnchor.constraint(equalTo: view.leadingAnchor),
+            tableView.trailingAnchor.constraint(equalTo: view.trailingAnchor)
+        ])
+
+        tableView.dataSource = self
+        tableView.delegate = self
+    }
+
+    func observeLogs() {
+        NotificationCenter.default.addObserver(
+            self,
+            selector: #selector(onLogsUpdated),
+            name: .networkLogStoreDidUpdate,
+            object: nil
+        )
+    }
+
+    @objc
+    func onLogsUpdated() {
+        loadItems()
+    }
+
+    func loadItems() {
+        let logs = NetworkLogStore.shared.getAllLogs()
+        items = buildItems(from: logs)
+        tableView.reloadData()
+    }
+
+    func buildItems(from logs: [NetworkLog]) -> [APIItem] {
+        var counts: [String: APIItem] = [:]
+
+        for log in logs {
+            let host = log.request.url?.host ?? "-"
+            let path = endpointPath(from: log.request.url)
+            let key = "\(host)|\(path)"
+
+            if let existing = counts[key] {
+                counts[key] = APIItem(host: host, path: path, count: existing.count + 1)
+            } else {
+                counts[key] = APIItem(host: host, path: path, count: 1)
+            }
+        }
+
+        return counts.values.sorted { lhs, rhs in
+            if lhs.host == rhs.host {
+                return lhs.path < rhs.path
+            }
+            return lhs.host < rhs.host
+        }
+    }
+
+    func endpointPath(from url: URL?) -> String {
+        let path = url?.path ?? "/"
+        if path == "/" || path.isEmpty {
+            return "/"
+        }
+        return path.hasPrefix("/") ? String(path.dropFirst()) : path
+    }
+}
+
+extension ApiListViewController: UITableViewDataSource {
+
+    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
+        items.count
+    }
+
+    func tableView(
+        _ tableView: UITableView,
+        cellForRowAt indexPath: IndexPath
+    ) -> UITableViewCell {
+        let cell = tableView.dequeueReusableCell(withIdentifier: "ApiCell")
+            ?? UITableViewCell(style: .subtitle, reuseIdentifier: "ApiCell")
+
+        let item = items[indexPath.row]
+        cell.textLabel?.text = item.path
+        cell.textLabel?.font = .systemFont(ofSize: 15, weight: .medium)
+        cell.detailTextLabel?.text = "\(item.host) • \(item.count) calls"
+        cell.detailTextLabel?.textColor = .secondaryLabel
+        cell.selectionStyle = .none
+
+        return cell
+    }
+}
+
+extension ApiListViewController: UITableViewDelegate {
+
+    func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
+        let item = items[indexPath.row]
+        let title = item.path == "/" ? item.host : item.path
+
+        let filter: (NetworkLog) -> Bool = { log in
+            let host = log.request.url?.host ?? "-"
+            let path = self.endpointPath(from: log.request.url)
+            return host == item.host && path == item.path
+        }
+
+        let detailVC = LogListViewController(filter: filter, title: title)
+        navigationController?.pushViewController(detailVC, animated: true)
+    }
+}
+
+private struct APIItem {
+    let host: String
+    let path: String
+    let count: Int
+}