feat: add support for sqlite3

main.go

@@ -13,10 +13,12 @@ import (
 	"time"
 
 	"github.com/codegangsta/negroni"
+	"github.com/gobuffalo/packr/v2"
 	"github.com/gorilla/mux"
 	"github.com/jinzhu/gorm"
 	"github.com/joho/godotenv"
 	"github.com/rs/cors"
+	"github.com/rubenv/sql-migrate"
 	uuid "github.com/satori/go.uuid"
 	ini "gopkg.in/ini.v1"
 
@@ -118,25 +120,46 @@ func readConfig() *models.Config {
 func main() {
 	// Read Config
 	config := readConfig()
+	// Enable line numbers in logging
+	log.SetFlags(log.LstdFlags | log.Lshortfile)
 
 	// Connect to database
 	db, err := gorm.Open(config.DbDialect, utils.MakeConnectionString(config))
+	var migrateDo func()
+	if config.DbDialect == "sqlite3" {
+		db.DB().Exec("PRAGMA foreign_keys = ON;")
+		migrations := &migrate.PackrMigrationSource{
+			Box: packr.New("migrations", "./migrations/sqlite3"),
+		}
+		migrateDo = func() {
+			n, err := migrate.Exec(db.DB(), "sqlite3", migrations, migrate.Up)
+			if err != nil {
+				log.Fatal(err)
+			}
+			log.Printf("Applied %d migrations!\n", n)
+		}
+	} else {
+		migrateDo = func() {
+			db.AutoMigrate(&models.Alias{})
+			db.AutoMigrate(&models.Summary{})
+			db.AutoMigrate(&models.SummaryItem{})
+			db.AutoMigrate(&models.User{})
+			db.AutoMigrate(&models.Heartbeat{}).AddForeignKey("user_id", "users(id)", "RESTRICT", "RESTRICT")
+			db.AutoMigrate(&models.SummaryItem{}).AddForeignKey("summary_id", "summaries(id)", "CASCADE", "CASCADE")
+		}
+	}
 	db.LogMode(config.IsDev())
 	db.DB().SetMaxIdleConns(int(config.DbMaxConn))
 	db.DB().SetMaxOpenConns(int(config.DbMaxConn))
 	if err != nil {
+		log.Println(err)
 		log.Fatal("Could not connect to database.")
 	}
 	// TODO: Graceful shutdown
 	defer db.Close()
 
 	// Migrate database schema
-	db.AutoMigrate(&models.User{})
-	db.AutoMigrate(&models.Alias{})
-	db.AutoMigrate(&models.Summary{})
-	db.AutoMigrate(&models.SummaryItem{})
-	db.AutoMigrate(&models.Heartbeat{}).AddForeignKey("user_id", "users(id)", "RESTRICT", "RESTRICT")
-	db.AutoMigrate(&models.SummaryItem{}).AddForeignKey("summary_id", "summaries(id)", "CASCADE", "CASCADE")
+	migrateDo()
 
 	// Custom migrations and initial data
 	addDefaultUser(db, config)