dnote/pkg/server/database/migrate/main.go
2025-10-04 00:25:12 -07:00

72 lines
1.7 KiB
Go

/* Copyright (C) 2019, 2020, 2021, 2022, 2023, 2024, 2025 Dnote contributors
*
* This file is part of Dnote.
*
* Dnote is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Dnote is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with Dnote. If not, see <https://www.gnu.org/licenses/>.
*/
package main
import (
"flag"
"fmt"
"os"
"github.com/dnote/dnote/pkg/server/config"
"github.com/dnote/dnote/pkg/server/database"
"github.com/joho/godotenv"
"github.com/pkg/errors"
"github.com/rubenv/sql-migrate"
)
var (
migrationDir = flag.String("migrationDir", "../migrations", "the path to the directory with migraiton files")
)
func init() {
fmt.Println("Migrating Dnote database...")
// Load env
if os.Getenv("GO_ENV") != "PRODUCTION" {
if err := godotenv.Load("../../.env.dev"); err != nil {
panic(err)
}
}
}
func main() {
flag.Parse()
c := config.Load()
db := database.Open(c)
migrations := &migrate.FileMigrationSource{
Dir: *migrationDir,
}
migrate.SetTable("migrations")
sqlDB, err := db.DB()
if err != nil {
panic(errors.Wrap(err, "getting underlying sql.DB"))
}
n, err := migrate.Exec(sqlDB, "postgres", migrations, migrate.Up)
if err != nil {
panic(errors.Wrap(err, "executing migrations"))
}
fmt.Printf("Applied %d migrations\n", n)
}