mysql.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524
  1. //go:build !nomysql
  2. // +build !nomysql
  3. package dataprovider
  4. import (
  5. "context"
  6. "crypto/x509"
  7. "database/sql"
  8. "errors"
  9. "fmt"
  10. "strings"
  11. "time"
  12. // we import go-sql-driver/mysql here to be able to disable MySQL support using a build tag
  13. _ "github.com/go-sql-driver/mysql"
  14. "github.com/drakkan/sftpgo/v2/logger"
  15. "github.com/drakkan/sftpgo/v2/version"
  16. "github.com/drakkan/sftpgo/v2/vfs"
  17. )
  18. const (
  19. mysqlResetSQL = "DROP TABLE IF EXISTS `{{api_keys}}` CASCADE;" +
  20. "DROP TABLE IF EXISTS `{{folders_mapping}}` CASCADE;" +
  21. "DROP TABLE IF EXISTS `{{admins}}` CASCADE;" +
  22. "DROP TABLE IF EXISTS `{{folders}}` CASCADE;" +
  23. "DROP TABLE IF EXISTS `{{shares}}` CASCADE;" +
  24. "DROP TABLE IF EXISTS `{{users}}` CASCADE;" +
  25. "DROP TABLE IF EXISTS `{{schema_version}}` CASCADE;"
  26. mysqlInitialSQL = "CREATE TABLE `{{schema_version}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `version` integer NOT NULL);" +
  27. "CREATE TABLE `{{admins}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `username` varchar(255) NOT NULL UNIQUE, " +
  28. "`description` varchar(512) NULL, `password` varchar(255) NOT NULL, `email` varchar(255) NULL, `status` integer NOT NULL, " +
  29. "`permissions` longtext NOT NULL, `filters` longtext NULL, `additional_info` longtext NULL);" +
  30. "CREATE TABLE `{{folders}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `name` varchar(255) NOT NULL UNIQUE, " +
  31. "`description` varchar(512) NULL, `path` varchar(512) NULL, `used_quota_size` bigint NOT NULL, " +
  32. "`used_quota_files` integer NOT NULL, `last_quota_update` bigint NOT NULL, `filesystem` longtext NULL);" +
  33. "CREATE TABLE `{{users}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `username` varchar(255) NOT NULL UNIQUE, " +
  34. "`status` integer NOT NULL, `expiration_date` bigint NOT NULL, `description` varchar(512) NULL, `password` longtext NULL, " +
  35. "`public_keys` longtext NULL, `home_dir` varchar(512) NOT NULL, `uid` integer NOT NULL, `gid` integer NOT NULL, " +
  36. "`max_sessions` integer NOT NULL, `quota_size` bigint NOT NULL, `quota_files` integer NOT NULL, " +
  37. "`permissions` longtext NOT NULL, `used_quota_size` bigint NOT NULL, `used_quota_files` integer NOT NULL, " +
  38. "`last_quota_update` bigint NOT NULL, `upload_bandwidth` integer NOT NULL, `download_bandwidth` integer NOT NULL, " +
  39. "`last_login` bigint NOT NULL, `filters` longtext NULL, `filesystem` longtext NULL, `additional_info` longtext NULL);" +
  40. "CREATE TABLE `{{folders_mapping}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `virtual_path` varchar(512) NOT NULL, " +
  41. "`quota_size` bigint NOT NULL, `quota_files` integer NOT NULL, `folder_id` integer NOT NULL, `user_id` integer NOT NULL);" +
  42. "ALTER TABLE `{{folders_mapping}}` ADD CONSTRAINT `{{prefix}}unique_mapping` UNIQUE (`user_id`, `folder_id`);" +
  43. "ALTER TABLE `{{folders_mapping}}` ADD CONSTRAINT `{{prefix}}folders_mapping_folder_id_fk_folders_id` FOREIGN KEY (`folder_id`) REFERENCES `{{folders}}` (`id`) ON DELETE CASCADE;" +
  44. "ALTER TABLE `{{folders_mapping}}` ADD CONSTRAINT `{{prefix}}folders_mapping_user_id_fk_users_id` FOREIGN KEY (`user_id`) REFERENCES `{{users}}` (`id`) ON DELETE CASCADE;" +
  45. "INSERT INTO {{schema_version}} (version) VALUES (10);"
  46. mysqlV11SQL = "CREATE TABLE `{{api_keys}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, `name` varchar(255) NOT NULL, `key_id` varchar(50) NOT NULL UNIQUE," +
  47. "`api_key` varchar(255) NOT NULL UNIQUE, `scope` integer NOT NULL, `created_at` bigint NOT NULL, `updated_at` bigint NOT NULL, `last_use_at` bigint NOT NULL, " +
  48. "`expires_at` bigint NOT NULL, `description` longtext NULL, `admin_id` integer NULL, `user_id` integer NULL);" +
  49. "ALTER TABLE `{{api_keys}}` ADD CONSTRAINT `{{prefix}}api_keys_admin_id_fk_admins_id` FOREIGN KEY (`admin_id`) REFERENCES `{{admins}}` (`id`) ON DELETE CASCADE;" +
  50. "ALTER TABLE `{{api_keys}}` ADD CONSTRAINT `{{prefix}}api_keys_user_id_fk_users_id` FOREIGN KEY (`user_id`) REFERENCES `{{users}}` (`id`) ON DELETE CASCADE;"
  51. mysqlV11DownSQL = "DROP TABLE `{{api_keys}}` CASCADE;"
  52. mysqlV12SQL = "ALTER TABLE `{{admins}}` ADD COLUMN `created_at` bigint DEFAULT 0 NOT NULL;" +
  53. "ALTER TABLE `{{admins}}` ALTER COLUMN `created_at` DROP DEFAULT;" +
  54. "ALTER TABLE `{{admins}}` ADD COLUMN `updated_at` bigint DEFAULT 0 NOT NULL;" +
  55. "ALTER TABLE `{{admins}}` ALTER COLUMN `updated_at` DROP DEFAULT;" +
  56. "ALTER TABLE `{{admins}}` ADD COLUMN `last_login` bigint DEFAULT 0 NOT NULL;" +
  57. "ALTER TABLE `{{admins}}` ALTER COLUMN `last_login` DROP DEFAULT;" +
  58. "ALTER TABLE `{{users}}` ADD COLUMN `created_at` bigint DEFAULT 0 NOT NULL;" +
  59. "ALTER TABLE `{{users}}` ALTER COLUMN `created_at` DROP DEFAULT;" +
  60. "ALTER TABLE `{{users}}` ADD COLUMN `updated_at` bigint DEFAULT 0 NOT NULL;" +
  61. "ALTER TABLE `{{users}}` ALTER COLUMN `updated_at` DROP DEFAULT;" +
  62. "CREATE INDEX `{{prefix}}users_updated_at_idx` ON `{{users}}` (`updated_at`);"
  63. mysqlV12DownSQL = "ALTER TABLE `{{admins}}` DROP COLUMN `updated_at`;" +
  64. "ALTER TABLE `{{admins}}` DROP COLUMN `created_at`;" +
  65. "ALTER TABLE `{{admins}}` DROP COLUMN `last_login`;" +
  66. "ALTER TABLE `{{users}}` DROP COLUMN `created_at`;" +
  67. "ALTER TABLE `{{users}}` DROP COLUMN `updated_at`;"
  68. mysqlV13SQL = "ALTER TABLE `{{users}}` ADD COLUMN `email` varchar(255) NULL;"
  69. mysqlV13DownSQL = "ALTER TABLE `{{users}}` DROP COLUMN `email`;"
  70. mysqlV14SQL = "CREATE TABLE `{{shares}}` (`id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY, " +
  71. "`share_id` varchar(60) NOT NULL UNIQUE, `name` varchar(255) NOT NULL, `description` varchar(512) NULL, " +
  72. "`scope` integer NOT NULL, `paths` longtext NOT NULL, `created_at` bigint NOT NULL, " +
  73. "`updated_at` bigint NOT NULL, `last_use_at` bigint NOT NULL, `expires_at` bigint NOT NULL, " +
  74. "`password` longtext NULL, `max_tokens` integer NOT NULL, `used_tokens` integer NOT NULL, " +
  75. "`allow_from` longtext NULL, `user_id` integer NOT NULL);" +
  76. "ALTER TABLE `{{shares}}` ADD CONSTRAINT `{{prefix}}shares_user_id_fk_users_id` " +
  77. "FOREIGN KEY (`user_id`) REFERENCES `{{users}}` (`id`) ON DELETE CASCADE;"
  78. mysqlV14DownSQL = "DROP TABLE `{{shares}}` CASCADE;"
  79. )
  80. // MySQLProvider auth provider for MySQL/MariaDB database
  81. type MySQLProvider struct {
  82. dbHandle *sql.DB
  83. }
  84. func init() {
  85. version.AddFeature("+mysql")
  86. }
  87. func initializeMySQLProvider() error {
  88. var err error
  89. dbHandle, err := sql.Open("mysql", getMySQLConnectionString(false))
  90. if err == nil {
  91. providerLog(logger.LevelDebug, "mysql database handle created, connection string: %#v, pool size: %v",
  92. getMySQLConnectionString(true), config.PoolSize)
  93. dbHandle.SetMaxOpenConns(config.PoolSize)
  94. if config.PoolSize > 0 {
  95. dbHandle.SetMaxIdleConns(config.PoolSize)
  96. } else {
  97. dbHandle.SetMaxIdleConns(2)
  98. }
  99. dbHandle.SetConnMaxLifetime(240 * time.Second)
  100. provider = &MySQLProvider{dbHandle: dbHandle}
  101. } else {
  102. providerLog(logger.LevelError, "error creating mysql database handler, connection string: %#v, error: %v",
  103. getMySQLConnectionString(true), err)
  104. }
  105. return err
  106. }
  107. func getMySQLConnectionString(redactedPwd bool) string {
  108. var connectionString string
  109. if config.ConnectionString == "" {
  110. password := config.Password
  111. if redactedPwd {
  112. password = "[redacted]"
  113. }
  114. connectionString = fmt.Sprintf("%v:%v@tcp([%v]:%v)/%v?charset=utf8mb4&interpolateParams=true&timeout=10s&parseTime=true&tls=%v&writeTimeout=10s&readTimeout=10s",
  115. config.Username, password, config.Host, config.Port, config.Name, getSSLMode())
  116. } else {
  117. connectionString = config.ConnectionString
  118. }
  119. return connectionString
  120. }
  121. func (p *MySQLProvider) checkAvailability() error {
  122. return sqlCommonCheckAvailability(p.dbHandle)
  123. }
  124. func (p *MySQLProvider) validateUserAndPass(username, password, ip, protocol string) (User, error) {
  125. return sqlCommonValidateUserAndPass(username, password, ip, protocol, p.dbHandle)
  126. }
  127. func (p *MySQLProvider) validateUserAndTLSCert(username, protocol string, tlsCert *x509.Certificate) (User, error) {
  128. return sqlCommonValidateUserAndTLSCertificate(username, protocol, tlsCert, p.dbHandle)
  129. }
  130. func (p *MySQLProvider) validateUserAndPubKey(username string, publicKey []byte) (User, string, error) {
  131. return sqlCommonValidateUserAndPubKey(username, publicKey, p.dbHandle)
  132. }
  133. func (p *MySQLProvider) updateQuota(username string, filesAdd int, sizeAdd int64, reset bool) error {
  134. return sqlCommonUpdateQuota(username, filesAdd, sizeAdd, reset, p.dbHandle)
  135. }
  136. func (p *MySQLProvider) getUsedQuota(username string) (int, int64, error) {
  137. return sqlCommonGetUsedQuota(username, p.dbHandle)
  138. }
  139. func (p *MySQLProvider) setUpdatedAt(username string) {
  140. sqlCommonSetUpdatedAt(username, p.dbHandle)
  141. }
  142. func (p *MySQLProvider) updateLastLogin(username string) error {
  143. return sqlCommonUpdateLastLogin(username, p.dbHandle)
  144. }
  145. func (p *MySQLProvider) updateAdminLastLogin(username string) error {
  146. return sqlCommonUpdateAdminLastLogin(username, p.dbHandle)
  147. }
  148. func (p *MySQLProvider) userExists(username string) (User, error) {
  149. return sqlCommonGetUserByUsername(username, p.dbHandle)
  150. }
  151. func (p *MySQLProvider) addUser(user *User) error {
  152. return sqlCommonAddUser(user, p.dbHandle)
  153. }
  154. func (p *MySQLProvider) updateUser(user *User) error {
  155. return sqlCommonUpdateUser(user, p.dbHandle)
  156. }
  157. func (p *MySQLProvider) deleteUser(user *User) error {
  158. return sqlCommonDeleteUser(user, p.dbHandle)
  159. }
  160. func (p *MySQLProvider) dumpUsers() ([]User, error) {
  161. return sqlCommonDumpUsers(p.dbHandle)
  162. }
  163. func (p *MySQLProvider) getRecentlyUpdatedUsers(after int64) ([]User, error) {
  164. return sqlCommonGetRecentlyUpdatedUsers(after, p.dbHandle)
  165. }
  166. func (p *MySQLProvider) getUsers(limit int, offset int, order string) ([]User, error) {
  167. return sqlCommonGetUsers(limit, offset, order, p.dbHandle)
  168. }
  169. func (p *MySQLProvider) dumpFolders() ([]vfs.BaseVirtualFolder, error) {
  170. return sqlCommonDumpFolders(p.dbHandle)
  171. }
  172. func (p *MySQLProvider) getFolders(limit, offset int, order string) ([]vfs.BaseVirtualFolder, error) {
  173. return sqlCommonGetFolders(limit, offset, order, p.dbHandle)
  174. }
  175. func (p *MySQLProvider) getFolderByName(name string) (vfs.BaseVirtualFolder, error) {
  176. ctx, cancel := context.WithTimeout(context.Background(), defaultSQLQueryTimeout)
  177. defer cancel()
  178. return sqlCommonGetFolderByName(ctx, name, p.dbHandle)
  179. }
  180. func (p *MySQLProvider) addFolder(folder *vfs.BaseVirtualFolder) error {
  181. return sqlCommonAddFolder(folder, p.dbHandle)
  182. }
  183. func (p *MySQLProvider) updateFolder(folder *vfs.BaseVirtualFolder) error {
  184. return sqlCommonUpdateFolder(folder, p.dbHandle)
  185. }
  186. func (p *MySQLProvider) deleteFolder(folder *vfs.BaseVirtualFolder) error {
  187. return sqlCommonDeleteFolder(folder, p.dbHandle)
  188. }
  189. func (p *MySQLProvider) updateFolderQuota(name string, filesAdd int, sizeAdd int64, reset bool) error {
  190. return sqlCommonUpdateFolderQuota(name, filesAdd, sizeAdd, reset, p.dbHandle)
  191. }
  192. func (p *MySQLProvider) getUsedFolderQuota(name string) (int, int64, error) {
  193. return sqlCommonGetFolderUsedQuota(name, p.dbHandle)
  194. }
  195. func (p *MySQLProvider) adminExists(username string) (Admin, error) {
  196. return sqlCommonGetAdminByUsername(username, p.dbHandle)
  197. }
  198. func (p *MySQLProvider) addAdmin(admin *Admin) error {
  199. return sqlCommonAddAdmin(admin, p.dbHandle)
  200. }
  201. func (p *MySQLProvider) updateAdmin(admin *Admin) error {
  202. return sqlCommonUpdateAdmin(admin, p.dbHandle)
  203. }
  204. func (p *MySQLProvider) deleteAdmin(admin *Admin) error {
  205. return sqlCommonDeleteAdmin(admin, p.dbHandle)
  206. }
  207. func (p *MySQLProvider) getAdmins(limit int, offset int, order string) ([]Admin, error) {
  208. return sqlCommonGetAdmins(limit, offset, order, p.dbHandle)
  209. }
  210. func (p *MySQLProvider) dumpAdmins() ([]Admin, error) {
  211. return sqlCommonDumpAdmins(p.dbHandle)
  212. }
  213. func (p *MySQLProvider) validateAdminAndPass(username, password, ip string) (Admin, error) {
  214. return sqlCommonValidateAdminAndPass(username, password, ip, p.dbHandle)
  215. }
  216. func (p *MySQLProvider) apiKeyExists(keyID string) (APIKey, error) {
  217. return sqlCommonGetAPIKeyByID(keyID, p.dbHandle)
  218. }
  219. func (p *MySQLProvider) addAPIKey(apiKey *APIKey) error {
  220. return sqlCommonAddAPIKey(apiKey, p.dbHandle)
  221. }
  222. func (p *MySQLProvider) updateAPIKey(apiKey *APIKey) error {
  223. return sqlCommonUpdateAPIKey(apiKey, p.dbHandle)
  224. }
  225. func (p *MySQLProvider) deleteAPIKey(apiKey *APIKey) error {
  226. return sqlCommonDeleteAPIKey(apiKey, p.dbHandle)
  227. }
  228. func (p *MySQLProvider) getAPIKeys(limit int, offset int, order string) ([]APIKey, error) {
  229. return sqlCommonGetAPIKeys(limit, offset, order, p.dbHandle)
  230. }
  231. func (p *MySQLProvider) dumpAPIKeys() ([]APIKey, error) {
  232. return sqlCommonDumpAPIKeys(p.dbHandle)
  233. }
  234. func (p *MySQLProvider) updateAPIKeyLastUse(keyID string) error {
  235. return sqlCommonUpdateAPIKeyLastUse(keyID, p.dbHandle)
  236. }
  237. func (p *MySQLProvider) shareExists(shareID, username string) (Share, error) {
  238. return sqlCommonGetShareByID(shareID, username, p.dbHandle)
  239. }
  240. func (p *MySQLProvider) addShare(share *Share) error {
  241. return sqlCommonAddShare(share, p.dbHandle)
  242. }
  243. func (p *MySQLProvider) updateShare(share *Share) error {
  244. return sqlCommonUpdateShare(share, p.dbHandle)
  245. }
  246. func (p *MySQLProvider) deleteShare(share *Share) error {
  247. return sqlCommonDeleteShare(share, p.dbHandle)
  248. }
  249. func (p *MySQLProvider) getShares(limit int, offset int, order, username string) ([]Share, error) {
  250. return sqlCommonGetShares(limit, offset, order, username, p.dbHandle)
  251. }
  252. func (p *MySQLProvider) dumpShares() ([]Share, error) {
  253. return sqlCommonDumpShares(p.dbHandle)
  254. }
  255. func (p *MySQLProvider) updateShareLastUse(shareID string, numTokens int) error {
  256. return sqlCommonUpdateShareLastUse(shareID, numTokens, p.dbHandle)
  257. }
  258. func (p *MySQLProvider) close() error {
  259. return p.dbHandle.Close()
  260. }
  261. func (p *MySQLProvider) reloadConfig() error {
  262. return nil
  263. }
  264. // initializeDatabase creates the initial database structure
  265. func (p *MySQLProvider) initializeDatabase() error {
  266. dbVersion, err := sqlCommonGetDatabaseVersion(p.dbHandle, false)
  267. if err == nil && dbVersion.Version > 0 {
  268. return ErrNoInitRequired
  269. }
  270. if errors.Is(err, sql.ErrNoRows) {
  271. return errSchemaVersionEmpty
  272. }
  273. initialSQL := strings.ReplaceAll(mysqlInitialSQL, "{{schema_version}}", sqlTableSchemaVersion)
  274. initialSQL = strings.ReplaceAll(initialSQL, "{{admins}}", sqlTableAdmins)
  275. initialSQL = strings.ReplaceAll(initialSQL, "{{folders}}", sqlTableFolders)
  276. initialSQL = strings.ReplaceAll(initialSQL, "{{users}}", sqlTableUsers)
  277. initialSQL = strings.ReplaceAll(initialSQL, "{{folders_mapping}}", sqlTableFoldersMapping)
  278. initialSQL = strings.ReplaceAll(initialSQL, "{{prefix}}", config.SQLTablesPrefix)
  279. return sqlCommonExecSQLAndUpdateDBVersion(p.dbHandle, strings.Split(initialSQL, ";"), 10)
  280. }
  281. //nolint:dupl
  282. func (p *MySQLProvider) migrateDatabase() error {
  283. dbVersion, err := sqlCommonGetDatabaseVersion(p.dbHandle, true)
  284. if err != nil {
  285. return err
  286. }
  287. switch version := dbVersion.Version; {
  288. case version == sqlDatabaseVersion:
  289. providerLog(logger.LevelDebug, "sql database is up to date, current version: %v", version)
  290. return ErrNoInitRequired
  291. case version < 10:
  292. err = fmt.Errorf("database version %v is too old, please see the upgrading docs", version)
  293. providerLog(logger.LevelError, "%v", err)
  294. logger.ErrorToConsole("%v", err)
  295. return err
  296. case version == 10:
  297. return updateMySQLDatabaseFromV10(p.dbHandle)
  298. case version == 11:
  299. return updateMySQLDatabaseFromV11(p.dbHandle)
  300. case version == 12:
  301. return updateMySQLDatabaseFromV12(p.dbHandle)
  302. case version == 13:
  303. return updateMySQLDatabaseFromV13(p.dbHandle)
  304. default:
  305. if version > sqlDatabaseVersion {
  306. providerLog(logger.LevelError, "database version %v is newer than the supported one: %v", version,
  307. sqlDatabaseVersion)
  308. logger.WarnToConsole("database version %v is newer than the supported one: %v", version,
  309. sqlDatabaseVersion)
  310. return nil
  311. }
  312. return fmt.Errorf("database version not handled: %v", version)
  313. }
  314. }
  315. func (p *MySQLProvider) revertDatabase(targetVersion int) error {
  316. dbVersion, err := sqlCommonGetDatabaseVersion(p.dbHandle, true)
  317. if err != nil {
  318. return err
  319. }
  320. if dbVersion.Version == targetVersion {
  321. return errors.New("current version match target version, nothing to do")
  322. }
  323. switch dbVersion.Version {
  324. case 14:
  325. return downgradeMySQLDatabaseFromV14(p.dbHandle)
  326. case 13:
  327. return downgradeMySQLDatabaseFromV13(p.dbHandle)
  328. case 12:
  329. return downgradeMySQLDatabaseFromV12(p.dbHandle)
  330. case 11:
  331. return downgradeMySQLDatabaseFromV11(p.dbHandle)
  332. default:
  333. return fmt.Errorf("database version not handled: %v", dbVersion.Version)
  334. }
  335. }
  336. func (p *MySQLProvider) resetDatabase() error {
  337. sql := strings.ReplaceAll(mysqlResetSQL, "{{schema_version}}", sqlTableSchemaVersion)
  338. sql = strings.ReplaceAll(sql, "{{admins}}", sqlTableAdmins)
  339. sql = strings.ReplaceAll(sql, "{{folders}}", sqlTableFolders)
  340. sql = strings.ReplaceAll(sql, "{{users}}", sqlTableUsers)
  341. sql = strings.ReplaceAll(sql, "{{folders_mapping}}", sqlTableFoldersMapping)
  342. sql = strings.ReplaceAll(sql, "{{api_keys}}", sqlTableAPIKeys)
  343. sql = strings.ReplaceAll(sql, "{{shares}}", sqlTableShares)
  344. return sqlCommonExecSQLAndUpdateDBVersion(p.dbHandle, strings.Split(sql, ";"), 0)
  345. }
  346. func updateMySQLDatabaseFromV10(dbHandle *sql.DB) error {
  347. if err := updateMySQLDatabaseFrom10To11(dbHandle); err != nil {
  348. return err
  349. }
  350. return updateMySQLDatabaseFromV11(dbHandle)
  351. }
  352. func updateMySQLDatabaseFromV11(dbHandle *sql.DB) error {
  353. if err := updateMySQLDatabaseFrom11To12(dbHandle); err != nil {
  354. return err
  355. }
  356. return updateMySQLDatabaseFromV12(dbHandle)
  357. }
  358. func updateMySQLDatabaseFromV12(dbHandle *sql.DB) error {
  359. if err := updateMySQLDatabaseFrom12To13(dbHandle); err != nil {
  360. return err
  361. }
  362. return updateMySQLDatabaseFromV13(dbHandle)
  363. }
  364. func updateMySQLDatabaseFromV13(dbHandle *sql.DB) error {
  365. return updateMySQLDatabaseFrom13To14(dbHandle)
  366. }
  367. func downgradeMySQLDatabaseFromV14(dbHandle *sql.DB) error {
  368. if err := downgradeMySQLDatabaseFrom14To13(dbHandle); err != nil {
  369. return err
  370. }
  371. return downgradeMySQLDatabaseFromV13(dbHandle)
  372. }
  373. func downgradeMySQLDatabaseFromV13(dbHandle *sql.DB) error {
  374. if err := downgradeMySQLDatabaseFrom13To12(dbHandle); err != nil {
  375. return err
  376. }
  377. return downgradeMySQLDatabaseFromV12(dbHandle)
  378. }
  379. func downgradeMySQLDatabaseFromV12(dbHandle *sql.DB) error {
  380. if err := downgradeMySQLDatabaseFrom12To11(dbHandle); err != nil {
  381. return err
  382. }
  383. return downgradeMySQLDatabaseFromV11(dbHandle)
  384. }
  385. func downgradeMySQLDatabaseFromV11(dbHandle *sql.DB) error {
  386. return downgradeMySQLDatabaseFrom11To10(dbHandle)
  387. }
  388. func updateMySQLDatabaseFrom13To14(dbHandle *sql.DB) error {
  389. logger.InfoToConsole("updating database version: 13 -> 14")
  390. providerLog(logger.LevelInfo, "updating database version: 13 -> 14")
  391. sql := strings.ReplaceAll(mysqlV14SQL, "{{shares}}", sqlTableShares)
  392. sql = strings.ReplaceAll(sql, "{{users}}", sqlTableUsers)
  393. sql = strings.ReplaceAll(sql, "{{prefix}}", config.SQLTablesPrefix)
  394. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 14)
  395. }
  396. func downgradeMySQLDatabaseFrom14To13(dbHandle *sql.DB) error {
  397. logger.InfoToConsole("downgrading database version: 14 -> 13")
  398. providerLog(logger.LevelInfo, "downgrading database version: 14 -> 13")
  399. sql := strings.ReplaceAll(mysqlV14DownSQL, "{{shares}}", sqlTableShares)
  400. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 13)
  401. }
  402. func updateMySQLDatabaseFrom12To13(dbHandle *sql.DB) error {
  403. logger.InfoToConsole("updating database version: 12 -> 13")
  404. providerLog(logger.LevelInfo, "updating database version: 12 -> 13")
  405. sql := strings.ReplaceAll(mysqlV13SQL, "{{users}}", sqlTableUsers)
  406. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 13)
  407. }
  408. func downgradeMySQLDatabaseFrom13To12(dbHandle *sql.DB) error {
  409. logger.InfoToConsole("downgrading database version: 13 -> 12")
  410. providerLog(logger.LevelInfo, "downgrading database version: 13 -> 12")
  411. sql := strings.ReplaceAll(mysqlV13DownSQL, "{{users}}", sqlTableUsers)
  412. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 12)
  413. }
  414. func updateMySQLDatabaseFrom11To12(dbHandle *sql.DB) error {
  415. logger.InfoToConsole("updating database version: 11 -> 12")
  416. providerLog(logger.LevelInfo, "updating database version: 11 -> 12")
  417. sql := strings.ReplaceAll(mysqlV12SQL, "{{users}}", sqlTableUsers)
  418. sql = strings.ReplaceAll(sql, "{{admins}}", sqlTableAdmins)
  419. sql = strings.ReplaceAll(sql, "{{prefix}}", config.SQLTablesPrefix)
  420. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 12)
  421. }
  422. func downgradeMySQLDatabaseFrom12To11(dbHandle *sql.DB) error {
  423. logger.InfoToConsole("downgrading database version: 12 -> 11")
  424. providerLog(logger.LevelInfo, "downgrading database version: 12 -> 11")
  425. sql := strings.ReplaceAll(mysqlV12DownSQL, "{{users}}", sqlTableUsers)
  426. sql = strings.ReplaceAll(sql, "{{admins}}", sqlTableAdmins)
  427. sql = strings.ReplaceAll(sql, "{{prefix}}", config.SQLTablesPrefix)
  428. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 11)
  429. }
  430. func updateMySQLDatabaseFrom10To11(dbHandle *sql.DB) error {
  431. logger.InfoToConsole("updating database version: 10 -> 11")
  432. providerLog(logger.LevelInfo, "updating database version: 10 -> 11")
  433. sql := strings.ReplaceAll(mysqlV11SQL, "{{users}}", sqlTableUsers)
  434. sql = strings.ReplaceAll(sql, "{{admins}}", sqlTableAdmins)
  435. sql = strings.ReplaceAll(sql, "{{api_keys}}", sqlTableAPIKeys)
  436. sql = strings.ReplaceAll(sql, "{{prefix}}", config.SQLTablesPrefix)
  437. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 11)
  438. }
  439. func downgradeMySQLDatabaseFrom11To10(dbHandle *sql.DB) error {
  440. logger.InfoToConsole("downgrading database version: 11 -> 10")
  441. providerLog(logger.LevelInfo, "downgrading database version: 11 -> 10")
  442. sql := strings.ReplaceAll(mysqlV11DownSQL, "{{api_keys}}", sqlTableAPIKeys)
  443. return sqlCommonExecSQLAndUpdateDBVersion(dbHandle, strings.Split(sql, ";"), 10)
  444. }