mirror of
https://github.com/thomiceli/opengist.git
synced 2024-12-23 13:02:39 +00:00
22052bd38f
* Add a setting to allow accessing individual gists without auth This is a middle ground between the existing setting "Require Login", which requires login to do anything at all, and having it off, which shows a public list of gists and more generally allows discovering info about the users/gists of the instance without login. The idea of this setting is that it is "require login" for everything except individual gists. Fixes #228. Co-authored-by: Thomas Miceli <tho.miceli@gmail.com>
92 lines
1.9 KiB
Go
92 lines
1.9 KiB
Go
package db
|
|
|
|
import (
|
|
"errors"
|
|
"slices"
|
|
"strings"
|
|
|
|
msqlite "github.com/glebarez/go-sqlite"
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/thomiceli/opengist/internal/config"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
)
|
|
|
|
var db *gorm.DB
|
|
|
|
func Setup(dbPath string, sharedCache bool) error {
|
|
var err error
|
|
journalMode := strings.ToUpper(config.C.SqliteJournalMode)
|
|
|
|
if !slices.Contains([]string{"DELETE", "TRUNCATE", "PERSIST", "MEMORY", "WAL", "OFF"}, journalMode) {
|
|
log.Warn().Msg("Invalid SQLite journal mode: " + journalMode)
|
|
}
|
|
|
|
sharedCacheStr := ""
|
|
if sharedCache {
|
|
sharedCacheStr = "&cache=shared"
|
|
}
|
|
|
|
if db, err = gorm.Open(sqlite.Open(dbPath+"?_fk=true&_journal_mode="+journalMode+sharedCacheStr), &gorm.Config{
|
|
Logger: logger.Default.LogMode(logger.Silent),
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = db.SetupJoinTable(&Gist{}, "Likes", &Like{}); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = db.SetupJoinTable(&User{}, "Liked", &Like{}); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = db.AutoMigrate(&User{}, &Gist{}, &SSHKey{}, &AdminSetting{}, &Invitation{}); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = ApplyMigrations(db); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Default admin setting values
|
|
return initAdminSettings(map[string]string{
|
|
SettingDisableSignup: "0",
|
|
SettingRequireLogin: "0",
|
|
SettingAllowGistsWithoutLogin: "0",
|
|
SettingDisableLoginForm: "0",
|
|
SettingDisableGravatar: "0",
|
|
})
|
|
}
|
|
|
|
func Close() error {
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return sqlDB.Close()
|
|
}
|
|
|
|
func CountAll(table interface{}) (int64, error) {
|
|
var count int64
|
|
err := db.Model(table).Count(&count).Error
|
|
return count, err
|
|
}
|
|
|
|
func IsUniqueConstraintViolation(err error) bool {
|
|
var sqliteErr *msqlite.Error
|
|
if errors.As(err, &sqliteErr) && sqliteErr.Code() == 2067 {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func Ping() error {
|
|
sql, err := db.DB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return sql.Ping()
|
|
}
|