ovpn-certman/services/db.go

package services

import (
	"errors"
	"log"

	"git.klink.asia/paul/certman/models"
	"github.com/jinzhu/gorm"
)

// Error Definitions
var (
	ErrNotImplemented = errors.New("Not implemented")
)

type DBConfig struct {
	Type string
	DSN  string
	Log  bool
}

// DB is a wrapper around gorm.DB to provide custom methods
type DB struct {
	*gorm.DB

	conf *DBConfig
}

func NewDB(conf *DBConfig) *DB {
	// Establish connection
	db, err := gorm.Open(conf.Type, conf.DSN)
	if err != nil {
		log.Fatalf("Could not open database: %s", err.Error())
	}

	// Migrate models
	db.AutoMigrate(models.User{}, models.Client{})
	db.LogMode(conf.Log)

	return &DB{
		DB:   db,
		conf: conf,
	}
}

// CountUsers returns the number of Users in the datastore
func (db *DB) CountUsers() (uint, error) {
	return 0, ErrNotImplemented
}

// CreateUser inserts a user into the datastore
func (db *DB) CreateUser(*models.User) (*models.User, error) {
	return nil, ErrNotImplemented
}

// ListUsers returns a slice of 'count' users, starting at 'offset'
func (db *DB) ListUsers(count, offset int) ([]*models.User, error) {
	var users = make([]*models.User, 0)

	return users, ErrNotImplemented
}

// GetUserByID returns a single user by ID
func (db *DB) GetUserByID(id uint) (*models.User, error) {
	return nil, ErrNotImplemented
}

// GetUserByEmail returns a single user by email
func (db *DB) GetUserByEmail(email string) (*models.User, error) {
	return nil, ErrNotImplemented
}

// DeleteUser removes a user from the datastore
func (db *DB) DeleteUser(id uint) error {
	return ErrNotImplemented
}