NCManageDatabase+Account.swift 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486
  1. //
  2. // NCManageDatabase+Account.swift
  3. // Nextcloud
  4. //
  5. // Created by Henrik Storch on 30.11.21.
  6. // Copyright © 2021 Marino Faggiana. All rights reserved.
  7. //
  8. import Foundation
  9. import RealmSwift
  10. import NCCommunication
  11. extension NCManageDatabase {
  12. @objc func copyObject(account: tableAccount) -> tableAccount {
  13. return tableAccount.init(value: account)
  14. }
  15. @objc func addAccount(_ account: String, urlBase: String, user: String, password: String) {
  16. let realm = try! Realm()
  17. do {
  18. try realm.safeWrite {
  19. let addObject = tableAccount()
  20. addObject.account = account
  21. // Brand
  22. if NCBrandOptions.shared.use_default_auto_upload {
  23. addObject.autoUpload = true
  24. addObject.autoUploadImage = true
  25. addObject.autoUploadVideo = true
  26. addObject.autoUploadWWAnVideo = true
  27. }
  28. CCUtility.setPassword(account, password: password)
  29. addObject.urlBase = urlBase
  30. addObject.user = user
  31. addObject.userId = user
  32. realm.add(addObject, update: .all)
  33. }
  34. } catch let error {
  35. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  36. }
  37. }
  38. @objc func updateAccount(_ account: tableAccount) {
  39. let realm = try! Realm()
  40. do {
  41. try realm.safeWrite {
  42. realm.add(account, update: .all)
  43. }
  44. } catch let error {
  45. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  46. }
  47. }
  48. @objc func deleteAccount(_ account: String) {
  49. let realm = try! Realm()
  50. do {
  51. try realm.safeWrite {
  52. let result = realm.objects(tableAccount.self).filter("account == %@", account)
  53. realm.delete(result)
  54. }
  55. } catch let error {
  56. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  57. }
  58. }
  59. @objc func getActiveAccount() -> tableAccount? {
  60. let realm = try! Realm()
  61. guard let result = realm.objects(tableAccount.self).filter("active == true").first else {
  62. return nil
  63. }
  64. return tableAccount.init(value: result)
  65. }
  66. @objc func getAccounts() -> [String]? {
  67. let realm = try! Realm()
  68. let results = realm.objects(tableAccount.self).sorted(byKeyPath: "account", ascending: true)
  69. if results.count > 0 {
  70. return Array(results.map { $0.account })
  71. }
  72. return nil
  73. }
  74. @objc func getAccount(predicate: NSPredicate) -> tableAccount? {
  75. let realm = try! Realm()
  76. guard let result = realm.objects(tableAccount.self).filter(predicate).first else {
  77. return nil
  78. }
  79. return tableAccount.init(value: result)
  80. }
  81. @objc func getAllAccount() -> [tableAccount] {
  82. let realm = try! Realm()
  83. let sorted = [SortDescriptor(keyPath: "active", ascending: false), SortDescriptor(keyPath: "user", ascending: true)]
  84. let results = realm.objects(tableAccount.self).sorted(by: sorted)
  85. return Array(results.map { tableAccount.init(value: $0) })
  86. }
  87. @objc func getAllAccountOrderAlias() -> [tableAccount] {
  88. let realm = try! Realm()
  89. let sorted = [SortDescriptor(keyPath: "active", ascending: false), SortDescriptor(keyPath: "alias", ascending: true), SortDescriptor(keyPath: "user", ascending: true)]
  90. let results = realm.objects(tableAccount.self).sorted(by: sorted)
  91. return Array(results.map { tableAccount.init(value: $0) })
  92. }
  93. @objc func getAccountAutoUploadFileName() -> String {
  94. let realm = try! Realm()
  95. guard let result = realm.objects(tableAccount.self).filter("active == true").first else {
  96. return ""
  97. }
  98. if result.autoUploadFileName.count > 0 {
  99. return result.autoUploadFileName
  100. } else {
  101. return NCBrandOptions.shared.folderDefaultAutoUpload
  102. }
  103. }
  104. @objc func getAccountAutoUploadDirectory(urlBase: String, account: String) -> String {
  105. let realm = try! Realm()
  106. guard let result = realm.objects(tableAccount.self).filter("active == true").first else {
  107. return ""
  108. }
  109. if result.autoUploadDirectory.count > 0 {
  110. // FIX change webdav -> /dav/files/
  111. if result.autoUploadDirectory.contains("/webdav") {
  112. return NCUtilityFileSystem.shared.getHomeServer(account: account)
  113. } else {
  114. return result.autoUploadDirectory
  115. }
  116. } else {
  117. return NCUtilityFileSystem.shared.getHomeServer(account: account)
  118. }
  119. }
  120. @objc func getAccountAutoUploadPath(urlBase: String, account: String) -> String {
  121. let cameraFileName = self.getAccountAutoUploadFileName()
  122. let cameraDirectory = self.getAccountAutoUploadDirectory(urlBase: urlBase, account: account)
  123. let folderPhotos = CCUtility.stringAppendServerUrl(cameraDirectory, addFileName: cameraFileName)!
  124. return folderPhotos
  125. }
  126. @discardableResult
  127. @objc func setAccountActive(_ account: String) -> tableAccount? {
  128. let realm = try! Realm()
  129. var accountReturn = tableAccount()
  130. do {
  131. try realm.safeWrite {
  132. let results = realm.objects(tableAccount.self)
  133. for result in results {
  134. if result.account == account {
  135. result.active = true
  136. accountReturn = result
  137. } else {
  138. result.active = false
  139. }
  140. }
  141. }
  142. } catch let error {
  143. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  144. return nil
  145. }
  146. return tableAccount.init(value: accountReturn)
  147. }
  148. @objc func removePasswordAccount(_ account: String) {
  149. let realm = try! Realm()
  150. do {
  151. try realm.safeWrite {
  152. if let result = realm.objects(tableAccount.self).filter("account == %@", account).first {
  153. result.password = "********"
  154. }
  155. }
  156. } catch let error {
  157. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  158. }
  159. }
  160. @objc func setAccountAutoUploadProperty(_ property: String, state: Bool) {
  161. let realm = try! Realm()
  162. do {
  163. try realm.safeWrite {
  164. if let result = realm.objects(tableAccount.self).filter("active == true").first {
  165. if (tableAccount().objectSchema.properties.contains { $0.name == property }) {
  166. result[property] = state
  167. }
  168. }
  169. }
  170. } catch let error {
  171. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  172. }
  173. }
  174. @objc func setAccountAutoUploadFileName(_ fileName: String?) {
  175. let realm = try! Realm()
  176. do {
  177. try realm.safeWrite {
  178. if let result = realm.objects(tableAccount.self).filter("active == true").first {
  179. if let fileName = fileName {
  180. result.autoUploadFileName = fileName
  181. } else {
  182. result.autoUploadFileName = self.getAccountAutoUploadFileName()
  183. }
  184. }
  185. }
  186. } catch let error {
  187. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  188. }
  189. }
  190. @objc func setAccountAutoUploadDirectory(_ serverUrl: String?, urlBase: String, account: String) {
  191. let realm = try! Realm()
  192. do {
  193. try realm.safeWrite {
  194. if let result = realm.objects(tableAccount.self).filter("active == true").first {
  195. if let serverUrl = serverUrl {
  196. result.autoUploadDirectory = serverUrl
  197. } else {
  198. result.autoUploadDirectory = self.getAccountAutoUploadDirectory(urlBase: urlBase, account: account)
  199. }
  200. }
  201. }
  202. } catch let error {
  203. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  204. }
  205. }
  206. @objc func setAccountUserProfile(_ userProfile: NCCommunicationUserProfile) -> tableAccount? {
  207. let realm = try! Realm()
  208. var returnAccount = tableAccount()
  209. do {
  210. guard let activeAccount = self.getActiveAccount() else {
  211. return nil
  212. }
  213. try realm.safeWrite {
  214. guard let result = realm.objects(tableAccount.self).filter("account == %@", activeAccount.account).first else {
  215. return
  216. }
  217. result.address = userProfile.address
  218. result.backend = userProfile.backend
  219. result.backendCapabilitiesSetDisplayName = userProfile.backendCapabilitiesSetDisplayName
  220. result.backendCapabilitiesSetPassword = userProfile.backendCapabilitiesSetPassword
  221. result.displayName = userProfile.displayName
  222. result.email = userProfile.email
  223. result.enabled = userProfile.enabled
  224. result.groups = userProfile.groups.joined(separator: ",")
  225. result.language = userProfile.language
  226. result.lastLogin = userProfile.lastLogin
  227. result.locale = userProfile.locale
  228. result.organisation = userProfile.organisation
  229. result.phone = userProfile.phone
  230. result.quota = userProfile.quota
  231. result.quotaFree = userProfile.quotaFree
  232. result.quotaRelative = userProfile.quotaRelative
  233. result.quotaTotal = userProfile.quotaTotal
  234. result.quotaUsed = userProfile.quotaUsed
  235. result.storageLocation = userProfile.storageLocation
  236. result.subadmin = userProfile.subadmin.joined(separator: ",")
  237. result.twitter = userProfile.twitter
  238. result.userId = userProfile.userId
  239. result.website = userProfile.website
  240. returnAccount = result
  241. }
  242. } catch let error {
  243. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  244. }
  245. return tableAccount.init(value: returnAccount)
  246. }
  247. @objc func setAccountUserProfileHC(businessSize: String, businessType: String, city: String, organisation: String, country: String, role: String, zip: String) -> tableAccount? {
  248. let realm = try! Realm()
  249. var returnAccount = tableAccount()
  250. do {
  251. guard let activeAccount = self.getActiveAccount() else {
  252. return nil
  253. }
  254. try realm.safeWrite {
  255. guard let result = realm.objects(tableAccount.self).filter("account == %@", activeAccount.account).first else {
  256. return
  257. }
  258. result.businessSize = businessSize
  259. result.businessType = businessType
  260. result.city = city
  261. result.organisation = organisation
  262. result.country = country
  263. result.role = role
  264. result.zip = zip
  265. returnAccount = result
  266. }
  267. } catch let error {
  268. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  269. }
  270. return tableAccount.init(value: returnAccount)
  271. }
  272. /*
  273. #if !EXTENSION
  274. @objc func setAccountHCFeatures(_ features: HCFeatures) -> tableAccount? {
  275. let realm = try! Realm()
  276. var returnAccount = tableAccount()
  277. do {
  278. guard let account = self.getAccountActive() else {
  279. return nil
  280. }
  281. try realm.write {
  282. guard let result = realm.objects(tableAccount.self).filter("account == %@", account.account).first else {
  283. return
  284. }
  285. result.hcIsTrial = features.isTrial
  286. result.hcTrialExpired = features.trialExpired
  287. result.hcTrialRemainingSec = features.trialRemainingSec
  288. if features.trialEndTime > 0 {
  289. result.hcTrialEndTime = Date(timeIntervalSince1970: features.trialEndTime) as NSDate
  290. } else {
  291. result.hcTrialEndTime = nil
  292. }
  293. result.hcAccountRemoveExpired = features.accountRemoveExpired
  294. result.hcAccountRemoveRemainingSec = features.accountRemoveRemainingSec
  295. if features.accountRemoveTime > 0 {
  296. result.hcAccountRemoveTime = Date(timeIntervalSince1970: features.accountRemoveTime) as NSDate
  297. } else {
  298. result.hcAccountRemoveTime = nil
  299. }
  300. result.hcNextGroupExpirationGroup = features.nextGroupExpirationGroup
  301. result.hcNextGroupExpirationGroupExpired = features.nextGroupExpirationGroupExpired
  302. if features.nextGroupExpirationExpiresTime > 0 {
  303. result.hcNextGroupExpirationExpiresTime = Date(timeIntervalSince1970: features.nextGroupExpirationExpiresTime) as NSDate
  304. } else {
  305. result.hcNextGroupExpirationExpiresTime = nil
  306. }
  307. result.hcNextGroupExpirationExpires = features.nextGroupExpirationExpires
  308. returnAccount = result
  309. }
  310. } catch let error {
  311. print("[LOG] Could not write to database: ", error)
  312. }
  313. return tableAccount.init(value: returnAccount)
  314. }
  315. #endif
  316. */
  317. @objc func setAccountMediaPath(_ path: String, account: String) {
  318. let realm = try! Realm()
  319. do {
  320. try realm.safeWrite {
  321. if let result = realm.objects(tableAccount.self).filter("account == %@", account).first {
  322. result.mediaPath = path
  323. }
  324. }
  325. } catch let error {
  326. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  327. }
  328. }
  329. @objc func setAccountUserStatus(userStatusClearAt: NSDate?, userStatusIcon: String?, userStatusMessage: String?, userStatusMessageId: String?, userStatusMessageIsPredefined: Bool, userStatusStatus: String?, userStatusStatusIsUserDefined: Bool, account: String) {
  330. let realm = try! Realm()
  331. do {
  332. try realm.safeWrite {
  333. if let result = realm.objects(tableAccount.self).filter("account == %@", account).first {
  334. result.userStatusClearAt = userStatusClearAt
  335. result.userStatusIcon = userStatusIcon
  336. result.userStatusMessage = userStatusMessage
  337. result.userStatusMessageId = userStatusMessageId
  338. result.userStatusMessageIsPredefined = userStatusMessageIsPredefined
  339. result.userStatusStatus = userStatusStatus
  340. result.userStatusStatusIsUserDefined = userStatusStatusIsUserDefined
  341. }
  342. }
  343. } catch let error {
  344. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  345. }
  346. }
  347. @objc func setAccountAlias(_ alias: String?) {
  348. let realm = try! Realm()
  349. let alias = alias?.trimmingCharacters(in: .whitespacesAndNewlines)
  350. do {
  351. try realm.safeWrite {
  352. if let result = realm.objects(tableAccount.self).filter("active == true").first {
  353. if let alias = alias {
  354. result.alias = alias
  355. } else {
  356. result.alias = ""
  357. }
  358. }
  359. }
  360. } catch let error {
  361. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  362. }
  363. }
  364. @objc func setAccountColorFiles(lightColorBackground: String, darkColorBackground: String) {
  365. let realm = try! Realm()
  366. do {
  367. try realm.safeWrite {
  368. if let result = realm.objects(tableAccount.self).filter("active == true").first {
  369. result.lightColorBackground = lightColorBackground
  370. result.darkColorBackground = darkColorBackground
  371. }
  372. }
  373. } catch let error {
  374. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  375. }
  376. }
  377. }