memory.go 67 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593
  1. // Copyright (C) 2019-2022 Nicola Murino
  2. //
  3. // This program is free software: you can redistribute it and/or modify
  4. // it under the terms of the GNU Affero General Public License as published
  5. // by the Free Software Foundation, version 3.
  6. //
  7. // This program is distributed in the hope that it will be useful,
  8. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. // GNU Affero General Public License for more details.
  11. //
  12. // You should have received a copy of the GNU Affero General Public License
  13. // along with this program. If not, see <https://www.gnu.org/licenses/>.
  14. package dataprovider
  15. import (
  16. "crypto/x509"
  17. "errors"
  18. "fmt"
  19. "os"
  20. "path/filepath"
  21. "sort"
  22. "sync"
  23. "time"
  24. "github.com/drakkan/sftpgo/v2/internal/logger"
  25. "github.com/drakkan/sftpgo/v2/internal/util"
  26. "github.com/drakkan/sftpgo/v2/internal/vfs"
  27. )
  28. var (
  29. errMemoryProviderClosed = errors.New("memory provider is closed")
  30. )
  31. type memoryProviderHandle struct {
  32. // configuration file to use for loading users
  33. configFile string
  34. sync.Mutex
  35. isClosed bool
  36. // slice with ordered usernames
  37. usernames []string
  38. // map for users, username is the key
  39. users map[string]User
  40. // slice with ordered group names
  41. groupnames []string
  42. // map for group, group name is the key
  43. groups map[string]Group
  44. // map for virtual folders, folder name is the key
  45. vfolders map[string]vfs.BaseVirtualFolder
  46. // slice with ordered folder names
  47. vfoldersNames []string
  48. // map for admins, username is the key
  49. admins map[string]Admin
  50. // slice with ordered admins
  51. adminsUsernames []string
  52. // map for API keys, keyID is the key
  53. apiKeys map[string]APIKey
  54. // slice with ordered API keys KeyID
  55. apiKeysIDs []string
  56. // map for shares, shareID is the key
  57. shares map[string]Share
  58. // slice with ordered shares shareID
  59. sharesIDs []string
  60. // map for event actions, name is the key
  61. actions map[string]BaseEventAction
  62. // slice with ordered actions
  63. actionsNames []string
  64. // map for event actions, name is the key
  65. rules map[string]EventRule
  66. // slice with ordered rules
  67. rulesNames []string
  68. }
  69. // MemoryProvider defines the auth provider for a memory store
  70. type MemoryProvider struct {
  71. dbHandle *memoryProviderHandle
  72. }
  73. func initializeMemoryProvider(basePath string) {
  74. configFile := ""
  75. if util.IsFileInputValid(config.Name) {
  76. configFile = config.Name
  77. if !filepath.IsAbs(configFile) {
  78. configFile = filepath.Join(basePath, configFile)
  79. }
  80. }
  81. provider = &MemoryProvider{
  82. dbHandle: &memoryProviderHandle{
  83. isClosed: false,
  84. usernames: []string{},
  85. users: make(map[string]User),
  86. groupnames: []string{},
  87. groups: make(map[string]Group),
  88. vfolders: make(map[string]vfs.BaseVirtualFolder),
  89. vfoldersNames: []string{},
  90. admins: make(map[string]Admin),
  91. adminsUsernames: []string{},
  92. apiKeys: make(map[string]APIKey),
  93. apiKeysIDs: []string{},
  94. shares: make(map[string]Share),
  95. sharesIDs: []string{},
  96. actions: make(map[string]BaseEventAction),
  97. actionsNames: []string{},
  98. rules: make(map[string]EventRule),
  99. rulesNames: []string{},
  100. configFile: configFile,
  101. },
  102. }
  103. if err := provider.reloadConfig(); err != nil {
  104. logger.Error(logSender, "", "unable to load initial data: %v", err)
  105. logger.ErrorToConsole("unable to load initial data: %v", err)
  106. }
  107. }
  108. func (p *MemoryProvider) checkAvailability() error {
  109. p.dbHandle.Lock()
  110. defer p.dbHandle.Unlock()
  111. if p.dbHandle.isClosed {
  112. return errMemoryProviderClosed
  113. }
  114. return nil
  115. }
  116. func (p *MemoryProvider) close() error {
  117. p.dbHandle.Lock()
  118. defer p.dbHandle.Unlock()
  119. if p.dbHandle.isClosed {
  120. return errMemoryProviderClosed
  121. }
  122. p.dbHandle.isClosed = true
  123. return nil
  124. }
  125. func (p *MemoryProvider) validateUserAndTLSCert(username, protocol string, tlsCert *x509.Certificate) (User, error) {
  126. var user User
  127. if tlsCert == nil {
  128. return user, errors.New("TLS certificate cannot be null or empty")
  129. }
  130. user, err := p.userExists(username)
  131. if err != nil {
  132. providerLog(logger.LevelWarn, "error authenticating user %#v: %v", username, err)
  133. return user, err
  134. }
  135. return checkUserAndTLSCertificate(&user, protocol, tlsCert)
  136. }
  137. func (p *MemoryProvider) validateUserAndPass(username, password, ip, protocol string) (User, error) {
  138. user, err := p.userExists(username)
  139. if err != nil {
  140. providerLog(logger.LevelWarn, "error authenticating user %#v: %v", username, err)
  141. return user, err
  142. }
  143. return checkUserAndPass(&user, password, ip, protocol)
  144. }
  145. func (p *MemoryProvider) validateUserAndPubKey(username string, pubKey []byte, isSSHCert bool) (User, string, error) {
  146. var user User
  147. if len(pubKey) == 0 {
  148. return user, "", errors.New("credentials cannot be null or empty")
  149. }
  150. user, err := p.userExists(username)
  151. if err != nil {
  152. providerLog(logger.LevelWarn, "error authenticating user %#v: %v", username, err)
  153. return user, "", err
  154. }
  155. return checkUserAndPubKey(&user, pubKey, isSSHCert)
  156. }
  157. func (p *MemoryProvider) validateAdminAndPass(username, password, ip string) (Admin, error) {
  158. admin, err := p.adminExists(username)
  159. if err != nil {
  160. providerLog(logger.LevelWarn, "error authenticating admin %#v: %v", username, err)
  161. return admin, ErrInvalidCredentials
  162. }
  163. err = admin.checkUserAndPass(password, ip)
  164. return admin, err
  165. }
  166. func (p *MemoryProvider) updateAPIKeyLastUse(keyID string) error {
  167. p.dbHandle.Lock()
  168. defer p.dbHandle.Unlock()
  169. if p.dbHandle.isClosed {
  170. return errMemoryProviderClosed
  171. }
  172. apiKey, err := p.apiKeyExistsInternal(keyID)
  173. if err != nil {
  174. return err
  175. }
  176. apiKey.LastUseAt = util.GetTimeAsMsSinceEpoch(time.Now())
  177. p.dbHandle.apiKeys[apiKey.KeyID] = apiKey
  178. return nil
  179. }
  180. func (p *MemoryProvider) setUpdatedAt(username string) {
  181. p.dbHandle.Lock()
  182. defer p.dbHandle.Unlock()
  183. if p.dbHandle.isClosed {
  184. return
  185. }
  186. user, err := p.userExistsInternal(username)
  187. if err != nil {
  188. return
  189. }
  190. user.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  191. p.dbHandle.users[user.Username] = user
  192. setLastUserUpdate()
  193. }
  194. func (p *MemoryProvider) updateLastLogin(username string) error {
  195. p.dbHandle.Lock()
  196. defer p.dbHandle.Unlock()
  197. if p.dbHandle.isClosed {
  198. return errMemoryProviderClosed
  199. }
  200. user, err := p.userExistsInternal(username)
  201. if err != nil {
  202. return err
  203. }
  204. user.LastLogin = util.GetTimeAsMsSinceEpoch(time.Now())
  205. p.dbHandle.users[user.Username] = user
  206. return nil
  207. }
  208. func (p *MemoryProvider) updateAdminLastLogin(username string) error {
  209. p.dbHandle.Lock()
  210. defer p.dbHandle.Unlock()
  211. if p.dbHandle.isClosed {
  212. return errMemoryProviderClosed
  213. }
  214. admin, err := p.adminExistsInternal(username)
  215. if err != nil {
  216. return err
  217. }
  218. admin.LastLogin = util.GetTimeAsMsSinceEpoch(time.Now())
  219. p.dbHandle.admins[admin.Username] = admin
  220. return nil
  221. }
  222. func (p *MemoryProvider) updateTransferQuota(username string, uploadSize, downloadSize int64, reset bool) error {
  223. p.dbHandle.Lock()
  224. defer p.dbHandle.Unlock()
  225. if p.dbHandle.isClosed {
  226. return errMemoryProviderClosed
  227. }
  228. user, err := p.userExistsInternal(username)
  229. if err != nil {
  230. providerLog(logger.LevelError, "unable to update transfer quota for user %#v error: %v", username, err)
  231. return err
  232. }
  233. if reset {
  234. user.UsedUploadDataTransfer = uploadSize
  235. user.UsedDownloadDataTransfer = downloadSize
  236. } else {
  237. user.UsedUploadDataTransfer += uploadSize
  238. user.UsedDownloadDataTransfer += downloadSize
  239. }
  240. user.LastQuotaUpdate = util.GetTimeAsMsSinceEpoch(time.Now())
  241. providerLog(logger.LevelDebug, "transfer quota updated for user %#v, ul increment: %v dl increment: %v is reset? %v",
  242. username, uploadSize, downloadSize, reset)
  243. p.dbHandle.users[user.Username] = user
  244. return nil
  245. }
  246. func (p *MemoryProvider) updateQuota(username string, filesAdd int, sizeAdd int64, reset bool) error {
  247. p.dbHandle.Lock()
  248. defer p.dbHandle.Unlock()
  249. if p.dbHandle.isClosed {
  250. return errMemoryProviderClosed
  251. }
  252. user, err := p.userExistsInternal(username)
  253. if err != nil {
  254. providerLog(logger.LevelError, "unable to update quota for user %#v error: %v", username, err)
  255. return err
  256. }
  257. if reset {
  258. user.UsedQuotaSize = sizeAdd
  259. user.UsedQuotaFiles = filesAdd
  260. } else {
  261. user.UsedQuotaSize += sizeAdd
  262. user.UsedQuotaFiles += filesAdd
  263. }
  264. user.LastQuotaUpdate = util.GetTimeAsMsSinceEpoch(time.Now())
  265. providerLog(logger.LevelDebug, "quota updated for user %#v, files increment: %v size increment: %v is reset? %v",
  266. username, filesAdd, sizeAdd, reset)
  267. p.dbHandle.users[user.Username] = user
  268. return nil
  269. }
  270. func (p *MemoryProvider) getUsedQuota(username string) (int, int64, int64, int64, error) {
  271. p.dbHandle.Lock()
  272. defer p.dbHandle.Unlock()
  273. if p.dbHandle.isClosed {
  274. return 0, 0, 0, 0, errMemoryProviderClosed
  275. }
  276. user, err := p.userExistsInternal(username)
  277. if err != nil {
  278. providerLog(logger.LevelError, "unable to get quota for user %#v error: %v", username, err)
  279. return 0, 0, 0, 0, err
  280. }
  281. return user.UsedQuotaFiles, user.UsedQuotaSize, user.UsedUploadDataTransfer, user.UsedDownloadDataTransfer, err
  282. }
  283. func (p *MemoryProvider) addUser(user *User) error {
  284. // we can query virtual folder while validating a user
  285. // so we have to check without holding the lock
  286. err := ValidateUser(user)
  287. if err != nil {
  288. return err
  289. }
  290. p.dbHandle.Lock()
  291. defer p.dbHandle.Unlock()
  292. if p.dbHandle.isClosed {
  293. return errMemoryProviderClosed
  294. }
  295. _, err = p.userExistsInternal(user.Username)
  296. if err == nil {
  297. return fmt.Errorf("username %#v already exists", user.Username)
  298. }
  299. user.ID = p.getNextID()
  300. user.LastQuotaUpdate = 0
  301. user.UsedQuotaSize = 0
  302. user.UsedQuotaFiles = 0
  303. user.UsedUploadDataTransfer = 0
  304. user.UsedDownloadDataTransfer = 0
  305. user.LastLogin = 0
  306. user.FirstUpload = 0
  307. user.FirstDownload = 0
  308. user.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  309. user.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  310. user.VirtualFolders = p.joinUserVirtualFoldersFields(user)
  311. for idx := range user.Groups {
  312. if err = p.addUserFromGroupMapping(user.Username, user.Groups[idx].Name); err != nil {
  313. return err
  314. }
  315. }
  316. p.dbHandle.users[user.Username] = user.getACopy()
  317. p.dbHandle.usernames = append(p.dbHandle.usernames, user.Username)
  318. sort.Strings(p.dbHandle.usernames)
  319. return nil
  320. }
  321. func (p *MemoryProvider) updateUser(user *User) error {
  322. // we can query virtual folder while validating a user
  323. // so we have to check without holding the lock
  324. err := ValidateUser(user)
  325. if err != nil {
  326. return err
  327. }
  328. p.dbHandle.Lock()
  329. defer p.dbHandle.Unlock()
  330. if p.dbHandle.isClosed {
  331. return errMemoryProviderClosed
  332. }
  333. u, err := p.userExistsInternal(user.Username)
  334. if err != nil {
  335. return err
  336. }
  337. for _, oldFolder := range u.VirtualFolders {
  338. p.removeRelationFromFolderMapping(oldFolder.Name, u.Username, "")
  339. }
  340. for idx := range u.Groups {
  341. if err = p.removeUserFromGroupMapping(u.Username, u.Groups[idx].Name); err != nil {
  342. return err
  343. }
  344. }
  345. user.VirtualFolders = p.joinUserVirtualFoldersFields(user)
  346. for idx := range user.Groups {
  347. if err = p.addUserFromGroupMapping(user.Username, user.Groups[idx].Name); err != nil {
  348. return err
  349. }
  350. }
  351. user.LastQuotaUpdate = u.LastQuotaUpdate
  352. user.UsedQuotaSize = u.UsedQuotaSize
  353. user.UsedQuotaFiles = u.UsedQuotaFiles
  354. user.UsedUploadDataTransfer = u.UsedUploadDataTransfer
  355. user.UsedDownloadDataTransfer = u.UsedDownloadDataTransfer
  356. user.LastLogin = u.LastLogin
  357. user.FirstDownload = u.FirstDownload
  358. user.FirstUpload = u.FirstUpload
  359. user.CreatedAt = u.CreatedAt
  360. user.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  361. user.ID = u.ID
  362. // pre-login and external auth hook will use the passed *user so save a copy
  363. p.dbHandle.users[user.Username] = user.getACopy()
  364. setLastUserUpdate()
  365. return nil
  366. }
  367. func (p *MemoryProvider) deleteUser(user User, softDelete bool) error {
  368. p.dbHandle.Lock()
  369. defer p.dbHandle.Unlock()
  370. if p.dbHandle.isClosed {
  371. return errMemoryProviderClosed
  372. }
  373. u, err := p.userExistsInternal(user.Username)
  374. if err != nil {
  375. return err
  376. }
  377. for _, oldFolder := range u.VirtualFolders {
  378. p.removeRelationFromFolderMapping(oldFolder.Name, u.Username, "")
  379. }
  380. for idx := range u.Groups {
  381. if err = p.removeUserFromGroupMapping(u.Username, u.Groups[idx].Name); err != nil {
  382. return err
  383. }
  384. }
  385. delete(p.dbHandle.users, user.Username)
  386. // this could be more efficient
  387. p.dbHandle.usernames = make([]string, 0, len(p.dbHandle.users))
  388. for username := range p.dbHandle.users {
  389. p.dbHandle.usernames = append(p.dbHandle.usernames, username)
  390. }
  391. sort.Strings(p.dbHandle.usernames)
  392. p.deleteAPIKeysWithUser(user.Username)
  393. p.deleteSharesWithUser(user.Username)
  394. return nil
  395. }
  396. func (p *MemoryProvider) updateUserPassword(username, password string) error {
  397. p.dbHandle.Lock()
  398. defer p.dbHandle.Unlock()
  399. if p.dbHandle.isClosed {
  400. return errMemoryProviderClosed
  401. }
  402. user, err := p.userExistsInternal(username)
  403. if err != nil {
  404. return err
  405. }
  406. user.Password = password
  407. p.dbHandle.users[username] = user
  408. return nil
  409. }
  410. func (p *MemoryProvider) dumpUsers() ([]User, error) {
  411. p.dbHandle.Lock()
  412. defer p.dbHandle.Unlock()
  413. users := make([]User, 0, len(p.dbHandle.usernames))
  414. var err error
  415. if p.dbHandle.isClosed {
  416. return users, errMemoryProviderClosed
  417. }
  418. for _, username := range p.dbHandle.usernames {
  419. u := p.dbHandle.users[username]
  420. user := u.getACopy()
  421. p.addVirtualFoldersToUser(&user)
  422. users = append(users, user)
  423. }
  424. return users, err
  425. }
  426. func (p *MemoryProvider) dumpFolders() ([]vfs.BaseVirtualFolder, error) {
  427. p.dbHandle.Lock()
  428. defer p.dbHandle.Unlock()
  429. folders := make([]vfs.BaseVirtualFolder, 0, len(p.dbHandle.vfoldersNames))
  430. if p.dbHandle.isClosed {
  431. return folders, errMemoryProviderClosed
  432. }
  433. for _, f := range p.dbHandle.vfolders {
  434. folders = append(folders, f)
  435. }
  436. return folders, nil
  437. }
  438. func (p *MemoryProvider) getRecentlyUpdatedUsers(after int64) ([]User, error) {
  439. if getLastUserUpdate() < after {
  440. return nil, nil
  441. }
  442. p.dbHandle.Lock()
  443. defer p.dbHandle.Unlock()
  444. if p.dbHandle.isClosed {
  445. return nil, errMemoryProviderClosed
  446. }
  447. users := make([]User, 0, 10)
  448. for _, username := range p.dbHandle.usernames {
  449. u := p.dbHandle.users[username]
  450. if u.UpdatedAt < after {
  451. continue
  452. }
  453. user := u.getACopy()
  454. p.addVirtualFoldersToUser(&user)
  455. if len(user.Groups) > 0 {
  456. groupMapping := make(map[string]Group)
  457. for idx := range user.Groups {
  458. group, err := p.groupExistsInternal(user.Groups[idx].Name)
  459. if err != nil {
  460. continue
  461. }
  462. groupMapping[group.Name] = group
  463. }
  464. user.applyGroupSettings(groupMapping)
  465. }
  466. user.SetEmptySecretsIfNil()
  467. users = append(users, user)
  468. }
  469. return users, nil
  470. }
  471. func (p *MemoryProvider) getUsersForQuotaCheck(toFetch map[string]bool) ([]User, error) {
  472. users := make([]User, 0, 30)
  473. p.dbHandle.Lock()
  474. defer p.dbHandle.Unlock()
  475. if p.dbHandle.isClosed {
  476. return users, errMemoryProviderClosed
  477. }
  478. for _, username := range p.dbHandle.usernames {
  479. if needFolders, ok := toFetch[username]; ok {
  480. u := p.dbHandle.users[username]
  481. user := u.getACopy()
  482. if needFolders {
  483. p.addVirtualFoldersToUser(&user)
  484. }
  485. if len(user.Groups) > 0 {
  486. groupMapping := make(map[string]Group)
  487. for idx := range user.Groups {
  488. group, err := p.groupExistsInternal(user.Groups[idx].Name)
  489. if err != nil {
  490. continue
  491. }
  492. groupMapping[group.Name] = group
  493. }
  494. user.applyGroupSettings(groupMapping)
  495. }
  496. user.SetEmptySecretsIfNil()
  497. user.PrepareForRendering()
  498. users = append(users, user)
  499. }
  500. }
  501. return users, nil
  502. }
  503. func (p *MemoryProvider) getUsers(limit int, offset int, order string) ([]User, error) {
  504. users := make([]User, 0, limit)
  505. var err error
  506. p.dbHandle.Lock()
  507. defer p.dbHandle.Unlock()
  508. if p.dbHandle.isClosed {
  509. return users, errMemoryProviderClosed
  510. }
  511. if limit <= 0 {
  512. return users, err
  513. }
  514. itNum := 0
  515. if order == OrderASC {
  516. for _, username := range p.dbHandle.usernames {
  517. itNum++
  518. if itNum <= offset {
  519. continue
  520. }
  521. u := p.dbHandle.users[username]
  522. user := u.getACopy()
  523. p.addVirtualFoldersToUser(&user)
  524. user.PrepareForRendering()
  525. users = append(users, user)
  526. if len(users) >= limit {
  527. break
  528. }
  529. }
  530. } else {
  531. for i := len(p.dbHandle.usernames) - 1; i >= 0; i-- {
  532. itNum++
  533. if itNum <= offset {
  534. continue
  535. }
  536. username := p.dbHandle.usernames[i]
  537. u := p.dbHandle.users[username]
  538. user := u.getACopy()
  539. p.addVirtualFoldersToUser(&user)
  540. user.PrepareForRendering()
  541. users = append(users, user)
  542. if len(users) >= limit {
  543. break
  544. }
  545. }
  546. }
  547. return users, err
  548. }
  549. func (p *MemoryProvider) userExists(username string) (User, error) {
  550. p.dbHandle.Lock()
  551. defer p.dbHandle.Unlock()
  552. if p.dbHandle.isClosed {
  553. return User{}, errMemoryProviderClosed
  554. }
  555. user, err := p.userExistsInternal(username)
  556. if err != nil {
  557. return user, err
  558. }
  559. p.addVirtualFoldersToUser(&user)
  560. return user, nil
  561. }
  562. func (p *MemoryProvider) userExistsInternal(username string) (User, error) {
  563. if val, ok := p.dbHandle.users[username]; ok {
  564. return val.getACopy(), nil
  565. }
  566. return User{}, util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", username))
  567. }
  568. func (p *MemoryProvider) groupExistsInternal(name string) (Group, error) {
  569. if val, ok := p.dbHandle.groups[name]; ok {
  570. return val.getACopy(), nil
  571. }
  572. return Group{}, util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", name))
  573. }
  574. func (p *MemoryProvider) actionExistsInternal(name string) (BaseEventAction, error) {
  575. if val, ok := p.dbHandle.actions[name]; ok {
  576. return val.getACopy(), nil
  577. }
  578. return BaseEventAction{}, util.NewRecordNotFoundError(fmt.Sprintf("event action %q does not exist", name))
  579. }
  580. func (p *MemoryProvider) ruleExistsInternal(name string) (EventRule, error) {
  581. if val, ok := p.dbHandle.rules[name]; ok {
  582. return val.getACopy(), nil
  583. }
  584. return EventRule{}, util.NewRecordNotFoundError(fmt.Sprintf("event rule %q does not exist", name))
  585. }
  586. func (p *MemoryProvider) addAdmin(admin *Admin) error {
  587. p.dbHandle.Lock()
  588. defer p.dbHandle.Unlock()
  589. if p.dbHandle.isClosed {
  590. return errMemoryProviderClosed
  591. }
  592. err := admin.validate()
  593. if err != nil {
  594. return err
  595. }
  596. _, err = p.adminExistsInternal(admin.Username)
  597. if err == nil {
  598. return fmt.Errorf("admin %#v already exists", admin.Username)
  599. }
  600. admin.ID = p.getNextAdminID()
  601. admin.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  602. admin.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  603. admin.LastLogin = 0
  604. p.dbHandle.admins[admin.Username] = admin.getACopy()
  605. p.dbHandle.adminsUsernames = append(p.dbHandle.adminsUsernames, admin.Username)
  606. sort.Strings(p.dbHandle.adminsUsernames)
  607. return nil
  608. }
  609. func (p *MemoryProvider) updateAdmin(admin *Admin) error {
  610. p.dbHandle.Lock()
  611. defer p.dbHandle.Unlock()
  612. if p.dbHandle.isClosed {
  613. return errMemoryProviderClosed
  614. }
  615. err := admin.validate()
  616. if err != nil {
  617. return err
  618. }
  619. a, err := p.adminExistsInternal(admin.Username)
  620. if err != nil {
  621. return err
  622. }
  623. admin.ID = a.ID
  624. admin.CreatedAt = a.CreatedAt
  625. admin.LastLogin = a.LastLogin
  626. admin.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  627. p.dbHandle.admins[admin.Username] = admin.getACopy()
  628. return nil
  629. }
  630. func (p *MemoryProvider) deleteAdmin(admin Admin) error {
  631. p.dbHandle.Lock()
  632. defer p.dbHandle.Unlock()
  633. if p.dbHandle.isClosed {
  634. return errMemoryProviderClosed
  635. }
  636. _, err := p.adminExistsInternal(admin.Username)
  637. if err != nil {
  638. return err
  639. }
  640. delete(p.dbHandle.admins, admin.Username)
  641. // this could be more efficient
  642. p.dbHandle.adminsUsernames = make([]string, 0, len(p.dbHandle.admins))
  643. for username := range p.dbHandle.admins {
  644. p.dbHandle.adminsUsernames = append(p.dbHandle.adminsUsernames, username)
  645. }
  646. sort.Strings(p.dbHandle.adminsUsernames)
  647. p.deleteAPIKeysWithAdmin(admin.Username)
  648. return nil
  649. }
  650. func (p *MemoryProvider) adminExists(username string) (Admin, error) {
  651. p.dbHandle.Lock()
  652. defer p.dbHandle.Unlock()
  653. if p.dbHandle.isClosed {
  654. return Admin{}, errMemoryProviderClosed
  655. }
  656. return p.adminExistsInternal(username)
  657. }
  658. func (p *MemoryProvider) adminExistsInternal(username string) (Admin, error) {
  659. if val, ok := p.dbHandle.admins[username]; ok {
  660. return val.getACopy(), nil
  661. }
  662. return Admin{}, util.NewRecordNotFoundError(fmt.Sprintf("admin %#v does not exist", username))
  663. }
  664. func (p *MemoryProvider) dumpAdmins() ([]Admin, error) {
  665. p.dbHandle.Lock()
  666. defer p.dbHandle.Unlock()
  667. admins := make([]Admin, 0, len(p.dbHandle.admins))
  668. if p.dbHandle.isClosed {
  669. return admins, errMemoryProviderClosed
  670. }
  671. for _, admin := range p.dbHandle.admins {
  672. admins = append(admins, admin)
  673. }
  674. return admins, nil
  675. }
  676. func (p *MemoryProvider) getAdmins(limit int, offset int, order string) ([]Admin, error) {
  677. admins := make([]Admin, 0, limit)
  678. p.dbHandle.Lock()
  679. defer p.dbHandle.Unlock()
  680. if p.dbHandle.isClosed {
  681. return admins, errMemoryProviderClosed
  682. }
  683. if limit <= 0 {
  684. return admins, nil
  685. }
  686. itNum := 0
  687. if order == OrderASC {
  688. for _, username := range p.dbHandle.adminsUsernames {
  689. itNum++
  690. if itNum <= offset {
  691. continue
  692. }
  693. a := p.dbHandle.admins[username]
  694. admin := a.getACopy()
  695. admin.HideConfidentialData()
  696. admins = append(admins, admin)
  697. if len(admins) >= limit {
  698. break
  699. }
  700. }
  701. } else {
  702. for i := len(p.dbHandle.adminsUsernames) - 1; i >= 0; i-- {
  703. itNum++
  704. if itNum <= offset {
  705. continue
  706. }
  707. username := p.dbHandle.adminsUsernames[i]
  708. a := p.dbHandle.admins[username]
  709. admin := a.getACopy()
  710. admin.HideConfidentialData()
  711. admins = append(admins, admin)
  712. if len(admins) >= limit {
  713. break
  714. }
  715. }
  716. }
  717. return admins, nil
  718. }
  719. func (p *MemoryProvider) updateFolderQuota(name string, filesAdd int, sizeAdd int64, reset bool) error {
  720. p.dbHandle.Lock()
  721. defer p.dbHandle.Unlock()
  722. if p.dbHandle.isClosed {
  723. return errMemoryProviderClosed
  724. }
  725. folder, err := p.folderExistsInternal(name)
  726. if err != nil {
  727. providerLog(logger.LevelError, "unable to update quota for folder %#v error: %v", name, err)
  728. return err
  729. }
  730. if reset {
  731. folder.UsedQuotaSize = sizeAdd
  732. folder.UsedQuotaFiles = filesAdd
  733. } else {
  734. folder.UsedQuotaSize += sizeAdd
  735. folder.UsedQuotaFiles += filesAdd
  736. }
  737. folder.LastQuotaUpdate = util.GetTimeAsMsSinceEpoch(time.Now())
  738. p.dbHandle.vfolders[name] = folder
  739. return nil
  740. }
  741. func (p *MemoryProvider) getGroups(limit, offset int, order string, minimal bool) ([]Group, error) {
  742. p.dbHandle.Lock()
  743. defer p.dbHandle.Unlock()
  744. if p.dbHandle.isClosed {
  745. return nil, errMemoryProviderClosed
  746. }
  747. if limit <= 0 {
  748. return nil, nil
  749. }
  750. groups := make([]Group, 0, limit)
  751. itNum := 0
  752. if order == OrderASC {
  753. for _, name := range p.dbHandle.groupnames {
  754. itNum++
  755. if itNum <= offset {
  756. continue
  757. }
  758. g := p.dbHandle.groups[name]
  759. group := g.getACopy()
  760. p.addVirtualFoldersToGroup(&group)
  761. group.PrepareForRendering()
  762. groups = append(groups, group)
  763. if len(groups) >= limit {
  764. break
  765. }
  766. }
  767. } else {
  768. for i := len(p.dbHandle.groupnames) - 1; i >= 0; i-- {
  769. itNum++
  770. if itNum <= offset {
  771. continue
  772. }
  773. name := p.dbHandle.groupnames[i]
  774. g := p.dbHandle.groups[name]
  775. group := g.getACopy()
  776. p.addVirtualFoldersToGroup(&group)
  777. group.PrepareForRendering()
  778. groups = append(groups, group)
  779. if len(groups) >= limit {
  780. break
  781. }
  782. }
  783. }
  784. return groups, nil
  785. }
  786. func (p *MemoryProvider) getGroupsWithNames(names []string) ([]Group, error) {
  787. p.dbHandle.Lock()
  788. defer p.dbHandle.Unlock()
  789. if p.dbHandle.isClosed {
  790. return nil, errMemoryProviderClosed
  791. }
  792. groups := make([]Group, 0, len(names))
  793. for _, name := range names {
  794. if val, ok := p.dbHandle.groups[name]; ok {
  795. group := val.getACopy()
  796. p.addVirtualFoldersToGroup(&group)
  797. groups = append(groups, group)
  798. }
  799. }
  800. return groups, nil
  801. }
  802. func (p *MemoryProvider) getUsersInGroups(names []string) ([]string, error) {
  803. p.dbHandle.Lock()
  804. defer p.dbHandle.Unlock()
  805. if p.dbHandle.isClosed {
  806. return nil, errMemoryProviderClosed
  807. }
  808. var users []string
  809. for _, name := range names {
  810. if val, ok := p.dbHandle.groups[name]; ok {
  811. group := val.getACopy()
  812. users = append(users, group.Users...)
  813. }
  814. }
  815. return users, nil
  816. }
  817. func (p *MemoryProvider) groupExists(name string) (Group, error) {
  818. p.dbHandle.Lock()
  819. defer p.dbHandle.Unlock()
  820. if p.dbHandle.isClosed {
  821. return Group{}, errMemoryProviderClosed
  822. }
  823. group, err := p.groupExistsInternal(name)
  824. if err != nil {
  825. return group, err
  826. }
  827. p.addVirtualFoldersToGroup(&group)
  828. return group, nil
  829. }
  830. func (p *MemoryProvider) addGroup(group *Group) error {
  831. if err := group.validate(); err != nil {
  832. return err
  833. }
  834. p.dbHandle.Lock()
  835. defer p.dbHandle.Unlock()
  836. if p.dbHandle.isClosed {
  837. return errMemoryProviderClosed
  838. }
  839. _, err := p.groupExistsInternal(group.Name)
  840. if err == nil {
  841. return fmt.Errorf("group %#v already exists", group.Name)
  842. }
  843. group.ID = p.getNextGroupID()
  844. group.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  845. group.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  846. group.VirtualFolders = p.joinGroupVirtualFoldersFields(group)
  847. p.dbHandle.groups[group.Name] = group.getACopy()
  848. p.dbHandle.groupnames = append(p.dbHandle.groupnames, group.Name)
  849. sort.Strings(p.dbHandle.groupnames)
  850. return nil
  851. }
  852. func (p *MemoryProvider) updateGroup(group *Group) error {
  853. if err := group.validate(); err != nil {
  854. return err
  855. }
  856. p.dbHandle.Lock()
  857. defer p.dbHandle.Unlock()
  858. if p.dbHandle.isClosed {
  859. return errMemoryProviderClosed
  860. }
  861. g, err := p.groupExistsInternal(group.Name)
  862. if err != nil {
  863. return err
  864. }
  865. for _, oldFolder := range g.VirtualFolders {
  866. p.removeRelationFromFolderMapping(oldFolder.Name, "", g.Name)
  867. }
  868. group.VirtualFolders = p.joinGroupVirtualFoldersFields(group)
  869. group.CreatedAt = g.CreatedAt
  870. group.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  871. group.ID = g.ID
  872. p.dbHandle.groups[group.Name] = group.getACopy()
  873. return nil
  874. }
  875. func (p *MemoryProvider) deleteGroup(group Group) error {
  876. p.dbHandle.Lock()
  877. defer p.dbHandle.Unlock()
  878. if p.dbHandle.isClosed {
  879. return errMemoryProviderClosed
  880. }
  881. g, err := p.groupExistsInternal(group.Name)
  882. if err != nil {
  883. return err
  884. }
  885. if len(g.Users) > 0 {
  886. return util.NewValidationError(fmt.Sprintf("the group %#v is referenced, it cannot be removed", group.Name))
  887. }
  888. for _, oldFolder := range g.VirtualFolders {
  889. p.removeRelationFromFolderMapping(oldFolder.Name, "", g.Name)
  890. }
  891. delete(p.dbHandle.groups, group.Name)
  892. // this could be more efficient
  893. p.dbHandle.groupnames = make([]string, 0, len(p.dbHandle.groups))
  894. for name := range p.dbHandle.groups {
  895. p.dbHandle.groupnames = append(p.dbHandle.groupnames, name)
  896. }
  897. sort.Strings(p.dbHandle.groupnames)
  898. return nil
  899. }
  900. func (p *MemoryProvider) dumpGroups() ([]Group, error) {
  901. p.dbHandle.Lock()
  902. defer p.dbHandle.Unlock()
  903. groups := make([]Group, 0, len(p.dbHandle.groups))
  904. var err error
  905. if p.dbHandle.isClosed {
  906. return groups, errMemoryProviderClosed
  907. }
  908. for _, name := range p.dbHandle.groupnames {
  909. g := p.dbHandle.groups[name]
  910. group := g.getACopy()
  911. p.addVirtualFoldersToGroup(&group)
  912. groups = append(groups, group)
  913. }
  914. return groups, err
  915. }
  916. func (p *MemoryProvider) getUsedFolderQuota(name string) (int, int64, error) {
  917. p.dbHandle.Lock()
  918. defer p.dbHandle.Unlock()
  919. if p.dbHandle.isClosed {
  920. return 0, 0, errMemoryProviderClosed
  921. }
  922. folder, err := p.folderExistsInternal(name)
  923. if err != nil {
  924. providerLog(logger.LevelError, "unable to get quota for folder %#v error: %v", name, err)
  925. return 0, 0, err
  926. }
  927. return folder.UsedQuotaFiles, folder.UsedQuotaSize, err
  928. }
  929. func (p *MemoryProvider) joinGroupVirtualFoldersFields(group *Group) []vfs.VirtualFolder {
  930. var folders []vfs.VirtualFolder
  931. for idx := range group.VirtualFolders {
  932. folder := &group.VirtualFolders[idx]
  933. f, err := p.addOrUpdateFolderInternal(&folder.BaseVirtualFolder, "", group.Name, 0, 0, 0)
  934. if err == nil {
  935. folder.BaseVirtualFolder = f
  936. folders = append(folders, *folder)
  937. }
  938. }
  939. return folders
  940. }
  941. func (p *MemoryProvider) addVirtualFoldersToGroup(group *Group) {
  942. if len(group.VirtualFolders) > 0 {
  943. var folders []vfs.VirtualFolder
  944. for idx := range group.VirtualFolders {
  945. folder := &group.VirtualFolders[idx]
  946. baseFolder, err := p.folderExistsInternal(folder.Name)
  947. if err != nil {
  948. continue
  949. }
  950. folder.BaseVirtualFolder = baseFolder.GetACopy()
  951. folders = append(folders, *folder)
  952. }
  953. group.VirtualFolders = folders
  954. }
  955. }
  956. func (p *MemoryProvider) addActionsToRule(rule *EventRule) {
  957. var actions []EventAction
  958. for idx := range rule.Actions {
  959. action := &rule.Actions[idx]
  960. baseAction, err := p.actionExistsInternal(action.Name)
  961. if err != nil {
  962. continue
  963. }
  964. baseAction.Options.SetEmptySecretsIfNil()
  965. action.BaseEventAction = baseAction
  966. actions = append(actions, *action)
  967. }
  968. rule.Actions = actions
  969. }
  970. func (p *MemoryProvider) addRuleToActionMapping(ruleName, actionName string) error {
  971. a, err := p.actionExistsInternal(actionName)
  972. if err != nil {
  973. return util.NewGenericError(fmt.Sprintf("action %q does not exist", actionName))
  974. }
  975. if !util.Contains(a.Rules, ruleName) {
  976. a.Rules = append(a.Rules, ruleName)
  977. p.dbHandle.actions[actionName] = a
  978. }
  979. return nil
  980. }
  981. func (p *MemoryProvider) removeRuleFromActionMapping(ruleName, actionName string) error {
  982. a, err := p.actionExistsInternal(actionName)
  983. if err != nil {
  984. providerLog(logger.LevelWarn, "action %q does not exist, cannot remove from mapping", actionName)
  985. return nil
  986. }
  987. if util.Contains(a.Rules, ruleName) {
  988. var rules []string
  989. for _, r := range a.Rules {
  990. if r != ruleName {
  991. rules = append(rules, r)
  992. }
  993. }
  994. a.Rules = rules
  995. p.dbHandle.actions[actionName] = a
  996. }
  997. return nil
  998. }
  999. func (p *MemoryProvider) addUserFromGroupMapping(username, groupname string) error {
  1000. g, err := p.groupExistsInternal(groupname)
  1001. if err != nil {
  1002. return err
  1003. }
  1004. if !util.Contains(g.Users, username) {
  1005. g.Users = append(g.Users, username)
  1006. p.dbHandle.groups[groupname] = g
  1007. }
  1008. return nil
  1009. }
  1010. func (p *MemoryProvider) removeUserFromGroupMapping(username, groupname string) error {
  1011. g, err := p.groupExistsInternal(groupname)
  1012. if err != nil {
  1013. return err
  1014. }
  1015. if util.Contains(g.Users, username) {
  1016. var users []string
  1017. for _, u := range g.Users {
  1018. if u != username {
  1019. users = append(users, u)
  1020. }
  1021. }
  1022. g.Users = users
  1023. p.dbHandle.groups[groupname] = g
  1024. }
  1025. return nil
  1026. }
  1027. func (p *MemoryProvider) joinUserVirtualFoldersFields(user *User) []vfs.VirtualFolder {
  1028. var folders []vfs.VirtualFolder
  1029. for idx := range user.VirtualFolders {
  1030. folder := &user.VirtualFolders[idx]
  1031. f, err := p.addOrUpdateFolderInternal(&folder.BaseVirtualFolder, user.Username, "", 0, 0, 0)
  1032. if err == nil {
  1033. folder.BaseVirtualFolder = f
  1034. folders = append(folders, *folder)
  1035. }
  1036. }
  1037. return folders
  1038. }
  1039. func (p *MemoryProvider) addVirtualFoldersToUser(user *User) {
  1040. if len(user.VirtualFolders) > 0 {
  1041. var folders []vfs.VirtualFolder
  1042. for idx := range user.VirtualFolders {
  1043. folder := &user.VirtualFolders[idx]
  1044. baseFolder, err := p.folderExistsInternal(folder.Name)
  1045. if err != nil {
  1046. continue
  1047. }
  1048. folder.BaseVirtualFolder = baseFolder.GetACopy()
  1049. folders = append(folders, *folder)
  1050. }
  1051. user.VirtualFolders = folders
  1052. }
  1053. }
  1054. func (p *MemoryProvider) removeRelationFromFolderMapping(folderName, username, groupname string) {
  1055. folder, err := p.folderExistsInternal(folderName)
  1056. if err == nil {
  1057. if username != "" {
  1058. var usernames []string
  1059. for _, user := range folder.Users {
  1060. if user != username {
  1061. usernames = append(usernames, user)
  1062. }
  1063. }
  1064. folder.Users = usernames
  1065. }
  1066. if groupname != "" {
  1067. var groups []string
  1068. for _, group := range folder.Groups {
  1069. if group != groupname {
  1070. groups = append(groups, group)
  1071. }
  1072. }
  1073. folder.Groups = groups
  1074. }
  1075. p.dbHandle.vfolders[folder.Name] = folder
  1076. }
  1077. }
  1078. func (p *MemoryProvider) updateFoldersMappingInternal(folder vfs.BaseVirtualFolder) {
  1079. p.dbHandle.vfolders[folder.Name] = folder
  1080. if !util.Contains(p.dbHandle.vfoldersNames, folder.Name) {
  1081. p.dbHandle.vfoldersNames = append(p.dbHandle.vfoldersNames, folder.Name)
  1082. sort.Strings(p.dbHandle.vfoldersNames)
  1083. }
  1084. }
  1085. func (p *MemoryProvider) addOrUpdateFolderInternal(baseFolder *vfs.BaseVirtualFolder, username, groupname string,
  1086. usedQuotaSize int64, usedQuotaFiles int, lastQuotaUpdate int64,
  1087. ) (vfs.BaseVirtualFolder, error) {
  1088. folder, err := p.folderExistsInternal(baseFolder.Name)
  1089. if err == nil {
  1090. // exists
  1091. folder.MappedPath = baseFolder.MappedPath
  1092. folder.Description = baseFolder.Description
  1093. folder.FsConfig = baseFolder.FsConfig.GetACopy()
  1094. if username != "" && !util.Contains(folder.Users, username) {
  1095. folder.Users = append(folder.Users, username)
  1096. }
  1097. if groupname != "" && !util.Contains(folder.Groups, groupname) {
  1098. folder.Groups = append(folder.Groups, groupname)
  1099. }
  1100. p.updateFoldersMappingInternal(folder)
  1101. return folder, nil
  1102. }
  1103. if _, ok := err.(*util.RecordNotFoundError); ok {
  1104. folder = baseFolder.GetACopy()
  1105. folder.ID = p.getNextFolderID()
  1106. folder.UsedQuotaSize = usedQuotaSize
  1107. folder.UsedQuotaFiles = usedQuotaFiles
  1108. folder.LastQuotaUpdate = lastQuotaUpdate
  1109. if username != "" {
  1110. folder.Users = []string{username}
  1111. }
  1112. if groupname != "" {
  1113. folder.Groups = []string{groupname}
  1114. }
  1115. p.updateFoldersMappingInternal(folder)
  1116. return folder, nil
  1117. }
  1118. return folder, err
  1119. }
  1120. func (p *MemoryProvider) folderExistsInternal(name string) (vfs.BaseVirtualFolder, error) {
  1121. if val, ok := p.dbHandle.vfolders[name]; ok {
  1122. return val, nil
  1123. }
  1124. return vfs.BaseVirtualFolder{}, util.NewRecordNotFoundError(fmt.Sprintf("folder %#v does not exist", name))
  1125. }
  1126. func (p *MemoryProvider) getFolders(limit, offset int, order string, minimal bool) ([]vfs.BaseVirtualFolder, error) {
  1127. folders := make([]vfs.BaseVirtualFolder, 0, limit)
  1128. var err error
  1129. p.dbHandle.Lock()
  1130. defer p.dbHandle.Unlock()
  1131. if p.dbHandle.isClosed {
  1132. return folders, errMemoryProviderClosed
  1133. }
  1134. if limit <= 0 {
  1135. return folders, err
  1136. }
  1137. itNum := 0
  1138. if order == OrderASC {
  1139. for _, name := range p.dbHandle.vfoldersNames {
  1140. itNum++
  1141. if itNum <= offset {
  1142. continue
  1143. }
  1144. f := p.dbHandle.vfolders[name]
  1145. folder := f.GetACopy()
  1146. folder.PrepareForRendering()
  1147. folders = append(folders, folder)
  1148. if len(folders) >= limit {
  1149. break
  1150. }
  1151. }
  1152. } else {
  1153. for i := len(p.dbHandle.vfoldersNames) - 1; i >= 0; i-- {
  1154. itNum++
  1155. if itNum <= offset {
  1156. continue
  1157. }
  1158. name := p.dbHandle.vfoldersNames[i]
  1159. f := p.dbHandle.vfolders[name]
  1160. folder := f.GetACopy()
  1161. folder.PrepareForRendering()
  1162. folders = append(folders, folder)
  1163. if len(folders) >= limit {
  1164. break
  1165. }
  1166. }
  1167. }
  1168. return folders, err
  1169. }
  1170. func (p *MemoryProvider) getFolderByName(name string) (vfs.BaseVirtualFolder, error) {
  1171. p.dbHandle.Lock()
  1172. defer p.dbHandle.Unlock()
  1173. if p.dbHandle.isClosed {
  1174. return vfs.BaseVirtualFolder{}, errMemoryProviderClosed
  1175. }
  1176. folder, err := p.folderExistsInternal(name)
  1177. if err != nil {
  1178. return vfs.BaseVirtualFolder{}, err
  1179. }
  1180. return folder.GetACopy(), nil
  1181. }
  1182. func (p *MemoryProvider) addFolder(folder *vfs.BaseVirtualFolder) error {
  1183. err := ValidateFolder(folder)
  1184. if err != nil {
  1185. return err
  1186. }
  1187. p.dbHandle.Lock()
  1188. defer p.dbHandle.Unlock()
  1189. if p.dbHandle.isClosed {
  1190. return errMemoryProviderClosed
  1191. }
  1192. _, err = p.folderExistsInternal(folder.Name)
  1193. if err == nil {
  1194. return fmt.Errorf("folder %#v already exists", folder.Name)
  1195. }
  1196. folder.ID = p.getNextFolderID()
  1197. folder.Users = nil
  1198. p.dbHandle.vfolders[folder.Name] = folder.GetACopy()
  1199. p.dbHandle.vfoldersNames = append(p.dbHandle.vfoldersNames, folder.Name)
  1200. sort.Strings(p.dbHandle.vfoldersNames)
  1201. return nil
  1202. }
  1203. func (p *MemoryProvider) updateFolder(folder *vfs.BaseVirtualFolder) error {
  1204. err := ValidateFolder(folder)
  1205. if err != nil {
  1206. return err
  1207. }
  1208. p.dbHandle.Lock()
  1209. defer p.dbHandle.Unlock()
  1210. if p.dbHandle.isClosed {
  1211. return errMemoryProviderClosed
  1212. }
  1213. f, err := p.folderExistsInternal(folder.Name)
  1214. if err != nil {
  1215. return err
  1216. }
  1217. folder.ID = f.ID
  1218. folder.LastQuotaUpdate = f.LastQuotaUpdate
  1219. folder.UsedQuotaFiles = f.UsedQuotaFiles
  1220. folder.UsedQuotaSize = f.UsedQuotaSize
  1221. folder.Users = f.Users
  1222. p.dbHandle.vfolders[folder.Name] = folder.GetACopy()
  1223. // now update the related users
  1224. for _, username := range folder.Users {
  1225. user, err := p.userExistsInternal(username)
  1226. if err == nil {
  1227. var folders []vfs.VirtualFolder
  1228. for idx := range user.VirtualFolders {
  1229. userFolder := &user.VirtualFolders[idx]
  1230. if folder.Name == userFolder.Name {
  1231. userFolder.BaseVirtualFolder = folder.GetACopy()
  1232. }
  1233. folders = append(folders, *userFolder)
  1234. }
  1235. user.VirtualFolders = folders
  1236. p.dbHandle.users[user.Username] = user
  1237. }
  1238. }
  1239. return nil
  1240. }
  1241. func (p *MemoryProvider) deleteFolder(f vfs.BaseVirtualFolder) error {
  1242. p.dbHandle.Lock()
  1243. defer p.dbHandle.Unlock()
  1244. if p.dbHandle.isClosed {
  1245. return errMemoryProviderClosed
  1246. }
  1247. folder, err := p.folderExistsInternal(f.Name)
  1248. if err != nil {
  1249. return err
  1250. }
  1251. for _, username := range folder.Users {
  1252. user, err := p.userExistsInternal(username)
  1253. if err == nil {
  1254. var folders []vfs.VirtualFolder
  1255. for idx := range user.VirtualFolders {
  1256. userFolder := &user.VirtualFolders[idx]
  1257. if folder.Name != userFolder.Name {
  1258. folders = append(folders, *userFolder)
  1259. }
  1260. }
  1261. user.VirtualFolders = folders
  1262. p.dbHandle.users[user.Username] = user
  1263. }
  1264. }
  1265. for _, groupname := range folder.Groups {
  1266. group, err := p.groupExistsInternal(groupname)
  1267. if err == nil {
  1268. var folders []vfs.VirtualFolder
  1269. for idx := range group.VirtualFolders {
  1270. groupFolder := &group.VirtualFolders[idx]
  1271. if folder.Name != groupFolder.Name {
  1272. folders = append(folders, *groupFolder)
  1273. }
  1274. }
  1275. group.VirtualFolders = folders
  1276. p.dbHandle.groups[group.Name] = group
  1277. }
  1278. }
  1279. delete(p.dbHandle.vfolders, folder.Name)
  1280. p.dbHandle.vfoldersNames = []string{}
  1281. for name := range p.dbHandle.vfolders {
  1282. p.dbHandle.vfoldersNames = append(p.dbHandle.vfoldersNames, name)
  1283. }
  1284. sort.Strings(p.dbHandle.vfoldersNames)
  1285. return nil
  1286. }
  1287. func (p *MemoryProvider) apiKeyExistsInternal(keyID string) (APIKey, error) {
  1288. if val, ok := p.dbHandle.apiKeys[keyID]; ok {
  1289. return val.getACopy(), nil
  1290. }
  1291. return APIKey{}, util.NewRecordNotFoundError(fmt.Sprintf("API key %#v does not exist", keyID))
  1292. }
  1293. func (p *MemoryProvider) apiKeyExists(keyID string) (APIKey, error) {
  1294. p.dbHandle.Lock()
  1295. defer p.dbHandle.Unlock()
  1296. if p.dbHandle.isClosed {
  1297. return APIKey{}, errMemoryProviderClosed
  1298. }
  1299. return p.apiKeyExistsInternal(keyID)
  1300. }
  1301. func (p *MemoryProvider) addAPIKey(apiKey *APIKey) error {
  1302. err := apiKey.validate()
  1303. if err != nil {
  1304. return err
  1305. }
  1306. p.dbHandle.Lock()
  1307. defer p.dbHandle.Unlock()
  1308. if p.dbHandle.isClosed {
  1309. return errMemoryProviderClosed
  1310. }
  1311. _, err = p.apiKeyExistsInternal(apiKey.KeyID)
  1312. if err == nil {
  1313. return fmt.Errorf("API key %#v already exists", apiKey.KeyID)
  1314. }
  1315. if apiKey.User != "" {
  1316. if _, err := p.userExistsInternal(apiKey.User); err != nil {
  1317. return util.NewValidationError(fmt.Sprintf("related user %#v does not exists", apiKey.User))
  1318. }
  1319. }
  1320. if apiKey.Admin != "" {
  1321. if _, err := p.adminExistsInternal(apiKey.Admin); err != nil {
  1322. return util.NewValidationError(fmt.Sprintf("related admin %#v does not exists", apiKey.User))
  1323. }
  1324. }
  1325. apiKey.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1326. apiKey.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1327. apiKey.LastUseAt = 0
  1328. p.dbHandle.apiKeys[apiKey.KeyID] = apiKey.getACopy()
  1329. p.dbHandle.apiKeysIDs = append(p.dbHandle.apiKeysIDs, apiKey.KeyID)
  1330. sort.Strings(p.dbHandle.apiKeysIDs)
  1331. return nil
  1332. }
  1333. func (p *MemoryProvider) updateAPIKey(apiKey *APIKey) error {
  1334. err := apiKey.validate()
  1335. if err != nil {
  1336. return err
  1337. }
  1338. p.dbHandle.Lock()
  1339. defer p.dbHandle.Unlock()
  1340. if p.dbHandle.isClosed {
  1341. return errMemoryProviderClosed
  1342. }
  1343. k, err := p.apiKeyExistsInternal(apiKey.KeyID)
  1344. if err != nil {
  1345. return err
  1346. }
  1347. if apiKey.User != "" {
  1348. if _, err := p.userExistsInternal(apiKey.User); err != nil {
  1349. return util.NewValidationError(fmt.Sprintf("related user %#v does not exists", apiKey.User))
  1350. }
  1351. }
  1352. if apiKey.Admin != "" {
  1353. if _, err := p.adminExistsInternal(apiKey.Admin); err != nil {
  1354. return util.NewValidationError(fmt.Sprintf("related admin %#v does not exists", apiKey.User))
  1355. }
  1356. }
  1357. apiKey.ID = k.ID
  1358. apiKey.KeyID = k.KeyID
  1359. apiKey.Key = k.Key
  1360. apiKey.CreatedAt = k.CreatedAt
  1361. apiKey.LastUseAt = k.LastUseAt
  1362. apiKey.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1363. p.dbHandle.apiKeys[apiKey.KeyID] = apiKey.getACopy()
  1364. return nil
  1365. }
  1366. func (p *MemoryProvider) deleteAPIKey(apiKey APIKey) error {
  1367. p.dbHandle.Lock()
  1368. defer p.dbHandle.Unlock()
  1369. if p.dbHandle.isClosed {
  1370. return errMemoryProviderClosed
  1371. }
  1372. _, err := p.apiKeyExistsInternal(apiKey.KeyID)
  1373. if err != nil {
  1374. return err
  1375. }
  1376. delete(p.dbHandle.apiKeys, apiKey.KeyID)
  1377. p.updateAPIKeysOrdering()
  1378. return nil
  1379. }
  1380. func (p *MemoryProvider) getAPIKeys(limit int, offset int, order string) ([]APIKey, error) {
  1381. apiKeys := make([]APIKey, 0, limit)
  1382. p.dbHandle.Lock()
  1383. defer p.dbHandle.Unlock()
  1384. if p.dbHandle.isClosed {
  1385. return apiKeys, errMemoryProviderClosed
  1386. }
  1387. if limit <= 0 {
  1388. return apiKeys, nil
  1389. }
  1390. itNum := 0
  1391. if order == OrderDESC {
  1392. for i := len(p.dbHandle.apiKeysIDs) - 1; i >= 0; i-- {
  1393. itNum++
  1394. if itNum <= offset {
  1395. continue
  1396. }
  1397. keyID := p.dbHandle.apiKeysIDs[i]
  1398. k := p.dbHandle.apiKeys[keyID]
  1399. apiKey := k.getACopy()
  1400. apiKey.HideConfidentialData()
  1401. apiKeys = append(apiKeys, apiKey)
  1402. if len(apiKeys) >= limit {
  1403. break
  1404. }
  1405. }
  1406. } else {
  1407. for _, keyID := range p.dbHandle.apiKeysIDs {
  1408. itNum++
  1409. if itNum <= offset {
  1410. continue
  1411. }
  1412. k := p.dbHandle.apiKeys[keyID]
  1413. apiKey := k.getACopy()
  1414. apiKey.HideConfidentialData()
  1415. apiKeys = append(apiKeys, apiKey)
  1416. if len(apiKeys) >= limit {
  1417. break
  1418. }
  1419. }
  1420. }
  1421. return apiKeys, nil
  1422. }
  1423. func (p *MemoryProvider) dumpAPIKeys() ([]APIKey, error) {
  1424. p.dbHandle.Lock()
  1425. defer p.dbHandle.Unlock()
  1426. apiKeys := make([]APIKey, 0, len(p.dbHandle.apiKeys))
  1427. if p.dbHandle.isClosed {
  1428. return apiKeys, errMemoryProviderClosed
  1429. }
  1430. for _, k := range p.dbHandle.apiKeys {
  1431. apiKeys = append(apiKeys, k)
  1432. }
  1433. return apiKeys, nil
  1434. }
  1435. func (p *MemoryProvider) deleteAPIKeysWithUser(username string) {
  1436. found := false
  1437. for k, v := range p.dbHandle.apiKeys {
  1438. if v.User == username {
  1439. delete(p.dbHandle.apiKeys, k)
  1440. found = true
  1441. }
  1442. }
  1443. if found {
  1444. p.updateAPIKeysOrdering()
  1445. }
  1446. }
  1447. func (p *MemoryProvider) deleteAPIKeysWithAdmin(username string) {
  1448. found := false
  1449. for k, v := range p.dbHandle.apiKeys {
  1450. if v.Admin == username {
  1451. delete(p.dbHandle.apiKeys, k)
  1452. found = true
  1453. }
  1454. }
  1455. if found {
  1456. p.updateAPIKeysOrdering()
  1457. }
  1458. }
  1459. func (p *MemoryProvider) deleteSharesWithUser(username string) {
  1460. found := false
  1461. for k, v := range p.dbHandle.shares {
  1462. if v.Username == username {
  1463. delete(p.dbHandle.shares, k)
  1464. found = true
  1465. }
  1466. }
  1467. if found {
  1468. p.updateSharesOrdering()
  1469. }
  1470. }
  1471. func (p *MemoryProvider) updateAPIKeysOrdering() {
  1472. // this could be more efficient
  1473. p.dbHandle.apiKeysIDs = make([]string, 0, len(p.dbHandle.apiKeys))
  1474. for keyID := range p.dbHandle.apiKeys {
  1475. p.dbHandle.apiKeysIDs = append(p.dbHandle.apiKeysIDs, keyID)
  1476. }
  1477. sort.Strings(p.dbHandle.apiKeysIDs)
  1478. }
  1479. func (p *MemoryProvider) updateSharesOrdering() {
  1480. // this could be more efficient
  1481. p.dbHandle.sharesIDs = make([]string, 0, len(p.dbHandle.shares))
  1482. for shareID := range p.dbHandle.shares {
  1483. p.dbHandle.sharesIDs = append(p.dbHandle.sharesIDs, shareID)
  1484. }
  1485. sort.Strings(p.dbHandle.sharesIDs)
  1486. }
  1487. func (p *MemoryProvider) shareExistsInternal(shareID, username string) (Share, error) {
  1488. if val, ok := p.dbHandle.shares[shareID]; ok {
  1489. if username != "" && val.Username != username {
  1490. return Share{}, util.NewRecordNotFoundError(fmt.Sprintf("Share %#v does not exist", shareID))
  1491. }
  1492. return val.getACopy(), nil
  1493. }
  1494. return Share{}, util.NewRecordNotFoundError(fmt.Sprintf("Share %#v does not exist", shareID))
  1495. }
  1496. func (p *MemoryProvider) shareExists(shareID, username string) (Share, error) {
  1497. p.dbHandle.Lock()
  1498. defer p.dbHandle.Unlock()
  1499. if p.dbHandle.isClosed {
  1500. return Share{}, errMemoryProviderClosed
  1501. }
  1502. return p.shareExistsInternal(shareID, username)
  1503. }
  1504. func (p *MemoryProvider) addShare(share *Share) error {
  1505. err := share.validate()
  1506. if err != nil {
  1507. return err
  1508. }
  1509. p.dbHandle.Lock()
  1510. defer p.dbHandle.Unlock()
  1511. if p.dbHandle.isClosed {
  1512. return errMemoryProviderClosed
  1513. }
  1514. _, err = p.shareExistsInternal(share.ShareID, share.Username)
  1515. if err == nil {
  1516. return fmt.Errorf("share %#v already exists", share.ShareID)
  1517. }
  1518. if _, err := p.userExistsInternal(share.Username); err != nil {
  1519. return util.NewValidationError(fmt.Sprintf("related user %#v does not exists", share.Username))
  1520. }
  1521. if !share.IsRestore {
  1522. share.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1523. share.UpdatedAt = share.CreatedAt
  1524. share.LastUseAt = 0
  1525. share.UsedTokens = 0
  1526. }
  1527. if share.CreatedAt == 0 {
  1528. share.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1529. }
  1530. if share.UpdatedAt == 0 {
  1531. share.UpdatedAt = share.CreatedAt
  1532. }
  1533. p.dbHandle.shares[share.ShareID] = share.getACopy()
  1534. p.dbHandle.sharesIDs = append(p.dbHandle.sharesIDs, share.ShareID)
  1535. sort.Strings(p.dbHandle.sharesIDs)
  1536. return nil
  1537. }
  1538. func (p *MemoryProvider) updateShare(share *Share) error {
  1539. err := share.validate()
  1540. if err != nil {
  1541. return err
  1542. }
  1543. p.dbHandle.Lock()
  1544. defer p.dbHandle.Unlock()
  1545. if p.dbHandle.isClosed {
  1546. return errMemoryProviderClosed
  1547. }
  1548. s, err := p.shareExistsInternal(share.ShareID, share.Username)
  1549. if err != nil {
  1550. return err
  1551. }
  1552. if _, err := p.userExistsInternal(share.Username); err != nil {
  1553. return util.NewValidationError(fmt.Sprintf("related user %#v does not exists", share.Username))
  1554. }
  1555. share.ID = s.ID
  1556. share.ShareID = s.ShareID
  1557. if !share.IsRestore {
  1558. share.UsedTokens = s.UsedTokens
  1559. share.CreatedAt = s.CreatedAt
  1560. share.LastUseAt = s.LastUseAt
  1561. share.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1562. }
  1563. if share.CreatedAt == 0 {
  1564. share.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1565. }
  1566. if share.UpdatedAt == 0 {
  1567. share.UpdatedAt = share.CreatedAt
  1568. }
  1569. p.dbHandle.shares[share.ShareID] = share.getACopy()
  1570. return nil
  1571. }
  1572. func (p *MemoryProvider) deleteShare(share Share) error {
  1573. p.dbHandle.Lock()
  1574. defer p.dbHandle.Unlock()
  1575. if p.dbHandle.isClosed {
  1576. return errMemoryProviderClosed
  1577. }
  1578. _, err := p.shareExistsInternal(share.ShareID, share.Username)
  1579. if err != nil {
  1580. return err
  1581. }
  1582. delete(p.dbHandle.shares, share.ShareID)
  1583. p.updateSharesOrdering()
  1584. return nil
  1585. }
  1586. func (p *MemoryProvider) getShares(limit int, offset int, order, username string) ([]Share, error) {
  1587. p.dbHandle.Lock()
  1588. defer p.dbHandle.Unlock()
  1589. if p.dbHandle.isClosed {
  1590. return []Share{}, errMemoryProviderClosed
  1591. }
  1592. if limit <= 0 {
  1593. return []Share{}, nil
  1594. }
  1595. shares := make([]Share, 0, limit)
  1596. itNum := 0
  1597. if order == OrderDESC {
  1598. for i := len(p.dbHandle.sharesIDs) - 1; i >= 0; i-- {
  1599. shareID := p.dbHandle.sharesIDs[i]
  1600. s := p.dbHandle.shares[shareID]
  1601. if s.Username != username {
  1602. continue
  1603. }
  1604. itNum++
  1605. if itNum <= offset {
  1606. continue
  1607. }
  1608. share := s.getACopy()
  1609. share.HideConfidentialData()
  1610. shares = append(shares, share)
  1611. if len(shares) >= limit {
  1612. break
  1613. }
  1614. }
  1615. } else {
  1616. for _, shareID := range p.dbHandle.sharesIDs {
  1617. s := p.dbHandle.shares[shareID]
  1618. if s.Username != username {
  1619. continue
  1620. }
  1621. itNum++
  1622. if itNum <= offset {
  1623. continue
  1624. }
  1625. share := s.getACopy()
  1626. share.HideConfidentialData()
  1627. shares = append(shares, share)
  1628. if len(shares) >= limit {
  1629. break
  1630. }
  1631. }
  1632. }
  1633. return shares, nil
  1634. }
  1635. func (p *MemoryProvider) dumpShares() ([]Share, error) {
  1636. p.dbHandle.Lock()
  1637. defer p.dbHandle.Unlock()
  1638. shares := make([]Share, 0, len(p.dbHandle.shares))
  1639. if p.dbHandle.isClosed {
  1640. return shares, errMemoryProviderClosed
  1641. }
  1642. for _, s := range p.dbHandle.shares {
  1643. shares = append(shares, s)
  1644. }
  1645. return shares, nil
  1646. }
  1647. func (p *MemoryProvider) updateShareLastUse(shareID string, numTokens int) error {
  1648. p.dbHandle.Lock()
  1649. defer p.dbHandle.Unlock()
  1650. if p.dbHandle.isClosed {
  1651. return errMemoryProviderClosed
  1652. }
  1653. share, err := p.shareExistsInternal(shareID, "")
  1654. if err != nil {
  1655. return err
  1656. }
  1657. share.LastUseAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1658. share.UsedTokens += numTokens
  1659. p.dbHandle.shares[share.ShareID] = share
  1660. return nil
  1661. }
  1662. func (p *MemoryProvider) getDefenderHosts(from int64, limit int) ([]DefenderEntry, error) {
  1663. return nil, ErrNotImplemented
  1664. }
  1665. func (p *MemoryProvider) getDefenderHostByIP(ip string, from int64) (DefenderEntry, error) {
  1666. return DefenderEntry{}, ErrNotImplemented
  1667. }
  1668. func (p *MemoryProvider) isDefenderHostBanned(ip string) (DefenderEntry, error) {
  1669. return DefenderEntry{}, ErrNotImplemented
  1670. }
  1671. func (p *MemoryProvider) updateDefenderBanTime(ip string, minutes int) error {
  1672. return ErrNotImplemented
  1673. }
  1674. func (p *MemoryProvider) deleteDefenderHost(ip string) error {
  1675. return ErrNotImplemented
  1676. }
  1677. func (p *MemoryProvider) addDefenderEvent(ip string, score int) error {
  1678. return ErrNotImplemented
  1679. }
  1680. func (p *MemoryProvider) setDefenderBanTime(ip string, banTime int64) error {
  1681. return ErrNotImplemented
  1682. }
  1683. func (p *MemoryProvider) cleanupDefender(from int64) error {
  1684. return ErrNotImplemented
  1685. }
  1686. func (p *MemoryProvider) addActiveTransfer(transfer ActiveTransfer) error {
  1687. return ErrNotImplemented
  1688. }
  1689. func (p *MemoryProvider) updateActiveTransferSizes(ulSize, dlSize, transferID int64, connectionID string) error {
  1690. return ErrNotImplemented
  1691. }
  1692. func (p *MemoryProvider) removeActiveTransfer(transferID int64, connectionID string) error {
  1693. return ErrNotImplemented
  1694. }
  1695. func (p *MemoryProvider) cleanupActiveTransfers(before time.Time) error {
  1696. return ErrNotImplemented
  1697. }
  1698. func (p *MemoryProvider) getActiveTransfers(from time.Time) ([]ActiveTransfer, error) {
  1699. return nil, ErrNotImplemented
  1700. }
  1701. func (p *MemoryProvider) addSharedSession(session Session) error {
  1702. return ErrNotImplemented
  1703. }
  1704. func (p *MemoryProvider) deleteSharedSession(key string) error {
  1705. return ErrNotImplemented
  1706. }
  1707. func (p *MemoryProvider) getSharedSession(key string) (Session, error) {
  1708. return Session{}, ErrNotImplemented
  1709. }
  1710. func (p *MemoryProvider) cleanupSharedSessions(sessionType SessionType, before int64) error {
  1711. return ErrNotImplemented
  1712. }
  1713. func (p *MemoryProvider) getEventActions(limit, offset int, order string, minimal bool) ([]BaseEventAction, error) {
  1714. p.dbHandle.Lock()
  1715. defer p.dbHandle.Unlock()
  1716. if p.dbHandle.isClosed {
  1717. return nil, errMemoryProviderClosed
  1718. }
  1719. if limit <= 0 {
  1720. return nil, nil
  1721. }
  1722. actions := make([]BaseEventAction, 0, limit)
  1723. itNum := 0
  1724. if order == OrderASC {
  1725. for _, name := range p.dbHandle.actionsNames {
  1726. itNum++
  1727. if itNum <= offset {
  1728. continue
  1729. }
  1730. a := p.dbHandle.actions[name]
  1731. action := a.getACopy()
  1732. action.PrepareForRendering()
  1733. actions = append(actions, action)
  1734. if len(actions) >= limit {
  1735. break
  1736. }
  1737. }
  1738. } else {
  1739. for i := len(p.dbHandle.actionsNames) - 1; i >= 0; i-- {
  1740. itNum++
  1741. if itNum <= offset {
  1742. continue
  1743. }
  1744. name := p.dbHandle.actionsNames[i]
  1745. a := p.dbHandle.actions[name]
  1746. action := a.getACopy()
  1747. action.PrepareForRendering()
  1748. actions = append(actions, action)
  1749. if len(actions) >= limit {
  1750. break
  1751. }
  1752. }
  1753. }
  1754. return actions, nil
  1755. }
  1756. func (p *MemoryProvider) dumpEventActions() ([]BaseEventAction, error) {
  1757. p.dbHandle.Lock()
  1758. defer p.dbHandle.Unlock()
  1759. if p.dbHandle.isClosed {
  1760. return nil, errMemoryProviderClosed
  1761. }
  1762. actions := make([]BaseEventAction, 0, len(p.dbHandle.actions))
  1763. for _, name := range p.dbHandle.actionsNames {
  1764. a := p.dbHandle.actions[name]
  1765. action := a.getACopy()
  1766. actions = append(actions, action)
  1767. }
  1768. return actions, nil
  1769. }
  1770. func (p *MemoryProvider) eventActionExists(name string) (BaseEventAction, error) {
  1771. p.dbHandle.Lock()
  1772. defer p.dbHandle.Unlock()
  1773. if p.dbHandle.isClosed {
  1774. return BaseEventAction{}, errMemoryProviderClosed
  1775. }
  1776. return p.actionExistsInternal(name)
  1777. }
  1778. func (p *MemoryProvider) addEventAction(action *BaseEventAction) error {
  1779. err := action.validate()
  1780. if err != nil {
  1781. return err
  1782. }
  1783. p.dbHandle.Lock()
  1784. defer p.dbHandle.Unlock()
  1785. if p.dbHandle.isClosed {
  1786. return errMemoryProviderClosed
  1787. }
  1788. _, err = p.actionExistsInternal(action.Name)
  1789. if err == nil {
  1790. return fmt.Errorf("event action %q already exists", action.Name)
  1791. }
  1792. action.ID = p.getNextActionID()
  1793. action.Rules = nil
  1794. p.dbHandle.actions[action.Name] = action.getACopy()
  1795. p.dbHandle.actionsNames = append(p.dbHandle.actionsNames, action.Name)
  1796. sort.Strings(p.dbHandle.actionsNames)
  1797. return nil
  1798. }
  1799. func (p *MemoryProvider) updateEventAction(action *BaseEventAction) error {
  1800. err := action.validate()
  1801. if err != nil {
  1802. return err
  1803. }
  1804. p.dbHandle.Lock()
  1805. defer p.dbHandle.Unlock()
  1806. if p.dbHandle.isClosed {
  1807. return errMemoryProviderClosed
  1808. }
  1809. oldAction, err := p.actionExistsInternal(action.Name)
  1810. if err != nil {
  1811. return fmt.Errorf("event action %s does not exist", action.Name)
  1812. }
  1813. action.ID = oldAction.ID
  1814. action.Name = oldAction.Name
  1815. action.Rules = nil
  1816. if len(oldAction.Rules) > 0 {
  1817. var relatedRules []string
  1818. for _, ruleName := range oldAction.Rules {
  1819. rule, err := p.ruleExistsInternal(ruleName)
  1820. if err == nil {
  1821. relatedRules = append(relatedRules, ruleName)
  1822. rule.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1823. p.dbHandle.rules[ruleName] = rule
  1824. setLastRuleUpdate()
  1825. }
  1826. }
  1827. action.Rules = relatedRules
  1828. }
  1829. p.dbHandle.actions[action.Name] = action.getACopy()
  1830. return nil
  1831. }
  1832. func (p *MemoryProvider) deleteEventAction(action BaseEventAction) error {
  1833. p.dbHandle.Lock()
  1834. defer p.dbHandle.Unlock()
  1835. if p.dbHandle.isClosed {
  1836. return errMemoryProviderClosed
  1837. }
  1838. oldAction, err := p.actionExistsInternal(action.Name)
  1839. if err != nil {
  1840. return fmt.Errorf("event action %s does not exist", action.Name)
  1841. }
  1842. if len(oldAction.Rules) > 0 {
  1843. return util.NewValidationError(fmt.Sprintf("action %s is referenced, it cannot be removed", oldAction.Name))
  1844. }
  1845. delete(p.dbHandle.actions, action.Name)
  1846. // this could be more efficient
  1847. p.dbHandle.actionsNames = make([]string, 0, len(p.dbHandle.actions))
  1848. for name := range p.dbHandle.actions {
  1849. p.dbHandle.actionsNames = append(p.dbHandle.actionsNames, name)
  1850. }
  1851. sort.Strings(p.dbHandle.actionsNames)
  1852. return nil
  1853. }
  1854. func (p *MemoryProvider) getEventRules(limit, offset int, order string) ([]EventRule, error) {
  1855. p.dbHandle.Lock()
  1856. defer p.dbHandle.Unlock()
  1857. if p.dbHandle.isClosed {
  1858. return nil, errMemoryProviderClosed
  1859. }
  1860. if limit <= 0 {
  1861. return nil, nil
  1862. }
  1863. itNum := 0
  1864. rules := make([]EventRule, 0, limit)
  1865. if order == OrderASC {
  1866. for _, name := range p.dbHandle.rulesNames {
  1867. itNum++
  1868. if itNum <= offset {
  1869. continue
  1870. }
  1871. r := p.dbHandle.rules[name]
  1872. rule := r.getACopy()
  1873. p.addActionsToRule(&rule)
  1874. rule.PrepareForRendering()
  1875. rules = append(rules, rule)
  1876. if len(rules) >= limit {
  1877. break
  1878. }
  1879. }
  1880. } else {
  1881. for i := len(p.dbHandle.rulesNames) - 1; i >= 0; i-- {
  1882. itNum++
  1883. if itNum <= offset {
  1884. continue
  1885. }
  1886. name := p.dbHandle.rulesNames[i]
  1887. r := p.dbHandle.rules[name]
  1888. rule := r.getACopy()
  1889. p.addActionsToRule(&rule)
  1890. rule.PrepareForRendering()
  1891. rules = append(rules, rule)
  1892. if len(rules) >= limit {
  1893. break
  1894. }
  1895. }
  1896. }
  1897. return rules, nil
  1898. }
  1899. func (p *MemoryProvider) dumpEventRules() ([]EventRule, error) {
  1900. p.dbHandle.Lock()
  1901. defer p.dbHandle.Unlock()
  1902. if p.dbHandle.isClosed {
  1903. return nil, errMemoryProviderClosed
  1904. }
  1905. rules := make([]EventRule, 0, len(p.dbHandle.rules))
  1906. for _, name := range p.dbHandle.rulesNames {
  1907. r := p.dbHandle.rules[name]
  1908. rule := r.getACopy()
  1909. p.addActionsToRule(&rule)
  1910. rules = append(rules, rule)
  1911. }
  1912. return rules, nil
  1913. }
  1914. func (p *MemoryProvider) getRecentlyUpdatedRules(after int64) ([]EventRule, error) {
  1915. if getLastRuleUpdate() < after {
  1916. return nil, nil
  1917. }
  1918. p.dbHandle.Lock()
  1919. defer p.dbHandle.Unlock()
  1920. if p.dbHandle.isClosed {
  1921. return nil, errMemoryProviderClosed
  1922. }
  1923. rules := make([]EventRule, 0, 10)
  1924. for _, name := range p.dbHandle.rulesNames {
  1925. r := p.dbHandle.rules[name]
  1926. if r.UpdatedAt < after {
  1927. continue
  1928. }
  1929. rule := r.getACopy()
  1930. p.addActionsToRule(&rule)
  1931. rules = append(rules, rule)
  1932. }
  1933. return rules, nil
  1934. }
  1935. func (p *MemoryProvider) eventRuleExists(name string) (EventRule, error) {
  1936. p.dbHandle.Lock()
  1937. defer p.dbHandle.Unlock()
  1938. if p.dbHandle.isClosed {
  1939. return EventRule{}, errMemoryProviderClosed
  1940. }
  1941. rule, err := p.ruleExistsInternal(name)
  1942. if err != nil {
  1943. return rule, err
  1944. }
  1945. p.addActionsToRule(&rule)
  1946. return rule, nil
  1947. }
  1948. func (p *MemoryProvider) addEventRule(rule *EventRule) error {
  1949. if err := rule.validate(); err != nil {
  1950. return err
  1951. }
  1952. p.dbHandle.Lock()
  1953. defer p.dbHandle.Unlock()
  1954. if p.dbHandle.isClosed {
  1955. return errMemoryProviderClosed
  1956. }
  1957. _, err := p.ruleExistsInternal(rule.Name)
  1958. if err == nil {
  1959. return fmt.Errorf("event rule %q already exists", rule.Name)
  1960. }
  1961. rule.ID = p.getNextRuleID()
  1962. rule.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1963. rule.UpdatedAt = rule.CreatedAt
  1964. for idx := range rule.Actions {
  1965. if err := p.addRuleToActionMapping(rule.Name, rule.Actions[idx].Name); err != nil {
  1966. return err
  1967. }
  1968. }
  1969. sort.Slice(rule.Actions, func(i, j int) bool {
  1970. return rule.Actions[i].Order < rule.Actions[j].Order
  1971. })
  1972. p.dbHandle.rules[rule.Name] = rule.getACopy()
  1973. p.dbHandle.rulesNames = append(p.dbHandle.rulesNames, rule.Name)
  1974. sort.Strings(p.dbHandle.rulesNames)
  1975. setLastRuleUpdate()
  1976. return nil
  1977. }
  1978. func (p *MemoryProvider) updateEventRule(rule *EventRule) error {
  1979. if err := rule.validate(); err != nil {
  1980. return err
  1981. }
  1982. p.dbHandle.Lock()
  1983. defer p.dbHandle.Unlock()
  1984. if p.dbHandle.isClosed {
  1985. return errMemoryProviderClosed
  1986. }
  1987. oldRule, err := p.ruleExistsInternal(rule.Name)
  1988. if err != nil {
  1989. return err
  1990. }
  1991. for idx := range oldRule.Actions {
  1992. if err = p.removeRuleFromActionMapping(rule.Name, oldRule.Actions[idx].Name); err != nil {
  1993. return err
  1994. }
  1995. }
  1996. for idx := range rule.Actions {
  1997. if err = p.addRuleToActionMapping(rule.Name, rule.Actions[idx].Name); err != nil {
  1998. return err
  1999. }
  2000. }
  2001. rule.ID = oldRule.ID
  2002. rule.CreatedAt = oldRule.CreatedAt
  2003. rule.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  2004. sort.Slice(rule.Actions, func(i, j int) bool {
  2005. return rule.Actions[i].Order < rule.Actions[j].Order
  2006. })
  2007. p.dbHandle.rules[rule.Name] = rule.getACopy()
  2008. setLastRuleUpdate()
  2009. return nil
  2010. }
  2011. func (p *MemoryProvider) deleteEventRule(rule EventRule, softDelete bool) error {
  2012. p.dbHandle.Lock()
  2013. defer p.dbHandle.Unlock()
  2014. if p.dbHandle.isClosed {
  2015. return errMemoryProviderClosed
  2016. }
  2017. oldRule, err := p.ruleExistsInternal(rule.Name)
  2018. if err != nil {
  2019. return err
  2020. }
  2021. if len(oldRule.Actions) > 0 {
  2022. for idx := range oldRule.Actions {
  2023. if err = p.removeRuleFromActionMapping(rule.Name, oldRule.Actions[idx].Name); err != nil {
  2024. return err
  2025. }
  2026. }
  2027. }
  2028. delete(p.dbHandle.rules, rule.Name)
  2029. p.dbHandle.rulesNames = make([]string, 0, len(p.dbHandle.rules))
  2030. for name := range p.dbHandle.rules {
  2031. p.dbHandle.rulesNames = append(p.dbHandle.rulesNames, name)
  2032. }
  2033. sort.Strings(p.dbHandle.rulesNames)
  2034. setLastRuleUpdate()
  2035. return nil
  2036. }
  2037. func (p *MemoryProvider) getTaskByName(name string) (Task, error) {
  2038. return Task{}, ErrNotImplemented
  2039. }
  2040. func (p *MemoryProvider) addTask(name string) error {
  2041. return ErrNotImplemented
  2042. }
  2043. func (p *MemoryProvider) updateTask(name string, version int64) error {
  2044. return ErrNotImplemented
  2045. }
  2046. func (p *MemoryProvider) updateTaskTimestamp(name string) error {
  2047. return ErrNotImplemented
  2048. }
  2049. func (p *MemoryProvider) setFirstDownloadTimestamp(username string) error {
  2050. p.dbHandle.Lock()
  2051. defer p.dbHandle.Unlock()
  2052. if p.dbHandle.isClosed {
  2053. return errMemoryProviderClosed
  2054. }
  2055. user, err := p.userExistsInternal(username)
  2056. if err != nil {
  2057. return err
  2058. }
  2059. if user.FirstDownload > 0 {
  2060. return util.NewGenericError(fmt.Sprintf("first download already set to %v",
  2061. util.GetTimeFromMsecSinceEpoch(user.FirstDownload)))
  2062. }
  2063. user.FirstDownload = util.GetTimeAsMsSinceEpoch(time.Now())
  2064. p.dbHandle.users[user.Username] = user
  2065. return nil
  2066. }
  2067. func (p *MemoryProvider) setFirstUploadTimestamp(username string) error {
  2068. p.dbHandle.Lock()
  2069. defer p.dbHandle.Unlock()
  2070. if p.dbHandle.isClosed {
  2071. return errMemoryProviderClosed
  2072. }
  2073. user, err := p.userExistsInternal(username)
  2074. if err != nil {
  2075. return err
  2076. }
  2077. if user.FirstUpload > 0 {
  2078. return util.NewGenericError(fmt.Sprintf("first upload already set to %v",
  2079. util.GetTimeFromMsecSinceEpoch(user.FirstUpload)))
  2080. }
  2081. user.FirstUpload = util.GetTimeAsMsSinceEpoch(time.Now())
  2082. p.dbHandle.users[user.Username] = user
  2083. return nil
  2084. }
  2085. func (p *MemoryProvider) getNextID() int64 {
  2086. nextID := int64(1)
  2087. for _, v := range p.dbHandle.users {
  2088. if v.ID >= nextID {
  2089. nextID = v.ID + 1
  2090. }
  2091. }
  2092. return nextID
  2093. }
  2094. func (p *MemoryProvider) getNextFolderID() int64 {
  2095. nextID := int64(1)
  2096. for _, v := range p.dbHandle.vfolders {
  2097. if v.ID >= nextID {
  2098. nextID = v.ID + 1
  2099. }
  2100. }
  2101. return nextID
  2102. }
  2103. func (p *MemoryProvider) getNextAdminID() int64 {
  2104. nextID := int64(1)
  2105. for _, a := range p.dbHandle.admins {
  2106. if a.ID >= nextID {
  2107. nextID = a.ID + 1
  2108. }
  2109. }
  2110. return nextID
  2111. }
  2112. func (p *MemoryProvider) getNextGroupID() int64 {
  2113. nextID := int64(1)
  2114. for _, g := range p.dbHandle.groups {
  2115. if g.ID >= nextID {
  2116. nextID = g.ID + 1
  2117. }
  2118. }
  2119. return nextID
  2120. }
  2121. func (p *MemoryProvider) getNextActionID() int64 {
  2122. nextID := int64(1)
  2123. for _, a := range p.dbHandle.actions {
  2124. if a.ID >= nextID {
  2125. nextID = a.ID + 1
  2126. }
  2127. }
  2128. return nextID
  2129. }
  2130. func (p *MemoryProvider) getNextRuleID() int64 {
  2131. nextID := int64(1)
  2132. for _, r := range p.dbHandle.rules {
  2133. if r.ID >= nextID {
  2134. nextID = r.ID + 1
  2135. }
  2136. }
  2137. return nextID
  2138. }
  2139. func (p *MemoryProvider) clear() {
  2140. p.dbHandle.Lock()
  2141. defer p.dbHandle.Unlock()
  2142. p.dbHandle.usernames = []string{}
  2143. p.dbHandle.users = make(map[string]User)
  2144. p.dbHandle.vfoldersNames = []string{}
  2145. p.dbHandle.vfolders = make(map[string]vfs.BaseVirtualFolder)
  2146. p.dbHandle.admins = make(map[string]Admin)
  2147. p.dbHandle.adminsUsernames = []string{}
  2148. p.dbHandle.apiKeys = make(map[string]APIKey)
  2149. p.dbHandle.apiKeysIDs = []string{}
  2150. p.dbHandle.shares = make(map[string]Share)
  2151. p.dbHandle.sharesIDs = []string{}
  2152. }
  2153. func (p *MemoryProvider) reloadConfig() error {
  2154. if p.dbHandle.configFile == "" {
  2155. providerLog(logger.LevelDebug, "no dump configuration file defined")
  2156. return nil
  2157. }
  2158. providerLog(logger.LevelDebug, "loading dump from file: %#v", p.dbHandle.configFile)
  2159. fi, err := os.Stat(p.dbHandle.configFile)
  2160. if err != nil {
  2161. providerLog(logger.LevelError, "error loading dump: %v", err)
  2162. return err
  2163. }
  2164. if fi.Size() == 0 {
  2165. err = errors.New("dump configuration file is invalid, its size must be > 0")
  2166. providerLog(logger.LevelError, "error loading dump: %v", err)
  2167. return err
  2168. }
  2169. if fi.Size() > 10485760 {
  2170. err = errors.New("dump configuration file is invalid, its size must be <= 10485760 bytes")
  2171. providerLog(logger.LevelError, "error loading dump: %v", err)
  2172. return err
  2173. }
  2174. content, err := os.ReadFile(p.dbHandle.configFile)
  2175. if err != nil {
  2176. providerLog(logger.LevelError, "error loading dump: %v", err)
  2177. return err
  2178. }
  2179. dump, err := ParseDumpData(content)
  2180. if err != nil {
  2181. providerLog(logger.LevelError, "error loading dump: %v", err)
  2182. return err
  2183. }
  2184. p.clear()
  2185. if err := p.restoreFolders(dump); err != nil {
  2186. return err
  2187. }
  2188. if err := p.restoreGroups(dump); err != nil {
  2189. return err
  2190. }
  2191. if err := p.restoreUsers(dump); err != nil {
  2192. return err
  2193. }
  2194. if err := p.restoreAdmins(dump); err != nil {
  2195. return err
  2196. }
  2197. if err := p.restoreAPIKeys(dump); err != nil {
  2198. return err
  2199. }
  2200. if err := p.restoreShares(dump); err != nil {
  2201. return err
  2202. }
  2203. if err := p.restoreEventActions(dump); err != nil {
  2204. return err
  2205. }
  2206. if err := p.restoreEventRules(dump); err != nil {
  2207. return err
  2208. }
  2209. providerLog(logger.LevelDebug, "config loaded from file: %#v", p.dbHandle.configFile)
  2210. return nil
  2211. }
  2212. func (p *MemoryProvider) restoreEventActions(dump BackupData) error {
  2213. for _, action := range dump.EventActions {
  2214. a, err := p.eventActionExists(action.Name)
  2215. action := action // pin
  2216. if err == nil {
  2217. action.ID = a.ID
  2218. err = UpdateEventAction(&action, ActionExecutorSystem, "")
  2219. if err != nil {
  2220. providerLog(logger.LevelError, "error updating event action %q: %v", action.Name, err)
  2221. return err
  2222. }
  2223. } else {
  2224. err = AddEventAction(&action, ActionExecutorSystem, "")
  2225. if err != nil {
  2226. providerLog(logger.LevelError, "error adding event action %q: %v", action.Name, err)
  2227. return err
  2228. }
  2229. }
  2230. }
  2231. return nil
  2232. }
  2233. func (p *MemoryProvider) restoreEventRules(dump BackupData) error {
  2234. for _, rule := range dump.EventRules {
  2235. r, err := p.eventRuleExists(rule.Name)
  2236. rule := rule // pin
  2237. if err == nil {
  2238. rule.ID = r.ID
  2239. err = UpdateEventRule(&rule, ActionExecutorSystem, "")
  2240. if err != nil {
  2241. providerLog(logger.LevelError, "error updating event rule %q: %v", rule.Name, err)
  2242. return err
  2243. }
  2244. } else {
  2245. err = AddEventRule(&rule, ActionExecutorSystem, "")
  2246. if err != nil {
  2247. providerLog(logger.LevelError, "error adding event rule %q: %v", rule.Name, err)
  2248. return err
  2249. }
  2250. }
  2251. }
  2252. return nil
  2253. }
  2254. func (p *MemoryProvider) restoreShares(dump BackupData) error {
  2255. for _, share := range dump.Shares {
  2256. s, err := p.shareExists(share.ShareID, "")
  2257. share := share // pin
  2258. share.IsRestore = true
  2259. if err == nil {
  2260. share.ID = s.ID
  2261. err = UpdateShare(&share, ActionExecutorSystem, "")
  2262. if err != nil {
  2263. providerLog(logger.LevelError, "error updating share %#v: %v", share.ShareID, err)
  2264. return err
  2265. }
  2266. } else {
  2267. err = AddShare(&share, ActionExecutorSystem, "")
  2268. if err != nil {
  2269. providerLog(logger.LevelError, "error adding share %#v: %v", share.ShareID, err)
  2270. return err
  2271. }
  2272. }
  2273. }
  2274. return nil
  2275. }
  2276. func (p *MemoryProvider) restoreAPIKeys(dump BackupData) error {
  2277. for _, apiKey := range dump.APIKeys {
  2278. if apiKey.Key == "" {
  2279. return fmt.Errorf("cannot restore an empty API key: %+v", apiKey)
  2280. }
  2281. k, err := p.apiKeyExists(apiKey.KeyID)
  2282. apiKey := apiKey // pin
  2283. if err == nil {
  2284. apiKey.ID = k.ID
  2285. err = UpdateAPIKey(&apiKey, ActionExecutorSystem, "")
  2286. if err != nil {
  2287. providerLog(logger.LevelError, "error updating API key %#v: %v", apiKey.KeyID, err)
  2288. return err
  2289. }
  2290. } else {
  2291. err = AddAPIKey(&apiKey, ActionExecutorSystem, "")
  2292. if err != nil {
  2293. providerLog(logger.LevelError, "error adding API key %#v: %v", apiKey.KeyID, err)
  2294. return err
  2295. }
  2296. }
  2297. }
  2298. return nil
  2299. }
  2300. func (p *MemoryProvider) restoreAdmins(dump BackupData) error {
  2301. for _, admin := range dump.Admins {
  2302. admin := admin // pin
  2303. admin.Username = config.convertName(admin.Username)
  2304. a, err := p.adminExists(admin.Username)
  2305. if err == nil {
  2306. admin.ID = a.ID
  2307. err = UpdateAdmin(&admin, ActionExecutorSystem, "")
  2308. if err != nil {
  2309. providerLog(logger.LevelError, "error updating admin %#v: %v", admin.Username, err)
  2310. return err
  2311. }
  2312. } else {
  2313. err = AddAdmin(&admin, ActionExecutorSystem, "")
  2314. if err != nil {
  2315. providerLog(logger.LevelError, "error adding admin %#v: %v", admin.Username, err)
  2316. return err
  2317. }
  2318. }
  2319. }
  2320. return nil
  2321. }
  2322. func (p *MemoryProvider) restoreGroups(dump BackupData) error {
  2323. for _, group := range dump.Groups {
  2324. group := group // pin
  2325. group.Name = config.convertName(group.Name)
  2326. g, err := p.groupExists(group.Name)
  2327. if err == nil {
  2328. group.ID = g.ID
  2329. err = UpdateGroup(&group, g.Users, ActionExecutorSystem, "")
  2330. if err != nil {
  2331. providerLog(logger.LevelError, "error updating group %#v: %v", group.Name, err)
  2332. return err
  2333. }
  2334. } else {
  2335. group.Users = nil
  2336. err = AddGroup(&group, ActionExecutorSystem, "")
  2337. if err != nil {
  2338. providerLog(logger.LevelError, "error adding group %#v: %v", group.Name, err)
  2339. return err
  2340. }
  2341. }
  2342. }
  2343. return nil
  2344. }
  2345. func (p *MemoryProvider) restoreFolders(dump BackupData) error {
  2346. for _, folder := range dump.Folders {
  2347. folder := folder // pin
  2348. folder.Name = config.convertName(folder.Name)
  2349. f, err := p.getFolderByName(folder.Name)
  2350. if err == nil {
  2351. folder.ID = f.ID
  2352. err = UpdateFolder(&folder, f.Users, f.Groups, ActionExecutorSystem, "")
  2353. if err != nil {
  2354. providerLog(logger.LevelError, "error updating folder %#v: %v", folder.Name, err)
  2355. return err
  2356. }
  2357. } else {
  2358. folder.Users = nil
  2359. err = AddFolder(&folder, ActionExecutorSystem, "")
  2360. if err != nil {
  2361. providerLog(logger.LevelError, "error adding folder %#v: %v", folder.Name, err)
  2362. return err
  2363. }
  2364. }
  2365. }
  2366. return nil
  2367. }
  2368. func (p *MemoryProvider) restoreUsers(dump BackupData) error {
  2369. for _, user := range dump.Users {
  2370. user := user // pin
  2371. user.Username = config.convertName(user.Username)
  2372. u, err := p.userExists(user.Username)
  2373. if err == nil {
  2374. user.ID = u.ID
  2375. err = UpdateUser(&user, ActionExecutorSystem, "")
  2376. if err != nil {
  2377. providerLog(logger.LevelError, "error updating user %#v: %v", user.Username, err)
  2378. return err
  2379. }
  2380. } else {
  2381. err = AddUser(&user, ActionExecutorSystem, "")
  2382. if err != nil {
  2383. providerLog(logger.LevelError, "error adding user %#v: %v", user.Username, err)
  2384. return err
  2385. }
  2386. }
  2387. }
  2388. return nil
  2389. }
  2390. // initializeDatabase does nothing, no initilization is needed for memory provider
  2391. func (p *MemoryProvider) initializeDatabase() error {
  2392. return ErrNoInitRequired
  2393. }
  2394. func (p *MemoryProvider) migrateDatabase() error {
  2395. return ErrNoInitRequired
  2396. }
  2397. func (p *MemoryProvider) revertDatabase(targetVersion int) error {
  2398. return errors.New("memory provider does not store data, revert not possible")
  2399. }
  2400. func (p *MemoryProvider) resetDatabase() error {
  2401. return errors.New("memory provider does not store data, reset not possible")
  2402. }