package query import ( "log" "path" // "time" "gorm.io/gorm" "gorm.io/gorm/logger" "gorm.io/driver/sqlite" "reichard.io/imagini/internal/config" "reichard.io/imagini/internal/models" ) func NewDB(c *config.Config) *gorm.DB { gormConfig := &gorm.Config{ PrepareStmt: true, Logger: logger.Default.LogMode(logger.Silent), } var db *gorm.DB if c.DBType == "SQLite" { dbLocation := path.Join(c.ConfigPath, "imagini.db") db, _ = gorm.Open(sqlite.Open(dbLocation), gormConfig) } else { log.Fatal("ERROR: Unsupported Database") } // Initialize Database db.AutoMigrate(&models.ServerSetting{}) db.AutoMigrate(&models.User{}) db.AutoMigrate(&models.MediaItem{}) db.AutoMigrate(&models.Tag{}) db.AutoMigrate(&models.Album{}) return db } func ItemsFromAlbum(db *gorm.DB, user models.User, album models.Album) []models.MediaItem { var mediaItems []models.MediaItem // db.Table("media_albums"). // Select("media_item.*"). // Joins("INNER JOIN media_items ON media_albums.ID = media_items.Albums"). // Where("media_albums.album_id = ? AND media_items.User = ?", albumID, userID). db. //Where("album = ? AND user = ?", albumID, userID). Find(&mediaItems) return mediaItems // db.Raw(` // SELECT // MediaItems.* // FROM // MediaAlbums // INNER JOIN MediaItems ON MediaAlbums.mediaID = MediaItems.mediaID // WHERE MediaAlbums.albumID = ? AND MediaItems.userID = ?`, albumID, userID) } // func ItemsFromTags(userID int, tagID int) []MediaItem { // return nil // } // // func IndexMediaItems(newItems []MediaItem) { // } // func PopulateTestData() { // user1 := User{Username: "Evan", Email: "evan@reichard.io", FirstName: "Evan", LastName: "Reichard", AuthType: "LDAP", Salt: "1234", HashedPWSalt: "1234"} // user2 := User{Username: "Ryan", Email: "ryan@example.com", FirstName: "Ryan", LastName: "Dunfrey", AuthType: "Local", Salt: "2345", HashedPWSalt: "2345"} // user3 := User{Username: "Bill", Email: "bill@example.com", FirstName: "Bill", LastName: "Smith", AuthType: "LDAP", Salt: "3456", HashedPWSalt: "3456"} // // mi1 := MediaItem{ // User: user1, // EXIFDate: time.Now(), // Latitude: "1234", // Longitude: "1234", // RelPath: "./1234.jpg", // Tags: []Tag{ // {Name: "Tag1"}, // {Name: "Tag2"}, // }, // Albums: []Album{ // {Name: "Album1"}, // {Name: "Album2"}, // }, // } // // mi2 := MediaItem{ // User: user2, // EXIFDate: time.Now(), // Latitude: "1234", // Longitude: "1234", // RelPath: "./1234.jpg", // Tags: []Tag{ // {Name: "Tag3"}, // {Name: "Tag4"}, // }, // Albums: []Album{ // {Name: "Album3"}, // {Name: "Album4"}, // }, // } // // mi3 := MediaItem{ // User: user3, // EXIFDate: time.Now(), // Latitude: "1234", // Longitude: "1234", // RelPath: "./1234.jpg", // Tags: []Tag{ // {Name: "Tag4"}, // {Name: "Tag5"}, // }, // Albums: []Album{ // {Name: "Album1"}, // {Name: "Album7"}, // }, // } // // // db.Create(&user1) // // db.Create(&user2) // // db.Create(&user3) // db.Create(&mi1) // db.Create(&mi2) // db.Create(&mi3) // }