@@ -10,29 +10,56 @@ import Foundation
1010import UIKit
1111import Alamofire
1212
13+enum ResultR: Int {
14+ case internalServerError = 500
15+ case exceededContent = 470
16+ case longContent = 471
17+ case incorrectContent = 472
18+ case impossibleContent = 473
19+ case invalidTocken = 498
20+ case alreadyRegistered = 409
21+ case invalidCode = 403
22+ case badAccess = 400
23+ case tooMuchToAdd = 406
24+ case notFound = 404
25+ case success1 = 204
26+ case success = 200
27+}
28+
1329class Model{
1430
1531 static let invalidTocken = 498
1632
17− static var hashTagTree: CompletionTree?
33+ static var hashTagTree: CompletionTree?
1834
1935 private static var isValidTocken: ((Int)->())? = { responce in
2036 print(responce)
2137 if responce == invalidTocken {
38+
2239 DataStorage.standard.setIsLoggedIn(value: false, with: 0)
2340 (UIApplication.shared.delegate as! AppDelegate).relaunch()
41+
42+ Model.logout {
43+ print("Logout Is Successful")
44+ }
2445 }
2546 }
2647
2748 private static let baseUrl = "https://valera-denis.herokuapp.com"
2849 static let decoder = JSONDecoder()
29−
50+
51+ static let authMeURL = URL(string: "\(baseUrl)/authentication/me")!
52+ static let deactivateURL = URL(string: "\(baseUrl)/deactivateAll")!
53+ static let registerMeURL = URL(string: "\(baseUrl)/authentication/register")!
54+ static let authVerifyURL = URL(string: "\(baseUrl)/authentication/verify")!
3055 static let authenticationURL = URL(string:"\(baseUrl)/authentication/login")!
56+ static let logOutURL = URL(string:"\(baseUrl)/authentication/logout")!
3157 static let postsLastURL = URL(string:"\(baseUrl)/posts/last")!
3258 static let postsForUserURL = URL(string:"\(baseUrl)/posts/forUser")!
3359 static let postsPublishURL = URL(string:"\(baseUrl)/posts/publish")!
3460 static let usersURL = URL(string:"\(baseUrl)/users")!
3561 static let usersAllURL = URL(string:"\(baseUrl)/users/all")!
62+ static let usersUpdateURL = URL(string:"\(baseUrl)/users/update")!
3663 static let channelsGetURL = URL(string: "\(baseUrl)/channels/get")!
3764 static let channelsUpdateURL = URL(string: "\(baseUrl)/channels/update")!
3865 static let channelsListURL = URL(string: "\(baseUrl)/channels")!
@@ -49,6 +76,7 @@ class Model{
4976 static let messagesGetGroupChatURL = URL(string: "\(baseUrl)/messages/get/groupChat")! //r
5077 static let messagesSendURL = URL(string: "\(baseUrl)/messages/send")!
5178 static let messagesGetUserChatURL = URL(string: "\(baseUrl)/messages/get/userChat")!
79+ static let facultySearchURL = URL(string: "\(baseUrl)/faculty/search")!
5280
5381
5482 struct QueryPosts<T: Codable>: Codable {
@@ -120,6 +148,8 @@ class Model{
120148 var lastName: String
121149 var firstName: String
122150 var id: Int
151+ var faculty: Faculty
152+ var email: String
123153
124154 func fullName() -> String {
125155 return "\(firstName) \(lastName)"
@@ -183,24 +213,108 @@ class Model{
183213 }
184214 }
185215
186− static func authenticate(with id: Int, completion: @escaping ((Bool)->())) {
216+ struct AuthStatus: Codable {
217+ var userStatus: String
187218
188− let json: [String:Any] = ["authenticationId" : id]
189− let jsonData = try? JSONSerialization.data(withJSONObject: json)
219+ init(){
220+ userStatus = "invalid"
221+ }
222+ }
223+
224+ static func logout(completion: @escaping (()->())){
225+ var request = URLRequest(url: logOutURL)
226+ request.httpMethod = "POST"
227+
228+ AF.request(request).response { (responce) in
229+ isValidTocken?(responce.response?.statusCode ?? 498)
230+
231+ if let code = responce.response?.statusCode, code == 498 {
232+ return
233+ }
234+
235+ completion()
236+ }
237+ }
238+
239+ /*
240+ "email": "vchernyshev@hse.ru",
241+ "middleName": "Algebrovich",
242+ "lastName": "Leonidov",
243+ "faculty": "cs.hse.ru/dse",
244+ "firstName": "Seva"
245+ */
246+
247+ struct NewRegistration: Codable{
248+ var email: String
249+ var firstName: String?
250+ var middleName: String?
251+ var lastName: String?
252+ var faculty: String?
253+ }
254+
255+ static func register(object: NewRegistration, completion: @escaping (()->())) {
256+ var request = URLRequest(url: registerMeURL)
257+ request.httpBody = try? JSONEncoder().encode(object)
190258
191− var request = URLRequest(url: authenticationURL)
192259 request.httpMethod = "POST"
260+ request.setValue("application/json; charset=utf-8", forHTTPHeaderField: "Content-Type")
193261
194− // insert json data to the request
195− request.httpBody = jsonData
262+ AF.request(request).response { (responce) in
263+
264+ guard let code = responce.response?.statusCode else { return }
265+ isValidTocken?(code)
266+
267+ if let code = responce.response?.statusCode, code == 498 {
268+ return
269+ }
270+
271+ if code == 204 {
272+
273+ let fields = responce.response?.allHeaderFields as? [String :String]
274+ let cookies = HTTPCookie.cookies(withResponseHeaderFields: fields!, for: responce.response!.url!)
275+ HTTPCookieStorage.shared.setCookie(cookies[0])
276+
277+ completion()
278+ }
279+ }
280+ }
281+
282+ // true only if everything is good
283+ static func verify(with code: Int, completion: @escaping ((Bool)->())){
284+ var request = URLRequest(url: authVerifyURL)
285+ request.httpMethod = "POST"
286+ request.httpBody = "\(code)".data(using: .utf8)
196287 request.setValue("application/json; charset=utf-8", forHTTPHeaderField: "Content-Type")
197288
198289 AF.request(request).response { (responce) in
199290
200− guard let realResponse = responce.response, realResponse.statusCode == 200 else
201− {
202− print("Not a 200 response")
291+ guard let statusCode = responce.response?.statusCode else { completion(false); return }
292+ isValidTocken?(statusCode)
293+
294+ if let code = responce.response?.statusCode, code == 498 {
295+ return
296+ }
297+
298+ if statusCode == 204 {
299+ // at this point cookies are set
300+ completion(true)
301+ } else {
203302 completion(false)
303+ }
304+ }
305+ }
306+
307+ // yes or no
308+ static func authenticateMe(completion: @escaping ((Bool)->())){
309+ var request = URLRequest(url: authMeURL)
310+ request.httpMethod = "POST"
311+
312+ AF.request(request).response { (responce) in
313+
314+ guard let statusCode = responce.response?.statusCode else { completion(false); return }
315+ isValidTocken?(statusCode)
316+
317+ if let code = responce.response?.statusCode, code == 498 {
204318 return
205319 }
206320
@@ -210,14 +324,45 @@ class Model{
210324 return
211325 }
212326
213− let fields = realResponse.allHeaderFields as? [String :String]
327+ DataStorage.standard.setIsLoggedIn(value: true, with: personId)
328+ completion(true)
329+ }
330+ }
331+
332+ // register, ok or invalid
333+ static func authenticate(with email: String, completion: @escaping ((AuthStatus)->())) {
334+
335+ let json: [String:Any] = ["authenticationEmail" : email]
336+ let jsonData = try? JSONSerialization.data(withJSONObject: json)
337+
338+ var request = URLRequest(url: authenticationURL)
339+ request.httpMethod = "POST"
340+
341+ // insert json data to the request
342+ request.httpBody = jsonData
343+ request.setValue("application/json; charset=utf-8", forHTTPHeaderField: "Content-Type")
344+
345+ AF.request(request).response { (response) in
214346
215− let cookies = HTTPCookie.cookies(withResponseHeaderFields: fields!, for: realResponse.url!)
347+ guard let realResponse = response.response, realResponse.statusCode == 200 else
348+ {
349+ print("Not a 200 response")
350+ completion(AuthStatus())
351+ return
352+ }
353+
354+ guard let json = response.data else { return }
355+ guard let res = try? decoder.decode(AuthStatus.self, from: json) else { return }
216356
217− HTTPCookieStorage.shared.setCookie(cookies[0])
357+ if realResponse.statusCode == 200 && res.userStatus == "registered"
358+ {
359+ // at this point cookies are set
360+ let fields = realResponse.allHeaderFields as? [String :String]
361+ let cookies = HTTPCookie.cookies(withResponseHeaderFields: fields!, for: realResponse.url!)
362+ HTTPCookieStorage.shared.setCookie(cookies[0])
363+ }
218364
219− DataStorage.standard.setIsLoggedIn(value: true, with: personId)
220− completion(true)
365+ completion(res)
221366 }
222367 }
223368
@@ -267,6 +412,10 @@ class Model{
267412
268413 isValidTocken?(response.response?.statusCode ?? 498)
269414
415+ if let code = response.response?.statusCode, code == 498 {
416+ return
417+ }
418+
270419 guard let json = response.data else { return }
271420
272421 guard let newQueery = try? decoder.decode(QueryPosts<Posts>.self, from: json) else { print("no")
@@ -278,7 +427,7 @@ class Model{
278427 }
279428
280429
281− static func createAndPublish(body: [Attachments], tags: [String]){
430+ static func createAndPublish(body: [Attachments], tags: [String], completion: @escaping ((ResultR)->())){
282431 let jsonUpd = CreatedPost(body: body, tags: tags)
283432 var request = URLRequest(url: postsPublishURL)
284433
@@ -292,6 +441,18 @@ class Model{
292441 (response) in
293442
294443 isValidTocken?(response.response?.statusCode ?? 498)
444+
445+ if let code = response.response?.statusCode, code == 498 {
446+ return
447+ }
448+
449+ if let code = response.response?.statusCode,
450+ let result = ResultR(rawValue: code)
451+ {
452+ completion(result)
453+ } else {
454+ completion(.internalServerError)
455+ }
295456 }
296457 }
297458
@@ -317,6 +478,10 @@ class Model{
317478
318479 isValidTocken?(response.response?.statusCode ?? 498)
319480
481+ if let code = response.response?.statusCode, code == 498 {
482+ return
483+ }
484+
320485 guard let json = response.data else { return }
321486
322487 guard let newPost = try? decoder.decode(QueryPosts<Posts>.self, from: json) else { return }
@@ -329,7 +494,7 @@ class Model{
329494
330495 static func getUsers(for ids: [Int], completion: @escaping (([Int:Users])->())){
331496 let json = "\(Set(ids))"
332− print(json)
497+
333498 var request = URLRequest(url: usersURL)
334499
335500 request.httpMethod = "POST"
@@ -343,11 +508,16 @@ class Model{
343508
344509 isValidTocken?(response.response?.statusCode ?? 498)
345510
511+ if let code = response.response?.statusCode, code == 498 {
512+ return
513+ }
514+
346515 guard let json = response.data else { return }
347516
348517 guard let users = try? decoder.decode([Users].self, from: json) else { return }
349518
350519 var dict: [Int:Users] = [:]
520+
351521 users.forEach({ (user) in
352522 dict[user.id] = user
353523 })
@@ -374,6 +544,9 @@ class Model{
374544
375545 isValidTocken?(response.response?.statusCode ?? 498)
376546
547+ if let code = response.response?.statusCode, code == 498 {
548+ return
549+ }
377550
378551 guard let json = response.data else { return }
379552
@@ -385,7 +558,7 @@ class Model{
385558 }
386559
387560
388− static func createChannel(with channel: Channels) {
561+ static func createChannel(with channel: Channels, completion: @escaping ((ResultR)->())) {
389562
390563 var request = URLRequest(url: channelsCreateURL)
391564 request.httpMethod = "POST"
@@ -396,6 +569,13 @@ class Model{
396569 (response) in
397570
398571 isValidTocken?(response.response?.statusCode ?? 498)
572+
573+ if let code = response.response?.statusCode, let result = ResultR(rawValue: code) {
574+ completion(result)
575+ return
576+ }
577+
578+ completion(ResultR.internalServerError)
399579 }
400580
401581 }
@@ -408,6 +588,9 @@ class Model{
408588 AF.request(request).responseJSON { (response) in
409589
410590 isValidTocken?(response.response?.statusCode ?? 498)
591+ if let code = response.response?.statusCode, code == 498 {
592+ return
593+ }
411594
412595 guard let json = response.data else { return }
413596
@@ -426,6 +609,9 @@ class Model{
426609
427610 AF.request(request).response { (response) in
428611 isValidTocken?(response.response?.statusCode ?? 498)
612+ if let code = response.response?.statusCode, code == 498 {
613+ return
614+ }
429615 }
430616 completion()
431617 }
@@ -441,6 +627,9 @@ class Model{
441627 (response) in
442628
443629 isValidTocken?(response.response?.statusCode ?? 498)
630+ if let code = response.response?.statusCode, code == 498 {
631+ return
632+ }
444633 }
445634 }
446635
@@ -463,6 +652,9 @@ class Model{
463652 }
464653
465654 isValidTocken?(response.response?.statusCode ?? 498)
655+ if let code = response.response?.statusCode, code == 498 {
656+ return
657+ }
466658 }
467659 }
468660
@@ -538,12 +730,13 @@ class Model{
538730 (response) in
539731
540732 isValidTocken?(response.response?.statusCode ?? 498)
733+ if let code = response.response?.statusCode, code == 498 {
734+ return
735+ }
541736
542737 guard let json = response.data else { return }
543738
544739 guard let newQueery = try? decoder.decode(QueryPosts<Posts>.self, from: json) else { return }
545−
546− // idUser = newQueery.users
547740 completion((newQueery.users, newQueery.response))
548741 }
549742 }
@@ -553,6 +746,9 @@ class Model{
553746 AF.request(URLRequest(url: hashTagTreeURL)).responseJSON {
554747 (response) in
555748 isValidTocken?(response.response?.statusCode ?? 498)
749+ if let code = response.response?.statusCode, code == 498 {
750+ return
751+ }
556752 guard let json = response.data else { return }
557753 guard let tree = try? decoder.decode(CompletionTree.self, from: json) else { return }
558754 completion(tree)
@@ -570,6 +766,10 @@ class Model{
570766 AF.request(request).responseJSON { response in
571767 isValidTocken?(response.response?.statusCode ?? 498)
572768
769+ if let code = response.response?.statusCode, code == 498 {
770+ return
771+ }
772+
573773 guard let json = response.data else { return }
574774 let dialogs = try! decoder.decode(QueryPosts<Dialog>.self, from: json)
575775
@@ -622,7 +822,12 @@ class Model{
622822
623823 struct UserChat: Codable {
624824 var user: Int
625− var lastMessage: LastMessage
825+ var lastMessage: LastMessage?
826+
827+ init(user: Int, lastMessage: LastMessage? = nil) {
828+ self.user = user
829+ self.lastMessage = lastMessage
830+ }
626831 }
627832
628833 struct Group: Codable {
@@ -736,6 +941,9 @@ class Model{
736941 AF.request(request).response { (response) in
737942
738943 isValidTocken?(response.response?.statusCode ?? 498)
944+ if let code = response.response?.statusCode, code == 498 {
945+ return
946+ }
739947
740948 guard let json = response.data else { return }
741949 let stringInt = String.init(data: json, encoding: String.Encoding.utf8)
@@ -745,7 +953,7 @@ class Model{
745953 }
746954 }
747955
748− static func getMessagesFor(typeOfChat: Model.Dialog, chat id: Int, exclusiveFrom: Int? = nil, limit l: Int = 10, direction: String = "backward", completion: @escaping (([LastMessage])->()))
956+ static func getMessagesFor(typeOfChat: Model.Dialog, chat id: Int, exclusiveFrom: Int? = nil, limit l: Int = 25, direction: String = "backward", completion: @escaping (([LastMessage])->()))
749957 {
750958 let req = GeneralRequest<Int>(direction: direction, limit: l, exclusiveFrom: exclusiveFrom, request: id)
751959 var request: URLRequest?
@@ -764,6 +972,9 @@ class Model{
764972 AF.request(request!).response { (response) in
765973
766974 isValidTocken?(response.response?.statusCode ?? 498)
975+ if let code = response.response?.statusCode, code == 498 {
976+ return
977+ }
767978
768979 guard let json = response.data else { return }
769980 guard let messages = try? decoder.decode([LastMessage].self, from: json) else { return }
@@ -781,6 +992,10 @@ class Model{
781992
782993 AF.request(request).response { (response) in
783994 isValidTocken?(response.response?.statusCode ?? 498)
995+
996+ if let code = response.response?.statusCode, code == 498 {
997+ return
998+ }
784999 }
7851000
7861001 completion()
@@ -795,6 +1010,10 @@ class Model{
7951010
7961011 AF.request(request).response { (response) in
7971012 isValidTocken?(response.response?.statusCode ?? 498)
1013+
1014+ if let code = response.response?.statusCode, code == 498 {
1015+ return
1016+ }
7981017 }
7991018
8001019 // completion()
@@ -837,7 +1056,7 @@ class Model{
8371056 }
8381057 }
8391058
840− static func sendMessage(message: SendMessage, completion: @escaping (()->())){
1059+ static func sendMessage(message: SendMessage, completion: @escaping ((ResultR)->())){
8411060
8421061 var request = URLRequest(url: messagesSendURL)
8431062 request.setValue("application/json; charset=utf-8", forHTTPHeaderField: "Content-Type")
@@ -847,6 +1066,84 @@ class Model{
8471066 AF.request(request).response { (response) in
8481067 isValidTocken?(response.response?.statusCode ?? 498)
8491068
1069+ if let code = response.response?.statusCode, code == 498 {
1070+ return
1071+ }
1072+
1073+ if let result = ResultR(rawValue: response.response!.statusCode){
1074+ completion(result)
1075+ } else {
1076+ completion(.internalServerError)
1077+ }
1078+ }
1079+ }
1080+
1081+ struct Faculty: Codable
1082+ {
1083+
1084+ var path: String
1085+ var url: String
1086+ var campusName: String
1087+ var campusCode: String
1088+ var name: String
1089+ var tags: [String]
1090+ var address: String
1091+
1092+ }
1093+
1094+ static func searchFaculty(string: String, completion: @escaping (([Model.Faculty])->()))
1095+ {
1096+ var request = URLRequest(url: facultySearchURL)
1097+ request.httpMethod = "POST"
1098+ request.httpBody = string.data(using: .utf8)
1099+ request.setValue("text/plain;charset=utf-8", forHTTPHeaderField: "Content-Type")
1100+
1101+ AF.request(request).response { (response) in
1102+ isValidTocken?(response.response?.statusCode ?? 498)
1103+
1104+ if let code = response.response?.statusCode, code == 498 {
1105+ return
1106+ }
1107+
1108+ guard let json = response.data else { return }
1109+ guard let faculties = try? decoder.decode([Faculty].self, from: json) else { return }
1110+
1111+ completion(faculties)
1112+ }
1113+ }
1114+
1115+ static func userUpdate(with newUser: NewRegistration, completion: @escaping ((Bool)->())) {
1116+ var request = URLRequest(url: usersUpdateURL)
1117+ request.setValue("application/json; charset=utf-8", forHTTPHeaderField: "Content-Type")
1118+ request.httpMethod = "POST"
1119+ request.httpBody = try? JSONEncoder().encode(newUser)
1120+
1121+ AF.request(request).response { (response) in
1122+ isValidTocken?(response.response?.statusCode ?? 498)
1123+
1124+ if let code = response.response?.statusCode, code == 498 {
1125+ return
1126+ }
1127+
1128+ if let code = response.response?.statusCode, code != 204{
1129+ completion (false)
1130+ } else {
1131+ completion(true)
1132+ }
1133+ }
1134+ }
1135+
1136+ static func deactivateAll(completion: @escaping (()->())){
1137+ var request = URLRequest(url: deactivateURL)
1138+ request.httpMethod = "DELETE"
1139+
1140+ AF.request(request).response { (responce) in
1141+ isValidTocken?(responce.response?.statusCode ?? 498)
1142+
1143+ if let code = responce.response?.statusCode, code == 498 {
1144+ return
1145+ }
1146+
8501147 completion()
8511148 }
8521149 }