imagini/internal/db/db.go

package db

import (
    "errors"
    "path"
    "fmt"

    log "github.com/sirupsen/logrus"
    // "gorm.io/gorm/logger"
    "gorm.io/driver/sqlite"
    "gorm.io/gorm"

    "reichard.io/imagini/internal/config"
    "reichard.io/imagini/graph/model"
)

type DBManager struct {
    db  *gorm.DB
}

func NewMgr(c *config.Config) *DBManager {
    gormConfig := &gorm.Config{
        PrepareStmt: true,
        // Logger: logger.Default.LogMode(logger.Silent),
    }

    // Create manager
    dbm := &DBManager{}

    if c.DBType == "SQLite" {
        dbLocation := path.Join(c.ConfigPath, "imagini.db")
        dbm.db, _ = gorm.Open(sqlite.Open(dbLocation), gormConfig)
    } else {
        log.Fatal("Unsupported Database")
    }

    // Initialize database
    dbm.db.AutoMigrate(&model.Device{})
    dbm.db.AutoMigrate(&model.User{})
    dbm.db.AutoMigrate(&model.MediaItem{})
    dbm.db.AutoMigrate(&model.Tag{})
    dbm.db.AutoMigrate(&model.Album{})

    // Determine whether to bootstrap
    var count int64
    dbm.db.Model(&model.User{}).Count(&count)
    if count == 0 {
        dbm.bootstrapDatabase()
    }

    return dbm
}

func (dbm *DBManager) bootstrapDatabase() {
    log.Info("[query] Bootstrapping database.")

    password := "admin"
    user := &model.User{
        Username: "admin",
        AuthType: "Local",
        Password: &password,
        Role: model.RoleAdmin,
    }

    err := dbm.CreateUser(user)

    if err != nil {
        log.Fatal("[query] Unable to bootstrap database.")
    }
}

func (dbm *DBManager) QueryBuilder(dest interface{}, params []byte) (int64, error) {
    // TODO:
    //  - Where Filters
    //  - Sort Filters
    //  - Paging Filters

    objType := fmt.Sprintf("%T", dest)
    if objType == "*[]model.MediaItem" {
        // TODO: Validate MediaItem Type
    } else {
        // Return Error
        return 0, errors.New("Invalid type")
    }

    var count int64
    err := dbm.db.Find(dest).Count(&count).Error;
    return count, err

    // Paging:
    //  - Regular Pagination:
    //      - /api/v1/MediaItems?page[limit]=50&page=2
    //  - Meta Count Only
    //      - /api/v1/MediaItems?page[limit]=0

    // Sorting:
    //  - Ascending Sort:
    //      - /api/v1/MediaItems?sort=created_at
    //  - Descending Sort:
    //      - /api/v1/MediaItems?sort=-created_at

    // Filters:
    //  - Greater Than / Less Than (created_at, updated_at, exif_date)
    //      - /api/v1/MediaItems?filter[created_at]>=2020-01-01&filter[created_at]<=2021-01-01
    //  - Long / Lat Range (latitude, longitude)
    //      - /api/v1/MediaItems?filter[latitude]>=71.1827&filter[latitude]<=72.0000&filter[longitude]>=100.000&filter[longitude]<=101.0000
    //  - Image / Video (media_type)
    //      - /api/v1/MediaItems?filter[media_type]=Image
    //  - Tags (tags)
    //      - /api/v1/MediaItems?filter[tags]=id1,id2,id3
    //  - Albums (albums)
    //      - /api/v1/MediaItems?filter[albums]=id1
}