NCManageDatabase+Activity.swift 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345
  1. //
  2. // NCManageDatabase+Activity.swift
  3. // Nextcloud
  4. //
  5. // Created by Henrik Storch on 30.11.21.
  6. // Copyright © 2021 Marino Faggiana. All rights reserved.
  7. //
  8. // Author Marino Faggiana <marino.faggiana@nextcloud.com>
  9. //
  10. // This program is free software: you can redistribute it and/or modify
  11. // it under the terms of the GNU General Public License as published by
  12. // the Free Software Foundation, either version 3 of the License, or
  13. // (at your option) any later version.
  14. //
  15. // This program is distributed in the hope that it will be useful,
  16. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  18. // GNU General Public License for more details.
  19. //
  20. // You should have received a copy of the GNU General Public License
  21. // along with this program. If not, see <http://www.gnu.org/licenses/>.
  22. //
  23. import Foundation
  24. import RealmSwift
  25. import NextcloudKit
  26. import SwiftyJSON
  27. class tableActivity: Object, DateCompareable {
  28. var dateKey: Date { date as Date }
  29. @objc dynamic var account = ""
  30. @objc dynamic var idPrimaryKey = ""
  31. @objc dynamic var action = "Activity"
  32. @objc dynamic var date = NSDate()
  33. @objc dynamic var idActivity: Int = 0
  34. @objc dynamic var app = ""
  35. @objc dynamic var type = ""
  36. @objc dynamic var user = ""
  37. @objc dynamic var subject = ""
  38. @objc dynamic var subjectRich = ""
  39. let subjectRichItem = List<tableActivitySubjectRich>()
  40. @objc dynamic var icon = ""
  41. @objc dynamic var link = ""
  42. @objc dynamic var message = ""
  43. @objc dynamic var objectType = ""
  44. @objc dynamic var objectId: Int = 0
  45. @objc dynamic var objectName = ""
  46. @objc dynamic var note = ""
  47. @objc dynamic var selector = ""
  48. @objc dynamic var verbose: Bool = false
  49. override static func primaryKey() -> String {
  50. return "idPrimaryKey"
  51. }
  52. }
  53. class tableActivityLatestId: Object {
  54. @objc dynamic var account = ""
  55. @objc dynamic var activityFirstKnown: Int = 0
  56. @objc dynamic var activityLastGiven: Int = 0
  57. override static func primaryKey() -> String {
  58. return "account"
  59. }
  60. }
  61. class tableActivityPreview: Object {
  62. @objc dynamic var account = ""
  63. @objc dynamic var filename = ""
  64. @objc dynamic var idPrimaryKey = ""
  65. @objc dynamic var idActivity: Int = 0
  66. @objc dynamic var source = ""
  67. @objc dynamic var link = ""
  68. @objc dynamic var mimeType = ""
  69. @objc dynamic var fileId: Int = 0
  70. @objc dynamic var view = ""
  71. @objc dynamic var isMimeTypeIcon: Bool = false
  72. override static func primaryKey() -> String {
  73. return "idPrimaryKey"
  74. }
  75. }
  76. class tableActivitySubjectRich: Object {
  77. @objc dynamic var account = ""
  78. @objc dynamic var idActivity: Int = 0
  79. @objc dynamic var idPrimaryKey = ""
  80. @objc dynamic var id = ""
  81. @objc dynamic var key = ""
  82. @objc dynamic var link = ""
  83. @objc dynamic var name = ""
  84. @objc dynamic var path = ""
  85. @objc dynamic var type = ""
  86. override static func primaryKey() -> String {
  87. return "idPrimaryKey"
  88. }
  89. }
  90. extension NCManageDatabase {
  91. func addActivity(_ activities: [NKActivity], account: String) {
  92. do {
  93. let realm = try Realm()
  94. try realm.write {
  95. for activity in activities {
  96. let addObjectActivity = tableActivity()
  97. addObjectActivity.account = account
  98. addObjectActivity.idActivity = activity.idActivity
  99. addObjectActivity.idPrimaryKey = account + String(activity.idActivity)
  100. addObjectActivity.date = activity.date
  101. addObjectActivity.app = activity.app
  102. addObjectActivity.type = activity.type
  103. addObjectActivity.user = activity.user
  104. addObjectActivity.subject = activity.subject
  105. if let subjectRich = activity.subjectRich,
  106. let json = JSON(subjectRich).array {
  107. addObjectActivity.subjectRich = json[0].stringValue
  108. if json.count > 1,
  109. let dict = json[1].dictionary {
  110. for (key, value) in dict {
  111. let addObjectActivitySubjectRich = tableActivitySubjectRich()
  112. let dict = value as JSON
  113. addObjectActivitySubjectRich.account = account
  114. if dict["id"].intValue > 0 {
  115. addObjectActivitySubjectRich.id = String(dict["id"].intValue)
  116. } else {
  117. addObjectActivitySubjectRich.id = dict["id"].stringValue
  118. }
  119. addObjectActivitySubjectRich.name = dict["name"].stringValue
  120. addObjectActivitySubjectRich.idPrimaryKey = account
  121. + String(activity.idActivity)
  122. + addObjectActivitySubjectRich.id
  123. + addObjectActivitySubjectRich.name
  124. addObjectActivitySubjectRich.key = key
  125. addObjectActivitySubjectRich.idActivity = activity.idActivity
  126. addObjectActivitySubjectRich.link = dict["link"].stringValue
  127. addObjectActivitySubjectRich.path = dict["path"].stringValue
  128. addObjectActivitySubjectRich.type = dict["type"].stringValue
  129. realm.add(addObjectActivitySubjectRich, update: .all)
  130. }
  131. }
  132. }
  133. if let previews = activity.previews,
  134. let json = JSON(previews).array {
  135. for preview in json {
  136. let addObjectActivityPreview = tableActivityPreview()
  137. addObjectActivityPreview.account = account
  138. addObjectActivityPreview.idActivity = activity.idActivity
  139. addObjectActivityPreview.fileId = preview["fileId"].intValue
  140. addObjectActivityPreview.filename = preview["filename"].stringValue
  141. addObjectActivityPreview.idPrimaryKey = account + String(activity.idActivity) + String(addObjectActivityPreview.fileId)
  142. addObjectActivityPreview.source = preview["source"].stringValue
  143. addObjectActivityPreview.link = preview["link"].stringValue
  144. addObjectActivityPreview.mimeType = preview["mimeType"].stringValue
  145. addObjectActivityPreview.view = preview["view"].stringValue
  146. addObjectActivityPreview.isMimeTypeIcon = preview["isMimeTypeIcon"].boolValue
  147. realm.add(addObjectActivityPreview, update: .all)
  148. }
  149. }
  150. addObjectActivity.icon = activity.icon
  151. addObjectActivity.link = activity.link
  152. addObjectActivity.message = activity.message
  153. addObjectActivity.objectType = activity.objectType
  154. addObjectActivity.objectId = activity.objectId
  155. addObjectActivity.objectName = activity.objectName
  156. realm.add(addObjectActivity, update: .all)
  157. }
  158. }
  159. } catch let error {
  160. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  161. }
  162. }
  163. func getActivity(predicate: NSPredicate, filterFileId: String?) -> (all: [tableActivity], filter: [tableActivity]) {
  164. do {
  165. let realm = try Realm()
  166. realm.refresh()
  167. let results = realm.objects(tableActivity.self).filter(predicate).sorted(byKeyPath: "idActivity", ascending: false)
  168. let allActivity = Array(results.map(tableActivity.init))
  169. guard let filterFileId = filterFileId else {
  170. return (all: allActivity, filter: allActivity)
  171. }
  172. // comments are loaded seperately, see NCManageDatabase.getComments
  173. let filtered = allActivity.filter({ String($0.objectId) == filterFileId && $0.type != "comments" })
  174. return (all: allActivity, filter: filtered)
  175. } catch let error as NSError {
  176. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  177. }
  178. return([], [])
  179. }
  180. func getActivitySubjectRich(account: String, idActivity: Int, key: String) -> tableActivitySubjectRich? {
  181. do {
  182. let realm = try Realm()
  183. realm.refresh()
  184. let results = realm.objects(tableActivitySubjectRich.self).filter("account == %@ && idActivity == %d && key == %@", account, idActivity, key).first
  185. return results.map { tableActivitySubjectRich.init(value: $0) }
  186. } catch let error as NSError {
  187. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  188. }
  189. return nil
  190. }
  191. func getActivitySubjectRich(account: String, idActivity: Int, id: String) -> tableActivitySubjectRich? {
  192. do {
  193. let realm = try Realm()
  194. realm.refresh()
  195. let results = realm.objects(tableActivitySubjectRich.self).filter("account == %@ && idActivity == %d && id == %@", account, idActivity, id)
  196. var activitySubjectRich = results.first
  197. if results.count == 2 {
  198. for result in results {
  199. if result.key == "newfile" {
  200. activitySubjectRich = result
  201. }
  202. }
  203. }
  204. return activitySubjectRich.map { tableActivitySubjectRich.init(value: $0) }
  205. } catch let error as NSError {
  206. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  207. }
  208. return nil
  209. }
  210. func getActivityPreview(account: String, idActivity: Int, orderKeysId: [String]) -> [tableActivityPreview] {
  211. var results: [tableActivityPreview] = []
  212. do {
  213. let realm = try Realm()
  214. realm.refresh()
  215. for id in orderKeysId {
  216. if let result = realm.objects(tableActivityPreview.self).filter("account == %@ && idActivity == %d && fileId == %d", account, idActivity, Int(id) ?? 0).first {
  217. results.append(result)
  218. }
  219. }
  220. return results
  221. } catch let error as NSError {
  222. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  223. }
  224. return []
  225. }
  226. func updateLatestActivityId(activityFirstKnown: Int, activityLastGiven: Int, account: String) {
  227. do {
  228. let realm = try Realm()
  229. try realm.write {
  230. let newRecentActivity = tableActivityLatestId()
  231. newRecentActivity.activityFirstKnown = activityFirstKnown
  232. newRecentActivity.activityLastGiven = activityLastGiven
  233. newRecentActivity.account = account
  234. realm.add(newRecentActivity, update: .all)
  235. }
  236. } catch {
  237. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  238. }
  239. }
  240. func getLatestActivityId(account: String) -> tableActivityLatestId? {
  241. do {
  242. let realm = try Realm()
  243. realm.refresh()
  244. return realm.objects(tableActivityLatestId.self).filter("account == %@", account).first
  245. } catch let error as NSError {
  246. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  247. }
  248. return nil
  249. }
  250. // MARK: -
  251. // MARK: Table Comments
  252. func addComments(_ comments: [NKComments], account: String, objectId: String) {
  253. do {
  254. let realm = try Realm()
  255. try realm.write {
  256. let results = realm.objects(tableComments.self).filter("account == %@ AND objectId == %@", account, objectId)
  257. realm.delete(results)
  258. for comment in comments {
  259. let object = tableComments()
  260. object.account = account
  261. object.actorDisplayName = comment.actorDisplayName
  262. object.actorId = comment.actorId
  263. object.actorType = comment.actorType
  264. object.creationDateTime = comment.creationDateTime as NSDate
  265. object.isUnread = comment.isUnread
  266. object.message = comment.message
  267. object.messageId = comment.messageId
  268. object.objectId = comment.objectId
  269. object.objectType = comment.objectType
  270. object.path = comment.path
  271. object.verb = comment.verb
  272. realm.add(object, update: .all)
  273. }
  274. }
  275. } catch let error {
  276. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  277. }
  278. }
  279. func getComments(account: String, objectId: String) -> [tableComments] {
  280. do {
  281. let realm = try Realm()
  282. realm.refresh()
  283. let results = realm.objects(tableComments.self).filter("account == %@ AND objectId == %@", account, objectId).sorted(byKeyPath: "creationDateTime", ascending: false)
  284. return Array(results.map(tableComments.init))
  285. } catch let error as NSError {
  286. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  287. }
  288. return []
  289. }
  290. }