mysql.go 20 KB

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