From 41f56ad05d12520bb90f91889fa979465d0b3d6b Mon Sep 17 00:00:00 2001
From: ᴜɴᴋɴᴡᴏɴ <u@gogs.io>
Date: Sat, 11 Apr 2020 20:18:05 +0800
Subject: login_source: migrate to GORM and add tests (#6090)

* Use GORM in all write paths

* Migrate to GORM

* Fix lint errors

* Use GORM  to init table

* dbutil: make writer detect error

* Add more tests

* Rename to clearTables

* db: finish adding tests

* osutil: add tests

* Fix load source files path
---
 internal/db/db.go | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'internal/db/db.go')

diff --git a/internal/db/db.go b/internal/db/db.go
index 1be2cc4b..77d78f53 100644
--- a/internal/db/db.go
+++ b/internal/db/db.go
@@ -124,7 +124,7 @@ func getLogWriter() (io.Writer, error) {
 
 var tables = []interface{}{
 	new(AccessToken),
-	new(LFSObject),
+	new(LFSObject), new(LoginSource),
 }
 
 func Init() error {
@@ -167,9 +167,14 @@ func Init() error {
 		return time.Now().UTC().Truncate(time.Microsecond)
 	}
 
+	sourceFiles, err := loadLoginSourceFiles(filepath.Join(conf.CustomDir(), "conf", "auth.d"))
+	if err != nil {
+		return errors.Wrap(err, "load login source files")
+	}
+
 	// Initialize stores, sorted in alphabetical order.
 	AccessTokens = &accessTokens{DB: db}
-	LoginSources = &loginSources{DB: db}
+	LoginSources = &loginSources{DB: db, files: sourceFiles}
 	LFS = &lfs{DB: db}
 	Perms = &perms{DB: db}
 	Repos = &repos{DB: db}
-- 
cgit v1.2.3