added source files

Author: Casperhr

Sources/Bugsnag/BugsnagMiddleware.swift

@@ -0,0 +1,43 @@
+import Vapor
+import HTTP
+
+public final class BugsnagMiddleware: Middleware {
+    
+    let drop: Droplet
+    let configuration: Configuration
+    let connectionManager: ConnectionMananger
+    public init(drop: Droplet) throws {
+        self.drop = drop
+        self.configuration = try Configuration(drop: drop)
+        self.connectionManager = ConnectionMananger(drop: drop, config: configuration)
+    }
+
+    public func respond(to request: Request, chainingTo next: Responder) throws -> Response {
+        
+        do {
+            return try next.respond(to: request)
+        } catch Abort.badRequest {
+            try report(status: .badRequest, message: "Bad request", request: request)
+            throw Abort.badRequest
+        } catch Abort.serverError {
+            try report(status: .internalServerError, message: "Server error", request: request)
+            throw Abort.serverError
+        } catch Abort.notFound {
+            try report(status: .notFound, message: "Not found", request: request)
+            throw Abort.notFound
+        } catch Abort.custom(let status, let message) {
+            try report(status: status, message: message, request: request)
+            throw Abort.custom(status: status, message: message)
+        }
+        // This can be incommented when Vapor updates
+        /*catch Abort.customWithCode(let status, let message, let code) {
+            try report(status: status, message: message, request: request)
+            throw Abort.customWithCode(status: status, message: message, code: code)
+        }*/
+    }
+    
+    public func report(status: Status, message: String, request: Request) throws {
+        _ = try connectionManager.post(status: status, message: message, request: request)
+    }
+}
+

Sources/Bugsnag/Configuration.swift

@@ -0,0 +1,57 @@
+import Foundation
+import Vapor
+import HTTP
+
+public struct Configuration {
+    
+    public enum Field: String {
+        case apiKey                 = "bugsnag.apiKey"
+        case notifyReleaseStages    = "bugsnag.notifyReleaseStages"
+        case endpoint               = "bugsnag.endpoint"
+        case filters                = "bugsnag.filters"
+        
+        var path: [String] {
+            return rawValue.components(separatedBy: ".")
+        }
+        
+        var error: Abort {
+            return .custom(status: .internalServerError,
+                           message: "Bugsnag error - \(rawValue) config is missing.")
+        }
+    }
+    
+    public let apiKey: String
+    public let notifyReleaseStages: [String]
+    public let endpoint: String
+    public let filters: [String]
+    public init(drop: Droplet) throws {
+        self.apiKey                 = try Configuration.extract(field: .apiKey, drop: drop)
+        self.notifyReleaseStages    = try Configuration.extract(field: .notifyReleaseStages, drop: drop)
+        self.endpoint               = try Configuration.extract(field: .endpoint, drop: drop)
+        self.filters                = try Configuration.extract(field: .filters, drop: drop)
+    }
+    
+    private static func extract(field: Field , drop: Droplet) throws -> [String] {
+        // Get array
+        guard let platforms = drop.config[field.path]?.array else {
+            throw field.error
+        }
+        
+        // Get from config and make sure all values are strings
+        return try platforms.map({
+            guard let string = $0.string else {
+                throw field.error
+            }
+            
+            return string
+        })
+    }
+    
+    private static func extract(field: Field , drop: Droplet) throws -> String {
+        guard let string = drop.config[field.path]?.string else {
+            throw field.error
+        }
+        
+        return string
+    }
+}

Sources/Bugsnag/ConnectionManager.swift

@@ -0,0 +1,101 @@
+import Vapor
+import HTTP
+import Foundation
+
+public final class ConnectionMananger {
+    
+    let drop: Droplet
+    let config: Configuration
+    
+    public init(drop: Droplet, config: Configuration) {
+        self.drop = drop
+        self.config = config
+    }
+    
+    func headers() -> [HeaderKey: String] {
+        
+        let headers = [
+            HeaderKey("Content-Type"): "application/json",
+        ]
+        
+        return headers
+    }
+    
+    func body(message: String, request: Request) throws -> JSON {
+        var code: [String: Node] = [:]
+        
+        var index = 0
+        for entry in Thread.callStackSymbols {
+            code[String(index)] = Node(entry)
+            
+            index = index + 1
+        }
+        
+        let stacktrace = Node([
+            Node([
+                "file": Node(message),
+                "lineNumber": 0,
+                "columnNumber": 0,
+                "method": "NA",
+                "code": Node(code)
+            ])
+        ])
+   
+        let app: Node = Node([
+            "releaseStage": Node(drop.environment.description),
+            "type": "Vapor"
+        ])
+        
+        var headers: [String: Node] = [:]
+        for (key, value) in request.headers {
+            headers[key.key] = Node(value)
+        }
+        
+        let metaData = Node([
+            "request": Node([
+                "method": Node(request.method.description),
+                "headers": Node(headers),
+                "params": request.parameters,
+                "url": Node(request.uri.path)
+            ])
+        ])
+        
+        
+        let event: Node = Node([
+            Node([
+                "payloadVersion": 2,
+                "exceptions": Node([
+                    Node([
+                        "errorClass": Node(message),
+                        "message": Node(message),
+                        "stacktrace": stacktrace
+                    ])
+                ]),
+                "app": app,
+                "severity": "error",
+                "metaData": metaData
+            ])
+        ])
+    
+        return try JSON(node: [
+            "apiKey": self.config.apiKey,
+            "notifier": Node([
+                    "name": "Bugsnag Vapor",
+                    "version": "1.0.11",
+                    "url": "https://github.com/nodes-vapor/bugsnag"
+            ]),
+            "events": event,
+        ])
+    }
+    
+    func post(json: JSON) throws -> Status {
+        let response = try drop.client.post(self.config.endpoint, headers: headers(), body: json.makeBody())
+        
+        return response.status
+    }
+    
+    func post(status: Status, message: String, request: Request) throws -> Status {
+        return try post(json: body(message: message, request: request))
+    }
+    
+}