mirror of
https://github.com/thomiceli/opengist.git
synced 2024-12-23 13:02:39 +00:00
34 lines
662 B
Go
34 lines
662 B
Go
package models
|
|
|
|
import (
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
)
|
|
|
|
var db *gorm.DB
|
|
|
|
func Setup(dbpath string) error {
|
|
var err error
|
|
|
|
if db, err = gorm.Open(sqlite.Open(dbpath+"?_fk=true"), &gorm.Config{
|
|
Logger: logger.Default.LogMode(logger.Silent),
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = db.AutoMigrate(&User{}, &SSHKey{}, &Gist{}, &AdminSetting{}); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Default admin setting values
|
|
return initAdminSettings(map[string]string{
|
|
SettingDisableSignup: "0",
|
|
})
|
|
}
|
|
|
|
func CountAll(table interface{}) (int64, error) {
|
|
var count int64
|
|
err := db.Model(table).Count(&count).Error
|
|
return count, err
|
|
}
|