Check status code

ngs · Jun 30, 2018 · 80b2677 · 80b2677
1 parent 6d20c0b
commit 80b2677
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/CI2Go/View Controllers/BuildsViewController.swift b/CI2Go/View Controllers/BuildsViewController.swift
@@ -160,10 +160,12 @@ class BuildsViewController: UITableViewController {
     }
 
     func loadUser() {
-        URLSession.shared.dataTask(endpoint: .me) { [weak self] (user, _, _, err) in
+        URLSession.shared.dataTask(endpoint: .me) { [weak self] (user, _, res, err) in
             guard let user = user else {
                 Crashlytics.sharedInstance().recordError(err ?? APIError.noData)
-                self?.logout()
+                if let res = res as? HTTPURLResponse, res.statusCode == 401 {
+                    self?.logout()
+                }
                 return
             }
             self?.currentUser = user