oef/orm/orm.go

package orm

import (
	"fmt"
	"net/http"
	"path"
	"reflect"
	"strings"

	"git.andreafazzi.eu/andrea/oef/config"
	"github.com/gorilla/sessions"
	"github.com/jinzhu/gorm"
	"github.com/jinzhu/inflection"

	_ "github.com/jinzhu/gorm/dialects/mysql"
)

type IDer interface {
	GetID() uint
}

type GetFn func(map[string]string, http.ResponseWriter, *http.Request) (interface{}, error)

var (
	fns    map[string]func(map[string]string, http.ResponseWriter, *http.Request) (interface{}, error)
	currDB *gorm.DB
	store  = sessions.NewCookieStore([]byte(config.Config.Keys.CookieStoreKey))
)

func init() {
	fns = make(map[string]func(map[string]string, http.ResponseWriter, *http.Request) (interface{}, error), 0)
}

func New(connection string) (*gorm.DB, error) {
	db, err := gorm.Open("mysql", connection)
	if err != nil {
		return nil, err
	}
	return db, nil
}

func AutoMigrate(models ...interface{}) {
	if err := currDB.AutoMigrate(models...).Error; err != nil {
		panic(err)
	}

}

func CreateCategories() {
	for _, name := range categories {
		var category Category
		if err := currDB.FirstOrCreate(&category, Category{Name: name}).Error; err != nil {
			panic(err)
		}
	}
}

func Use(db *gorm.DB) {
	currDB = db
}

func DB() *gorm.DB {
	return currDB
}

func MapHandlers(models []interface{}) error {
	for _, model := range models {
		name := inflection.Plural(strings.ToLower(ModelName(model)))
		for p, action := range map[string]string{
			"":            "ReadAll",
			"create/":     "Create",
			"{id}":        "Read",
			"{id}/update": "Update",
			"{id}/delete": "Delete",
		} {
			method := reflect.ValueOf(model).MethodByName(action)
			if !method.IsValid() {
				return fmt.Errorf("Action %s is not defined for model %s", action, name)
			}
			joinedPath := path.Join("/", name, p)
			if strings.HasSuffix(p, "/") {
				joinedPath += "/"
			}
			fns[joinedPath] = method.Interface().(func(map[string]string, http.ResponseWriter, *http.Request) (interface{}, error))
		}

	}

	return nil
}

func GetFunc(path string) (GetFn, error) {
	fn, ok := fns[path]
	if !ok {
		return nil, fmt.Errorf("Can't map path %s to any model methods.", path)
	}
	return fn, nil
}

func GetNothing(args map[string]string) (interface{}, error) {
	return nil, nil
}

func PostNothing(args map[string]string, w http.ResponseWriter, r *http.Request) (IDer, error) {
	return nil, nil
}

func ModelName(s interface{}) string {
	t := reflect.TypeOf(s)
	switch t.Kind() {
	case reflect.Ptr:
		elem := t.Elem()
		if strings.Contains(elem.String(), "[]") {
			return strings.Replace(elem.String(), "[]*orm.", "", -1)
		}
		return elem.Name()
	case reflect.Slice:
		return strings.Replace(t.Elem().String(), "*orm.", "", -1)
	default:
		return t.Name()
	}
}