Merge pull request #10 from llinardos/adds-patch-to-router

Adds patch to router.
depoon · May 20, 2019 · a8ee661 · a8ee661
2 parents 72fa9e7 + 4101733
commit a8ee661
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/Source/Localhost/LocalhostRouter.swift b/Source/Localhost/LocalhostRouter.swift
@@ -15,6 +15,9 @@ public protocol LocalhostRouter {
     func post(_ path: String, routeBlock: @escaping ((URLRequest) -> LocalhostServerResponse?))
     func delete(_ path: String, routeBlock: @escaping ((URLRequest) -> LocalhostServerResponse?))
     func put(_ path: String, routeBlock: @escaping ((URLRequest) -> LocalhostServerResponse?))
+    func patch(_ path: String, routeBlock: @escaping ((URLRequest) -> LocalhostServerResponse?))
+    func options(_ path: String, routeBlock: @escaping ((URLRequest) -> LocalhostServerResponse?))
+    func head(_ path: String, routeBlock: @escaping ((URLRequest) -> LocalhostServerResponse?))
 
     func startListening()
     func stopListening()

diff --git a/Source/Localhost/LocalhostServer.swift b/Source/Localhost/LocalhostServer.swift
@@ -47,6 +47,15 @@ extension LocalhostServer: LocalhostRouter {
         })
     }
 
+    public func patch(_ path: String, routeBlock: @escaping ((URLRequest) -> LocalhostServerResponse?)) {
+        self.server.add(path, block: {  [weak self] (req, res, next) in
+            self?.handleRoute(httpMethod: "PATCH",
+                              routeBlock: routeBlock,
+                              crRequest: req,
+                              crResponse: res)
+            }, recursive: false, method: .patch)
+    }
+
     public func head(_ path: String, routeBlock: @escaping ((URLRequest) -> LocalhostServerResponse?)) {
         self.server.head(path, block: {  [weak self] (req, res, next) in
             self?.handleRoute(httpMethod: "HEAD",